quadrilateralcowboy/tools/compilers/roqvq/codec.cpp

1703 lines
48 KiB
C++
Raw Permalink Normal View History

2020-06-12 21:06:25 +00:00
/*
===========================================================================
Doom 3 GPL Source Code
Copyright (C) 1999-2011 id Software LLC, a ZeniMax Media company.
This file is part of the Doom 3 GPL Source Code (?Doom 3 Source Code?).
Doom 3 Source Code is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
Doom 3 Source Code is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with Doom 3 Source Code. If not, see <http://www.gnu.org/licenses/>.
In addition, the Doom 3 Source Code is also subject to certain additional terms. You should have received a copy of these additional terms immediately following the terms and conditions of the GNU General Public License which accompanied the Doom 3 Source Code. If not, please request a copy in writing from id Software at the address below.
If you have questions concerning this license or the applicable additional terms, you may contact in writing id Software LLC, c/o ZeniMax Media Inc., Suite 120, Rockville, Maryland 20850 USA.
===========================================================================
*/
#include "../../../idlib/precompiled.h"
#pragma hdrstop
#include "codec.h"
// flibit wedged this in for Win32... eh. -flibit
#ifdef _WIN32
#undef HUGE_VAL
#define HUGE_VAL HUGE
#endif
2020-06-12 21:06:25 +00:00
float glimit( const float val ) {
if (val<0) return 0;
if (val>255) return 255;
return val;
}
codec::codec() {
int i;
common->Printf("init: initing.....\n");
codebooksize = 256;
codebook2 = (VQDATA **) Mem_ClearedAlloc(256*sizeof(VQDATA *));
for(i=0; i < 256; i++) {
codebook2[i] = (VQDATA *) Mem_ClearedAlloc(16*sizeof(VQDATA));
}
codebook4 = (VQDATA **) Mem_ClearedAlloc(256*sizeof(VQDATA *));
for(i=0; i < 256; i++) {
codebook4[i] = (VQDATA *) Mem_ClearedAlloc(64*sizeof(VQDATA));
}
previousImage[0] = 0;
previousImage[1] = 0;
image = 0;
whichFrame = 0;
qStatus = 0;
luti = 0;
overAmount = 0;
codebookmade = 0;
slop = 0;
}
codec::~codec()
{
common->Printf("codec: resetting\n");
if (qStatus) Mem_Free( qStatus);
if (luti) Mem_Free(luti);
if (previousImage[0]) delete previousImage[0];
if (previousImage[1]) delete previousImage[1];
qStatus = 0;
initRGBtab = 0;
previousImage[0] = 0;
whichFrame = 0;
luti = 0;
return;
}
/* Because Shellsort is a variation on Insertion Sort, it has the same
* inconsistency that I noted in the InsertionSort class. Notice where I
* subtract a move to compensate for calling a swap for visual purposes.
*/
void codec::Sort( float *list, int *intIndex, int numElements )
{
#define STRIDE_FACTOR 3 // good value for stride factor is not well-understood
// 3 is a fairly good choice (Sedgewick)
int c,d, stride;
bool found;
stride = 1;
while (stride <= numElements)
stride = stride*STRIDE_FACTOR +1;
while (stride>(STRIDE_FACTOR-1)) { // loop to sort for each value of stride
stride = stride / STRIDE_FACTOR;
for (c = stride; c < numElements; c++){
found = false;
d = c-stride;
while ((d >= 0) && !found) { // move to left until correct place
if (list[d]<list[d+stride]) {
float ftemp;
int itemp;
ftemp = list[d]; list[d] = list[d+stride]; list[d+stride] = ftemp;
itemp = intIndex[d]; intIndex[d] = intIndex[d+stride]; intIndex[d+stride] = itemp;
d -= stride; // jump by stride factor
} else
found = true;
}
}
}
}
void codec::Segment( int *alist, float *flist, int numElements, float rmse)
{
int x, y, yy, xx, numc, onf, index, temp, best, a0, a1, a2, a3, bpp, i, len;
byte find[16], *lineout, *cbook, *src, *dst;
float fy, fcr, fcb;
idFile *fpcb;
char cbFile[256], tempcb[256], temptb[256];
bool doopen;
float y0,y1,y2,y3,cr,cb;
doopen = false;
a0 = a1 = a2 = a3 = 0;
sprintf( tempcb, "%s.cb", theRoQ->CurrentFilename());
sprintf( temptb, "%s.tb", theRoQ->CurrentFilename());
onf = 0;
len = (int)strlen(tempcb);
for(x=0;x<len;x++) {
if (tempcb[x] == '\n') for(y=x;y<len;y++) if (tempcb[y] == '/') x = y+1;
cbFile[onf++] = tempcb[x];
}
cbFile[onf] = 0;
lineout = (byte *)Mem_ClearedAlloc( 4*1024 );
common->Printf("trying %s\n", cbFile);
fpcb = fileSystem->OpenFileRead( cbFile );
if ( !fpcb ) {
doopen = true;
common->Printf("failed....\n");
} else {
if ( dimension2 == 16 ) x = 3584; else x = 2560;
if ( fpcb->Read( lineout, x ) != x ) {
doopen = true;
common->Printf("failed....\n");
}
fileSystem->CloseFile( fpcb );
}
if ( doopen ) {
common->Printf("segment: making %s\n", cbFile);
numc = numElements;
if (numElements > numc) numc = numElements;
onf = 0;
for(x=0;x<256;x++) {
for(y=0;y<dimension2;y++) codebook2[x][y] = 0;
for(y=0;y<dimension4;y++) codebook4[x][y] = 0;
}
bpp = image->samplesPerPixel();
cbook = (byte *)Mem_ClearedAlloc( 3 * image->pixelsWide() * image->pixelsHigh());
float *snrBook = (float *)Mem_ClearedAlloc( image->pixelsWide() * image->pixelsHigh() );
dst = cbook;
int numEntries = 0;
for (i=0; i<numQuadCels; i++) {
if (qStatus[i].size == 8 && qStatus[i].rsnr >= MIN_SNR*4) {
for(y=qStatus[i].yat;y<qStatus[i].yat+8; y+=4) {
for(x=qStatus[i].xat;x<qStatus[i].xat+8; x+=4) {
if (qStatus[i].rsnr == 9999.0f) {
snrBook[numEntries] = 1.0f;
} else {
snrBook[numEntries] = qStatus[i].rsnr;
}
numEntries++;
for(yy=y;yy<(y+4);yy++) {
for(xx=x;xx<(x+4);xx++) {
src = image->bitmapData() + (yy*(bpp*image->pixelsWide())) + (xx*bpp);
memcpy( dst, src, 3); dst += 3;
}
}
}
}
}
}
common->Printf("segment: %d 4x4 cels to vq\n", numEntries );
VQ( numEntries, dimension4, cbook, snrBook, codebook4, true );
dst = cbook;
numEntries = 0;
for( i=0; i<256; i++ ) {
for(y=0;y<4;y+=2) {
for(x=0;x<4;x+=2) {
snrBook[numEntries] = 1.0f;
numEntries++;
for(yy=y;yy<(y+2);yy++) {
for(xx=x;xx<(x+2);xx++) {
dst[0] = codebook4[i][yy*12+xx*3+0];
dst[1] = codebook4[i][yy*12+xx*3+1];
dst[2] = codebook4[i][yy*12+xx*3+2];
dst += 3;
}
}
}
}
}
common->Printf("segment: %d 2x2 cels to vq\n", numEntries);
VQ( numEntries, dimension2, cbook, snrBook, codebook2, false );
Mem_Free(cbook);
Mem_Free(snrBook);
index = 0;
for( onf = 0; onf < 256; onf++ ) {
numc = 0; fcr = fcb = 0;
for( x = 0; x < 4; x++ ) {
fy = RMULT*(float)(codebook2[onf][numc+0]) +
GMULT*(float)(codebook2[onf][numc+1]) +
BMULT*(float)(codebook2[onf][numc+2]) + 0.5f;
if (fy<0) fy = 0; if (fy>255) fy = 255;
fcr += RIEMULT*(float)(codebook2[onf][numc+0]);
fcr += GIEMULT*(float)(codebook2[onf][numc+1]);
fcr += BIEMULT*(float)(codebook2[onf][numc+2]);
fcb += RQEMULT*(float)(codebook2[onf][numc+0]);
fcb += GQEMULT*(float)(codebook2[onf][numc+1]);
fcb += BQEMULT*(float)(codebook2[onf][numc+2]);
lineout[index++] = (byte)fy;
numc += 3;
}
fcr = (fcr/4)+128.5f; if (fcr<0) fcr = 0; if (fcr>255) fcr = 255;
fcb = (fcb/4)+128.5f; if (fcb<0) fcb = 0; if (fcb>255) fcr = 255;
//common->Printf(" fcr == %f, fcb == %f\n", fcr, fcb );
lineout[index++] = (byte)fcr;
lineout[index++] = (byte)fcb;
}
for(onf=0;onf<256;onf++) {
for(y=0;y<4;y+=2) {
for(x=0;x<4;x+=2) {
numc = 0;
for(yy=y;yy<(y+2);yy++) {
temp = (yy*dimension2)+x*(dimension2/4);
find[numc++] = (byte)(codebook4[onf][temp+0] + 0.50f);
find[numc++] = (byte)(codebook4[onf][temp+1] + 0.50f);
find[numc++] = (byte)(codebook4[onf][temp+2] + 0.50f);
find[numc++] = (byte)(codebook4[onf][temp+3] + 0.50f);
find[numc++] = (byte)(codebook4[onf][temp+4] + 0.50f);
find[numc++] = (byte)(codebook4[onf][temp+5] + 0.50f);
}
lineout[index++] = BestCodeword( find, dimension2, codebook2 );
}
}
}
fpcb = fileSystem->OpenFileWrite( cbFile );
common->Printf("made up %d entries\n", index);
fpcb->Write( lineout, index );
fileSystem->CloseFile( fpcb );
common->Printf("finished write\n");
}
for(y=0;y<256;y++) {
x = y*6;
y0 = (float)lineout[x++];
y1 = (float)lineout[x++];
y2 = (float)lineout[x++];
y3 = (float)lineout[x++];
cb = (float)lineout[x++]; cb -= 128;
cr = (float)lineout[x ]; cr -= 128;
x = 0;
codebook2[y][x++] = glimit( y0 + 1.40200f*cr );
codebook2[y][x++] = glimit( y0 - 0.34414f*cb - 0.71414f*cr );
codebook2[y][x++] = glimit( y0 + 1.77200f*cb );
codebook2[y][x++] = glimit( y1 + 1.40200f*cr );
codebook2[y][x++] = glimit( y1 - 0.34414f*cb - 0.71414f*cr );
codebook2[y][x++] = glimit( y1 + 1.77200f*cb );
codebook2[y][x++] = glimit( y2 + 1.40200f*cr );
codebook2[y][x++] = glimit( y2 - 0.34414f*cb - 0.71414f*cr );
codebook2[y][x++] = glimit( y2 + 1.77200f*cb );
codebook2[y][x++] = glimit( y3 + 1.40200f*cr );
codebook2[y][x++] = glimit( y3 - 0.34414f*cb - 0.71414f*cr );
codebook2[y][x++] = glimit( y3 + 1.77200f*cb );
}
index = 6*256;
for(onf=0;onf<256;onf++) {
for(y=0;y<4;y+=2) {
for(x=0;x<4;x+=2) {
best = lineout[index++];
numc = 0;
for(yy=y;yy<(y+2);yy++) {
temp = (yy*dimension2)+x*(dimension2/4);
codebook4[onf][temp+0] = codebook2[best][numc++]; //r
codebook4[onf][temp+1] = codebook2[best][numc++]; //g
codebook4[onf][temp+2] = codebook2[best][numc++]; //b
codebook4[onf][temp+3] = codebook2[best][numc++]; //r a
codebook4[onf][temp+4] = codebook2[best][numc++]; //g r
codebook4[onf][temp+5] = codebook2[best][numc++]; //b g
}
}
}
}
theRoQ->WriteCodeBook(lineout);
//PrepareCodeBook();
Mem_Free(lineout);
}
int codec::BestCodeword( unsigned char *tempvector, int dimension, VQDATA **codebook )
{
VQDATA dist;
VQDATA bestDist = HUGE_VAL;
2020-06-12 21:06:25 +00:00
VQDATA tempvq[64];
int bestIndex = -1;
for( int i=0; i<dimension; i++ ) {
tempvq[i] = tempvector[i];
}
for( int i=0; i<256; i++) {
dist = 0.0;
for( int x=0; x<dimension; x+=3 ) {
const VQDATA r0 = codebook[i][x];
const VQDATA r1 = tempvq[x];
const VQDATA g0 = codebook[i][x+1];
const VQDATA g1 = tempvq[x+1];
const VQDATA b0 = codebook[i][x+2];
const VQDATA b1 = tempvq[x+2];
dist += (r0-r1)*(r0-r1);
if (dist >= bestDist) {
continue;
}
dist += (g0-g1)*(g0-g1);
if (dist >= bestDist) {
continue;
}
dist += (b0-b1)*(b0-b1);
if (dist >= bestDist) {
continue;
}
}
if ( dist < bestDist ) {
bestDist = dist;
bestIndex = i;
}
}
return bestIndex;
}
void codec::SetPreviousImage( const char*filename, NSBitmapImageRep *timage )
{
if (previousImage[0]) {
delete previousImage[0];
}
if (previousImage[1]) {
delete previousImage[1];
}
common->Printf("setPreviousImage:%s\n", filename);
previousImage[0] = new NSBitmapImageRep( );
previousImage[1] = new NSBitmapImageRep( );
whichFrame=1;
*previousImage[0] = *timage;
*previousImage[1] = *timage;
pixelsHigh = previousImage[0]->pixelsHigh();
pixelsWide = previousImage[0]->pixelsWide();
common->Printf("setPreviousImage: %dx%d\n", pixelsWide, pixelsHigh );
}
void codec::MakePreviousImage( quadcel *pquad )
{
int i, dy, dx, pluck, size, ind, xx, yy, pWide;
int x, y;
byte *rgbmap, *idataA, *fccdictionary;
bool diff;
for(i=0;i<256;i++) { used2[i] = used4[i] = false; }
pWide = pixelsWide & 0xfff0;
if (!previousImage[0]) {
previousImage[0] = new NSBitmapImageRep( pWide, (pixelsHigh & 0xfff0) );
previousImage[1] = new NSBitmapImageRep( pWide, (pixelsHigh & 0xfff0) );
}
rgbmap = previousImage[(whichFrame&1)]->bitmapData();
if ((whichFrame&1) == 1) {
fccdictionary = previousImage[0]->bitmapData();
} else {
fccdictionary = previousImage[1]->bitmapData();
}
idataA = (byte *)Mem_Alloc( 16*16*4 );
for(i=0;i<numQuadCels;i++) {
diff = false;
size = pquad[i].size;
if (size) {
switch( pquad[i].status ) {
case DEP:
break;
case SLD:
ind = pquad[i].patten[0];
used4[ind] = true;
for( dy=0; dy<size; dy++ ) {
pluck = (((dy+pquad[i].yat)*pWide)+pquad[i].xat)*4;
for( dx=0; dx<size; dx++ ) {
xx = ((dy>>1)*dimension2)+(dx>>1)*(dimension2/4);
if (rgbmap[pluck+0] != codebook4[ind][xx+0]) diff = true;
if (rgbmap[pluck+1] != codebook4[ind][xx+1]) diff = true;
if (rgbmap[pluck+2] != codebook4[ind][xx+2]) diff = true;
if (dimension4 == 64 && rgbmap[pluck+3] != codebook4[ind][xx+3]) diff = true;
rgbmap[pluck+0] = (byte)codebook4[ind][xx+0];
rgbmap[pluck+1] = (byte)codebook4[ind][xx+1];
rgbmap[pluck+2] = (byte)codebook4[ind][xx+2];
if (dimension4 == 64)
rgbmap[pluck+3] = (byte)codebook4[ind][xx+3];
else
rgbmap[pluck+3] = 255;
pluck += 4;
}
}
if (diff == false && whichFrame) common->Printf("drawImage: SLD just changed the same thing\n");
break;
case PAT:
ind = pquad[i].patten[0];
used4[ind] = true;
for( dy=0; dy<size; dy++ ) {
pluck = (((dy+pquad[i].yat)*pWide)+pquad[i].xat)*4;
for( dx=0; dx<size; dx++ ) {
xx = (dy*size*(dimension2/4))+dx*(dimension2/4);
if (rgbmap[pluck+0] != codebook4[ind][xx+0]) diff = true;
if (rgbmap[pluck+1] != codebook4[ind][xx+1]) diff = true;
if (rgbmap[pluck+2] != codebook4[ind][xx+2]) diff = true;
if (dimension4 == 64 && rgbmap[pluck+3] != codebook4[ind][xx+3]) diff = true;
rgbmap[pluck+0] = (byte)codebook4[ind][xx+0];
rgbmap[pluck+1] = (byte)codebook4[ind][xx+1];
rgbmap[pluck+2] = (byte)codebook4[ind][xx+2];
if (dimension4 == 64)
rgbmap[pluck+3] = (byte)codebook4[ind][xx+3];
else
rgbmap[pluck+3] = 255;
pluck += 4;
}
}
if (diff == false && whichFrame) common->Printf("drawImage: PAT just changed the same thing\n");
break;
case CCC:
dx = 1;
for(yy=0;yy<4;yy+=2) {
for(xx=0;xx<4;xx+=2) {
ind = pquad[i].patten[dx++];
used2[ind] = true;
dy = 0;
for(y=yy;y<(yy+2);y++) {
for(x=xx;x<(xx+2);x++) {
pluck = (((y+pquad[i].yat)*pWide)+(pquad[i].xat+x))*4;
if (rgbmap[pluck+0] != codebook2[ind][dy+0]) diff = true;
if (rgbmap[pluck+1] != codebook2[ind][dy+1]) diff = true;
if (rgbmap[pluck+2] != codebook2[ind][dy+2]) diff = true;
if (dimension4 == 64 && rgbmap[pluck+3] != codebook2[ind][dy+3]) diff = true;
rgbmap[pluck+0] = (byte)codebook2[ind][dy+0];
rgbmap[pluck+1] = (byte)codebook2[ind][dy+1];
rgbmap[pluck+2] = (byte)codebook2[ind][dy+2];
if (dimension4 == 64) {
rgbmap[pluck+3] = (byte)codebook2[ind][dy+3];
dy += 4;
} else {
rgbmap[pluck+3] = 255;
dy += 3;
}
}
}
}
}
if (diff == false && whichFrame) {
/*
common->Printf("drawImage: CCC just changed the same thing\n");
common->Printf("sparseEncode: something is wrong here\n");
common->Printf("xat: %d\n", pquad[i].xat);
common->Printf("yat: %d\n", pquad[i].yat);
common->Printf("size %d\n", pquad[i].size);
common->Printf("type: %d\n", pquad[i].status);
common->Printf("motsnr: %0f\n", pquad[i].snr[FCC]);
common->Printf("cccsnr: %0f\n", pquad[i].snr[CCC]);
common->Printf("rmse: %0f\n", pquad[i].rsnr);
common->Printf("pat0: %0d\n", pquad[i].patten[1]);
common->Printf("pat1: %0d\n", pquad[i].patten[2]);
common->Printf("pat2: %0d\n", pquad[i].patten[3]);
common->Printf("pat3: %0d\n", pquad[i].patten[4]);
//exit(1);
*/
}
break;
case FCC:
dx = pquad[i].xat - ((pquad[i].domain >> 8 ) - 128);
dy = pquad[i].yat - ((pquad[i].domain & 0xff) - 128);
if (image->pixelsWide()==(image->pixelsHigh()*4)) dx = pquad[i].xat - ((pquad[i].domain >> 8 ) - 128)*2;
if (theRoQ->Scaleable()) {
dx = pquad[i].xat - ((pquad[i].domain >> 8 ) - 128)*2;
dy = pquad[i].yat - ((pquad[i].domain & 0xff) - 128)*2;
}
// if (pquad[i].yat == 0) common->Printf("dx = %d, dy = %d, xat = %d\n", dx, dy, pquad[i].xat);
ind = (dy*pWide+dx)*4;
for( dy=0; dy<size; dy++ ) {
pluck = (((dy+pquad[i].yat)*pWide)+pquad[i].xat)*4;
for( dx=0; dx<size; dx++ ) {
if (rgbmap[pluck+0] != fccdictionary[ind+0]) diff = true;
if (rgbmap[pluck+1] != fccdictionary[ind+1]) diff = true;
if (rgbmap[pluck+2] != fccdictionary[ind+2]) diff = true;
rgbmap[pluck+0] = fccdictionary[ind+0];
rgbmap[pluck+1] = fccdictionary[ind+1];
rgbmap[pluck+2] = fccdictionary[ind+2];
rgbmap[pluck+3] = fccdictionary[ind+3];
pluck += 4; ind += 4;
}
ind += (pWide - size)*4;
}
// if (diff == false && whichFrame) common->Printf("drawImage: FCC just changed the same thing\n");
break;
case MOT:
break;
default:
common->Error( "bad code!!\n");
break;
}
}
}
if (whichFrame == 0) {
memcpy( previousImage[1]->bitmapData(), previousImage[0]->bitmapData(), pWide*(pixelsHigh & 0xfff0)*4);
}
x = 0; y = 0;
for(i=0;i<256;i++) {
if (used4[i]) x++;
if (used2[i]) y++;
}
if (theRoQ->IsQuiet() == false) common->Printf("drawImage: used %d 4x4 and %d 2x2 VQ cels\n", x,y);
Mem_Free( idataA );
}
void codec::InitImages( void )
{
int x,y, index0, index1, temp;
float ftemp;
byte *lutimage;
numQuadCels = ((pixelsWide & 0xfff0)*(pixelsHigh & 0xfff0))/(MINSIZE*MINSIZE);
numQuadCels += numQuadCels/4 + numQuadCels/16;
if (qStatus) Mem_Free(qStatus);
qStatus = (quadcel *)Mem_ClearedAlloc(numQuadCels*sizeof (quadcel));
InitQStatus();
//
if (previousImage[0]) {
pixelsWide = previousImage[0]->pixelsWide();
pixelsHigh = previousImage[0]->pixelsHigh();
temp = ((whichFrame+1)&1);
if (!luti) luti = (byte *)Mem_Alloc(pixelsWide*pixelsHigh);
lutimage = previousImage[temp]->bitmapData();
if (theRoQ->IsQuiet() == false) {
common->Printf("initImage: remaking lut image using buffer %d\n", temp);
}
index0 = index1 = 0;
for(y=0;y<pixelsHigh; y++) {
for(x=0;x<pixelsWide; x++) {
ftemp = RMULT*lutimage[index0+0] + GMULT*lutimage[index0+1] + BMULT*lutimage[index0+2];
temp = (int)ftemp;
luti[index1] = temp;
index0 += previousImage[0]->samplesPerPixel();
index1++;
}
}
}
}
void codec::QuadX( int startX, int startY, int quadSize)
{
int startSize;
int bigx, bigy, lowx, lowy;
lowx = lowy = 0;
bigx = pixelsWide & 0xfff0;
bigy = pixelsHigh & 0xfff0;
if ( (startX >= lowx) && (startX+quadSize) <= (bigx) && (startY+quadSize) <= (bigy) && (startY >= lowy) && quadSize <= MAXSIZE) {
qStatus[onQuad].size = quadSize;
qStatus[onQuad].xat = startX;
qStatus[onQuad].yat = startY;
qStatus[onQuad].rsnr = 999999;
onQuad++;
}
if (quadSize != MINSIZE) {
startSize = quadSize>>1;
QuadX( startX , startY , startSize );
QuadX( startX+startSize , startY , startSize );
QuadX( startX , startY+startSize , startSize );
QuadX( startX+startSize , startY+startSize , startSize );
}
}
void codec::InitQStatus( void )
{
int i,x,y;
for (i=0; i<numQuadCels; i++)
qStatus[i].size = 0;
onQuad = 0;
for(y=0;y<pixelsHigh; y+=16) {
for(x=0;x<pixelsWide; x+=16) {
QuadX( x, y,16 );
}
}
}
void codec::VqData8( byte *cel, quadcel *pquad )
{
byte tempImage[8*8*4];
int x, y, i, best, temp;
i = 0;
for(y=0;y<4;y++) {
for(x=0;x<4;x++) {
temp = y*64+x*8;
tempImage[i++] = (cel[temp+0]+cel[temp+4]+cel[temp+32]+cel[temp+36])/4;
tempImage[i++] = (cel[temp+1]+cel[temp+5]+cel[temp+33]+cel[temp+37])/4;
tempImage[i++] = (cel[temp+2]+cel[temp+6]+cel[temp+34]+cel[temp+38])/4;
if (dimension4 == 64)
tempImage[i++] = (cel[temp+3]+cel[temp+7]+cel[temp+35]+cel[temp+39])/4;
}
}
pquad->patten[0] = best = BestCodeword( tempImage, dimension4, codebook4 );
for(y=0;y<8;y++) {
for(x=0;x<8;x++) {
temp = y*32+x*4;
i = ((y/2)*4*(dimension2/4))+(x/2)*(dimension2/4);
tempImage[temp+0] = (byte)codebook4[best][i+0];
tempImage[temp+1] = (byte)codebook4[best][i+1];
tempImage[temp+2] = (byte)codebook4[best][i+2];
if (dimension4 == 64)
tempImage[temp+3] = (byte)codebook4[best][i+3];
else
tempImage[temp+3] = 255;
}
}
pquad->snr[SLD] = Snr( cel, tempImage, 8 )+1.0;
}
void codec::VqData4( byte *cel, quadcel *pquad )
{
byte tempImage[64];
int i, best, bpp;
// if (theRoQ->makingVideo] && previousImage[0]) return self;
if (dimension4 == 64) bpp = 4; else bpp = 3;
for(i=0;i<16;i++) {
tempImage[i*bpp+0] = cel[i*4+0];
tempImage[i*bpp+1] = cel[i*4+1];
tempImage[i*bpp+2] = cel[i*4+2];
if (dimension4 == 64) tempImage[i*bpp+3] = cel[i*4+3];
}
pquad->patten[0] = best = BestCodeword( tempImage, dimension4, codebook4 );
for(i=0;i<16;i++) {
tempImage[i*4+0] = (byte)codebook4[best][i*bpp+0];
tempImage[i*4+1] = (byte)codebook4[best][i*bpp+1];
tempImage[i*4+2] = (byte)codebook4[best][i*bpp+2];
if (dimension4 == 64)
tempImage[i*4+3] = (byte)codebook4[best][i*bpp+3];
else
tempImage[i*4+3] = 255;
}
pquad->snr[PAT] = Snr( cel, tempImage, 4 );
}
void codec::VqData2( byte *cel, quadcel *pquad )
{
byte tempImage[16], tempOut[64];
int i, j, best,x,y,xx,yy,bpp;
if (dimension4 == 64) bpp = 4; else bpp = 3;
j = 1;
for(yy=0;yy<4;yy+=2) {
for(xx=0;xx<4;xx+=2) {
i = 0;
for(y=yy;y<(yy+2);y++) {
for(x=xx;x<(xx+2);x++) {
tempImage[i++] = cel[y*16+x*4+0];
tempImage[i++] = cel[y*16+x*4+1];
tempImage[i++] = cel[y*16+x*4+2];
if (dimension4 == 64) tempImage[i++] = cel[y*16+x*4+3];
}
}
pquad->patten[j++] = best = BestCodeword( tempImage, dimension2, codebook2 );
i = 0;
for(y=yy;y<(yy+2);y++) {
for(x=xx;x<(xx+2);x++) {
tempOut[y*16+x*4+0] = (byte)codebook2[best][i++];
tempOut[y*16+x*4+1] = (byte)codebook2[best][i++];
tempOut[y*16+x*4+2] = (byte)codebook2[best][i++];
if (dimension4 == 64)
tempOut[y*16+x*4+3] = (byte)codebook2[best][i++];
else
tempOut[y*16+x*4+3] = 255;
}
}
}
}
pquad->snr[CCC] = Snr( cel, tempOut, 4 );
}
void codec::IRGBtab(void)
{
initRGBtab++;
}
float codec::Snr( byte *old, byte *bnew, int size ) {
int i, j;
float fsnr;
register int ind;
ind = 0;
for(i=0; i<size; i++) {
for(j=0; j<size; j++) {
if (old[3]||bnew[3]) ind += RGBADIST( old, bnew );
old += 4; bnew += 4;
}
}
fsnr = (float)ind;
fsnr /= (size*size);
fsnr = (float)sqrt( fsnr );
return (fsnr);
}
int codec::ComputeMotionBlock( byte *old, byte *bnew, int size )
{
int i,j,snr;
if (dimension4==64) return 0; // do not use this for alpha pieces
snr = 0;
for(i=0; i<size; i++) {
for(j=0; j<size; j++) {
snr += RGBADIST( old, bnew );
old += 4; bnew += 4;
}
}
snr /= (size*size);
return ( snr <= MOTION_MIN );
}
void codec::FvqData( byte *bitmap, int size, int realx, int realy, quadcel *pquad, bool clamp)
{
int x, y, xLen, yLen, mblur0, ripl, bpp, fabort, temp1;
int lowX, lowY, onx, ony, sX, sY, depthx, depthy, breakHigh;
float lowestSNR, fmblur0;
byte *scale1;
byte *bitma2;
int searchY, searchX, xxMean, yyMean;
if ( !previousImage[0] || dimension4 == 64) {
return;
}
for(x=0; x<(size*size); x++) {
fmblur0 = RMULT*bitmap[x*4+0] + GMULT*bitmap[x*4+1] + BMULT*bitmap[x*4+2];
luty[x] = (byte)fmblur0;
}
if (!luti) {
pquad->domain = 0x8080;
pquad->snr[FCC] = 9999;
return;
}
ony = realy - (realy & 0xfff0);
onx = realx - (realx & 0xfff0);
xLen = previousImage[0]->pixelsWide();
yLen = previousImage[0]->pixelsHigh();
ripl = xLen-size;
breakHigh = 99999999;
fabort = 0;
lowX = lowY = -1;
depthx = depthy = 1;
searchY = 8; //16;
searchX = 8; //32;
//if (xLen == (yLen*4)) depthx = 2;
//if (theRoQ->Scaleable()) depthx = depthy = 2;
if (clamp) { searchX = searchY = 8; }
searchX = searchX*depthx;
searchY = searchY*depthy;
xxMean = dxMean*depthx;
yyMean = dyMean*depthy;
if (((realx-xxMean)+searchX)<0 ||(((realx-xxMean)-searchX)+depthx+size)>xLen || ((realy-yyMean)+searchY)<0 || (((realy-yyMean)-searchY)+depthy+size)>yLen) {
pquad->snr[FCC] = 9999;
return;
}
int sPsQ = -1;
for( sX=(((realx-xxMean)-searchX)+depthx); sX<=((realx-xxMean)+searchX) && !fabort; sX+=depthx ) {
for( sY=(((realy-yyMean)-searchY)+depthy); sY<=((realy-yyMean)+searchY) && breakHigh; sY+=depthy ) {
temp1 = xLen*sY+sX;
if ( sX >= 0 && (sX+size) <= xLen && sY >= 0 && (sY+size) <= yLen ) {
bpp = previousImage[0]->samplesPerPixel();
ripl = (xLen-size)*bpp;
mblur0 = 0;
bitma2 = bitmap;
scale1 = previousImage[((whichFrame+1)&1)]->bitmapData() + temp1*bpp;
// mblur0 = 0;
// bitma2 = luty;
// scale1 = luti + temp1;
for( y=0; y<size; y++) {
for( x=0; x<size; x++) {
mblur0 += RGBADIST( bitma2, scale1);
bitma2 += 4; scale1 += 4;
}
if (mblur0 > breakHigh) {
break;
}
scale1 += ripl;
}
if (breakHigh > mblur0) {
breakHigh = mblur0;
lowX = sX;
lowY = sY;
}
}
}
}
if (lowX != -1 && lowY != -1) {
bpp = previousImage[0]->samplesPerPixel();
ripl = (xLen-size)*bpp;
mblur0 = 0;
bitma2 = bitmap;
scale1 = previousImage[((whichFrame+1)&1)]->bitmapData() + (xLen*lowY+lowX)*bpp;
for( y=0; y<size; y++) {
for( x=0; x<size; x++) {
mblur0 += RGBADIST( bitma2, scale1 );
scale1 += 4; bitma2 += 4;
}
scale1 += ripl;
}
lowestSNR = (float)mblur0;
lowestSNR /= (size*size);
lowestSNR = (float)sqrt( lowestSNR );
sX = (realx-lowX+128);
sY = (realy-lowY+128);
if (depthx==2) {
sX = ((realx-lowX)/2+128);
}
if (depthy==2) {
sY = ((realy-lowY)/2+128);
}
pquad->domain = (sX<<8)+sY;
pquad->snr[FCC] = lowestSNR;
}
}
void codec::GetData( unsigned char *iData, int qSize, int startX, int startY, NSBitmapImageRep *bitmap)
{
int x,y,yoff,bpp,yend,xend;
byte *iPlane[5];
int r,g,b,a;
yend = qSize+startY;
xend = qSize+startX;
if (startY > bitmap->pixelsHigh()) return;
if (yend > bitmap->pixelsHigh()) yend = bitmap->pixelsHigh();
if (xend > bitmap->pixelsWide()) xend = bitmap->pixelsWide();
bpp = bitmap->samplesPerPixel();
if (bitmap->hasAlpha()) {
iPlane[0] = bitmap->bitmapData();
for(y=startY;y<yend;y++) {
yoff = y*bitmap->pixelsWide()*bpp;
for(x=startX;x<xend;x++) {
r = iPlane[0][yoff+(x*bpp)+0];
g = iPlane[0][yoff+(x*bpp)+1];
b = iPlane[0][yoff+(x*bpp)+2];
a = iPlane[0][yoff+(x*bpp)+3];
*iData++ = r; *iData++ = g; *iData++ = b; *iData++ = a;
}
}
} else {
iPlane[0] = bitmap->bitmapData();
for(y=startY;y<yend;y++) {
yoff = y*bitmap->pixelsWide()*bpp;
for(x=startX;x<xend;x++) {
r = iPlane[0][yoff+(x*bpp)+0];
g = iPlane[0][yoff+(x*bpp)+1];
b = iPlane[0][yoff+(x*bpp)+2];
*iData++ = r; *iData++ = g; *iData++ = b;
*iData++ = 255;
}
}
}
}
void codec::LowestQuad( quadcel*qtemp, int* status, float* snr, int bweigh)
{
float wtemp;
float quickadd[DEAD];
int i;
quickadd[CCC] = 1;
quickadd[SLD] = 1;
quickadd[MOT] = 1;
quickadd[FCC] = 1;
quickadd[PAT] = 1;
/*
if (slop > theRoQ->NormalFrameSize()) {
quickadd[CCC] = 0.5f;
quickadd[PAT] = 1.0f;
}
*/
wtemp = 99999;
for(i=(DEAD-1);i>0;i--) {
if ( qtemp->snr[i]*quickadd[i] < wtemp ) {
*status = i;
*snr = qtemp->snr[i];
wtemp = qtemp->snr[i]*quickadd[i];
}
}
if ( qtemp->mark ) *status = MOT;
}
int codec::GetCurrentQuadOutputSize( quadcel *pquad )
{
int totalbits, i, totalbytes;
int quickadd[DEAD+1];
totalbits = 0;
quickadd[DEP] = 2;
quickadd[SLD] = 10;
quickadd[PAT] = 10;
quickadd[CCC] = 34;
quickadd[MOT] = 2;
quickadd[FCC] = 10;
quickadd[DEAD] = 0;
for( i=0; i<numQuadCels; i++ ) {
if (pquad[i].size && pquad[i].size < 16 ) totalbits += quickadd[pquad[i].status];
}
totalbytes = (totalbits >> 3)+2;
return (totalbytes);
}
float codec::GetCurrentRMSE( quadcel *pquad )
{
int i, j;
double totalbits;
totalbits = 0;
j = 0;
for( i=0; i<numQuadCels; i++ ) {
if (pquad[i].size && pquad[i].status && pquad[i].status != DEAD) {
if (pquad[i].size == 8) { totalbits += pquad[i].rsnr*4; j += 4; }
if (pquad[i].size == 4) { totalbits += pquad[i].rsnr*1; j += 1; }
}
}
totalbits /= j;
return ((float)totalbits);
}
int codec::AddQuad( quadcel *pquad, int lownum )
{
int i, nx, nsize;
float newsnr, cmul;
byte *idataA, *idataB;
if (lownum != -1) {
if (pquad[lownum].size == 8) {
nx = 1; nsize = 4; cmul = 1;
} else {
nx = 5; nsize = 8; cmul = 4;
}
newsnr = 0;
idataA = (byte *)Mem_Alloc(8*8*4);
idataB = (byte *)Mem_Alloc(8*8*4);
for( i=lownum+1; i<lownum+(nx*4)+1; i+=nx ) {
pquad[i].size = nsize;
GetData( idataA, pquad[i].size, pquad[i].xat, pquad[i].yat, image );
VqData4( idataA ,&pquad[i] );
VqData2( idataA, &pquad[i] );
if (previousImage[0]) {
FvqData( idataA, pquad[i].size, pquad[i].xat, pquad[i].yat, &pquad[i], true );
GetData( idataB, pquad[i].size, pquad[i].xat, pquad[i].yat, previousImage[whichFrame&1] );
pquad[i].snr[MOT] = Snr( idataA, idataB, pquad[i].size );
if (ComputeMotionBlock( idataA, idataB, pquad[i].size ) && !theRoQ->IsLastFrame() && !detail) {
pquad[i].mark = true;
}
}
LowestQuad( &pquad[i] ,&pquad[i].status, &pquad[i].rsnr, true );
newsnr += pquad[i].rsnr;
}
Mem_Free(idataA); Mem_Free(idataB);
newsnr /= 4;
LowestQuad( &pquad[lownum], &pquad[lownum].status, &pquad[lownum].rsnr, false );
if ( pquad[lownum+nx*0+1].status == MOT && pquad[lownum+nx*1+1].status == MOT
&& pquad[lownum+nx*2+1].status == MOT && pquad[lownum+nx*3+1].status == MOT
&& nsize == 4) { newsnr = 9999; pquad[lownum].status = MOT; }
if ( pquad[lownum].rsnr > newsnr ) {
pquad[lownum].status = DEP;
pquad[lownum].rsnr = 0;
for( i=lownum+1; i<lownum+(nx*4)+1; i+=nx ) {
theRoQ->MarkQuadx( pquad[i].xat, pquad[i].yat, nsize, pquad[i].rsnr, qStatus[i].status );
}
} else {
theRoQ->MarkQuadx( pquad[lownum].xat, pquad[lownum].yat, nsize*2, pquad[lownum].rsnr, qStatus[lownum].status );
pquad[lownum+nx*0+1].status = 0;
pquad[lownum+nx*1+1].status = 0;
pquad[lownum+nx*2+1].status = 0;
pquad[lownum+nx*3+1].status = 0;
pquad[lownum+nx*0+1].size = 0;
pquad[lownum+nx*1+1].size = 0;
pquad[lownum+nx*2+1].size = 0;
pquad[lownum+nx*3+1].size = 0;
}
} else {
lownum = -1;
}
return lownum;
}
int codec::MotMeanX( void )
{
return dxMean;
}
int codec::MotMeanY( void )
{
return dyMean;
}
void codec::SparseEncode( void )
{
int i, j, osize, fsize, num[DEAD+1], *ilist, onf, ong, wtype, temp;
float *flist, sRMSE, numredo;
byte *idataA, *idataB;
osize = 8;
image = theRoQ->CurrentImage();
newImage = 0;
pixelsHigh = image->pixelsHigh();
pixelsWide = image->pixelsWide();
dimension2 = 12; dimension4 = 48;
if (image->hasAlpha()&&(theRoQ->ParamNoAlpha() == false)) { dimension2 = 16; dimension4 = 64; }
idataA = (byte *)Mem_Alloc( 16*16*4 );
idataB = (byte *)Mem_Alloc( 16*16*4 );
if (!previousImage[0]) common->Printf("sparseEncode: sparsely encoding a %d,%d image\n", pixelsWide, pixelsHigh);
InitImages();
flist = (float *)Mem_ClearedAlloc( (numQuadCels+1) *sizeof(float) );
ilist = (int *)Mem_ClearedAlloc( (numQuadCels+1) *sizeof(int ) );
fsize = 56*1024;
if (theRoQ->NumberOfFrames()>2) {
if (previousImage[0]) fsize = theRoQ->NormalFrameSize(); else fsize = theRoQ->FirstFrameSize();
if (theRoQ->HasSound() && fsize > 6000 && previousImage[0]) fsize = 6000;
}
fsize += (slop/50);
if (fsize > 64000) {
fsize = 64000;
}
if (previousImage[0] && fsize > theRoQ->NormalFrameSize()*2) {
fsize = theRoQ->NormalFrameSize()*2;
}
dxMean = dyMean = 0;
if (previousImage[0]) wtype = 1; else wtype = 0;
for( i=0; i<numQuadCels; i++ ) {
for(j=0;j<DEAD;j++) qStatus[i].snr[j] = 9999;
qStatus[i].mark = false;
if ( qStatus[i].size == osize ) {
if (previousImage[0]) {
GetData( idataA, qStatus[i].size, qStatus[i].xat, qStatus[i].yat, image );
GetData( idataB, qStatus[i].size, qStatus[i].xat, qStatus[i].yat, previousImage[whichFrame&1] );
qStatus[i].snr[MOT] = Snr( idataA, idataB, qStatus[i].size );
if (ComputeMotionBlock( idataA, idataB, qStatus[i].size ) && !theRoQ->IsLastFrame()) {
qStatus[i].mark = true;
}
if (!qStatus[i].mark) {
FvqData( idataA, qStatus[i].size, qStatus[i].xat, qStatus[i].yat, &qStatus[i], false );
}
}
LowestQuad( &qStatus[i], &qStatus[i].status, &qStatus[i].rsnr, wtype );
if (qStatus[i].rsnr < 9999)
theRoQ->MarkQuadx( qStatus[i].xat, qStatus[i].yat, qStatus[i].size, qStatus[i].rsnr, qStatus[i].status );
} else {
if ( qStatus[i].size < osize ) {
qStatus[i].status = 0;
qStatus[i].size = 0;
} else {
qStatus[i].status = DEP;
qStatus[i].rsnr = 0;
}
}
}
//
// the quad is complete, so status can now be used for quad decomposition
// the first thing to do is to set it up for all the 4x4 cels to get output
// and then recurse from there to see what's what
//
sRMSE = GetCurrentRMSE( qStatus );
if (theRoQ->IsQuiet() == false) {
common->Printf("sparseEncode: rmse of quad0 is %f, size is %d (meant to be %d)\n", sRMSE, GetCurrentQuadOutputSize(qStatus), fsize );
}
onf = 0;
for(i=0;i<numQuadCels;i++) {
if ( qStatus[i].size && qStatus[i].status != DEP) {
flist[onf] = qStatus[i].rsnr;
ilist[onf] = i;
onf++;
}
}
Sort( flist, ilist, onf );
Segment( ilist, flist, onf, GetCurrentRMSE( qStatus ));
temp = dxMean = dyMean = 0;
/*
for( i=0; i<numQuadCels; i++ ) {
if (qStatus[i].size && qStatus[i].status == FCC) {
dxMean += (qStatus[i].domain >> 8 ) - 128;
dyMean += (qStatus[i].domain & 0xff) - 128;
temp++;
}
}
if (temp) { dxMean /= temp; dyMean /= temp; }
*/
common->Printf("sparseEncode: dx/dy mean is %d,%d\n", dxMean, dyMean);
numredo = 0;
detail = false;
if (codebookmade && whichFrame>4) fsize -= 256;
temp = 0;
for( i=0; i<numQuadCels; i++ ) {
if ( qStatus[i].size == osize && qStatus[i].mark == false && qStatus[i].snr[MOT] > 0 ) {
GetData( idataA, qStatus[i].size, qStatus[i].xat, qStatus[i].yat, image );
if (osize == 8) VqData8( idataA, &qStatus[i] );
if (previousImage[0]) {
int dx,dy;
dx = (qStatus[i].domain >> 8 ) - 128 - dxMean + 8;
dy = (qStatus[i].domain & 0xff) - 128 - dyMean + 8;
if (dx<0||dx>15||dy<0||dy>15) {
qStatus[i].snr[FCC] = 9999;
temp++;
FvqData( idataA, qStatus[i].size, qStatus[i].xat, qStatus[i].yat, &qStatus[i], true );
dx = (qStatus[i].domain >> 8 ) - 128 - dxMean + 8;
dy = (qStatus[i].domain & 0xff) - 128 - dyMean + 8;
if ((dx<0||dx>15||dy<0||dy>15)&&qStatus[i].snr[FCC]!=9999&&qStatus[i].status==FCC) {
common->Printf("sparseEncode: something is wrong here, dx/dy is %d,%d after being clamped\n", dx, dy);
common->Printf("xat: %d\n", qStatus[i].xat);
common->Printf("yat: %d\n", qStatus[i].yat);
common->Printf("size %d\n", qStatus[i].size);
common->Printf("type: %d\n", qStatus[i].status);
common->Printf("mot: %04x\n", qStatus[i].domain);
common->Printf("motsnr: %0f\n", qStatus[i].snr[FCC]);
common->Printf("rmse: %0f\n", qStatus[i].rsnr);
common->Error("need to go away now\n");
}
}
}
LowestQuad( &qStatus[i], &qStatus[i].status, &qStatus[i].rsnr, wtype );
theRoQ->MarkQuadx( qStatus[i].xat, qStatus[i].yat, qStatus[i].size, qStatus[i].rsnr, qStatus[i].status );
/*
if (qStatus[i].status==FCC && qStatus[i].snr[FCC]>qStatus[i].snr[SLD]) {
common->Printf("sparseEncode: something is wrong here\n");
common->Printf("xat: %d\n", qStatus[i].xat);
common->Printf("yat: %d\n", qStatus[i].yat);
common->Printf("size %d\n", qStatus[i].size);
common->Printf("type: %d\n", qStatus[i].status);
common->Printf("mot: %04x\n", qStatus[i].domain);
common->Printf("motsnr: %0f\n", qStatus[i].snr[FCC]);
common->Printf("sldsnr: %0f\n", qStatus[i].snr[SLD]);
common->Printf("rmse: %0f\n", qStatus[i].rsnr);
//common->Error("need to go away now\n");
}
*/
}
}
if (theRoQ->IsQuiet() == false) {
common->Printf("sparseEncode: rmse of quad0 is %f, size is %d (meant to be %d)\n", GetCurrentRMSE( qStatus ), GetCurrentQuadOutputSize( qStatus ), fsize );
common->Printf("sparseEncode: %d outside fcc limits\n", temp);
}
onf = 0;
for(i=0;i<numQuadCels;i++) {
if ( qStatus[i].size && qStatus[i].status != DEP) {
flist[onf] = qStatus[i].rsnr;
ilist[onf] = i;
onf++;
}
}
Sort(flist, ilist, onf );
ong = 0; detail = false;
while ( GetCurrentQuadOutputSize(qStatus) < fsize && ong < onf && flist[ong] > 0 && qStatus[ilist[ong]].mark == false) {
// badsnr = [self getCurrentRMSE: qStatus];
osize = AddQuad( qStatus, ilist[ong++] );
// if ([self getCurrentRMSE: qStatus] >= badsnr) {
// break;
// }
}
if ( GetCurrentQuadOutputSize( qStatus ) < fsize) {
ong = 0;
while ( GetCurrentQuadOutputSize(qStatus) < fsize && ong < onf) {
// badsnr = [self getCurrentRMSE: qStatus];
i = ilist[ong++];
if (qStatus[i].mark) {
detail = false;
qStatus[i].mark = false;
GetData( idataA, qStatus[i].size, qStatus[i].xat, qStatus[i].yat, image );
if (qStatus[i].size == 8) VqData8( idataA, &qStatus[i] );
if (qStatus[i].size == 4) VqData4( idataA, &qStatus[i] );
if (qStatus[i].size == 4) VqData2( idataA, &qStatus[i] );
if (previousImage[0]) {
FvqData( idataA, qStatus[i].size, qStatus[i].xat, qStatus[i].yat, &qStatus[i], true );
}
LowestQuad( &qStatus[i], &qStatus[i].status, &qStatus[i].rsnr, wtype );
if (qStatus[i].rsnr <= MIN_SNR) {
break;
}
theRoQ->MarkQuadx( qStatus[i].xat, qStatus[i].yat, qStatus[i].size, qStatus[i].rsnr, qStatus[i].status );
}
// if ([self getCurrentRMSE: qStatus] >= badsnr) {
// break;
// }
}
ong = 0;
while ( GetCurrentQuadOutputSize( qStatus ) < fsize && ong < onf && flist[ong] > 0) {
// badsnr = [self getCurrentRMSE: qStatus];
i = ilist[ong++];
// if (qStatus[i].rsnr <= MIN_SNR) {
// break;
// }
detail = true;
osize = AddQuad( qStatus, i );
// if ([self getCurrentRMSE: qStatus] >= badsnr) {
// break;
// }
}
}
common->Printf("sparseEncode: rmse of frame %d is %f, size is %d\n", whichFrame, GetCurrentRMSE(qStatus), GetCurrentQuadOutputSize(qStatus) );
if (previousImage[0])
fsize = theRoQ->NormalFrameSize();
else
fsize = theRoQ->FirstFrameSize();
slop += (fsize - GetCurrentQuadOutputSize(qStatus));
if (theRoQ->IsQuiet() == false) {
for(i=0;i<DEAD;i++) num[i] = 0;
j = 0;
for( i=0; i<numQuadCels; i++ ) {
if (qStatus[i].size == 8 && qStatus[i].status) {
if (qStatus[i].status < DEAD) num[qStatus[i].status]++; j++;
}
}
common->Printf("sparseEncode: for 08x08 CCC = %d, FCC = %d, MOT = %d, SLD = %d, PAT = %d\n", num[CCC], num[FCC], num[MOT], num[SLD], num[PAT]);
for(i=0;i<DEAD;i++) num[i] = 0;
for( i=0; i<numQuadCels; i++ ) {
if (qStatus[i].size == 4 && qStatus[i].status) {
if (qStatus[i].status < DEAD) num[qStatus[i].status]++; j++;
}
}
common->Printf("sparseEncode: for 04x04 CCC = %d, FCC = %d, MOT = %d, SLD = %d, PAT = %d\n", num[CCC], num[FCC], num[MOT], num[SLD], num[PAT]);
common->Printf("sparseEncode: average RMSE = %f, numActiveQuadCels = %d, estSize = %d, slop = %d \n", GetCurrentRMSE(qStatus), j, GetCurrentQuadOutputSize(qStatus), slop);
}
theRoQ->WriteFrame( qStatus );
MakePreviousImage( qStatus );
Mem_Free(idataA);
Mem_Free(idataB);
Mem_Free(flist);
Mem_Free(ilist);
if (newImage) delete newImage;
whichFrame++;
}
void codec::EncodeNothing( void )
{
int i, j, osize, fsize, num[DEAD+1], *ilist, wtype;
float *flist, sRMSE;
byte *idataA, *idataB;
osize = 8;
image = theRoQ->CurrentImage();
newImage = 0;
pixelsHigh = image->pixelsHigh();
pixelsWide = image->pixelsWide();
dimension2 = 12; dimension4 = 48;
if (image->hasAlpha()&&(theRoQ->ParamNoAlpha() == false)) { dimension2 = 16; dimension4 = 64; }
idataA = (byte *)Mem_Alloc( 16*16*4 );
idataB = (byte *)Mem_Alloc( 16*16*4 );
if (!previousImage[0]) common->Printf("sparseEncode: sparsely encoding a %d,%d image\n", pixelsWide, pixelsHigh);
InitImages();
flist = (float *)Mem_ClearedAlloc( (numQuadCels+1) * sizeof(float) );
ilist = (int *)Mem_ClearedAlloc( (numQuadCels+1) * sizeof(int ) );
fsize = 56*1024;
if (theRoQ->NumberOfFrames()>2) {
if (previousImage[0]) fsize = theRoQ->NormalFrameSize(); else fsize = theRoQ->FirstFrameSize();
if (theRoQ->HasSound() && fsize > 6000 && previousImage[0]) fsize = 6000;
}
dxMean = dyMean = 0;
if (previousImage[0]) wtype = 1; else wtype = 0;
for( i=0; i<numQuadCels; i++ ) {
for(j=0;j<DEAD;j++) qStatus[i].snr[j] = 9999;
qStatus[i].mark = false;
if ( qStatus[i].size == osize ) {
if (previousImage[0]) {
GetData( idataA, qStatus[i].size, qStatus[i].xat, qStatus[i].yat, image );
GetData( idataB, qStatus[i].size, qStatus[i].xat, qStatus[i].yat, previousImage[whichFrame&1] );
qStatus[i].snr[MOT] = Snr( idataA, idataB, qStatus[i].size );
}
LowestQuad( &qStatus[i], &qStatus[i].status, &qStatus[i].rsnr, wtype );
if (qStatus[i].rsnr < 9999)
theRoQ->MarkQuadx( qStatus[i].xat, qStatus[i].yat, qStatus[i].size, qStatus[i].rsnr, qStatus[i].status );
} else {
if ( qStatus[i].size < osize ) {
qStatus[i].status = 0;
qStatus[i].size = 0;
} else {
qStatus[i].status = DEP;
qStatus[i].rsnr = 0;
}
}
}
//
// the quad is complete, so status can now be used for quad decomposition
// the first thing to do is to set it up for all the 4x4 cels to get output
// and then recurse from there to see what's what
//
sRMSE = GetCurrentRMSE( qStatus );
common->Printf("sparseEncode: rmse of frame %d is %f, size is %d\n", whichFrame, sRMSE, GetCurrentQuadOutputSize( qStatus ) );
if (theRoQ->IsQuiet() == false) {
for(i=0;i<DEAD;i++) num[i] = 0;
j = 0;
for( i=0; i<numQuadCels; i++ ) {
if (qStatus[i].size == 8 && qStatus[i].status) {
if (qStatus[i].status < DEAD) num[qStatus[i].status]++; j++;
}
}
common->Printf("sparseEncode: for 08x08 CCC = %d, FCC = %d, MOT = %d, SLD = %d, PAT = %d\n", num[CCC], num[FCC], num[MOT], num[SLD], num[PAT]);
for(i=0;i<DEAD;i++) num[i] = 0;
for( i=0; i<numQuadCels; i++ ) {
if (qStatus[i].size == 4 && qStatus[i].status) {
if (qStatus[i].status < DEAD) num[qStatus[i].status]++; j++;
}
}
common->Printf("sparseEncode: for 04x04 CCC = %d, FCC = %d, MOT = %d, SLD = %d, PAT = %d\n", num[CCC], num[FCC], num[MOT], num[SLD], num[PAT]);
common->Printf("sparseEncode: average RMSE = %f, numActiveQuadCels = %d, estSize = %d \n", GetCurrentRMSE(qStatus), j, GetCurrentQuadOutputSize(qStatus));
}
theRoQ->WriteFrame( qStatus );
MakePreviousImage( qStatus );
Mem_Free(idataA);
Mem_Free(idataB);
Mem_Free(flist);
Mem_Free(ilist);
if (newImage) delete newImage;
whichFrame++;
}
void codec::VQ( const int numEntries, const int dimension, const unsigned char *vectors, float *import, VQDATA **codebook, const bool optimize ) {
int startMsec = Sys_Milliseconds();
if (numEntries <= 256) {
//
// copy the entries into the codebooks
//
for( int i=0; i<numEntries; i++ ) {
for( int j=0; j<dimension; j++ ) {
codebook[i][j] = vectors[j+i*dimension];
}
}
return;
}
//
// okay, we need to wittle this down to less than 256 entries
//
// get rid of identical entries
int i, j, x, ibase, jbase;
bool *inuse = (bool *)_alloca( numEntries * sizeof(bool) );
float *snrs = (float *)_alloca( numEntries * sizeof(float) );
int *indexes = (int *)_alloca( numEntries * sizeof(int) );
int *indexet = (int *)_alloca( numEntries * sizeof(int) );
int numFinalEntries = numEntries;
for( i=0; i<numEntries; i++ ) {
inuse[i] = true;
snrs[i] = -1.0f;
indexes[i] = -1;
indexet[i] = -1;
}
for( i=0; i<numEntries-1; i++ ) {
for( j=i+1; j<numEntries; j++ ) {
if (inuse[i] && inuse[j]) {
if (!memcmp( &vectors[i*dimension], &vectors[j*dimension], dimension)) {
inuse[j] = false;
numFinalEntries--;
import[i] += import[j];
}
}
}
}
common->Printf("VQ: has %d entries to process\n", numFinalEntries );
//
// are we done?
//
int end;
if (numFinalEntries > 256) {
//
// find the closest two and eliminate one
//
double bestDist = HUGE_VAL;
2020-06-12 21:06:25 +00:00
double dist, simport;
int bestIndex = -1;
int bestOtherIndex = 0;
int aentries = 0;
for( i=0; i<numEntries-1; i++ ) {
if (inuse[i]) {
end = numEntries;
if (optimize) {
if (numFinalEntries>8192) {
end = i+32;
} else if (numFinalEntries>4096) {
end = i+64;
} else if (numFinalEntries>2048) {
end = i+128;
} else if (numFinalEntries>1024) {
end = i+256;
} else if (numFinalEntries>512) {
end = i+512;
}
if (end>numEntries) {
end = numEntries;
}
}
ibase = i*dimension;
for( j=i+1; j<end; j++ ) {
if (inuse[j]) {
dist = 0.0;
jbase = j*dimension;
for( x=0; x<dimension; x+=3 ) {
#if 0
r0 = (float)vectors[ibase+x];
r1 = (float)vectors[jbase+x];
g0 = (float)vectors[ibase+x+1];
g1 = (float)vectors[jbase+x+1];
b0 = (float)vectors[ibase+x+2];
b1 = (float)vectors[jbase+x+2];
dist += idMath::Sqrt16( (r0-r1)*(r0-r1) + (g0-g1)*(g0-g1) + (b0-b1)*(b0-b1) );
#else
// JDC: optimization
int dr = vectors[ibase+x] - vectors[jbase+x];
int dg = vectors[ibase+x+1] - vectors[jbase+x+1];
int db = vectors[ibase+x+2] - vectors[jbase+x+2];
dist += idMath::Sqrt16( dr * dr + dg * dg + db * db );
#endif
}
simport = import[i] * import[j];
dist *= simport;
if ( dist < bestDist ) {
bestDist = dist;
bestIndex = i;
bestOtherIndex = j;
}
}
}
snrs[aentries] = bestDist;
indexes[aentries] = bestIndex;
indexet[aentries] = bestOtherIndex;
aentries++;
}
}
//
// until we have reduced it to 256 entries, find one to toss
//
do {
bestDist = HUGE_VAL;
2020-06-12 21:06:25 +00:00
bestIndex = -1;
bestOtherIndex = -1;
if (optimize) {
for( i=0; i<aentries; i++ ) {
if (inuse[indexes[i]] && inuse[indexet[i]] ) {
if ( snrs[i] < bestDist ) {
bestDist = snrs[i];
bestIndex = indexes[i];
bestOtherIndex = indexet[i];
}
}
}
}
if (bestIndex == -1 || !optimize) {
bestDist = HUGE_VAL;
2020-06-12 21:06:25 +00:00
bestIndex = -1;
bestOtherIndex = 0;
aentries = 0;
for( i=0; i<numEntries-1; i++ ) {
if (!inuse[i]) {
continue;
}
end = numEntries;
if (optimize) {
if (numFinalEntries>8192) {
end = i+32;
} else if (numFinalEntries>4096) {
end = i+64;
} else if (numFinalEntries>2048) {
end = i+128;
} else if (numFinalEntries>1024) {
end = i+256;
} else if (numFinalEntries>512) {
end = i+512;
}
}
if (end>numEntries) {
end = numEntries;
}
ibase = i*dimension;
for( j=i+1; j<end; j++ ) {
if ( !inuse[j]) {
continue;
}
dist = 0.0;
jbase = j*dimension;
simport = import[i] * import[j];
float scaledBestDist = bestDist / simport;
for( x=0; x<dimension; x+=3 ) {
#if 0
r0 = (float)vectors[ibase+x];
r1 = (float)vectors[jbase+x];
g0 = (float)vectors[ibase+x+1];
g1 = (float)vectors[jbase+x+1];
b0 = (float)vectors[ibase+x+2];
b1 = (float)vectors[jbase+x+2];
dist += idMath::Sqrt16( (r0-r1)*(r0-r1) + (g0-g1)*(g0-g1) + (b0-b1)*(b0-b1) );
#else
// JDC: optimization
int dr = vectors[ibase+x] - vectors[jbase+x];
int dg = vectors[ibase+x+1] - vectors[jbase+x+1];
int db = vectors[ibase+x+2] - vectors[jbase+x+2];
dist += idMath::Sqrt16( dr * dr + dg * dg + db * db );
if ( dist > scaledBestDist ) {
break;
}
#endif
}
dist *= simport;
if ( dist < bestDist ) {
bestDist = dist;
bestIndex = i;
bestOtherIndex = j;
}
}
snrs[aentries] = bestDist;
indexes[aentries] = bestIndex;
indexet[aentries] = bestOtherIndex;
aentries++;
}
}
//
// and lose one
//
inuse[bestIndex] = false;
numFinalEntries--;
import[bestOtherIndex] += import[bestIndex];
if ((numFinalEntries&511)==0) {
common->Printf("VQ: has %d entries to process\n", numFinalEntries );
session->UpdateScreen();
}
} while (numFinalEntries > 256);
}
//
// copy the entries into the codebooks
//
int onEntry = 0;
for( i=0; i<numEntries; i++ ) {
if (inuse[i]) {
ibase = i*dimension;
for( x=0; x<dimension; x++ ) {
codebook[onEntry][x] = vectors[ibase + x];
}
if (onEntry == 0) {
common->Printf("First vq = %d\n ", i);
}
if (onEntry == 255) {
common->Printf("last vq = %d\n", i);
}
onEntry++;
}
}
int endMsec = Sys_Milliseconds();
common->Printf( "VQ took %i msec\n", endMsec - startMsec );
}