FFTW使用

FFTW使用

• FFTW Home Page
• GitHub - FFTW/fftw3

简介

FFTW 是一个 C 语言子程序库，用于计算一维或多维、任意输入大小、实数和复数数据（以及偶数/奇数数据，即离散余弦/正弦变换或 DCT/DST）的离散傅立叶变换 (DFT)。

官方编译库

• FFTW有现成编译好的库可直接下载

• 官方编译的动态库Windows没有.lib但是有.def 可使用以下命令导出.lib

  # 在def文件的当前目录下用命令行执行以下命令
  # lib 是Vs的lib.exe程序，如果没有环境变量请手动显式的使用全路径 如："D:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin\lib.exe"
  # x64 是指生成64位的.lib
  lib /machine:x64 /def:libfftw3-3.def
  lib /machine:x64 /def:libfftw3f-3.def
  lib /machine:x64 /def:libfftw3l-3.def
  

手动编译

• 下载源码 推荐Release Release fftw-3.3.10 · FFTW/fftw3 · GitHub

简单使用

#include "fftw/fftw3.h"
#pragma comment(lib, "fftw/x64/libfftw3-3.lib")
#pragma comment(lib, "fftw/x64/libfftw3l-3.lib")
#pragma comment(lib, "fftw/x64/libfftw3f-3.lib")

void fftw_test() {
	int N = 1024;
	fftw_complex *in, *out;
	fftw_plan p;
	in = (fftw_complex*)fftw_malloc(sizeof(fftw_complex) * N);
	out = (fftw_complex*)fftw_malloc(sizeof(fftw_complex) * N);
	p = fftw_plan_dft_1d(N, in, out, FFTW_FORWARD, FFTW_ESTIMATE);

	fftw_execute(p); /* repeat as needed */

	fftw_destroy_plan(p);
	fftw_free(in); fftw_free(out);
}
// 使用 FFTW 自带复数类型 fftw_complex 作为输入数组 
void fftw_test3() {
	double data[] = { 1,2,3,4,5,6,7,8,9,10 };
	int data_size = 10;

	fftw_complex *in_cplx = (fftw_complex*)fftw_malloc(sizeof(fftw_complex) * data_size);
	fftw_complex *out_cplx = (fftw_complex*)fftw_malloc(sizeof(fftw_complex) * data_size);

	for (int i = 0; i < data_size; i++) {
		in_cplx[i][0] = data[i];
		in_cplx[i][1] = 0;
	}
	fftw_plan fft_plan = fftw_plan_dft_1d(data_size, in_cplx, out_cplx, FFTW_FORWARD, FFTW_ESTIMATE);
	fftw_execute(fft_plan);

	for (int i = 0; i < data_size; i++) {
		printf("%f %f %f\n", data[i], out_cplx[i][0], out_cplx[i][1]);
	}

	fftw_destroy_plan(fft_plan);
	fftw_free(out_cplx);
}
// 使用 C++ 复数类型std::complex<double>作为输入数组
void fftw_test4() {
	int data_size = 1024;
	std::unique_ptr<std::complex<double>> in_cplx(new std::complex <double>[data_size]);
	std::unique_ptr<std::complex<double>> out_cplx(new std::complex <double>[data_size]);
	for (int i = 0; i < data_size; i++) {
		(in_cplx.get() + i)->_Val[0] = i;
		(in_cplx.get() + i)->_Val[1] = i;
	}

	fftw_plan fft_plan = fftw_plan_dft_1d(data_size, (fftw_complex*)in_cplx.get(), (fftw_complex*)out_cplx.get(), FFTW_FORWARD, FFTW_ESTIMATE);
	fftw_execute(fft_plan);

	for (int i = 0; i < data_size; i++) {
		printf("%f %f %f\n", (in_cplx.get() + i)->_Val[0], (out_cplx.get() + i)->_Val[0], (out_cplx.get()+i)->_Val[1]);
	}

	fftw_destroy_plan(fft_plan);
}