?? restore.cpp
字號:
//對結果圖像進行反FFT
fourn(fftSrc,nn,2,-1);
//確定歸一化因子
MaxNum = 0.0;
for (j = 0;j < lHeight ;j++)
{
for(i = 0;i < lLineBytes ;i++)
{
fftSrc[(2*lLineBytes)*j + 2*i + 1] =
sqrt(fftSrc[(2*lLineBytes)*j + 2*i + 1] * fftSrc[(2*lLineBytes)*j + 2*i + 1]\
+fftSrc[(2*lLineBytes)*j + 2*i + 2] * fftSrc[(2*lLineBytes)*j + 2*i + 2]);
if( MaxNum < fftSrc[(2*lLineBytes)*j + 2*i + 1])
MaxNum = fftSrc[(2*lLineBytes)*j + 2*i + 1];
}
}
//轉換為圖像，加噪
char point;
for (j = 0;j < lHeight ;j++)
{
for(i = 0;i < lLineBytes ;i++)
{
if ( i + j == ((int)((i+j)/8))*8)
{
point = -16;
}
else
{
point = 0;
}
// 指向源圖像倒數第j行，第i個象素的指針
lpSrc = (char *)lpDIBBits + lLineBytes * j + i;
*lpSrc = (unsigned char) (fftSrc[(2*lLineBytes)*j + 2*i + 1]*255.0/MaxNum + point);
}
}
delete fftSrc;
delete fftKernel;
// 返回
return true;
}
/*************************************************************************
*
* 函數名稱：
* WienerDIB()
*
* 參數:
* LPSTR lpDIBBits - 指向源DIB圖像指針
* LONG lWidth - 源圖像寬度（象素數）
* LONG lHeight - 源圖像高度（象素數）
*
* 返回值:
* BOOL - 維納濾波復原操作成功返回TRUE，否則返回FALSE。
*
* 說明:
* 該函數用來對DIB圖像進行維納濾波復原操作。
*
************************************************************************/
BOOL WINAPI WienerDIB (LPSTR lpDIBBits, LONG lWidth, LONG lHeight)
{
// 指向源圖像的指針
LPSTR lpSrc;
//循環變量
long i;
long j;
//像素值
unsigned char pixel;
// 圖像每行的字節數
LONG lLineBytes;
//用于做FFT的數組
double *fftSrc,*fftKernel,*fftNoise;
double a,b,c,d,e,f,multi;
//二維FFT的長度和寬度
unsigned long nn[3];
//圖像歸一化因子
double MaxNum;
// 計算圖像每行的字節數
lLineBytes = WIDTHBYTES(lWidth * 8);
double dPower = log((double)lLineBytes)/log(2.0);
if(dPower != (int) dPower)
{
return false;
}
dPower = log((double)lHeight)/log(2.0);
if(dPower != (int) dPower)
{
return false;
}
fftSrc = new double [lHeight*lLineBytes*2+1];
fftKernel = new double [lHeight*lLineBytes*2+1];
fftNoise = new double [lHeight*lLineBytes*2+1];
nn[1] = lHeight;
nn[2] = lLineBytes;
for (j = 0;j < lHeight ;j++)
{
for(i = 0;i < lLineBytes ;i++)
{
// 指向源圖像倒數第j行，第i個象素的指針
lpSrc = (char *)lpDIBBits + lLineBytes * j + i;
pixel = (unsigned char)*lpSrc;
fftSrc[(2*lLineBytes)*j + 2*i + 1] = (double)pixel;
fftSrc[(2*lLineBytes)*j + 2*i + 2] = 0.0;
if(i < 5 && j == 0)
{
fftKernel[(2*lLineBytes)*j + 2*i + 1] = 1/5.0;
}
else
{
fftKernel[(2*lLineBytes)*j + 2*i + 1] = 0.0;
}
fftKernel[(2*lLineBytes)*j + 2*i + 2] = 0.0;
if ( i + j == ((int)((i+j)/8))*8)
{
fftNoise [(2*lLineBytes)*j + 2*i + 1]= -16.0;
}
else
{
fftNoise [(2*lLineBytes)*j + 2*i + 1]= 0.0;
}
fftNoise[(2*lLineBytes)*j + 2*i + 2] = 0.0;
}
}
srand((unsigned)time(NULL));
//對源圖像進行FFT
fourn(fftSrc,nn,2,1);
//對卷積核圖像進行FFT
fourn(fftKernel,nn,2,1);
//對噪聲圖像進行FFT
fourn(fftNoise,nn,2,1);
for (i = 1;i <lHeight*lLineBytes*2;i+=2)
{
a = fftSrc[i];
b = fftSrc[i+1];
c = fftKernel[i];
d = fftKernel[i+1];
e = fftNoise[i];
f = fftNoise[i+1];
multi = (a*a + b*b)/(a*a + b*b - e*e - f*f);
if (c*c + d*d > 1e-3)
{
fftSrc[i] = ( a*c + b*d ) / ( c*c + d*d ) / multi;
fftSrc[i+1] = ( b*c - a*d ) / ( c*c + d*d )/multi;
}
}
//對結果圖像進行反FFT
fourn(fftSrc,nn,2,-1);
//確定歸一化因子
MaxNum = 0.0;
for (j = 0;j < lHeight ;j++)
{
for(i = 0;i < lLineBytes ;i++)
{
fftSrc[(2*lLineBytes)*j + 2*i + 1] =
sqrt(fftSrc[(2*lLineBytes)*j + 2*i + 1] * fftSrc[(2*lLineBytes)*j + 2*i + 1]\
+fftSrc[(2*lLineBytes)*j + 2*i + 2] * fftSrc[(2*lLineBytes)*j + 2*i + 2]);
if( MaxNum < fftSrc[(2*lLineBytes)*j + 2*i + 1])
MaxNum = fftSrc[(2*lLineBytes)*j + 2*i + 1];
}
}
//轉換為圖像
for (j = 0;j < lHeight ;j++)
{
for(i = 0;i < lLineBytes ;i++)
{
// 指向源圖像倒數第j行，第i個象素的指針
lpSrc = (char *)lpDIBBits + lLineBytes * j + i;
*lpSrc = (unsigned char) (fftSrc[(2*lLineBytes)*j + 2*i + 1]*255.0/MaxNum );
}
}
delete fftSrc;
delete fftKernel;
delete fftNoise;
// 返回
return true;
}
BOOL fourn(double * data, unsigned long nn[], int ndim, int isign)
{
int idim;
unsigned long i1,i2,i3,i2rev,i3rev,ip1,ip2,ip3,ifp1,ifp2;
unsigned long ibit,k1,k2,n,nprev,nrem,ntot;
double tempi,tempr;
double theta,wi,wpi,wpr,wr,wtemp;
for (ntot=1,idim=1;idim<=ndim;idim++)
ntot *= nn[idim];
nprev=1;
for (idim=ndim;idim>=1;idim--) {
n=nn[idim];
nrem=ntot/(n*nprev);
ip1=nprev << 1;
ip2=ip1*n;
ip3=ip2*nrem;
i2rev=1;
for (i2=1;i2<=ip2;i2+=ip1) {
if (i2 < i2rev) {
for (i1=i2;i1<=i2+ip1-2;i1+=2) {
for (i3=i1;i3<=ip3;i3+=ip2) {
i3rev=i2rev+i3-i2;
SWAP(data[i3],data[i3rev]);
SWAP(data[i3+1],data[i3rev+1]);
}
}
}
ibit=ip2 >> 1;
while (ibit >= ip1 && i2rev > ibit) {
i2rev -= ibit;
ibit >>= 1;
}
i2rev += ibit;
}
ifp1=ip1;
while (ifp1 < ip2) {
ifp2=ifp1 << 1;
theta=isign*6.28318530717959/(ifp2/ip1);
wtemp=sin(0.5*theta);
wpr = -2.0*wtemp*wtemp;
wpi=sin(theta);
wr=1.0;
wi=0.0;
for (i3=1;i3<=ifp1;i3+=ip1) {
for (i1=i3;i1<=i3+ip1-2;i1+=2) {
for (i2=i1;i2<=ip3;i2+=ifp2) {
k1=i2;
k2=k1+ifp1;
tempr=wr*data[k2]-wi*data[k2+1];
tempi=wr*data[k2+1]+wi*data[k2];
data[k2]=data[k1]-tempr;
data[k2+1]=data[k1+1]-tempi;
data[k1] += tempr;
data[k1+1] += tempi;
}
}
wr=(wtemp=wr)*wpr-wi*wpi+wr;
wi=wi*wpr+wtemp*wpi+wi;
}
ifp1=ifp2;
}
nprev *= n;
}
return true;
}
/*************************************************************************
*
* 函數名稱：
* RandomNoiseDIB()
*
* 參數:
* LPSTR lpDIBBits - 指向源DIB圖像指針
* LONG lWidth - 源圖像寬度（象素數，必須是4的倍數）
* LONG lHeight - 源圖像高度（象素數）
*
* 返回值:
* BOOL - 模糊操作成功返回TRUE，否則返回FALSE。
*
* 說明:
* 該函數用來對DIB圖像進行模糊操作。
*
************************************************************************/
BOOL WINAPI RandomNoiseDIB (LPSTR lpDIBBits, LONG lWidth, LONG lHeight)
{
// 指向源圖像的指針
LPSTR lpSrc;
//循環變量
long i;
long j;
// 圖像每行的字節數
LONG lLineBytes;
//像素值
unsigned char pixel;
//噪聲
BYTE NoisePoint;
// 計算圖像每行的字節數
lLineBytes = WIDTHBYTES(lWidth * 8);
//生成偽隨機種子
srand((unsigned)time(NULL));
//在圖像中加噪
for (j = 0;j < lHeight ;j++)
{
for(i = 0;i < lLineBytes ;i++)
{
NoisePoint=rand()/1024;
// 指向源圖像倒數第j行，第i個象素的指針
lpSrc = (char *)lpDIBBits + lLineBytes * j + i;
//取得像素值
pixel = (unsigned char)*lpSrc;
*lpSrc = (unsigned char)(pixel*224/256 + NoisePoint);
}
}
// 返回
return true;
}
/*************************************************************************
*
* 函數名稱：
* SaltNoiseDIB()
*
* 參數:
* LPSTR lpDIBBits - 指向源DIB圖像指針
* LONG lWidth - 源圖像寬度（象素數，必須是4的倍數）
* LONG lHeight - 源圖像高度（象素數）
*
* 返回值:
* BOOL - 模糊操作成功返回TRUE，否則返回FALSE。
*
* 說明:
* 該函數用來對DIB圖像進行模糊操作。
*
************************************************************************/
BOOL WINAPI SaltNoiseDIB (LPSTR lpDIBBits, LONG lWidth, LONG lHeight)
{
// 指向源圖像的指針
LPSTR lpSrc;
//循環變量
long i;
long j;
// 圖像每行的字節數
LONG lLineBytes;
// 計算圖像每行的字節數
lLineBytes = WIDTHBYTES(lWidth * 8);
//生成偽隨機種子
srand((unsigned)time(NULL));
//在圖像中加噪
for (j = 0;j < lHeight ;j++)
{
for(i = 0;i < lLineBytes ;i++)
{
if(rand()>31500)
{
// 指向源圖像倒數第j行，第i個象素的指針
lpSrc = (char *)lpDIBBits + lLineBytes * j + i;
//圖像中當前點置為黑
*lpSrc = 0;
}
}
}
// 返回
return true;
}
#undef SWAP
?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -