当前位置: 首页 > news >正文

基于STM32进行FFT滤波并计算插值DA输出

文章目录

  • 一、前言背景
  • 二、项目构思
    • 1. 确定FFT点数、采样率、采样点数
    • 2. 双缓存设计
  • 三、代码实现
    • 1. STM32CubeMX配置和HAL库初始化
    • 2. 核心代码
  • 四、效果展示和后话
  • 五、项目联想与扩展
    • 1. 倍频
    • 2. 降频
    • 3. 插值
      • 3.1 线性插值
      • 3.2 样条插值

一、前言背景

STM32 对 AD 采样信号进行快速傅里叶变换(FFT),以获取其频谱信息。
信号源为信号发生器(正点原子DM40A)产生的 100Hz 三角波:
100Hz三角波

AD 采样由定时器触发,触发频率为 1280Hz。AD 以该频率采样 100Hz 的三角波,并对采样数据进行 FFT。FFT 计算出的结果是复数形式(实部、虚部交替排列的数组,数组大小为 2 N 2N 2N N N N 为FFT的点数)的双边频谱:

三角波FFT

即在 [ − f s 2 , f s 2 ] [- \frac{f_{s}}{2} , \frac{f_{s}}{2}] [2fs,2fs] 频率范围内对称分布( f s f_{s} fs为采样率)。对应的计算结果存储在一个复数数组中,该数组的索引范围为 0 0 0 2 N − 1 2N-1 2N1。我们把 0 0 0 N N N 之间的部分,定义为正频率分量;相应的, N N N 2 N − 1 2N -1 2N1 之间的部分则为负频率分量。

通常,我们在频率分析时,会计算幅频相应:
三角波幅频相应

这个应用很常见,马上就是要寻找峰值,然后就可以确定信号的主要频率成分及其幅度;又或者计算相位谱,以分析各频率分量的相位关系等等。
但很少有见过对频率谱进行操作(如滤波、调制等等)后再通过逆傅里叶变换(IFFT)还原信号的应用。

于是笔者突发奇想,进行了一次小小的测试,目标是要从100Hz的三角波中滤波得到它的基波100Hz的正弦波并通过DA将波形输出至示波器展示,平台是STM32H743VIT6

二、项目构思

1. 确定FFT点数、采样率、采样点数

AD/DA 的触发源均由定时器提供,定时器源自 PLL 倍频/分频前的 25MHz 晶振。而 100Hz 的三角波由独立的信号发生器产生,因此 AD 采样时钟与信号发生器的输出之间可能存在一定的相位漂移或频率偏差。如果希望波形满足实时性要求,AD 和 DA 需要由同一触发信号驱动,以保证同步(当然,也可以让 DA 的触发频率高于 AD 的触发频率,因为使用片上AD/DA,本质都共享同一时钟源,理论上不会产生频率偏差或相位漂移)。

其次是FFT点数、采样率、采样点数的安排。对于 100Hz 的三角波信号,为了能够较好地表征其频谱信息,选择12.8倍于100Hz的采样率进行采样,即 f s = 100 × 12.8 = 1280 f_{s} = 100 \times 12.8 = 1280 fs=100×12.8=1280Hz。这一采样率远高于奈奎斯特采样定理规定的最低要求( 2 × 100 = 200 2 \times 100 = 200 2×100=200Hz),可以有效减少失真,特别是对于高次谐波较多的三角波信号。

在时域上,这意味着每个周期大约对应 12-13 个采样点。采用 128 点 FFT 进行变换,则 FFT 频率分辨率为:
f r e s = f s N = 1280 128 = 10 H z f_{res} = \frac{f_{s}}{N} = \frac{1280}{128} = 10Hz fres=Nfs=1281280=10Hz
即每个 FFT 频谱格宽度为10Hz。而100Hz是10Hz的整数倍,因此可以较好地反映100Hz三角波的各次谐波分量。

当然,在理论上,FFT 频率分辨率越高越好(即采样率越低或 FFT 点数越高),可以更精细地刻画信号的频谱结构。然而,我们的系统具有实时性要求,即需要在 AD 采样完成后,立即执行 FFT 变换、滤波、IFFT 处理,并将结果通过 DA 输出。因此,处理时间必须控制在 T T T 的时间窗口内:
T = A D 采样点数 × 1 采样率 T = AD采样点数 \times \frac{1}{采样率} T=AD采样点数×采样率1

