同态滤波程序

同态滤波程序
void CPro2View::OnMixfilter()
{ pFile=new CFile("e：
\\picture\\za1.bmp",CFile::modeReadWrite|CFile::typeBinary);
Dib.Read(pFile);
Dib.MakePalette();
Filter(&Dib); //调用滤波模块
pFile->Close();
Invalidate();
}
BOOL CPro2View::Filter(CDib* pDib)
{ CPro2Doc* pDoc=GetDocument(); //获取文档类指针
int i,j;
int m,n;
double nRadius=100.0; //滤波半径
int nGet=0;
double H=0.0;
int nWidth=pDib->m_lpBMIH->biWidth;
int nHeight=pDib->m_lpBMIH->biHeight;
switch(pDib->m_lpBMIH->biBitCount)
{
case 24：
m_nRGBBit=3;
break;
case 32：
m_nRGBBit=4;
break;
default：
m_nRGBBit=3;
}
CSize sizeImageSave=pDib->GetDibSaveDim(); //图像总体数据尺寸
double *pDbTemp;
BYTE *pBits;
int nRGB=0;
if(!m_pDbImage)
{ m_pDbImage=new double[nWidth*nHeight];
if(!m_pDbImage) return FALSE;
}
for(nRGB=0;nRGB<3;nRGB++)
{ for(j=0;j<nHeight;j++) //获取将要进行处理的图像数据
{ pDbTemp=m_pDbImage+j*nWidth;
pBits=pDib->m_lpImage+(nHeight-j-1)*sizeImageSave.cx+nRGB;
for(i=0;i<nWidth;i++)
{ pDbTemp[i]=(double)(*(pBits+i*m_nRGBBit)); //取出像素灰度值
}
}
for(m=0;m<nHeight;m++) //进行对数变换
{ pDbTemp=m_pDbImage+m*nWidth;
for(n=0;n<nWidth;n++)
{ if(pDbTemp[n]==0)
continue;
pDbTemp[n]=log(pDbTemp[n]);
}
}
//FFT变换
double dTmpOne;
double dTmpTwo;
int nTransWidth;
int nTransHeight;
//将宽变为4的整数倍
dTmpOne=log(nWidth)/log(2);
dTmpTwo=ceil(dTmpOne);
dTmpTwo=pow(2,dTmpTwo);
nTransWidth=(int)dTmpTwo;
//将高变为4的整数倍
dTmpOne=log(nHeight)/log(2);
dTmpTwo=ceil(dTmpOne);
dTmpTwo=pow(2,dTmpTwo);
nTransHeight=(int)dTmpTwo;
complex<double>* pCTData;
complex<double>* pCFData;
pCTData=new complex<double>[nTransWidth*nTransHeight];
pCFData=new complex<double>[nTransWidth*nTransHeight];
for(m=0;m<nTransHeight;m++)//时域复数清零
{ for(n=0;n<nTransWidth;n++)
{ pCTData[m*nTransWidth+n]=complex<double>(0,0);
}
}
for(m=0;m<nHeight;m++) //给时域复数赋值
{ pDbTemp=m_pDbImage+m*nWidth;
for(n=0;n<nWidth;n++)
{ pCTData[m*nTransWidth+n]=complex<double>(pDbTemp[n],0);
}
}
pDoc->FFT_2D(pCTData,nWidth,nHeight,pCFData); //调用FFT处理
for(m=0;m<nTransHeight;m++) //滤波
{ for(n=0;n<nTransWidth;n++)
{
nGet=m*m+n*n;
H=(double)nGet/(nGet+nRadius*nRadius);
pCFData[m*nTransWidth+n]=
complex<double>(H*pCFData[m*nTransWidth+n].real(),H*pCFData[m*nTrans Width+n].imag());
}
}
pDoc->IFFT_2D(pCFData,nWidth,nHeight,pCTData); //进行IFFT处理
double dTemp;
for(m=0;m<nHeight;m++)
{ pDbTemp=m_pDbImage+m*nWidth; //调整处理数据指针
for(n=0;n<nWidth;n++)
{
dTemp=pCTData[m*nTransWidth+n].real()*pCTData[m*nTransWidth+n].real() +pCTData[m*nTransWidth+n].imag()*pCTData[m*nTransWidth+n].imag();
dTemp=exp(dTemp); //2的幂变换
dTemp=sqrt(dTemp)/600;
dTemp=min(dTemp,255);
pDbTemp[n]=dTemp;
}
}
for(j=0;j<nHeight;j++)
{
pDbTemp=m_pDbImage+j*nWidth;
pBits=pDib->m_lpImage+(nHeight-j-1)*sizeImageSave.cx+nRGB;
for(i=0;i<nWidth;i++)
{ *(pBits+i*m_nRGBBit)=FloatToByte(pDbTemp[i]); //修改结果数据
}
}
}
return TRUE;
}。