HarmonicRepresentation.h

Go to the documentation of this file.

//=====================================================================================
//=====================================================================================
 
#pragma once
namespace krotos
{
    class HarmonicRepresentation
    {
      public:
        /* Constructor
         *  @param sampleRate sampling rate in Hz
         */
        HarmonicRepresentation(float sampleRate);
 
        Eigen::MatrixXf processSignal(std::vector<float>& inputSignal);
 
      private:
        struct HarmRepParams
        {
            const float threshold = 0.3f;
            const int nHarms = 20;
 
        } harmRep;
 
        struct ConfigParams
        {
            const int hopSize = 1103;
            const int winSize = 4410;
            const int fftSize = 32768;
        } config;
 
        struct SwipeParams
        {
            const float ERBStep = 0.1f;
            const float normOverlap = 0.2f;
            const float strenThresh = std::numeric_limits<float>::lowest();
            const float log2PitchStep = 0.0208f;
 
        } swipe;
 
        void replaceNanWithMedian(Eigen::VectorXf& inputVector);
 
        Eigen::VectorXf fevalbp(const Eigen::MatrixXf& bp, const Eigen::VectorXf& x_v);
 
        Eigen::VectorXi findIndices(const Eigen::VectorXf& x_v,
                                    float threshMin = -std::numeric_limits<float>::infinity(),
                                    float threshMax = std::numeric_limits<float>::infinity());
 
        std::vector<Eigen::MatrixXf> calculateReshaped2D(const Eigen::MatrixXf& multiplied, int tSize, int nFreqCorrs,
                                                         int nInharmCoeffs);
        std::vector<Eigen::MatrixXi> applyTransformation(const std::vector<Eigen::MatrixXf>& reshapedMultiplied2_d,
                                                         float binSize, int fSize);
        std::vector<Eigen::MatrixXi> calculateDistrIdcs(int tSize, int nFreqCorrs, int nInharmCoeffs, int fSize,
                                                        std::vector<Eigen::MatrixXi>& fSupIdcsTFHI2_d);
        std::vector<Eigen::MatrixXf> extractSubmatrices(const Eigen::MatrixXf& distr,
                                                        const std::vector<Eigen::MatrixXi>& distrIdcsTFHI2_d);
        std::vector<Eigen::MatrixXf> computeTotalErgTFI(const std::vector<Eigen::MatrixXf>& submatrices_cell, int tSize,
                                                        int nFreqCorrs, int nInharmCoeffs);
        std::pair<Eigen::VectorXf, Eigen::VectorXi> maxRowValues(const Eigen::MatrixXf& matrix);
 
        std::pair<Eigen::VectorXf, Eigen::VectorXf> m_pitchStrengthPair;
        Eigen::VectorXf m_times;
 
        const float m_sampleRate;
        float m_maxSwipepWinSize;
    };
} // namespace krotos