C++实现FFT快速变换

 [转载]在C++中调用FFTW - ZillyRex - 博客园 (cnblogs.com)

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

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

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

获取FFTW

在 FFTW Installation on Windows 中下载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...