PostProcessing.h

Go to the documentation of this file.

#ifndef GRT_POST_PROCESSING_HEADER
#define GRT_POST_PROCESSING_HEADER

#include "MLBase.h"

GRT_BEGIN_NAMESPACE

class GRT_API PostProcessing : public MLBase
{
public:
 PostProcessing(void);
 
 virtual ~PostProcessing(void);
 
 virtual bool deepCopyFrom(const PostProcessing *postProcessing){ return false; }
 
 bool copyBaseVariables(const PostProcessing *postProcessingModule);

 virtual bool process(const VectorFloat &inputVector){ return false; }
 
 virtual bool reset(){ return true; }
 
 virtual bool saveModelToFile(std::string filename) const;
 
 virtual bool loadModelFromFile(std::string filename);
 
 virtual bool saveModelToFile(std::fstream &file) const{ return false; }
 
 virtual bool loadModelFromFile(std::fstream &file){ return false; }
 
 std::string getPostProcessingType() const;
 
 UINT getPostProcessingInputMode() const;
 
 UINT getPostProcessingOutputMode() const;
 
 UINT getNumInputDimensions() const;
 
 UINT getNumOutputDimensions() const;
 
 bool getInitialized() const;
 
 bool getIsPostProcessingInputModePredictedClassLabel() const;
 
 bool getIsPostProcessingInputModeClassLikelihoods() const;
 
 bool getIsPostProcessingOutputModePredictedClassLabel() const;
 
 bool getIsPostProcessingOutputModeClassLikelihoods() const;
 
 VectorFloat getProcessedData() const;
 
 typedef std::map< std::string, PostProcessing*(*)() > StringPostProcessingMap;
 
 static PostProcessing* createInstanceFromString(std::string const &postProcessingType);
 
 PostProcessing* createNewInstance() const;
 
 enum PostprocessingInputModes{INPUT_MODE_NOT_SET=0,INPUT_MODE_PREDICTED_CLASS_LABEL,INPUT_MODE_CLASS_LIKELIHOODS};
 enum PostprocessingOutputModes{OUTPUT_MODE_NOT_SET=0,OUTPUT_MODE_PREDICTED_CLASS_LABEL,OUTPUT_MODE_CLASS_LIKELIHOODS};
 
protected:
 bool init();
 
 bool savePostProcessingSettingsToFile(std::fstream &file) const;
 
 bool loadPostProcessingSettingsFromFile(std::fstream &file);

 std::string postProcessingType;
 bool initialized;
 UINT postProcessingInputMode;
 UINT postProcessingOutputMode;
 VectorFloat processedData;
 
 static StringPostProcessingMap *getMap() {
 if( !stringPostProcessingMap ){ stringPostProcessingMap = new StringPostProcessingMap; } 
 return stringPostProcessingMap; 
 }

private:
 static StringPostProcessingMap *stringPostProcessingMap;
 static UINT numPostProcessingInstances;
};

//These two functions/classes are used to register any new PostProcessing Module with the PostProcessing base class
template< typename T > PostProcessing *newPostProcessingModuleInstance() { return new T; }

template< typename T > 
class RegisterPostProcessingModule : PostProcessing { 
public:
 RegisterPostProcessingModule( std::string const &newPostProcessingModuleName ) { 
 getMap()->insert( std::pair< std::string, PostProcessing*(*)() >(newPostProcessingModuleName, &newPostProcessingModuleInstance< T > ) );
 }
};

GRT_END_NAMESPACE

#endif //GRT_POST_PROCESSING_HEADER