modules/HMMModule.h

Go to the documentation of this file.

#ifndef HMMMODULE_H
#define HMMMODULE_H

#include "ProcessModule.h"
#include "../framedata/IFrameData.h"
#include "../classification/KMeanCluster.h"
#include "../classification/LeftToRightHMM.h"
#include "../util/ConfigManager.h"
#include <vector>
#include <list>
#include <utility>


class HMMModule : public ProcessModule
{
    public:
        HMMModule(AbstractModule *successor);

        virtual ~HMMModule();

        virtual void processFrameData(IFrameData *data);

        int addGesture(int nClusters, int nStates);

        void delGesture(size_t gestureID);

        void clearGestures();

        void trainGesture(size_t gestureID);

        void stopTraining();

        bool serializeGesture(ConfigManager &gestureFile, std::string gestureName, int gestureId);

        bool deserializeGesture(ConfigSection &section, std::string gestureName);

    private:
        typedef std::pair<KMeanCluster*, LeftToRightHMM*> EvalPair;
        typedef std::vector<EvalPair> Evaluators;
        typedef std::list<Vector3d> Gesture;

        void classifyRecordedGesture();
        void trainRecordedGesture();

        void addGesure(KMeanCluster *clusterizer, LeftToRightHMM *hmm);

        Evaluators m_gestureClassificators;
        bool m_hasGestureStarted;
        Gesture m_gesture;

        size_t m_trainGestureID;
};

#endif

---