modules/GestureFinder.cpp

Go to the documentation of this file.

#include "GestureFinder.h"
#include "../framedata/VectorFrameData.h"
#include "../framedata/SignalFrameData.h"
#include "../visualizer/GestureFinderGUI.h"
#include "../util/Error.h"
#include <iostream>

GestureFinder::GestureFinder(size_t maxFramesDropped, double threshold, AbstractModule *successor) :
    ProcessModule(successor), 
    m_droppedVectors(maxFramesDropped+1), // make sure first gesture will be recognized
    m_maxFramesDropped(maxFramesDropped), 
    m_threshold(threshold)
{
    setGUI(new GestureFinderGUI(this));
}

GestureFinder::GestureFinder(ConfigSection *config, AbstractModule *successor) :
    ProcessModule(successor)
{
    if (config == NULL) {
        throw Error("GestureFinder::GestureFinder(): Pointer to ConfigSection is NULL");
    }
    try {
        m_maxFramesDropped = config->get<int>("max frames dropped"); 
    } catch (Error e) { // TODO: right exception type
        config->setItem("max frames dropped", "5");
        m_maxFramesDropped = 5;
    }
    m_droppedVectors = m_maxFramesDropped+1;
    try {
        m_threshold = config->get<double>("threshold");
    } catch (Error e) {
        config->setItem("threshold", "1.2");
        m_threshold = 1.2;
    }

    setGUI(new GestureFinderGUI(this)); 
}

void GestureFinder::processFrameData(IFrameData *data)
{
    if (isSignal(data)) {
        passSignal(data);
        return;
    }

    assertFramedataType(data, "Vector3D", "GestureFinder::processFrameData");

    Vector3d acc = static_cast<VectorFrameData*>(data)->getData();

    // if vector length > threshold, pass frame on
    if ((acc[0] * acc[0] + acc[1] * acc[1] + acc[2] * acc[2]) > m_threshold * m_threshold) {
        // gesture started here
        if (m_droppedVectors > m_maxFramesDropped) {
            std::cout << "send Signal start gesture" << std::endl;
            ProcessModule::processFrameData(new SignalFrameData(SignalFrameData::SIG_START_GESTURE));
        }
        m_droppedVectors = 0;
        ProcessModule::processFrameData(data);
        return;
    }
    
    // if not, drop frame, increase counter and send signal if neccesary
    delete data;
    ++m_droppedVectors;
    if (m_droppedVectors == m_maxFramesDropped) {
        std::cout << "send Signal stop gesture" << std::endl;
        ProcessModule::processFrameData(new SignalFrameData(SignalFrameData::SIG_STOP_GESTURE));
    }
}

---