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