OnsetDetectionFunction.h

Go to the documentation of this file.

//=======================================================================
//=======================================================================

#ifndef __ONSETDETECTIONFUNCTION_H
#define __ONSETDETECTIONFUNCTION_H

#include "fftw3.h"
#include <vector>

//=======================================================================
enum OnsetDetectionFunctionType
{
    EnergyEnvelope,
    EnergyDifference,
    SpectralDifference,
    SpectralDifferenceHWR,
    PhaseDeviation,
    ComplexSpectralDifference,
    ComplexSpectralDifferenceHWR,
    HighFrequencyContent,
    HighFrequencySpectralDifference,
    HighFrequencySpectralDifferenceHWR
};

//=======================================================================
enum WindowType
{
    RectangularWindow,
    HanningWindow,
    HammingWindow,
    BlackmanWindow,
    TukeyWindow
};

//=======================================================================
class OnsetDetectionFunction
{
public:
    
    OnsetDetectionFunction(int hopSize_,int frameSize_,int onsetDetectionFunctionType_,int windowType);
    
    ~OnsetDetectionFunction();
    
    void initialise(int hopSize_,int frameSize_,int onsetDetectionFunctionType_,int windowType);
    
    double calculateOnsetDetectionFunctionSample(double *buffer);
    
    void setOnsetDetectionFunctionType(int onsetDetectionFunctionType_);
    
private:
    
    void performFFT();

    //=======================================================================
    double energyEnvelope();
    
    double energyDifference();
    
    double spectralDifference();
    
    double spectralDifferenceHWR();
    
    double phaseDeviation();
    
    double complexSpectralDifference();
    
    double complexSpectralDifferenceHWR();
    
    double highFrequencyContent();
    
    double highFrequencySpectralDifference();
    
    double highFrequencySpectralDifferenceHWR();

    //=======================================================================
    void calculateRectangularWindow();
    
    void calculateHanningWindow();
    
    void calclulateHammingWindow();
    
    void calculateBlackmanWindow();
    
    void calculateTukeyWindow();

    //=======================================================================
    double princarg(double phaseVal);
    
    
    double pi;                          
    int frameSize;                      
    int hopSize;                        
    int onsetDetectionFunctionType;     
    fftw_plan p;                        
    fftw_complex *complexIn;            
    fftw_complex *complexOut;           
    bool initialised;                   
    std::vector<double> frame;          
    std::vector<double> window;         
    double prevEnergySum;               
    std::vector<double> magSpec;        
    std::vector<double> prevMagSpec;    
    std::vector<double> phase;          
    std::vector<double> prevPhase;      
    std::vector<double> prevPhase2;     
};


#endif