#include "stdafx.h"
|
#include "pyramid.h"
|
#include <stdlib.h>
|
#include <memory.h>
|
|
CPyramid::CPyramid()
|
{
|
layerNum=0;
|
ImgLayer=NULL;
|
}
|
|
void CPyramid::Init(int s_row, int s_col, int levelNum)
|
{
|
int i,j,k;
|
|
//new pyramid
|
this->ImgLayer = new CMatrix[levelNum];
|
this->layerNum = levelNum;
|
|
//calculate level
|
int delta=1;
|
int level=levelNum-1;
|
|
//memory allocation
|
// for(i=0;i<levelNum;i++)
|
{
|
int row = s_row-delta+1;
|
int col = s_col-delta+1;
|
int len = row*col;
|
|
int * buf1D = (int*) malloc( len * sizeof(int));
|
buf1Dtemp=(int*) malloc( len * sizeof(int));
|
|
int** buf2D = (int**) malloc( row * sizeof(int*));
|
buf2Dtemp = (int**) malloc( row * sizeof(int*));
|
|
for(j=0;j<row;j++)
|
{
|
buf2D[j] = buf1D + col * j;
|
buf2Dtemp[j]= buf1Dtemp + col * j;
|
}
|
|
this->ImgLayer[level].Array1D= buf1D;
|
this->ImgLayer[level].Array2D= buf2D;
|
|
this->ImgLayer[level].row = row;
|
this->ImgLayer[level].col = col;
|
this->ImgLayer[level].len = len;
|
|
delta*=2;
|
level--;
|
}
|
|
|
for(k=1; k<levelNum; k++)
|
{
|
int row = s_row-delta+1;
|
int col = s_col-delta+1;
|
int len = row*col;
|
|
int * buf1D = (int*) malloc( len * sizeof(int));
|
|
int** buf2D = (int**) malloc( row * sizeof(int*));
|
|
for(i=0; i<row; i++)
|
{
|
buf2D[i]= buf1D + col * i;
|
}
|
|
|
this->ImgLayer[level].Array1D= buf1D;
|
this->ImgLayer[level].Array2D= buf2D;
|
|
this->ImgLayer[level].row = row;
|
this->ImgLayer[level].col = col;
|
this->ImgLayer[level].len = len;
|
|
delta*=2;
|
level--;
|
}
|
}
|
|
|
void CPyramid::BuildPyd(unsigned char* Source, int s_row, int s_col, int levelNum)
|
{
|
int i,j,k;
|
|
//calculate level
|
int delta=1;
|
int level=levelNum-1;
|
|
//memory allocation
|
// for(i=0;i<levelNum;i++)
|
{
|
int row = s_row-delta+1;
|
int col = s_col-delta+1;
|
int len = row*col;
|
|
for(i=0;i<len;i++)
|
{
|
this->ImgLayer[level].Array1D[i] = Source[i]*Source[i];
|
}
|
|
delta = delta<<1;
|
level--;
|
}
|
|
|
for(k=1; k<levelNum; k++)
|
{
|
int** preLevel2D=this->ImgLayer[level+1].Array2D;
|
|
int row = s_row-delta+1;
|
int col = s_col-delta+1;
|
int len = row*col;
|
|
memset(buf1Dtemp,0,s_row*s_col*sizeof(int));
|
|
int pre_delta=delta>>1;
|
for(i=0; i<this->ImgLayer[level+1].row;i++)
|
for(j=0; j<col; j++)
|
{
|
buf2Dtemp[i][j]=preLevel2D[i][j]+preLevel2D[i][j+pre_delta];
|
}
|
for(i=0; i<row; i++)
|
for(j=0; j<col; j++)
|
{
|
ImgLayer[level].Array2D[i][j]=buf2Dtemp[i][j]+buf2Dtemp[i+pre_delta][j];
|
}
|
|
delta = delta<<1;
|
level--;
|
}
|
|
}
|
|
|
void CPyramid::BuildPyd(int* Source, int s_row, int s_col, int levelNum)//, unsigned int *operation)
|
{
|
int i,j,k;
|
|
int delta=1;
|
int level=levelNum-1;
|
|
//memory allocation
|
// for(i=0;i<levelNum;i++)
|
{
|
int row = s_row-delta+1;
|
int col = s_col-delta+1;
|
int len = row*col;
|
|
for(i=0;i<len;i++)
|
{
|
this->ImgLayer[level].Array1D[i] = Source[i]*Source[i];
|
}
|
|
delta*=2;
|
level--;
|
}
|
|
|
for(k=1; k<levelNum; k++)
|
{
|
int** preLevel2D=this->ImgLayer[level+1].Array2D;
|
|
int row = s_row-delta+1;
|
int col = s_col-delta+1;
|
int len = row*col;
|
|
memset(buf1Dtemp,0,s_row*s_col*sizeof(int));
|
|
int pre_delta=delta>>1;
|
for(i=0; i<this->ImgLayer[level+1].row;i++)
|
for(j=0; j<col; j++)
|
{
|
buf2Dtemp[i][j]=preLevel2D[i][j]+preLevel2D[i][j+pre_delta];
|
}
|
for(i=0; i<row; i++)
|
for(j=0; j<col; j++)
|
{
|
ImgLayer[level].Array2D[i][j]=buf2Dtemp[i][j]+buf2Dtemp[i+pre_delta][j];
|
}
|
|
delta = delta<<1;
|
level--;
|
}
|
}
|
|
void CPyramid::ClearPyd(void)
|
{
|
int level=this->layerNum;
|
for(int i=0; i<level; i++)
|
{
|
free( this->ImgLayer[i].Array1D );
|
free( this->ImgLayer[i].Array2D );
|
}
|
free(buf1Dtemp);
|
free(buf2Dtemp);
|
}
|
|
//=================================================================
|
//=================================================================
|
|
void ClearPyd(CPyramid *p)
|
{
|
int level=p->layerNum;
|
for(int i=0; i<level; i++)
|
{
|
free( p->ImgLayer[i].Array1D );
|
free( p->ImgLayer[i].Array2D );
|
}
|
}
|