基于此,我选择 12.8 倍于原始信号频率(100Hz)的采样率,即 1280Hz,并进行 128 点 FFT。

处理完成后,DA 端以相同采样率重建信号,每个周期包含 12-13 个采样点,这会让波形看上去很数字。故需要通过模拟低通滤波器(LPF)滤除采样频率 1280Hz 及其倍频分量,使得输出信号平滑,得到 100Hz 的正弦波。

2. 双缓存设计

在 AD 采集 128 个点后,我们需要执行 FFT 变换、滤波、IFFT 处理,并将结果通过 DA 输出。在此过程中,如果仅使用单一缓存,会导致数据处理期间无法继续采样,造成数据丢失。为了解决这一问题,我们采用 双缓存(Ping-Pong Buffering) 机制,使得数据采集与处理能够并行进行,提高实时性。

具体设计如下:

  1. 建立两个缓冲区 BufferA 和 BufferB。
  2. 当 DMA 将 ADC 采样数据搬运至 BufferA 时,我们对 BufferB 进行 FFT 变换、滤波、IFFT 处理,并准备其数据供 DAC 输出。
  3. 当 BufferA 搬运完成时,DMA 产生中断,设置一个标志位,并切换缓冲区:
    • ADC 的 DMA 目标地址切换到 BufferB,开始采集新数据。
    • BufferA 处理完成的数据被传输至 DAC,作为 DAC的DMA 的数据源。
  4. 在下一个 DMA 传输周期,两个缓冲区的角色再次交换,如此交替进行,确保数据采集与处理不间断。

乒乓操作

三、代码实现

1. STM32CubeMX配置和HAL库初始化

ADC配置
ADC

ADC2
DAC配置(利用OPAMP跟随输出)
DAC1

DAC2

DAC3
并完成HAL库的初始化代码:

_Noreturn void Main(void)
{RetargetInit(&huart1);HAL_ADCEx_Calibration_Start(&hadc1, ADC_CALIB_OFFSET_LINEARITY, ADC_SINGLE_ENDED);bsp_Adc_Start(&hadc1);bsp_SetTimerFreq(&htim1, SAMPLERATE);HAL_TIM_Base_Start_IT(&htim1);HAL_DAC_Start_DMA(&hdac1, DAC_CHANNEL_1, (uint32_t *)dac_dma_buf, DAC_DMA_BUF_NUM, DAC_ALIGN_12B_R);HAL_OPAMP_Start(&hopamp1);while (true){process_fft_ifft();}
}

2. 核心代码

定义采样率、ADC采样点数、DAC的DMA缓存点数、FFT点数:

#define SAMPLERATE  ((100)*(12.8))#define ADC_DAC_NUM 128#ifndef DAC_DMA_BUF_NUM
#define DAC_DMA_BUF_NUM ADC_DAC_NUM
#endifextern uint16_t dac_dma_buf[DAC_DMA_BUF_NUM];#define ADC_DMA_BUF_NUM ADC_DAC_NUM#define ADCx 1
#if ADCx==1
extern uint16_t       adc_dma_buffer[ADC_DMA_BUF_NUM];
#elif ADCx==2
extern uint32_t       adc_dma_buffer[ADC_DMA_BUF_NUM];
#endif#define FFT_NUM ADC_DAC_NUM

最核心的代码在ADC的DMA的完成中断中执行(这里这样写是因为H7配置了MPU,需要考虑Cache未击中问题):
使用两个指针完成对双缓存的调度

