在C++中调用FFTW

FFTW是一个可以进行可变长度一维或多维DFT的开源C程序库,是目前最快的FFT算法实现。

本文简述了在Windows平台上,如何在C++中调用FFTW,所使用的IDE为Visual Studio 2017。

FFTW的详细信息可在 http://www.fftw.org 中查看

获取FFTW

http://www.fftw.org/install/windows.html 中下载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...
posted @ 2019-11-06 01:29  ZillyRex  阅读(...)  评论(... 编辑 收藏