FFTW是一个可以进行可变长度一维或多维DFT的开源C程序库,是目前最快的FFT算法实现。
本文简述了在Windows平台上,如何在C++中调用FFTW,所使用的IDE为Visual Studio 2017。
获取FFTW
生成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
#include "fftw3.h"
#pragma comment(lib, "libfftw3-3.lib")
//实部与虚部
#define REAL 0
#define IMAG 1
using namespace std;
int main() {
/*
*fftw_complex 是FFTW自定义的复数类
*引入则会使用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...