ClassificationChain.cpp

Go to the documentation of this file.

#include "ClassificationChain.h"

#include "modules/Smoother.h"
#include "modules/GestureFinder.h"
#include "modules/DirectionFilter.h"
#include "modules/HMMModule.h"
#include "modules/OutputModule.h"

#include "framedata/VectorFrameData.h"

#include "util/Warning.h"
#include "util/WarningWrongFrameDataType.h"
#include <iostream>

ClassificationChain::ClassificationChain(AbstractMotionReceiver *r, ConfigManager *cfgManager) :
    m_classificationChain(NULL)
{
    // Set section config and default values if not present
    if (!cfgManager->hasSection("Smoother")) {
        cfgManager->addEmptySection("Smoother");
        cfgManager->getSection("Smoother").setItem("window size", "10"); 
    }
    if (!cfgManager->hasSection("Direction Filter")) {
        cfgManager->addEmptySection("Direction Filter");
        cfgManager->getSection("Direction Filter").setItem("threshold", "0.2"); 
    }
    if (!cfgManager->hasSection("Gesture Finder")) {
        cfgManager->addEmptySection("Gesture Finder");
        cfgManager->getSection("Gesture Finder").setItem("window size", "10"); 
        cfgManager->getSection("Gesture Finder").setItem("threshold", "0.3"); 
    }

    r->addObserver(this);
    m_classificationChain = new Smoother(&cfgManager->getSection("Smoother"),
            new GestureFinder(&cfgManager->getSection("Gesture Finder"),
                new DirectionFilter(&cfgManager->getSection("Direction Filter"),
                    new HMMModule(
                        new OutputModule()
                    )
                )
            )
        );
}

ClassificationChain::~ClassificationChain()
{
    if (m_classificationChain != NULL)
        delete m_classificationChain;
}

void ClassificationChain::update(const Vector3d& v)
{
    try {
        m_classificationChain->processFrameData(new VectorFrameData(v));
    } catch (WarningWrongFrameDataType &w) {
        std::cerr << std::endl << w.what() << std::endl;
    } catch (Warning &w) {
        std::cerr << std::endl << "Warning:" << std::endl << w.what() << std::endl;
    }
}

---