当前位置: 代码迷 >> 高性能计算 >> 关于FFT,问个很弱的有关问题
  详细解决方案

关于FFT,问个很弱的有关问题

热度:3766   发布时间:2013-02-26 00:00:00.0
关于FFT,问个很弱的问题
最近在看FFT,
看的参考书是 C语言数值算法程序大全(第二版)  也就是英文的(Numerical Recipes in C)

其中这本书的 12.2 的有一个叫做 four1 的经典 FFT 函数,

请问谁有这个函数的测试程序。也就是main函数?



------解决方案--------------------------------------------------------

我把 four1 贴在这里吧


#include <math.h>
#define SWAP(a,b) tempr=(a);(a)=(b);(b)=tempr

void four1(double data[], unsigned long nn, int isign)
{
unsigned long n,mmax,m,j,istep,i;
double wtemp,wr,wpr,wpi,wi,theta;
double tempr,tempi;

n=nn << 1;
j=1;
for (i=1;i<n;i+=2) {
if (j > i) {
SWAP(data[j],data[i]);
SWAP(data[j+1],data[i+1]);
}
m=n >> 1;
while (m >= 2 && j > m) {
j -= m;
m >>= 1;
}
j += m;
}
mmax=2;
while (n > mmax) {
istep=mmax << 1;
theta=isign*(6.28318530717959/mmax);
wtemp=sin(0.5*theta);
wpr = -2.0*wtemp*wtemp;
wpi=sin(theta);
wr=1.0;
wi=0.0;
for (m=1;m<mmax;m+=2) {
for (i=m;i<=n;i+=istep) {
j=i+mmax;
tempr=wr*data[j]-wi*data[j+1];
tempi=wr*data[j+1]+wi*data[j];
data[j]=data[i]-tempr;
data[j+1]=data[i+1]-tempi;
data[i] += tempr;
data[i+1] += tempi;
}
wr=(wtemp=wr)*wpr-wi*wpi+wr;
wi=wi*wpr+wtemp*wpi+wi;
}
mmax=istep;
}
}
#undef SWAP
  相关解决方案