cannam@52: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
cannam@52: 
cannam@52: /*
cannam@52:     QM DSP Library
cannam@52: 
cannam@52:     Centre for Digital Music, Queen Mary, University of London.
cannam@52:     This file copyright 2008-2009 Matthew Davies and QMUL.
cannam@52:     All rights reserved.
cannam@52: */
cannam@52: 
cannam@52: 
cannam@52: #ifndef TEMPOTRACKV2_H
cannam@52: #define TEMPOTRACKV2_H
cannam@52: 
cannam@52: #include <vector>
cannam@52: 
cannam@52: using std::vector;
cannam@52: 
cannam@52: class TempoTrackV2  
cannam@52: {
cannam@52: public:
cannam@52:     TempoTrackV2();
cannam@52:     ~TempoTrackV2();
cannam@52: 
cannam@52:     void calculateBeatPeriod(const vector<double> &df,
cannam@53:                              vector<double> &beatPeriod,
cannam@53:                              vector<double> &tempi);
cannam@52: 
cannam@52:     void calculateBeats(const vector<double> &df,
cannam@52:                         const vector<double> &beatPeriod,
cannam@52:                         vector<double> &beats);
cannam@52: 
cannam@52: private:
cannam@52:     typedef vector<int> i_vec_t;
cannam@52:     typedef vector<vector<int> > i_mat_t;
cannam@52:     typedef vector<double> d_vec_t;
cannam@52:     typedef vector<vector<double> > d_mat_t;
cannam@52: 
cannam@52:     void adapt_thresh(d_vec_t &df);
cannam@52:     double mean_array(const d_vec_t &dfin, int start, int end);
cannam@52:     void filter_df(d_vec_t &df);
cannam@52:     void get_rcf(const d_vec_t &dfframe, const d_vec_t &wv, d_vec_t &rcf);
cannam@53:     void viterbi_decode(const d_mat_t &rcfmat, const d_vec_t &wv,
cannam@53:                         d_vec_t &bp, d_vec_t &tempi);
cannam@52:     double get_max_val(const d_vec_t &df);
cannam@52:     int get_max_ind(const d_vec_t &df);
cannam@52:     void normalise_vec(d_vec_t &df);
cannam@52: };
cannam@52: 
cannam@52: #endif