本人实VC初学者,网上找的FFT代码,运行结果不对,请高手指教
本人是VC++初学者,就是一个菜鸟,什么都不懂,现用VC++的MFC模块做快速傅里叶变换,自己不会编程,只好网上搜索了代码,下面是FFT变换的代码,不知道哪里有问题,运行结果不对,请大神帮帮指点迷雾,万分感谢。#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#define N 1024
typedef struct // 复数的定义
{double real;double img;}complex;
void fft(); // 快速傅里叶变换
void initW(); // 快速核变换
void change(); // 将x(n)倒置
void add(complex,complex,complex *);
void mul (complex,complex,complex *); //
//复数乘法
void sub(complex,complex,complex *);
void output();
complex x[N],*W;
int size_x=0;
double PI;
int main()
{int i;
void fft();
void initW();
void change();
void add(complex ,complex ,complex *);
void mul(complex ,complex ,complex *);
void sub(complex ,complex ,complex *);
void divi(complex ,complex ,complex *);
PI=atan(1)*4;
printf("please input the size of x:\n");
scanf("%d",&size_x);
printf("please input the date in x(N):\n");
//int x[i]={1,1+i,2+i,3+2i,1+2i,0,2,-1+i};
for(i=0;i<size_x;i++)
scanf("%lf%lf",&x[i].real,&x[i].img);
initW();
fft();
output();
return 0;}
void fft()
{int i=0,j=0,k=0,l=0;
complex up,down,product;
change();
for(i=0;i<log(size_x)/log(2);i++)
{l=1<<i;
for(j=0;j<size_x;j+=2*l)
{for(k=0;k<l;k++)
{mul(x[j+k+l],W[size_x*k/2/l],&product);
add(x[j+k],product,&up);
sub(x[j+k],product,&down);
x[j+k]=up;
x[j+k+l]=down;}
}}}
void initW()
{int i;
W=(complex*)malloc(sizeof(complex)
*size_x);
for(i=0;i<size_x;i++)
{W[i].real=cos(2*PI/size_x*i);
W[i].img=- 1*sin(2*PI/size_x*i);
}
}
void change()
{complex temp;
unsigned short i=0,j=0,k=0;
double t;
for(i=0;i<size_x;i++)
{k=1;j=0;
t=(log(size_x)/log(2));
while((t--)>0)
{j=j<<1;j|=(k&1);k=k>>1;}
if(j>1)
{temp=x[i];x[i]=x[j];x[j]=temp;}
}
}
void output()
{int i;
printf("the result are as follows \n");
for(i=0;i<size_x;i++)
{printf("%.4fi \n",x[i].real );
if(x[i].img>=0.0001)
printf("+%.4fi\n",x[i].img);else if(fabs(x[i].img)
<0.0001)
printf("\n");else printf("%.4fi\n",x[i].img);}
}
void add(complex a,complex b,
complex *c)
{c->real=a.real+b.real;
c->img=a.img+b.img;}
void mul(complex a,complex b,
complex *c)
{c->real=a.real*b.real- a.img*b.img;
c->img=a.real*b.img+a.img*b.real;}
void sub(complex a,complex b,
complex *c)
{c->real=a.real- b.real;
c->img=a.img- b.img;}