void signalTask(void)
{/* 半传输完成中断 */if((ADC_DMAx->LISR & ADC_DMA_Streamx_HC) != RESET){float32_t *target_buffer = (active_buffer) ? bufferA : bufferB;float32_t *processing_buffer = (active_buffer) ? bufferB : bufferA;SCB_InvalidateDCache_by_Addr((uint32_t *)(&adc_dma_buffer[0]), ADC_DMA_BUF_NUM);for (uint16_t i = 0; i < FFT_NUM / 2; ++i) {target_buffer[i] = adc_dma_buffer[i];dac_dma_buf[i] = (uint16_t)processing_buffer[i];}adc_state = 1;ADC_DMAx->LIFCR = ADC_DMA_Streamx_HC;}/* 传输完成中断 */if((ADC_DMAx->LISR & ADC_DMA_Streamx_TC) != RESET){float32_t *target_buffer = (active_buffer) ? bufferA : bufferB;float32_t *processing_buffer = (active_buffer) ? bufferB : bufferA;SCB_InvalidateDCache_by_Addr((uint32_t *)(&adc_dma_buffer[ADC_DMA_BUF_NUM / 2]), ADC_DMA_BUF_NUM);for (uint16_t i = 0; i < FFT_NUM / 2; ++i) {target_buffer[i + FFT_NUM / 2] = adc_dma_buffer[i + FFT_NUM / 2];dac_dma_buf[i + FFT_NUM / 2] = (uint16_t)processing_buffer[i + FFT_NUM / 2];}adc_state = 2;ADC_DMAx->LIFCR = ADC_DMA_Streamx_TC;}
}

根据adc_state的状态判断ADC的DMA是否完成,来执行FFT、滤波、IFFT处理 process_fft_ifft
完成处理置位标志位active_buffer

float32_t fft_input[FFT_NUM*2];void process_fft_ifft(void)
{if (adc_state == 2){float32_t *processing_buffer = (active_buffer) ? bufferA : bufferB;for (uint16_t i = 0; i < FFT_NUM; ++i){fft_input[i * 2] = processing_buffer[i];fft_input[i * 2 + 1] = 0;}arm_cfft_f32(&arm_cfft_sR_f32_len128, fft_input, 0, 1);uint16_t cutoff_bin = (uint16_t)((120.0 / SAMPLERATE) * (FFT_NUM * 2));for (uint16_t i = cutoff_bin; i < FFT_NUM/2; i++) {fft_input[2 * i] = 0.0f;fft_input[2 * i + 1] = 0.0f;fft_input[2 * (FFT_NUM - i)] = 0.0f;fft_input[2 * (FFT_NUM - i) + 1] = 0.0f;}arm_cfft_f32(&arm_cfft_sR_f32_len128, fft_input, 1, 1);for (uint16_t i = 0; i < FFT_NUM; i++) {processing_buffer[i] = fft_input[i * 2];}active_buffer = !active_buffer;adc_state = 0;}
}

这里的滤波方法是暴力滤波(优化方法:可以采用平滑过渡,例如用 Hanning 窗或 Hamming 窗在频域做过渡,而不是直接硬性截断)。
在滤波的时候,请不要忘记处理负频率:

for (uint16_t i = cutoff_bin; i < FFT_NUM/2; i++) {
// 正频率fft_input[2 * i] = 0.0f;fft_input[2 * i + 1] = 0.0f;
// 负频率	fft_input[2 * (FFT_NUM - i)] = 0.0f;fft_input[2 * (FFT_NUM - i) + 1] = 0.0f;
}

四、效果展示和后话

最终效果展示
这里我没有加上LPF(懒)

system

波形
滤波后的FFT结果

后话

FFT滤波时,我直接截断高频分量相当于乘以一个矩形窗,在时域上会导致高斯振铃(Gibbs 现象)(类似于时域上的方波傅里叶变换导致的振铃效应),后续优化可以采用平滑过渡,例如用 Hanning 窗或 Hamming 窗在频域做过渡,而不是直接硬性截断。

例如,使用 Hanning 窗 对截止频率附近的分量进行平滑处理:

for (uint16_t i = cutoff_bin; i < FFT_NUM/2; i++) {float window_coeff = 0.22 * (1 + cosf(M_PI * (i - cutoff_bin) / (FFT_NUM/2 - cutoff_bin))); // Hanning 窗fft_input[2 * i] *= window_coeff;fft_input[2 * i + 1] *= window_coeff;fft_input[2 * (FFT_NUM - i)] *= window_coeff;fft_input[2 * (FFT_NUM - i) + 1] *= window_coeff;
}

