demo.c

/*
@file: main.c
@author: ZZH
@date: 2022-11-05
@info: fft测试案例
*/
#include <stdio.h>
#include <string.h>
#include "fft.h"

#define N 128
#define FS 5000

int main(const int argc, const char** argv)
{
    //构建一个含有多频率成分的信号
    Number_t input[N];
    for (int i = 0;i < N;i++)
    {
        double arg = 2 * M_PI * i / FS;
        input[i] = sin(70 * arg) + sin(800 * arg) + sin(2200 * arg);
    }

    Complex_t out[N];
    
    //傅立叶正变换, 把input的信号变为复数频谱, 输出数据体积会比输入大一倍
    rfft(out, input, N);

    Number_t out2[N];

    //对复数频谱求模得到实数, 这些实数就是每个频率下的信号强度
    length(out2, out, N);

    //打印幅度谱, 方框内将下标解算为实际频率, 后跟信号强度
    for (int i = 0;i < N / 2;i++)
    {
        printf("[%lf]: %lf\r\n", i * (double) FS / N, out2[i]);
    }

    // memset(&out[N / 2], 0, sizeof(out) / 2);

    //傅立叶反变换, 输入刚才得到的复数频谱, 输出实数信号, 这里的out2理论上应该和最初的input是一样的信号, 但是幅值被放大了N倍
    irfft(out2, out, N);

    for (int i = 0;i < N;i++)
    {
        out2[i] /= N;
    }

    int8_t res = 0;
    //输出原始信号和频谱反变换回来的信号, 这两个信号应该是完全一样的才正确
    for (int i = 0;i < N;i++)
    {
        printf("orig[%d]: %lf, ifft[%d]: %lf\r\n", i, input[i], i, out2[i]);

        if (fabs(input[i] - out2[i]) > 0.001)//误差不超过0.001
            res = -1;//否则认为是还原失败
    }

    return res;
}