BTrack.h

Go to the documentation of this file.

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

#ifndef __BTRACK_H
#define __BTRACK_H

#include "OnsetDetectionFunction.h"

//=======================================================================
class BTrack {
    
public:
    
    //=======================================================================
    BTrack();
    
    BTrack(int hopSize_);
    
    BTrack(int hopSize_,int frameSize_);
    
    //=======================================================================
    void processAudioFrame(double *frame);
    
    void processOnsetDetectionFunctionSample(double sample);
   
    //=======================================================================
    int getHopSize();
    
    bool beatDueInCurrentFrame();

    double getCurrentTempoEstimate();
    
    double getLatestCumulativeScoreValue();
    
    //=======================================================================
    void setTempo(double tempo);
    
    void fixTempo(double tempo);
    
    void doNotFixTempo();
    
    //=======================================================================
    static double getBeatTimeInSeconds(long frameNumber,int hopSize,int fs);
    
    static double getBeatTimeInSeconds(int frameNumber,int hopSize,int fs);
    
        
private:
    
    void initialise(int hopSize_,int frameSize_);
    
    void setHopSize(int hopSize_);
    
    void resampleOnsetDetectionFunction();
    
    void updateCumulativeScore(double odfSample);
    
    void predictBeat();
    
    void calculateTempo();
    
    void adaptiveThreshold(double *x,int N);
    
    double calculateMeanOfArray(double *array,int startIndex,int endIndex);
    
    void normaliseArray(double *array,int N);
    
    void calculateBalancedACF(double *onsetDetectionFunction);
    
    void calculateOutputOfCombFilterBank();
    
    //=======================================================================

    OnsetDetectionFunction odf;
    
    //=======================================================================
    // buffers
    double *onsetDF;                        
    double resampledOnsetDF[512];           
    double *cumulativeScore;                
    double acf[512];                        
    double weightingVector[128];            
    double combFilterBankOutput[128];       
    double tempoObservationVector[41];      
    double delta[41];                       
    double prevDelta[41];                   
    double prevDeltaFixed[41];              
    double tempoTransitionMatrix[41][41];   
    //=======================================================================
    // parameters
    
    
    double tightness;                       
    double alpha;                           
    double beatPeriod;                      
    double tempo;                           
    double estimatedTempo;                  
    double latestCumulativeScoreValue;      
    double tempoToLagFactor;                
    int m0;                                 
    int beatCounter;                        
    int hopSize;                            
    int onsetDFBufferSize;                  
    bool tempoFixed;                        
    bool beatDueInFrame;                    
};

#endif