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