这样可以让高频分量逐渐衰减,而不是直接消失,减少 Gibbs 现象。
Hanning
此外,FFT 频域滤波更适合用于选取特定频率分量的滤波器。在实际应用中,我们通常希望滤波器具有一定的过渡带,而不是突变的截止频率。这是因为理想的 “砖墙” 低通滤波器(即瞬间截止所有高频分量)在物理上不可实现,实际滤波器通常需要一定的过渡区域,以减少时域上的振铃效应并改善信号的平滑性。

项目地址: https://github.com/mico845/STM32FFTFilter

五、项目联想与扩展

1. 倍频

在前文中,我们提到 DA的触发频率高于 AD 的触发频率。由于我们使用的是 片上 AD/DA,它们本质上共享同一时钟源,因此 理论上不会产生频率偏差或相位漂移。

在本项目中,我们采用双缓冲技术,理论上可以在 DA 输出缓存区中存储 100 Hz 的正弦波(通过滤波器得到的),然后以更高的触发频率进行输出,从而合成更高频率的正弦波信号。

说干就干,我决定进行实验测试,尝试让系统在输入100 Hz三角波的情况下,输出300 Hz正弦波,以实现更灵活的需求。

代码实现
这里修改DAC,让定时器2的Trigger Out event事件触发
定时器2触发
配置定时器的触发频率为DAC_FREQ

bsp_SetTimerFreq(&htim2, DAC_FREQ);
HAL_TIM_Base_Start_IT(&htim2);
#define SAMPLERATE  ((100)*(12.8))
#define DAC_FREQ    (SAMPLERATE * 3)

效果展示

三倍频三角波

2. 降频

基于倍频的原理,我们同样可以设计降频系统。其方法是保持采样频率不变,但增加采样点数,从而降低输出信号的频率。例如,若希望输出信号的频率降低至原来的一半,则需要将采样点数增加一倍。或者进行插值,增多DA输出点数。
说干就干,我决定进行实验测试,尝试让系统在输入100 Hz三角波的情况下,输出50Hz正弦波,以实现更灵活的需求。

采样率和DAC频率如下:

#define SAMPLERATE  ((100)*(12.8) * 2)
#define DAC_FREQ    ((100)*(12.8))

这里玩一下宏,利用宏拼接的方式,方便调整FFT点数:

#define FFT_NUM ADC_DAC_NUM#define __s_arm_cfft_f32(x) arm_cfft_sR_f32_len##x
#define _s_arm_cfft_f32(x) __s_arm_cfft_f32(x)
#define s_arm_cfft_f32 _s_arm_cfft_f32(FFT_NUM)

FFT和IFFT就变成了

arm_cfft_f32(&s_arm_cfft_f32, fft_input, 0, 1);
arm_cfft_f32(&s_arm_cfft_f32, fft_input, 1, 1);

效果展示
降频正弦波
但是,由于DA的触发频率低于AD的触发频率,导致采样数据在DA输出时无法准确对齐,进而引发相位偏移。为了解决这一问题,我决定采用插值的方法。

3. 插值

3.1 线性插值

利用插值算法,在保证 ADC 采样率不变的前提下,使 DAC 在采样点之间生成更多的插值点,从而平滑输出,提高信号的连续性。

我修改了DMA 完成中断的一些细节,以确保数据存储、处理和输出的正确:

