利用ARM NEON计算FFT

本文将介绍利用ARM NEON计算快速傅里叶变换。

1. 编写FFT测试程序rfft_neon.c

#include <stdio.h>
#include <stdlib.h>

#include "NE10.h"

#define FFT_SIZE 512

int main(void)
{
    ne10_float32_t data[FFT_SIZE ] = {}; 
    ne10_fft_cpx_float32_t spec[(FFT_SIZE / 2) + 1] = {};
    ne10_fft_r2c_cfg_float32_t fft_cfg; 

    fft_cfg = ne10_fft_alloc_r2c_float32(FFT_SIZE ); /* FFT相关数据结构体 */

    /* 产生数据 */
    int i = 0;
    for (i = 0; i < FFT_SIZE ; i++)
    {
        data[i] = (ne10_float32_t)rand() / RAND_MAX * 50.0f;
    }

    ne10_fft_r2c_1d_float32_neon(spec, data, fft_cfg); /* FFT运算 */

    ne10_fft_destroy_r2c_float32(fft_cfg); /* 释放内存 */

    return 0;
}

2. 编译

/* arm-linux-gnueabi-gcc rfft_neon.c -I/your/include/path -L/your/lib/path -lNE10 -lm -static -mfpu=neon -o rfft_neon */