网上已有的示例一般是一维的傅立叶变换,此处给出二维傅立叶变换的例子。
需要注意的是,二维傅立叶变换时,也是将二维数组数据存储在一个一维数组里,只需要对数据进行重组就可以得到自己想要的数据了。
示例:
#include "fftw3.h"
#include <stdio.h>
#include <iostream>
using namespace std;
#define N 3
int main()
{
int i;
fftw_complex *in, *out;
fftw_plan p;
in = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * N * N);
out = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * N * N);
if (in == NULL || out == NULL)
{
printf("ERROR!");
}
else
{
for (i=0; i<N*N; i++)
{
in[i][0] = i+1;
in[i][1] = 0;
}
}
p = fftw_plan_dft_2d(N, N, in, out, FFTW_FORWARD, FFTW_ESTIMATE);
fftw_execute(p);
fftw_destroy_plan(p);
fftw_cleanup();
for (i=0; i<N*N; i++)
{
printf("%f, %fi\n", in[i][0], in[i][1]);
}
cout << endl;
for (i=0; i<N*N; i++)
{
printf("%f, %fi\n", out[i][0], out[i][1]);
}
if (in!=NULL)
fftw_free(in);
if (out!=NULL)
fftw_free(out);
return 0;
}
运行结果: