/************FFT***********/
#include <stdio.h>
#include <math.h>
#include <stdlib.h>
#define N 1000
typedef struct//复数的定义
{
double real;//复数的实部
double img;//复数的虚部
}complex;
void fft(); /*快速傅里叶变换*/
void ifft(); /*快速傅里叶逆变换*/
void initW();//快速核变换
void change();//将x(n)倒置
void add(complex ,complex ,complex *); /*复数加法*/
void mul(complex ,complex ,complex *); /*复数乘法*/
void sub(complex ,complex ,complex *); /*复数减法*/
void divi(complex ,complex ,complex *);/*复数除法*/
void output(); /*输出结果*/
complex x[N], *W; /*输出序列的值*/
int size_x=0; /*输入序列的长度，只限2的N次方*/
double PI;
int main()
{
int i,method;
system("cls");
PI=atan(1)*4; /*pi等于4乘以1.0的正切值*/
printf("Please input the size of x:\n"); /*输入序列的长度*/
scanf("%d",&size_x);
printf("Please input the data in x[N]:(such as:5 6)\n"); /*输入序列对应的值*/
for(i=0;i<size_x;i++)
scanf("%lf %lf",&x[i].real,&x[i].img);
initW();
/*选择FFT或逆FFT运算*/
printf("Use FFT(0) or IFFT(1)?\n");
scanf("%d",&method);
if(method==0)
fft();
else
ifft();
output();
return 0;
}
/*进行基-2 FFT运算*/
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; //左移符号,把相关值的二进制位向左移动，右边空出的全部补0
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 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<size_x;j+= 2*l ) /*一组蝶形运算*/
{
for(k=0;k<l;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<size_x;i++)
{
W[i].real=cos(2*PI/size_x*i); //1,0,-1,0
W[i].img=-1*sin(2*PI/size_x*i); //0,-1,0,1
}
}
void change() /*变址计算，将x(n)码位倒置*/
{
complex temp;
unsigned short i=0,j=0,k=0; //16位
double t;
for(i=0;i<size_x;i++)
{
k=i;
j=0;
t=(log(size_x)/log(2)); //t相当于公式中的M.
while( (t--)>0 )
{
j=j<<1; //左移符号,把相关值的二进制位向左移动，右边空出的全部补0
j|=(k&1); //&按位“与”运算符 ;|“按位或”
k=k>>1; //“>>”代表右移运算符
}
if(j>i)
{
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("%.4f",x[i].real);
if(x[i].img>=0.0001)
printf("+%.4fj\n",x[i].img);
else if(fabs(x[i].img)<0.0001)
printf("\n");
else
printf("%.4fj\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;
}
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);
}
