MovementTrajectoryFeatures.h

Go to the documentation of this file.

#ifndef GRT_MOVEMENT_TRAJECTORY_FEATURES_HEADER
#define GRT_MOVEMENT_TRAJECTORY_FEATURES_HEADER

#include "../../CoreModules/FeatureExtraction.h"
#include "../../Util/Util.h"

GRT_BEGIN_NAMESPACE
    
struct AngleMagnitude{
    AngleMagnitude(){
        angle = 0;
        magnitude = 0;
    }
    Float angle;
    Float magnitude;
};
typedef struct AngleMagnitude AngleMagnitude;
    
class MovementTrajectoryFeatures : public FeatureExtraction{
public:
    MovementTrajectoryFeatures(UINT trajectoryLength=100,UINT numCentroids=10,UINT featureMode=CENTROID_VALUE,UINT numHistogramBins=10,UINT numDimensions = 1,bool useTrajStartAndEndValues = false,bool useWeightedMagnitudeValues = true);
 
    MovementTrajectoryFeatures(const MovementTrajectoryFeatures &rhs);
    
    virtual ~MovementTrajectoryFeatures();
    
    MovementTrajectoryFeatures& operator=(const MovementTrajectoryFeatures &rhs);

    virtual bool deepCopyFrom(const FeatureExtraction *featureExtraction);
    
    virtual bool computeFeatures(const VectorFloat &inputVector);
    
    virtual bool reset();
    
    virtual bool saveModelToFile( std::string filename ) const;
    
    virtual bool loadModelFromFile( std::string filename );
    
    virtual bool saveModelToFile( std::fstream &file ) const;
    
    virtual bool loadModelFromFile( std::fstream &file );

    bool init(UINT trajectoryLength,UINT numCentroids,UINT featureMode,UINT numHistogramBins,UINT numDimensions,bool useTrajStartAndEndValues,bool useWeightedMagnitudeValues);
    
 VectorFloat update(Float x);
    
    VectorFloat update(const VectorFloat &x);
    
    CircularBuffer< VectorFloat > getTrajectoryData();
    
    MatrixDouble getCentroids();
    
    UINT getFeatureMode();
    
    //Tell the compiler we are using the following functions from the MLBase class to stop hidden virtual function warnings
    using MLBase::train;
    using MLBase::train_;
    using MLBase::predict;
    using MLBase::predict_;

protected:
    
    UINT trajectoryLength;
    UINT numCentroids;
    UINT featureMode;
    UINT numHistogramBins;
    bool useTrajStartAndEndValues;
    bool useWeightedMagnitudeValues;
    CircularBuffer< VectorFloat > trajectoryDataBuffer;
    MatrixDouble centroids;
    
    static RegisterFeatureExtractionModule< MovementTrajectoryFeatures > registerModule;
    
public:
    enum FeatureModes{CENTROID_VALUE=0,NORMALIZED_CENTROID_VALUE,CENTROID_DERIVATIVE,CENTROID_ANGLE_2D};
};

GRT_END_NAMESPACE

#endif //GRT_MOVEMENT_TRAJECTORY_FEATURES_HEADER