from scipy import signal import numpy as np import matplotlib.pyplot as plt plt.rcParams["font.family"] = 'Arial Unicode MS' original_sig = np.loadtxt("resources/unbalanced.txt") original_sig -= np.mean(original_sig) N = len(original_sig) pi = np.pi f2_jw = np.fft.fft(original_sig) f2_jw = np.fft.fftshift(f2_jw) jw_list = [complex(0, 1) * 2 * pi / N * item for item in np.linspace(-N/2, N/2, N, endpoint=False)] f1_jw = [] for i, (item1, item2) in enumerate(zip(f2_jw, jw_list)): if abs(item2) != 0: f1_jw.append(item1/item2) else: f1_jw.append(complex(0, 0)) f1_jw = np.array(f1_jw) * 1000 # m到mm的量纲转换 f1_jw = np.fft.ifftshift(f1_jw) vel_sig = np.fft.ifft(f1_jw).real fs = 8192 dt = 1/fs vel_sig *= dt # 实际采样频率为8192而非1,因此积分结果要乘以dt t_axis = [i * dt for i in range(len(original_sig))] result = signal.detrend(vel_sig) plt.figure(figsize=(12, 3)) plt.subplot(121) plt.plot(t_axis, vel_sig, label="频域积分计算得到的速度信号") plt.legend(loc="upper right") plt.subplot(122) plt.plot(t_axis, result, label="频域积分后去趋势得到的速度信号") plt.legend(loc="upper right") plt.show()将这段代码使用C语言进行编写,原始样本长度为512,为实数,在进行FFT处理之前,原始样本以一维数组表示复数,实部与虚部交替存在,虚部全部设置为0,FFT函数调用DSP库中的arm_cfft_f32函数
时间: 2024-03-27 15:36:20 浏览: 67
根据您提供的Python代码,我编写了相应的C语言代码,其中使用arm_cfft_f32函数进行FFT计算,并使用arm_rfft_f32函数进行实数FFT计算。同时,由于原始样本以一维数组表示复数,实部与虚部交替存在,虚部全部设置为0,因此在进行FFT计算时,需要将原始样本的虚部设置为0。具体实现如下:
```c
#include "arm_math.h"
#include <stdio.h>
#define N 512
int main()
{
// 读取原始样本
float original_sig[N];
FILE *fp = fopen("resources/unbalanced.txt", "r");
for (int i = 0; i < N; ++i) {
fscanf(fp, "%f", &original_sig[i]);
}
fclose(fp);
// 去趋势
arm_mean_f32(original_sig, N, &mean);
for (int i = 0; i < N; ++i) {
original_sig[i] -= mean;
}
// 使用一维数组表示复数,实部与虚部交替存在,虚部全部设置为0
float complex_sig[2 * N];
for (int i = 0; i < N; ++i) {
complex_sig[2 * i] = original_sig[i];
complex_sig[2 * i + 1] = 0;
}
// 进行FFT
arm_cfft_f32(&arm_cfft_sR_f32_len512, complex_sig, 0, 1);
arm_cmplx_mag_f32(complex_sig, original_sig, N);
// 进行频域积分
float jw_list[N];
float dt = 1.0 / 8192;
for (int i = 0; i < N; ++i) {
if (i < N / 2) {
jw_list[i] = 2 * ARM_PI / N * (i - N / 2);
} else {
jw_list[i] = 2 * ARM_PI / N * (i - N / 2);
}
}
for (int i = 0; i < N; ++i) {
if (i == 0) {
complex_sig[2 * i] = 0;
complex_sig[2 * i + 1] = 0;
} else {
complex_sig[2 * i] /= jw_list[i];
complex_sig[2 * i + 1] /= jw_list[i];
}
}
// 进行实数FFT计算
arm_rfft_instance_f32 rfft;
arm_rfft_init_f32(&rfft, N, 0, 1);
arm_rfft_f32(&rfft, complex_sig, original_sig);
// 转换为速度信号
for (int i = 0; i < N; ++i) {
original_sig[i] /= (2 * ARM_PI * (i == 0 ? 1 : jw_list[i]));
original_sig[i] *= 1000; // m到mm的量纲转换
original_sig[i] *= dt; // 实际采样频率为8192而非1,因此积分结果要乘以dt
}
// 去趋势
float mean;
arm_mean_f32(original_sig, N, &mean);
arm_offset_f32(original_sig, -mean, original_sig, N);
// 输出结果
for (int i = 0; i < N; ++i) {
printf("%f\n", original_sig[i]);
}
return 0;
}
```
需要注意的是,由于实数FFT计算得到的结果是对称的,因此在进行频域积分时,需要将虚部全部设置为0,并且jw_list需要进行特殊处理以保证积分结果的正确性。另外,为了方便计算,此处使用了arm_cmplx_mag_f32函数计算FFT的模值,而非手动计算模值。
阅读全文