C++实现快速傅里叶变换(FFT)算法

最新推荐文章于 2025-05-31 09:28:02 发布

smartisong

最新推荐文章于 2025-05-31 09:28:02 发布

阅读量3.9k

点赞数

CC 4.0 BY-SA版权

分类专栏：信号处理 C/C++ 文章标签：图像处理信号处理

本文链接：https://blog.csdn.net/erkuoge6464/article/details/119488037

C/C++ 同时被 2 个专栏收录

12 篇文章

订阅专栏

信号处理

7 篇文章

订阅专栏

本文详细介绍了如何使用C++实现快速傅里叶变换(FFT)算法，包括数据处理和逆变换的过程。通过Danielson-Lanczos引理进行复数序列的变换，适用于信号处理和数值计算等领域。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

C++实现快速傅里叶变换(FFT)算法


#include
#define DOUBLE_PI 6.283185307179586476925286766559

// 快速傅里叶变换
// data 长度为 (2 * 2^n), data 的偶位为实数部分, data 的奇位为虚数部分
// isInverse表示是否为逆变换
void FFT(double * data, int n, bool isInverse = false)
{
	int mmax, m, j, step, i;
	double temp;
	double theta, sin_htheta, sin_theta, pwr, wr, wi, tempr, tempi;
	n = 2 * (1 << n);
	int nn = n >> 1;
	// 长度为1的傅里叶变换, 位置交换过程
	j = 1;
	for(i = 1; i < n; i += 2)
	{
		if(j > i)
		{
			temp = data[j - 1];
			data[j - 1] = data[i - 1];
			data[i - 1] = temp;
			data[j] = temp;
			data[j] = data[i];
			data[i] = temp;
		}
	// 相反的二进制加法
		m = nn;
		while(m >= 2 && j > m)
		{
			j -= m;
			m >>= 1;
		}
			j += m;
	}
	// Danielson - Lanczos 引理应用
	mmax = 2;
	while(n > mmax)
	{
		step = mmax << 1;
		theta = DOUBLE_PI / mmax;
		if(isInverse)
		{
		theta = -theta;
		}
		sin_htheta = sin(0.5 * theta);
		sin_theta = sin(theta);
		pwr = -2.0 * sin_htheta * sin_htheta;
		wr = 1.0;
		wi = 0.0;
	for(m = 1; m < mmax; m += 2)
	{
		for(i = m; i <= n; i += step)
		{
		j = i + mmax;
		tempr = wr * data[j - 1] - wi * data[j];
		tempi = wr * data[j] + wi * data[j - 1];
		data[j - 1] = data[i - 1] - tempr;
		data[j] = data[i] - tempi;
		data[i - 1] += tempr;
		data[i] += tempi;
		}
		sin_htheta = wr;
		wr = sin_htheta * pwr - wi * sin_theta + wr;
		wi = wi * pwr + sin_htheta * sin_theta + wi;
	}
	mmax = step;
	}
}