aboutsummaryrefslogtreecommitdiff
path: root/src/codec2_fft.h
diff options
context:
space:
mode:
authordrowe67 <[email protected]>2023-07-20 08:59:48 +0930
committerGitHub <[email protected]>2023-07-20 08:59:48 +0930
commit06d4c11e699b0351765f10398abb4f663a984f36 (patch)
tree33e22af0814c5b6c3d676f096ae8c2ac8a3ed9f0 /src/codec2_fft.h
parent6588e77f38bdebd7adffe091b22e7760d95d0ccb (diff)
parent4d6c143c0abec15e1d6ed1fd95d36f80e6cb7df8 (diff)
Merge pull request #3 from drowe67/dr-cleanup21.2.0
Cleanup Part 2
Diffstat (limited to 'src/codec2_fft.h')
-rw-r--r--src/codec2_fft.h110
1 files changed, 52 insertions, 58 deletions
diff --git a/src/codec2_fft.h b/src/codec2_fft.h
index 66e8ca2..1952864 100644
--- a/src/codec2_fft.h
+++ b/src/codec2_fft.h
@@ -9,97 +9,91 @@
#define DRIVERS_FREEDV_CODEC2_FFT_H_
#include <assert.h>
-#include <stdlib.h>
+#include <math.h>
#include <stdio.h>
+#include <stdlib.h>
#include <string.h>
-#include <math.h>
#ifndef FDV_ARM_MATH
- #define USE_KISS_FFT
+#define USE_KISS_FFT
#else
- #include "arm_math.h"
- #include "arm_const_structs.h"
+#include "arm_const_structs.h"
+#include "arm_math.h"
#endif
-#include "defines.h"
#include "comp.h"
+#include "defines.h"
-
-typedef COMP codec2_fft_cpx;
+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;
+#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;
+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)
-{
-
+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);
+ 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
+ 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)
-{
+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);
+ 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);
+ 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);
+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)
-{
-
+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);
+ 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);
- }
+ 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