tm_stm32_fft.h

#ifndef TM_FFT_H
#define TM_FFT_H 100

/* C++ detection */
#ifdef __cplusplus
extern "C" {
#endif

#include "stm32fxxx_hal.h"
#include "defines.h"

#include "arm_math.h"
#include "arm_const_structs.h"

/* Memory allocation function */
#ifndef LIB_ALLOC_FUNC
#define LIB_ALLOC_FUNC    malloc
#endif

/* Memory free function */
#ifndef LIB_FREE_FUNC
#define LIB_FREE_FUNC     free
#endif

typedef struct {
    float32_t* Input;               
    float32_t* Output;              
    uint16_t FFT_Size;              
    uint8_t UseMalloc;              
    uint16_t Count;                 
    const arm_cfft_instance_f32* S; 
    float32_t MaxValue;             
    uint32_t MaxIndex;              
} TM_FFT_F32_t;

uint8_t TM_FFT_Init_F32(TM_FFT_F32_t* FFT, uint16_t FFT_Size, uint8_t use_malloc);

void TM_FFT_SetBuffers_F32(TM_FFT_F32_t* FFT, float32_t* InputBuffer, float32_t* OutputBuffer);

uint8_t TM_FFT_AddToBuffer(TM_FFT_F32_t* FFT, float32_t sampleValue);

uint8_t TM_FFT_AddToBufferWithImag(TM_FFT_F32_t* FFT, float32_t Real, float32_t Imag);

void TM_FFT_Process_F32(TM_FFT_F32_t* FFT);

void TM_FFT_Free_F32(TM_FFT_F32_t* FFT);

#define TM_FFT_GetMaxValue(FFT)            ((FFT)->MaxValue)

#define TM_FFT_GetMaxIndex(FFT)            ((FFT)->MaxIndex)

#define TM_FFT_GetFFTSize(FFT)             ((FFT)->FFT_Size)

#define TM_FFT_GetFromBuffer(FFT, index)   ((FFT)->Output[(uint16_t)(index)])

/* C++ detection */
#ifdef __cplusplus
}
#endif

#endif