AudioESP32FFT.h

#pragma once
 
#include "AudioFFT.h"
#include "fft.h"
 
namespace audio_tools {
 
class FFTDriverESP32FFT : public FFTDriver {
    public:
        bool begin(int len) override {
            this->len = len;
            if (p_fft_object==nullptr) p_fft_object = fft_init(len, FFT_REAL, FFT_FORWARD, NULL, NULL);
            assert(p_fft_object!=nullptr);
            return p_fft_object!=nullptr;
        }
        void end()override{
            if (p_fft_object!=nullptr) fft_destroy(p_fft_object);
        }
        void setValue(int idx, int value) override{
            p_fft_object->input[idx]  = value; 
        }
 
        void fft() override{
            fft_execute(p_fft_object);
        };
 
        float magnitude(int idx) override {
            return sqrt(pow(p_fft_object->output[2*idx],2) + pow(p_fft_object->output[2*idx+1],2));
        }
 
        float magnitudeFast(int idx) override {
            return (pow(p_fft_object->output[2*idx],2) + pow(p_fft_object->output[2*idx+1],2));
        }
 
        virtual bool isValid() override{ return p_fft_object!=nullptr; }
 
        fft_config_t *p_fft_object=nullptr;
        int len;
 
};
 
class AudioESP32FFT : public AudioFFTBase {
    public:
        AudioESP32FFT():AudioFFTBase(new FFTDriverESP32FFT()) {}
 
        float* array() {
            return driverEx()->p_fft_object->output;
        }
 
        FFTDriverESP32FFT* driverEx() {
            return (FFTDriverESP32FFT*)driver();
        }
};
 
}