void signalTask(void)
{/* 半传输完成中断 */if((ADC_DMAx->LISR & ADC_DMA_Streamx_HC) != RESET){float32_t *target_buffer = (active_buffer) ? bufferA : bufferB;float32_t *processing_buffer = (active_buffer) ? bufferB : bufferA;SCB_InvalidateDCache_by_Addr((uint32_t *)(&adc_dma_buffer[0]), ADC_DMA_BUF_NUM);for (uint16_t i = 0; i < ADC_DMA_BUF_NUM / 2; ++i) {target_buffer[i] = adc_dma_buffer[i];}for (uint16_t i = 0; i < DAC_DMA_BUF_NUM / 2; ++i) {dac_dma_buf[i] = (uint16_t)processing_buffer[i];}adc_state = 1;ADC_DMAx->LIFCR = ADC_DMA_Streamx_HC;}/* 传输完成中断 */if((ADC_DMAx->LISR & ADC_DMA_Streamx_TC) != RESET){float32_t *target_buffer = (active_buffer) ? bufferA : bufferB;float32_t *processing_buffer = (active_buffer) ? bufferB : bufferA;SCB_InvalidateDCache_by_Addr((uint32_t *)(&adc_dma_buffer[ADC_DMA_BUF_NUM / 2]), ADC_DMA_BUF_NUM);for (uint16_t i = 0; i < ADC_DMA_BUF_NUM / 2; ++i) {target_buffer[i + ADC_DMA_BUF_NUM / 2] = adc_dma_buffer[i + ADC_DMA_BUF_NUM / 2];}for (uint16_t i = 0; i < DAC_DMA_BUF_NUM / 2; ++i) {dac_dma_buf[i + DAC_DMA_BUF_NUM / 2] = (uint16_t)processing_buffer[i + DAC_DMA_BUF_NUM / 2];}adc_state = 2;ADC_DMAx->LIFCR = ADC_DMA_Streamx_TC;}
}

接下来,我打算使用线性插值做第一次尝试:

线性插值(Linear Interpolation,简称 Lerp)是一种用于在已知数据点之间估算新数据点的数学方法。它的基本思想是:如果两个已知点之间的数据是平滑变化的,我们可以用一条直线来近似它们之间的过渡,并计算中间值。

先简单了解一下线性插值算法:
假设有两个已知点:

  • A ( x 1 , y 1 ) A(x_{1}, y_{1}) A(x1,y1)
  • B ( x 2 , y 2 ) B(x_{2}, y_{2}) B(x2,y2)

你想知道在这两个点之间,某个位置 x x x处的 y y y是多少。
在数学上,线性插值使用下面的公式:
y = y 1 + ( x − x 1 ) ( x 2 − x 1 ) × ( y 2 − y 1 ) y = y_{1} + \frac{(x - x_{1})}{(x_{2} - x_{1})} \times (y_{2} - y_{1}) y=y1+(x2x1)(xx1)×(y2y1)
解释一下这个公式,也就是计算 x x x x 1 x_{1} x1 x 2 x_{2} x2之间的相对位置:
f r a c = ( x − x 1 ) ( x 2 − x 1 ) frac = \frac{(x - x_{1})}{(x_{2} - x_{1})} frac=(x2x1)(xx1)
这个值介于 0 0 0 1 1 1之间,表示 x x x x 1 x_{1} x1 x 2 x_{2} x2之间的比例
然后依照这个比例,计算出 y y y方向的偏移量:
Δ y = ( y 2 − y 1 ) × f r a c \Delta y = (y_{2} - y_{1}) \times frac Δy=(y2y1)×frac

最终得到插值的结果:
y = y 1 + Δ y y = y1 + \Delta y y=y1+Δy

在我们的这个系统中,也就是改成数组的方式来解释,可以直接写成:

	uint16_t w = output_size / input_size;// 省略 ... index = (float)i / w;idx = (uint16_t)index;frac = index - idx;

我来解释一下,假设 input_size = 4output_size = 8,那么 w = output_size / input_size = 8 / 4 = 2
output 数组的索引 i07,然后 index = i / 2 变为:

	i = 0  →  index = 0 / 2 = 0.0i = 1  →  index = 1 / 2 = 0.5i = 2  →  index = 2 / 2 = 1.0i = 3  →  index = 3 / 2 = 1.5i = 4  →  index = 4 / 2 = 2.0i = 5  →  index = 5 / 2 = 2.5i = 6  →  index = 6 / 2 = 3.0i = 7  →  index = 7 / 2 = 3.5

相应的idxfrac也就是:

	i = 0  →  index = 0.0,  idx = 0,  frac = 0.0i = 1  →  index = 0.5,  idx = 0,  frac = 0.5i = 2  →  index = 1.0,  idx = 1,  frac = 0.0i = 3  →  index = 1.5,  idx = 1,  frac = 0.5i = 4  →  index = 2.0,  idx = 2,  frac = 0.0i = 5  →  index = 2.5,  idx = 2,  frac = 0.5i = 6  →  index = 3.0,  idx = 3,  frac = 0.0i = 7  →  index = 3.5,  idx = 3,  frac = 0.5

