Convolver.h

#pragma once
#define ENABLE_NE10_FIR_FLOAT_NEON // Define needed for Ne10 library
#include <libraries/ne10/NE10.h> // neon library
#include <vector>
 
class ConvolverChannel
{
public:
        ConvolverChannel(const std::vector<float>& ir, unsigned int blockSize) {setup(ir, blockSize);};
        int setup(const std::vector<float>& ir, unsigned int blockSize);
        void process(ne10_float32_t* filterOut, const ne10_float32_t* filterIn);
        void cleanup();
private:
        ne10_fir_instance_f32_t firFilter;
        unsigned int blockSize;
        ne10_float32_t* firFilterCoeff = nullptr;
        ne10_float32_t* firFilterState = nullptr;
};
 
#include <string>
class Convolver
{
public:
        Convolver() {};
        ~Convolver() { cleanup(); };
        int setup(const std::string& filename, unsigned int blockSize, unsigned int maxLength = 0);
        int setup(const std::vector<std::vector<float>>& irs, unsigned int blockSize);
        void process(float* out, const float* in, unsigned int frames, unsigned int channel = 0);
        void processInterleaved(float* out, const float* in, unsigned int frames, unsigned int outChannels, unsigned int inChannels);
        void cleanup();
        unsigned int getChannels() { return convolverChannels.size(); }
private:
        void doProcessInterleaved(float* out, const float* in, unsigned int frames, unsigned int outChannels, unsigned int inChannels, unsigned int channel);
        std::vector<ConvolverChannel> convolverChannels;
        ne10_float32_t* filterIn = nullptr;
        ne10_float32_t* filterOut = nullptr;
};