【kissfft】使用kiss_fft做FFT与iFFT

长时间没有使用kissfft有点忘记API的使用了，这里记录一下最最基本的使用。 FFT与iFFt

FFT

使用FFT的时候先初始化kiss_fft_cfg，其中第二个参数0/1表示是做fft还是ifft

kiss_fft_state = kiss_fft_alloc(nfft, 0, 0, 0);

构建输入参数

kiss_fft输入参数是复数，但是我们时域的数据为实数，所以需要构建等长的复数参数，然后把时域信号赋值到real部分,image部分置0.

iFFT

kiss_fft_state = kiss_fft_alloc(nfft, 1, 0, 0);

iFFT的时候，可能kissfft没有做scaling,所以结果需要除以N。

最后的结果是输出复数的real部分。

实例代码

static kiss_fft_scalar rand_scalar(void) {
    kiss_fft_scalar s = (kiss_fft_scalar) ((rand() + 10) % 256);
    return s / 256.;
}

static void print_fft_result(kiss_fft_cpx *x, int n) {
    int l = 0;
    for (int i = 0; i < n; i++) {
        printf("(%4.4f+%4.4fi), ", x[i].r, x[i].i);
        if (l++ == 7) {
            l = 0;
            printf("\n");
        }
    }
    printf("\n");
}

void test_fft() {
    int i = 0;
    int nfft = 20;
    kiss_fft_cpx cin[nfft];
    kiss_fft_cpx cout[nfft];
    kiss_fft_cpx sout[nfft];
    kiss_fft_cfg kiss_fft_state;

    kiss_fft_scalar zero;
    memset(&zero, 0, sizeof(zero));

    for (i = 0; i < nfft; ++i) {
        cin[i].r = rand_scalar();
        cin[i].i = zero;
    }
    printf("\n");

    printf(" init data for kiss_fft (cin): \n");
    print_fft_result(cin, nfft);

    memset(cout, 0, sizeof(short) * nfft);
    memset(sout, 0, sizeof(short) * nfft);

    kiss_fft_state = kiss_fft_alloc(nfft, 0, 0, 0);

    kiss_fft(kiss_fft_state, cin, cout);
    kiss_fft_free(kiss_fft_state);

    printf(" results from kiss_fft (cout): \n");
    print_fft_result(cout, nfft);

    kiss_fft_state = kiss_fft_alloc(nfft, 1, 0, 0);

    for (i = 0; i < nfft; i++) {
        cout[i].r /= nfft;
        cout[i].i /= nfft;
    }
    // case A.
    kiss_fft(kiss_fft_state, cout, sout);

    // end case
    kiss_fft_free(kiss_fft_state);

    printf(" results from kiss_ifft (sout): \n");
    print_fft_result(sout, nfft);
    printf("\n");
}

int main(int argc, char *const argv[]) {

    test_fft();

    return 0;
}

结果如下：