McLeod.h

Go to the documentation of this file.

/*
  ==============================================================================
 
    McLeod.h
    Created: 4 Nov 2016 4:02:43pm
    Author:  KrotosMacMini
 
  ==============================================================================
*/
 
#pragma once
 
namespace krotos
{
    class McLeod
    {
      public:
        McLeod();
        McLeod(double sampleRate, int bufferSize);
        ~McLeod();
 
        void initialise(double newSampleRate, int newBufferSize);
 
        float getPitch(std::vector<float> InputBuffer);
        float getPitch(const float* InputBuffer, int numSamples);
 
      private:
        float getPitchInternal();
 
        const float DEFAULT_SAMPLE_RATE{44100};
        const int MAX_BUFFER_SIZE{1024};
        const float SMALL_CUTOFF{0.5};
        const double CUTOFF{0.97};          // 0.97 is default
        const float LOWER_PITCH_CUTOFF{40}; // Lowest pitch estimation [Hz]
        const double PI{3.14159265358979323846};
        const double TWO_PI{2 * PI};
 
        int bufferSize;
        int peakPositionsIndex;
        int periodEstimatesIndex;
        int amplitudeEstimatesIndex;
 
        float interpolatedX, interpolatedY;
 
        double sampleRate;
 
        std::vector<int> peakPositions;
        std::vector<float> mcLeodBuffer;
        std::vector<float> periodEstimates;
        std::vector<float> amplitudeEstimates;
        std::vector<float> window;
 
        void calculateNormalisedSquareDifferenceFunction(std::vector<float> buffer);
        void peakPicking();
        void parabolicInterpolation(int tauEstimation);
    };
} // namespace krotos