精简的fft源码

fucks

1. void fft(const double* src_x, const double* src_y, double* dst_x, double* dst_y, int k) {
   
2.     int i, n;
   
3.     double temp, temp_x, temp_y;
   
4.     double* buffer;
   
5.     if (0 == k) {
   
6.         *dst_x = *src_x;
   
7.         *dst_y = *src_y;
   
8.         return;
   
9.     }
   
10.     n = 1 << (k - 1);
    
11.     buffer = (double*)malloc(4 * n * sizeof(double));
    
12.     for (i = 0; i < n; i++) {
    
13.         buffer[i] = src_x[i * 2];
    
14.         buffer[i + n] = src_y[i * 2];
    
15.         buffer[i + 2 * n] = src_x[i * 2 + 1];
    
16.         buffer[i + 3 * n] = src_y[i * 2 + 1];
    
17.     }
    
18.     fft((const double*)buffer + 2 * n, (const double*)buffer + 3 * n, dst_x, dst_y, k - 1);
    
19.     fft((const double*)buffer, (const double*)buffer + n, (double*)buffer + 2 * n, (double*)buffer + 3 * n, k - 1);
    
20.     for (i = 0; i < n; i++) {
    
21.         temp = i * M_PI / n;
    
22.         temp_x = cos(temp);
    
23.         temp_y = sin(temp);
    
24.         buffer[i] = dst_y[i] * temp_y + dst_x[i] * temp_x;
    
25.         buffer[i + n] = dst_y[i] * temp_x - dst_x[i] * temp_y;
    
26.         dst_x[i] = buffer[i + 2 * n] + buffer[i];
    
27.         dst_y[i] = buffer[i + 3 * n] + buffer[i + n];
    
28.         dst_x[i + n] = buffer[i + 2 * n] - buffer[i];
    
29.         dst_y[i + n] = buffer[i + 3 * n] - buffer[i + n];
    
30.     }
    
31.     free(buffer);
    
32. }

复制代码

fucks

参数说明：
const double* src_x, 输入数据实部的指针
const double* src_y, 输入数据虚部的指针
double* dst_x, 输出数据实部的指针
double* dst_y, 输出数据虚部的指针
int k, FFT点数，比如1024点，k=10，4096点，k=12

fucks

由于使用了malloc和free函数，单片机上慎用！除非你自实现malloc和free函数功能

xad1974

大神也跑过来了

听我瞎扯蛋

看不懂也先顶一下

fucks

xad1974 发表于 2020-5-20 07:54
大神也跑过来了

受疫情影响，没钱吃饭，只好去食堂蹭不花钱的紫菜蛋汤嘛

sinmin

好，收藏了                           

a123123

您就是那个数学大牛！

fire

cool