FFTW函数库应用实例

时间：2010-05-21 来源：wangfanlion

下面这个程序是利用FFTW函数库编写的基于C/C++的FFT测试程序，多线程和MPI编程实例在本博客中也都可以找到：

#include<fftw3.h>
#include<stdio.h>
#define N 8
int main()
{
int i;

fftw_complex *in,*out;
fftw_plan p;
in = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * N);
out = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * N);

if((in==NULL)||(out==NULL))
{
printf("Error:insufficient available memory\n");
}
else
{

for(i=0; i<N; i++)/*测试数据 */
{
in[i][0] = i+1;
in[i][1] = 0;
}
}

p = fftw_plan_dft_1d(N, in, out, FFTW_FORWARD,FFTW_ESTIMATE);

fftw_execute(p); /* repeat as needed */
fftw_destroy_plan(p);
fftw_cleanup();
for(i=0;i<N;i++)/*OUTPUT*/
{
printf("%f,%fi\n",in[i][0],in[i][1]);
}
printf("\n");
for(i=0;i<N;i++)/*OUTPUT*/
{
printf("%f,%fi\n",out[i][0],out[i][1]);
    }

         if(in!=NULL) fftw_free(in);
         if(out!=NULL) fftw_free(out);
         return 0;
     }