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 GRT_API MovementTrajectoryFeatures : public FeatureExtraction{
public:
 enum FeatureModes{CENTROID_VALUE=0,NORMALIZED_CENTROID_VALUE,CENTROID_DERIVATIVE,CENTROID_ANGLE_2D};

 MovementTrajectoryFeatures(const UINT trajectoryLength=100,const UINT numCentroids=10,const UINT featureMode=CENTROID_VALUE,const UINT numHistogramBins=10,const UINT numDimensions = 1,const bool useTrajStartAndEndValues = false,const 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 save( std::fstream &file ) const;
 
 virtual bool load( std::fstream &file );
 
 bool init(const UINT trajectoryLength,const UINT numCentroids,const UINT featureMode,const UINT numHistogramBins,const UINT numDimensions,const bool useTrajStartAndEndValues,const bool useWeightedMagnitudeValues);
 
 VectorFloat update(const Float x);
 
 VectorFloat update(const VectorFloat &x);
 
 CircularBuffer< VectorFloat > getTrajectoryData() const;
 
 MatrixDouble getCentroids() const;
 
 UINT getFeatureMode() const;
 
 //Tell the compiler we are using the following functions from the MLBase class to stop hidden virtual function warnings
 using MLBase::save;
 using MLBase::load;
 using MLBase::train;
 using MLBase::train_;
 using MLBase::predict;
 using MLBase::predict_;

 static std::string getId();
 
protected:
 
 UINT trajectoryLength;
 UINT numCentroids;
 UINT featureMode;
 UINT numHistogramBins;
 bool useTrajStartAndEndValues;
 bool useWeightedMagnitudeValues;
 CircularBuffer< VectorFloat > trajectoryDataBuffer;
 MatrixDouble centroids;
 
private:
 static RegisterFeatureExtractionModule< MovementTrajectoryFeatures > registerModule;
 static std::string id;
};

GRT_END_NAMESPACE

#endif //GRT_MOVEMENT_TRAJECTORY_FEATURES_HEADER