使用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 =