#ifndef FFT_h
#define FFT_h

#include "stm32f4xx_hal.h"
#include <stdbool.h>
#include <math.h>
#include "functions.h"
#include "wm8731.h"
#include "screen_layout.h"

#define FFT_SIZE  512                                            // specify the size of the calculated FFT
#define FFT_USEFUL_SIZE 512                                     // size after FFT cropping
#define FFT_DOUBLE_SIZE_BUFFER (FFT_SIZE * 2)                    // Buffer size for FFT calculation
#define FFT_MIN 4.0f                                             // MIN threshold of FFT signal
#define FFT_TARGET 6.0f                                          // average threshold of the FFT signal
#define FFT_COMPRESS_INTERVAL 0.9f                               // compress interval of the FFT signal
#define FFT_MAX 6.0f                                             // MAX FFT signal threshold
#define FFT_STEP_COEFF 10.0f                                     // step coefficient for auto-calibration of the FFT signal (more - slower)
#define FFT_HZ_IN_PIXEL (float32_t)((float32_t)TRX_SAMPLERATE / (float32_t)LAY_FFT_PRINT_SIZE)     // hertz per pixel
#define FFT_BW_BRIGHTNESS 10																		 // pixel brightness on bw bar
#define FFT_SCALE_LINES_BRIGHTNESS 0.4f													 // pixel brightness on scale lines
#define FFT_MAX_GRID_NUMBER	13
#define ZOOMFFT_DECIM_STAGES 5
#define FFT_DMA_MAX_BLOCK 65000
#define FFT_TX_MIN_LEVEL 3.0f     															 //fft tx minimum scale level
#define FFT_MAX_TOP_SCALE 30 //maximum scale parameter

// Public variables
extern uint32_t FFT_buff_index;
extern bool FFT_buff_current;
extern bool FFT_need_fft;
extern bool FFT_new_buffer_ready;
extern float32_t FFTInput_I_A[FFT_SIZE];
extern float32_t FFTInput_Q_A[FFT_SIZE];
extern float32_t FFTInput_I_B[FFT_SIZE];
extern float32_t FFTInput_Q_B[FFT_SIZE];
extern uint16_t FFT_FPS;
extern uint16_t FFT_FPS_Last;

// Public methods
extern void FFT_Init(void);              // FFT initialization
extern void FFT_PreInit(void);													 // FFT precalculation
extern void FFT_bufferPrepare(void);										 // FFT Buffer process
extern void FFT_Reset(void);             // reset FFT
extern void FFT_doFFT(void);             // FFT calculation
extern bool FFT_printFFT(void);          // FFT output
extern void FFT_afterPrintFFT(void);		 // FFT output after callback
extern void FFT_printWaterfallDMA(void); // waterfall output
extern uint32_t getFreqOnFFTPosition(uint16_t position); //get frequency from pixel X position

#endif