整理后的fft程序(已验证)

#define N 1000
typedef struct
{
double real;
double img;
}complex;

void fft(); /*快速傅里叶变换*/
void ifft(); /*快速傅里叶逆变换*/
void initW();
void change();

void add(complex,complex,complex *); /*复数加法*/
void mul(complex,complex,complex *); /*复数乘法*/
void sub(complex,complex,complex *); /*复数减法*/
void divi(complex,complex,complex *);/*复数除法*/


complex x[N], *W;/*输出序列的值*/
int size_x=0;/*输入序列的长度，只限2的N次方*/
double PI;

void fft()
{
int i=0,j=0,k=0,l=0;
complex up,down,product;
change();
for(i=0;i{
l=1<for(j=0;j{
for(k=0;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 ifft()
{
int i=0,j=0,k=0,l=size_x;
complex up,down;
for(i=0;i<(int)(log(size_x)/log(2));i++) /*蝶形运算*/
{
l/=2;
for(j=0;j{
for(k=0;k{
add(x[j+k],x[j+k+l],&up);
up.real/=2;up.img/=2;
sub(x[j+k],x[j+k+l],&down);
down.real/=2;down.img/=2;
divi(down,W[size_x*k/2/l],&down);
x[j+k]=up;
x[j+k+l]=down;
}
}
}
change();
}

void initW()
{
int i;
W=(complex *)malloc(sizeof(complex)*size_x);
for(i=0;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{
k=i;j=0;
t=(log(size_x)/log(2));
while((t--)>0)
{
j=j<<1;
j|=(k&1);
k=k>>1;
}
if(j>i)
{
temp=x[i];
x[i]=x[j];
x[j]=temp;
}
}
}


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;
}

void divi(complex a,complex b,complex *c)
{
c->real=(a.real*b.real+a.img*b.img)/(b.real*b.real+b.img*b.img);
c->img=(a.img*b.real-a.real*b.img)/(b.real*b.real+b.img*b.img);
}


int main() //调用函数
{
int size_x;//定义采样数 只能是2的次方数 不够的补零 譬如需要512数据 只有500个 最后的12个设为

0
complex x[N];//采样数组
fft();//FFT变换得到频域序列
filter();//自定义滤波函数，把不需要的高次项清零，x[i]=0,i>2。
ifft();//FFT反变换得到滤波完成后的时域序列
}