Fft.h

#pragma once
#include <libraries/ne10/NE10.h>
#include <vector>
#include <cmath>
#include <libraries/math_neon/math_neon.h>

 
class Fft
{
public:
        Fft(){};
        Fft(unsigned int length){ setup(length); };
        ~Fft(){ cleanup(); };
        int setup(unsigned int length);
        void cleanup();
        void fft();
        void fft(const std::vector<float>& input);
        void ifft();
        void ifft(const std::vector<float>& reInput, const std::vector<float>& imInput);
        float& fdr(unsigned int n) { return frequencyDomain[n].r; };
        float& fdi(unsigned int n) { return frequencyDomain[n].i; };
        float fda(unsigned int n) { return sqrtf_neon(fdr(n) * fdr(n) + fdi(n) * fdi(n)); };
        float& td(unsigned int n) { return timeDomain[n]; };
        static bool isPowerOfTwo(unsigned int n);
        static unsigned int roundUpToPowerOfTwo(unsigned int n);
private:
        ne10_float32_t* timeDomain = nullptr;
        ne10_fft_cpx_float32_t* frequencyDomain = nullptr;
        ne10_fft_r2c_cfg_float32_t cfg = nullptr;
        unsigned int length;
};