在C++中调用FFTW

fftw是一个可以进行可变长度一维或多维dft的开源c程序库，是目前最快的fft算法实现。

本文简述了在windows平台上，如何在c++中调用fftw，所使用的ide为visual studio 2017。

fftw的详细信息可在 中查看

获取fftw

在 中下载32位或64位版并解压

生成lib文件

打开vs自带的 “x86_x64 cross tools command prompt” ，cd到之前的解压文件夹

键入命令

lib /machine:x86 /def:libfftw3-3.def

如果是64位则键入命令

lib /machine:x64 /def:libfftw3-3.def

以上命令产生了lib文件libfftw3-3.lib，应不同精度要求也可对libfftw3f-3.def及libfftw3l-3.def文件使用相同命令来产生相应lib文件

在c++中调用fftw

将下列三个文件粘贴到与cpp文件相同的目录下

• fftw3.h
• libfftw3-3.dll
• libfftw3-3.lib

在cpp文件中加入代码

#include "fftw3.h"
#pragma comment(lib, "libfftw3-3.lib")

测试代码

#include <iostream>

#include "fftw3.h"
#pragma comment(lib, "libfftw3-3.lib")

//实部与虚部
#define real 0
#define imag 1

using namespace std;

int main() {
    /*
    *fftw_complex 是fftw自定义的复数类
    *引入<complex>则会使用stl的复数类
    */
    fftw_complex x[5];
    fftw_complex y[5];

    for (int i = 0; i < 5; i++) {
        x[i][real] = i;
        x[i][imag] = 0;
    }
    
    //定义plan，包含序列长度、输入序列、输出序列、变换方向、变换模式
    fftw_plan plan = fftw_plan_dft_1d(5, x, y, fftw_forward, fftw_estimate);

    //对于每个plan，应当"一次定义 多次使用"，同一plan的运算速度极快
    fftw_execute(plan);

    for (int i = 0; i < 5; i++) {
        cout << y[i][real] << "  " << y[i][imag] << endl;
    }

    //销毁plan
    fftw_destroy_plan(plan);

    cout << "\npress enter to exit..." << endl;
    cin.get();

    return 0;
}

输出结果

10  0
-2.5  3.44095
-2.5  0.812299
-2.5  -0.812299
-2.5  -3.44095

press enter to exit...