DFT+IDFT+FFT+IFFT的c++实现

#include<iostream.h> #include<math.h> #include <stdio.h> #include <stdlib.h> #include <fstream.h> #include "head.h" void reverse_seq (double tp[2048],double tq[2048]) { int lh=1024; int j=lh; int n1=2046; double t=0; for(int i=1;i<=n1;i++) { if(i<j) { t=tp[i]; tp[i]=tp[j]; tp[j]=t; t=tq[i]; tq[i]=tq[j]; tq[j]=t; } int k=lh; while(j>=k) { j=j-k; k=k/2; } j+=k; } ofstream out4("\\c++\\reverse.dat"); if(!out4) { cout<<"创建文件失败，程序将退出！"<<endl; return; } else { for(int r=0;r<2048;r++) { out4<<"tq["<<r<<"]="<<tq[r]; out4<<endl; } } out4.close(); } void fft() { double pi=3.1415926; double c[4096],d[4096]; for(int t=0;t<2048;t++) { c[t]=0.0; d[t]=0.0; } int m=11; reverse_seq (a,b); for(int l=1;l<=m;l++) { int B=(int)pow(2,l-1); for(int j=0;j<=B-1;j++) { int p=j*int(pow(2,m-l)); for(int k=j;k<2048; k+=(int)(pow(2,l))) { double q=2*pi/2048; double s=q*p; double temp=cos(s); s=sin(s); c[k]=a[k]; d[k]=b[k]; c[k+B]=a[k+B]; d[k+B]=b[k+B]; a[k]=c[k]+temp*c[k+B]+s*d[k+B]; b[k]=d[k]+temp*d[k+B]-s*c[k+B]; a[k+B]=c[k]-temp*c[k+B]-s*d[k+B]; b[k+B]=d[k]-temp*d[k+B]+s*c[k+B]; } } } ofstream out5("\\c++\\fft.dat"); if(!out5) { cout<<"创建文件失败，程序将退出！"<<endl; return; } else { for(int r=0;r<2048;r++) { out5<<a[r]<<" "<<b[r]; out5<<endl; } } out5.close(); } void ifft() { double pi=3.1415926; double c[4096],d[4096]; for(int t=0;t<2048;t++) { c[t]=0.0; d[t]=0.0; } int m=11; reverse_seq (a,b); for(int l=1;l<=m;l++) { int B=(int)pow(2,l-1); for(int j=0;j<=B-1;j++) { int p=j*int(pow(2,m-l)); for(int k=j;k<2048; k+=(int)(pow(2,l))) { double q=2*pi/2048; double s=q*p; double temp=cos(s); s=sin(s); c[k]=a[k]; d[k]=b[k]; c[k+B]=a[k+B]; d[k+B]=b[k+B]; a[k]=c[k]+temp*c[k+B]-s*d[k+B]; b[k]=d[k]+temp*d[k+B]+s*c[k+B]; a[k+B]=c[k]-temp*c[k+B]+s*d[k+B]; b[k+B]=d[k]-temp*d[k+B]-s*c[k+B]; } } } ofstream out6("\\c++\\ifft.dat"); if(!out6) { cout<<"创建文件失败，程序将退出！"<<endl; return; } else { for(int r=0;r<2048;r++) { a[r]=a[r]/2048; b[r]=b[r]/2048; out6<<a[r]<<" "<<b[r]; out6<<endl; } } out6.close(); }