作者：gnuhpc
出處：http://www.cnblogs.com/gnuhpc/
#include "cv.h"
#include "highgui.h"
#include "cxcore.h"
void cvShiftDFT(CvArr *src_arr,CvArr *dst_arr)
{
CvMat * tmp;
CvMat q1stub,q2stub;
CvMat q3stub,q4stub;
CvMat d1stub,d2stub;
CvMat d3stub,d4stub;
CvMat *q1,*q2,*q3,*q4;
CvMat *d1,*d2,*d3,*d4;
CvSize size = cvGetSize(src_arr);
CvSize dst_size = cvGetSize(dst_arr);
int cx,cy;
if ((dst_size.width!= size.width)||(dst_size.height!=size.height))
{
cvError(CV_StsUnmatchedSizes,"cvShiftDFT",
"Source and Destination arrays must have the same sizes",
__FILE__,__LINE__);
}
if (src_arr == dst_arr)
{
tmp = cvCreateMat(size.height/2,size.width/2,cvGetElemType(src_arr));
}
cx=size.width/2;//取出圖像的原點
cy=size.height/2;
q1=cvGetSubRect(src_arr,&q1stub,cvRect(0,0,cx,cy));
//取出圖像的第一象限，由q1指針指向它
q2=cvGetSubRect(src_arr,&q2stub,cvRect(cx,0,cx,cy));
//取出圖像的第二象限，由q2指針指向它
q3=cvGetSubRect(src_arr,&q3stub,cvRect(cx,cy,cx,cy));
//取出圖像的第三象限，由q3指針指向它
q4=cvGetSubRect(src_arr,&q4stub,cvRect(0,cy,cx,cy));
//取出圖像的第四象限，由q4指針指向它
d1=cvGetSubRect(src_arr,&d1stub,cvRect(0,0,cx,cy));
d2=cvGetSubRect(src_arr,&d2stub,cvRect(cx,0,cx,cy));
d3=cvGetSubRect(src_arr,&d3stub,cvRect(cy,cy,cx,cy));
d4=cvGetSubRect(src_arr,&d4stub,cvRect(0,cy,cx,cy));
if (src_arr!=dst_arr)
{
if (!CV_ARE_TYPES_EQ(q1,d1))
{
cvError(CV_StsUnmatchedSizes,"cvShiftDFT",
"Source and Destination arrays must have the same sizes",
__FILE__,__LINE__);
}
//以圖像中心為原點，調整傅里葉變換圖像的四個象限區，
//即第一與第三象限交換，第二與第四象限交換
cvCopy(q3,d1,0);
cvCopy(q4,d2,0);
cvCopy(q1,d3,0);
cvCopy(q2,d4,0);
}
else
{//若源矩陣和目的矩陣相同則直接在源矩陣中進行操作
cvCopy(q3,tmp,0);
cvCopy(q1,q3,0);
cvCopy(tmp,q1,0);
cvCopy(q4,tmp,0);
cvCopy(q2,q4,0);
cvCopy(tmp,q2,0);
}
}
int main(int argc,char ** argv)
{
const char* filename =(argc>=2?argv[1]:"lena.jpg");
IplImage *im;
IplImage *realInput,*imaginaryInput,*complexInput;
IplImage *image_Re,*image_Im;
int dft_M,dft_N;
CvMat *dft_A;
CvMat tmp;
double m,M;
im = cvLoadImage(filename, CV_LOAD_IMAGE_GRAYSCALE );//加載圖像
if (!im)
{
return -1;
}
//分配空間
realInput = cvCreateImage(cvGetSize(im),IPL_DEPTH_64F,1);//單通道
imaginaryInput =cvCreateImage(cvGetSize(im),IPL_DEPTH_64F,1);//單通道
complexInput = cvCreateImage(cvGetSize(im),IPL_DEPTH_64F,2);//雙通道
cvScale(im,realInput,1.0,0.0);
//#define cvScale cvConvertScale=>readInput=im
cvZero(imaginaryInput);
//清空這個圖像的內容
cvMerge(realInput,imaginaryInput,NULL,NULL,complexInput);
//混合這兩個圖像作為complexInput的兩個通道
/*得到最優DFT尺寸 */
dft_M = cvGetOptimalDFTSize(im->height-1);
dft_N = cvGetOptimalDFTSize(im->width-1);
dft_A = cvCreateMat(dft_M,dft_N,CV_64FC2);
image_Re = cvCreateImage(cvSize(dft_N,dft_M),IPL_DEPTH_64F,1);//實部
image_Im = cvCreateImage(cvSize(dft_N,dft_M),IPL_DEPTH_64F,1);//虛部
cvGetSubRect(dft_A,&tmp,cvRect(0,0,im->width,im->height));
cvCopy(complexInput,&tmp,NULL);
if( dft_A->cols > im->width )//若得到的最優DFT尺寸在寬度上大於原圖，則重新取
{
cvGetSubRect(dft_A,&tmp,cvRect(im->width,0,dft_A->cols-im->width,im->height));
cvZero(&tmp);
}
cvDFT(dft_A,dft_A,CV_DXT_FORWARD,complexInput->height);
cvNamedWindow("win",0);
cvNamedWindow("magnitude",0);
cvShowImage("win",im);
//分割出實部和虛部
cvSplit(dft_A,image_Re,image_Im,0,0);
//計算功率譜 Mag=sqrt(Re^2+Im^2)
cvPow(image_Re,image_Re,2.0);
cvPow(image_Im,image_Im,2.0);
cvAdd(image_Re,image_Im,image_Re,NULL);//image_Re<=image_Re+image_Im
cvPow(image_Re,image_Re,0.5);
//計算log(1+Mag)
cvAddS(image_Re,cvScalarAll(1.0),image_Re,NULL);
cvLog(image_Re,image_Re);
cvShiftDFT(image_Re,image_Re);
cvMinMaxLoc(image_Re,&m,&M,NULL,NULL,NULL);
cvScale(image_Re,image_Re,1.0/(M-m),1.0*(-m)/(M-m));
cvShowImage("magnitude",image_Re);
cvWaitKey(-1);
return 0;
}