MovementDetector.h

Go to the documentation of this file.

#ifndef GRT_MOVEMENT_DETECTOR_HEADER
#define GRT_MOVEMENT_DETECTOR_HEADER

#include "../../CoreModules/MLBase.h"

GRT_BEGIN_NAMESPACE

class MovementDetector : public MLBase{
public:
    MovementDetector( const UINT numDimensions = 1, const Float upperThreshold = 1, const Float lowerThreshold = 0.9, const Float gamma = 0.95, const UINT searchTimeout = 0);
    
    virtual ~MovementDetector();
    
    virtual bool predict_( VectorFloat &input );
    
    virtual bool clear();
    
    virtual bool reset();
    
    virtual bool saveModelToFile( std::fstream &file ) const;
    
    virtual bool loadModelFromFile( std::fstream &file );
    
    Float getUpperThreshold() const;
    
    Float getLowerThreshold() const;
    
    Float getMovementIndex() const;
    
    Float getGamma() const;
    
    bool getMovementDetected() const;
    
    bool getNoMovementDetect() const;
    
    UINT getState() const;
    
    UINT getSearchTimeout() const;
    
    bool setUpperThreshold(const Float upperThreshold);
    
    bool setLowerThreshold(const Float lowerThreshold);
    
    bool setGamma(const Float gamma);
    
    bool setSearchTimeout(const UINT searchTimeout);
    
    //Tell the compiler we are using the base class train method to stop hidden virtual function warnings
    using MLBase::saveModelToFile;
    using MLBase::loadModelFromFile;
    
protected:

    UINT state;
    UINT searchTimeout;
    Float upperThreshold;
    Float lowerThreshold;
    Float movementIndex;
    Float gamma;
    bool firstSample;
    bool movementDetected;
    bool noMovementDetected;
    Timer searchTimer;
    VectorFloat lastSample;
    
public:
    enum SearchStates{SEARCHING_FOR_MOVEMENT=0,SEARCHING_FOR_NO_MOVEMENT,SEARCH_TIMEOUT};
    
};

GRT_END_NAMESPACE

#endif //GRT_MOVEMENT_DETECTOR_HEADER