frac等于0时,该点为采样点;反之,该点为需要插入的点。
我们手动计算i=1时的frac,根据公式可以得到frac = (1 - 0)/(2 - 0) = 0.5,和我们的计算结果也是吻合的。

从公式上理解,也就是
y = y 1 + ( x − x 1 ) ( x 2 − x 1 ) × ( y 2 − y 1 ) = y 1 + ( y 2 − y 1 ) × f r a c = y 1 × ( 1 − f r a c ) + f r a c × y 2 y = y_{1} + \frac{(x - x_{1})}{(x_{2} - x_{1})} \times (y_{2} - y_{1}) = y_{1} + (y_{2} - y_{1}) \times frac = y_{1} \times (1 - frac) + frac \times y_{2} y=y1+(x2x1)(xx1)×(y2y1)=y1+(y2y1)×frac=y1×(1frac)+frac×y2
代码上也就是:

	output[i] = (1 - frac) * input[idx] + frac * input[idx + 1];

故,我们完成线性拟合函数:

void lerp(uint16_t *input, uint16_t *output, uint16_t input_size, uint16_t output_size)
{uint16_t w = output_size / input_size;float index;uint16_t idx;float frac;for (uint16_t i = 0; i < output_size; i++){index = (float)i / w;idx = (uint16_t)index;frac = index - idx;if (idx < input_size - 1){output[i] = (uint16_t)((1 - frac) * input[idx] + frac * input[idx + 1]);}else{output[i] = input[input_size - 1];}}
}

调用函数lerp

        arm_cfft_f32(&s_arm_cfft_f32, fft_input, 1, 1);uint16_t interpolated_output[DAC_DMA_BUF_NUM];uint16_t interpolated_input[FFT_NUM];for (uint16_t i = 0; i < FFT_NUM; i++) {interpolated_input[i] = fft_input[2 * i];}lerp(interpolated_input, interpolated_output, FFT_NUM, DAC_DMA_BUF_NUM);for (uint16_t i = 0; i < DAC_DMA_BUF_NUM; i++) {processing_buffer[i] = interpolated_output[i];}active_buffer = !active_buffer;adc_state = 0;

这是对系统的采样率,DA触发频率,AD/DA的DMA缓存区大小配置:

#define INTERP_FACTOR 2#define SAMPLERATE  ((100)*(12.8))
#define DAC_FREQ    (SAMPLERATE * INTERP_FACTOR)#define ADC_DAC_NUM 128#ifndef DAC_DMA_BUF_NUM
#define DAC_DMA_BUF_NUM (ADC_DAC_NUM * INTERP_FACTOR)#define FFT_NUM ADC_DAC_NUM

效果展示

线性插值算法

3.2 样条插值


http://www.mrgr.cn/news/95507.html

相关文章:

  • PDF文件转Markdown,基于开源项目marker
  • 面试复习-基础网络+运维知识
  • c++ STL
  • 【机器学习】机器学习工程实战-第2章 项目开始前
  • LLM - 重排序(Rerank)
  • 【计算机网络】网络简介
  • C语言入门教程100讲(8)算术运算符
  • C语言入门教程100讲(3)代码注释
  • C语言入门教程100讲(7)类型转换
  • 【前端】Visual Studio Code安装配置教程:下载、汉化、常用组件、基本操作
  • C语言入门教程100讲(4)输入输出
  • 【AI学习笔记】Coze平台实现将Excel文档批量导入数据库全过程
  • C++学习之网盘项目单例模式
  • MySQL表的增加、查询、修改、删除的基础操作
  • 【HDLBits】Circuits—Combinational Logic合集(包含答案推导过程)
  • 分享最近前端面试遇到的一些问题
  • 【模板】计算几何入门
  • python:music21 构建 LSTM+GAN 模型生成爵士风格音乐
  • 挂谷猜想的证明错误百出
  • 使用flask_restful快速构建接口