diff options
| author | Marin Ivanov <[email protected]> | 2025-07-25 10:17:14 +0300 |
|---|---|---|
| committer | Marin Ivanov <[email protected]> | 2026-01-18 20:09:26 +0200 |
| commit | 0168586485e6310c598713c911b1dec5618d61a1 (patch) | |
| tree | 6aabc2a12ef8fef70683f5389bea00f948015f77 /src/codec2_fft.h | |
* codec2 cut-down version 1.2.0
* Remove codebook and generation of sources
* remove c2dec c2enc binaries
* prepare for emscripten
Diffstat (limited to 'src/codec2_fft.h')
| -rw-r--r-- | src/codec2_fft.h | 99 |
1 files changed, 99 insertions, 0 deletions
diff --git a/src/codec2_fft.h b/src/codec2_fft.h new file mode 100644 index 0000000..cea8f35 --- /dev/null +++ b/src/codec2_fft.h @@ -0,0 +1,99 @@ +/* + * codec2_fft.h + * + * Created on: 17.09.2016 + * Author: danilo + */ + +#ifndef DRIVERS_FREEDV_CODEC2_FFT_H_ +#define DRIVERS_FREEDV_CODEC2_FFT_H_ + +// #include <assert.h> +#include <math.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#ifndef FDV_ARM_MATH +#define USE_KISS_FFT +#else +#include "arm_const_structs.h" +#include "arm_math.h" +#endif + +#include "comp.h" +#include "defines.h" + +typedef COMP codec2_fft_cpx; +#include "kiss_fftr.h" + +#ifdef USE_KISS_FFT +#include "kiss_fft.h" +typedef kiss_fftr_cfg codec2_fftr_cfg; +typedef kiss_fft_cfg codec2_fft_cfg; +typedef kiss_fft_scalar codec2_fft_scalar; +#else +typedef float32_t codec2_fft_scalar; +typedef struct { + arm_rfft_fast_instance_f32* instance; + int inverse; +} codec2_fftr_struct; + +typedef codec2_fftr_struct* codec2_fftr_cfg; + +typedef struct { + const arm_cfft_instance_f32* instance; + int inverse; +} codec2_fft_struct; +typedef codec2_fft_struct* codec2_fft_cfg; +#endif + +static inline void codec2_fftr(codec2_fftr_cfg cfg, codec2_fft_scalar* in, + codec2_fft_cpx* out) { +#ifdef USE_KISS_FFT + kiss_fftr(cfg, in, (kiss_fft_cpx*)out); +#else + arm_rfft_fast_f32(cfg->instance, in, (float*)out, cfg->inverse); + out->imag = 0; // remove out[FFT_ENC/2]->real stored in out[0].imag +#endif +} + +static inline void codec2_fftri(codec2_fftr_cfg cfg, codec2_fft_cpx* in, + codec2_fft_scalar* out) { +#ifdef USE_KISS_FFT + kiss_fftri(cfg, (kiss_fft_cpx*)in, out); +#else + arm_rfft_fast_f32(cfg->instance, (float*)in, out, cfg->inverse); + // arm_scale_f32(out,cfg->instance->fftLenRFFT,out,cfg->instance->fftLenRFFT); +#endif +} + +codec2_fft_cfg codec2_fft_alloc(int nfft, int inverse_fft, void* mem, + size_t* lenmem); +codec2_fftr_cfg codec2_fftr_alloc(int nfft, int inverse_fft, void* mem, + size_t* lenmem); +void codec2_fft_free(codec2_fft_cfg cfg); +void codec2_fftr_free(codec2_fftr_cfg cfg); + +static inline void codec2_fft(codec2_fft_cfg cfg, codec2_fft_cpx* in, + codec2_fft_cpx* out) { +#ifdef USE_KISS_FFT + kiss_fft(cfg, (kiss_fft_cpx*)in, (kiss_fft_cpx*)out); +#else + memcpy(out, in, cfg->instance->fftLen * 2 * sizeof(float)); + arm_cfft_f32(cfg->instance, (float*)out, cfg->inverse, 1); + // TODO: this is not nice, but for now required to keep changes minimal + // however, since main goal is to reduce the memory usage + // we should convert to an in place interface + // on PC like platforms the overhead of using the "inplace" kiss_fft calls + // is neglectable compared to the gain in memory usage on STM32 platforms + if (cfg->inverse) { + arm_scale_f32((float*)out, cfg->instance->fftLen, (float*)out, + cfg->instance->fftLen * 2); + } +#endif +} + +void codec2_fft_inplace(codec2_fft_cfg cfg, codec2_fft_cpx* inout); + +#endif |
