使用fftw进行二维傅立叶变换

网上已有的示例一般是一维的傅立叶变换，此处给出二维傅立叶变换的例子。
需要注意的是，二维傅立叶变换时，也是将二维数组数据存储在一个一维数组里，只需要对数据进行重组就可以得到自己想要的数据了。

示例：

#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;

}

运行结果：