// SPIHTCoder.cpp: implementation of the CSPIHTCoder class.
//
//////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "SPIHTCoder.h"
#include "mainfrm.h"
#include "ImgPro.h"
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
CSPIHTCoder::CSPIHTCoder()
{
strFileNameOut = "c:\\out.spi";
strFileNameIn = "c:\\out.spi";
pNewlyAppended = NULL;
cAction = TO_DISK;
SPIHTFile = NULL;
MR = NULL;
M = NULL;
LIS = NULL;
LIP = NULL;
LSP = NULL;
CompressedFileInfo.pFirstByte = NULL;
pByte = NULL;
}
CSPIHTCoder::~CSPIHTCoder()
{
if(LSP!=NULL)
delete LSP;
if(LIP!=NULL)
delete LIP;
if(LIS!=NULL)
delete LIS;
if(M!=NULL)
delete M;
if(MR!=NULL)
delete MR;
/* if(pByte!=NULL){
delete[] pByte;
pByte = NULL;
}*/
}
void CSPIHTCoder::PutBit(char bit)
{
static unsigned long lTotalBits = (unsigned long)(nYDim*nXDim*BitRate);
if((zeroes+ones)>lTotalBits){
stop = 1;
return;
}
if (bit==1){
outputbyte |= mask;
ones++;
}
else
zeroes++;
mask >>= 1;
static long lByteCount = -1;
if (mask==0) {
lByteCount++;
*pByte = outputbyte;
pByte++;
outputbyte = 0;
mask = 0x80;
}
}
void CSPIHTCoder::DumpBuffer()
{
if(mask!=0)
*pByte = outputbyte;
}
void CSPIHTCoder::Initialization()
{
/* Create and initialize list LSP */
CLinkList::ListType d;
LSP = new CLinkList;
LSP->Create();
/* Create and initialize list LIP
LIP = new CLinkList;
LIP->Create();
int i,j;
for(j=0;j<(nYDim>>level);j++)
for(i=0;i<(nXDim>>level);i++){
d.x = i;
d.y = j;
LIP->Append(d);
} */
/* Create and initialize list LIP*/
LIP = new CLinkList;
LIP->Create();
d.x = 0;
d.y = 0;
LIP->Append(d);
d.x = 1;
d.y = 0;
LIP->Append(d);
d.x = 0;
d.y = 1;
LIP->Append(d);
d.x = 1;
d.y = 1;
LIP->Append(d);
/* Create and initialize list LIS
LIS = new CLinkList;
LIS->Create();
for(j=0;j<(nYDim>>(level+1));j++){
for(i=(nXDim>>(level+1));i<(nXDim>>level);i++){
d.x = i;
d.y = j;
d.type = TYPE_A;
LIS->Append(d);
}
}
for(j=(nYDim>>(level+1));j<(nYDim>>level);j++){
for(i=0;i<(nXDim>>(level+1));i++){
d.x = i;
d.y = j;
d.type = TYPE_A;
LIS->Append(d);
}
}
for(j=(nYDim>>(level+1));j<(nYDim>>level);j++){
for(i=(nXDim>>(level+1));i<(nXDim>>level);i++){
d.x = i;
d.y = j;
d.type = TYPE_A;
LIS->Append(d);
}
} */
/* Create and initialize list LIS */
LIS = new CLinkList;
LIS->Create();
d.x = 1;
d.y = 0;
d.type = TYPE_A;
LIS->Append(d);
d.x = 0;
d.y = 1;
LIS->Append(d);
d.x = 1;
d.y = 1;
LIS->Append(d);
}
BOOL CSPIHTCoder::ZeroTree(CMatrix2D<ElementType> *m, int x, int y, int threshold)
{
int i, j, min_x, max_x, min_y, max_y;
ElementType temp, max;
char stop;
stop = 0;
if ((x==0) && (y==0)) {
temp = m->m[0][0];
m->m[0][0] = 0;
max = m->MaxMagnitude();
m->m[0][0] = temp;
if (max>=threshold) stop = 1;
}
else {
min_x = x << 1;
min_y = y << 1;
max_x = (x+1) << 1;
max_y = (y+1) << 1;
if ((min_x==m->col) || (min_y==m->row))
{
return (1);
}
max = 0;
while ((max_y<=m->row) && (max_x<=m->col)) {
for (i=min_y; i<max_y; i++) {
for (j=min_x; j<max_x; j++) {
temp = abs(m->m[i][j]);
if (temp>=threshold) {
stop = 1;
break;
}
}
if (stop==1) break;
}
if (stop==1) break;
min_x <<= 1;
max_x <<= 1;
min_y <<= 1;
max_y <<= 1;
}
}
if (stop==1) return (0);
return (1);//original
//return (0);
}
BOOL CSPIHTCoder::TestSubset(CMatrix2D<ElementType> *m, int x, int y, int threshold, unsigned char type)
{
int i, j, stop;
stop = 0;
if(type==TYPE_A)
return(!ZeroTree(m,x,y,threshold));
else
{
for(i=2*x; i<2*x+2; i++)
{
for(j=2*y; j<2*y+2; j++)
{
if(!ZeroTree(m,i,j,threshold))
{
stop = 1;
break;
}
}
if(stop==1)
break;
}
if(stop==1)
return 1;
else
return 0;
}
}
void CSPIHTCoder::SortingPass1(CMatrix2D<ElementType> *m, int threshold)
{
int temp, s, i, j;
char found;
CLinkList::ListType d, d1;
if(stop==1)
return;
pNewlyAppended = LSP->end;
/* Deal with elements in the LIP */
LIP->Reset();
while(LIP->current!=NULL)
{
d = LIP->GetCurrentElement(&found);
if(found==0){
MessageBox(NULL,"Can't retrieve current list element..",
"Warning!",MB_OK);
return;
}
temp = m->m[d.y][d.x];
if(abs(temp)<threshold) {
PutBit(0);
if(stop==1)
return;
LIP->current = LIP->current->next;
}
else
{
PutBit(1);
if(temp>0)
PutBit(1);
else
PutBit(0);
if(stop==1)
return;
LIP->RemoveCurrentElement();
LSP->Append(d);
}
}
LIP->Reset();
/* Deal with elements in the LIS */
LIS->Reset();
while(LIS->current!=NULL)
{
d = LIS->GetCurrentElement(&found);
if(found==0){
MessageBox(NULL,"Can't retrieve current list element.",
NULL,MB_OK);
return;
}
s = TestSubset(m,d.x,d.y,threshold,d.type);
if(d.type==TYPE_A)
{
PutBit((char)s);
if(stop==1)
return;
if(s==1)
{
for(j=2*d.y; j<2*d.y+2; j++) {
for(i=2*d.x; i<2*d.x+2; i++)
{
d1.x = i;
d1.y = j;
temp = m->m[j][i];
if(abs(temp)>=threshold)
{
PutBit(1);
if(stop==1)
return;
LSP->Append(d1);
if(temp>0)
PutBit(1);
else
PutBit(0);
if(stop==1)
return;
}
else {
PutBit(0);
if(stop==1)
return;
LIP->Append(d1);
}
}
}
if(4*d.x+4<=m->col && 4*d.y+4<=m->row)
{
d1 = d;
d1.type = TYPE_B;
LIS->Append(d1);
}
LIS->RemoveCurrentElement();
}
else
LIS->current = LIS->current->next;
}
else
{
PutBit((char)s);
if(stop==1)
return;
if(s==1)
{
for(j=2*d.y;j<2*d.y+2;j++)
for(i=2*d.x;i<2*d.x+2;i++)
{
d1.x = i;
d1.y = j;
d1.type = TYPE_A;
LIS->Append(d1);
}
LIS->RemoveCurrentElement();
}
else
LIS->current = LIS->current->next;
}
}
LIS->Reset();
}
void CSPIHTCoder::RefinementPass1(CMatrix2D<ElementType> *m, int threshold)
{
CLinkList::ListElement *p;
int temp;
if(pNewlyAppended==NULL)
return;
p = LSP->head;
while(p!=pNewlyAppended->next)
{
temp = threshold & abs(m->m[p->data.y][p->data.x]);
if(temp==0)
PutBit(0);
else
PutBit(1);
if(stop==1)
return;
p = p->next;
}
}
void CSPIHTCoder::Encode(ElementType *example)
{
int threshold, pass_count;
/*
* Build work matrix.
*/
header.height = nXDim;
header.width = nYDim;
header.BitRate = BitRate;
header.level = level;
header.selection = selection;
M = new CMatrix2D<ElementType>;
M->Create(header.height,header.width);
if (M==NULL) {
MessageBox(NULL,"Failed to create matrix!","Warning!",MB_OK);
return;
}
M->LoadData(example,nYDim,nXDim);
/*
* Prepare output file.
*/
int nInit = (int)(floor(log10(M->MaxMagnitude())/log10(2)));
header.threshold = 1 << nInit;
threshold = header.threshold;
CompressedFileInfo.header = header;
CompressedFileInfo.lCounts = int(ceil(BitRate*nXDim*nYDim/8.0));
CompressedFileInfo.pFirstByte = new unsigned char[CompressedFileInfo.lCounts];
pByte = CompressedFileInfo.pFirstByte;
outputbyte = 0;
ma
