modules/Smoother.cpp

Go to the documentation of this file.

#include "Smoother.h"
#include <string>
#include "../framedata/VectorFrameData.h"
#include "../visualizer/SmootherGUI.h"
#include "../util/Error.h"

Smoother::Smoother(size_t windowSize, AbstractModule *successor) :
    ProcessModule(successor), m_windowSize(windowSize)
{
    setGUI(new SmootherGUI(this));
}

Smoother::Smoother(ConfigSection *config, AbstractModule *successor) :
    ProcessModule(successor)
{
    if (config == NULL) {
        throw Error("Smoother::Smoother(): Pointer to ConfigSection is NULL");
    }
    try {
        m_windowSize = config->get<int>("window size");
    } catch (Error e) { // TODO: ErrorNoConfigValue or sth
        config->setItem("window size", "10");
        m_windowSize = 10;
    }

    setGUI(new SmootherGUI(this));
}

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

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

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

    // append datum to window
    m_frameHistory.push_back(tmp);
    while (m_frameHistory.size() > m_windowSize) {
        m_frameHistory.pop_front();
    }

    // calculate average
    VectorList::reverse_iterator it =  m_frameHistory.rbegin();
    ++it; // skip last element as it is tmp
    for (; it != m_frameHistory.rend(); ++it) {
        tmp += *it;
    }
    tmp /= static_cast<double>(m_frameHistory.size());

    // gui output
    static_cast<SmootherGUI*>(getGUI())->addData(tmp);

    // send to next module
    static_cast<VectorFrameData*>(data)->setData(tmp);
    ProcessModule::processFrameData(data);
}

---