ClassLabelTimeoutFilter.h

Go to the documentation of this file.

#ifndef GRT_CLASS_LABEL_TIMEOUT_FILTER_HEADER
#define GRT_CLASS_LABEL_TIMEOUT_FILTER_HEADER

#include "../Util/GRTCommon.h"
#include "../CoreModules/PostProcessing.h"

GRT_BEGIN_NAMESPACE
 
class GRT_API ClassLabelAndTimer{
public:
 ClassLabelAndTimer(){
 classLabel = 0;
 }
 ClassLabelAndTimer(UINT classLabel,unsigned long timeoutDuration){
 this->classLabel = classLabel;
 timer.start(timeoutDuration);
 }
 
 //Setters
 bool set(UINT classLabel,unsigned long timeoutDuration){
 if( classLabel > 0 && timeoutDuration > 0 ){
 this->classLabel = classLabel;
 timer.start(timeoutDuration);
 return true;
 }
 return false;
 }
 
 //Getters
 UINT getClassLabel(){ return classLabel; }
 bool timerReached(){ return timer.timerReached(); }
 unsigned long getRemainingTime(){ return timer.getMilliSeconds(); }
 
protected:
 UINT classLabel;
 Timer timer;
};
 
class GRT_API ClassLabelTimeoutFilter : public PostProcessing{
public:
 ClassLabelTimeoutFilter(unsigned long timeoutDuration = 1000,UINT filterMode = ALL_CLASS_LABELS);
 
 ClassLabelTimeoutFilter(const ClassLabelTimeoutFilter &rhs);
 
 virtual ~ClassLabelTimeoutFilter();
 
 ClassLabelTimeoutFilter& operator=(const ClassLabelTimeoutFilter &rhs);
 
 virtual bool deepCopyFrom(const PostProcessing *postProcessing);
 
 virtual bool process(const VectorDouble &inputVector);
 
 virtual bool reset();
 
 virtual bool saveModelToFile( std::string filename ) const;
 
 virtual bool saveModelToFile( std::fstream &file ) const;
 
 virtual bool loadModelFromFile( std::string filename );
 
 virtual bool loadModelFromFile( std::fstream &file );

 bool init(unsigned long timeoutDuration,UINT filterMode = ALL_CLASS_LABELS); 
 
 UINT filter(UINT predictedClassLabel);
 
 UINT getFilteredClassLabel(){ return filteredClassLabel; }
 
 bool isTimeoutActive();
 
 bool setTimeoutDuration(unsigned long timeoutDuration);
 
 bool setFilterMode(UINT filterMode);
 
 enum FilterModes{ALL_CLASS_LABELS=0,INDEPENDENT_CLASS_LABELS};
 
protected:
 UINT filteredClassLabel;
 UINT filterMode;
 unsigned long timeoutDuration;
 Vector< ClassLabelAndTimer > classLabelTimers;
 
 static RegisterPostProcessingModule< ClassLabelTimeoutFilter > registerModule;
};

GRT_END_NAMESPACE

#endif //GRT_CLASS_LABEL_TIMEOUT_FILTER_HEADER