aboutsummaryrefslogtreecommitdiff
path: root/src/codec2_fft.h
diff options
context:
space:
mode:
authorMarin Ivanov <[email protected]>2025-07-25 10:17:14 +0300
committerMarin Ivanov <[email protected]>2026-01-18 20:09:26 +0200
commit0168586485e6310c598713c911b1dec5618d61a1 (patch)
tree6aabc2a12ef8fef70683f5389bea00f948015f77 /src/codec2_fft.h
Initial commitHEADmaster
* 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.h99
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