view SparseHMM.h @ 130:080fe18f5ebf fixedlag

refactored Viterbi

* perhaps I even discovered a bug (probablity sum was not reset for every frame)

/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */

/*
    pYIN - A fundamental frequency estimator for monophonic audio
    Centre for Digital Music, Queen Mary, University of London.
    
    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public License as
    published by the Free Software Foundation; either version 2 of the
    License, or (at your option) any later version.  See the file
    COPYING included with this distribution for more information.
*/

#ifndef _SPARSEHMM_H_
#define _SPARSEHMM_H_

#include <vector>
#include <queue>
#include <cstdio>

using std::vector;
using std::deque;
using std::pair;

class SparseHMM
{
public:
    SparseHMM();
    virtual const std::vector<double>
                           calculateObsProb(const vector<pair<double, double> >);
    virtual void           build();
    const std::vector<int> decodeViterbi(std::vector<vector<double> > obs);
    void                   initialise(vector<double> firstObs);
    int                    process(vector<double> newObs);
    vector<int>            finalise();
    // "sparse" HMM definition
    int m_nState;
    int m_nTrans;
    vector<double> m_init;
    vector<size_t> m_from;
    vector<size_t> m_to;
    vector<double> m_transProb;

    // variables for decoding
    deque<double> m_scale;
    deque<vector<int> > m_psi;
    vector<double> m_delta;
    vector<double> m_oldDelta;
};

#endif