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( const std::string &id = "" );
 
 virtual ~PostProcessing(void);
    
 virtual bool deepCopyFrom(const PostProcessing *postProcessing){ return false; }
    
 bool copyBaseVariables(const PostProcessing *postProcessingModule);

    virtual bool process(const VectorFloat &inputVector){ return false; }
    
 UINT getPostProcessingInputMode() const;
    
 UINT getPostProcessingOutputMode() 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* create(const std::string &postProcessingType);
    
    PostProcessing* create() 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};

    GRT_DEPRECATED_MSG( "createNewInstance is deprecated, use create instead.", PostProcessing* createNewInstance() const );
    GRT_DEPRECATED_MSG( "createInstanceFromString is deprecated, use create instead.", static PostProcessing* createInstanceFromString( const std::string &id ) );
    GRT_DEPRECATED_MSG( "getPostProcessingType is deprecated, use getId() instead", std::string getPostProcessingType() const );
    
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;
};

template< typename T >  PostProcessing *createNewPostProcessingModule() { return new T; } 

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

GRT_END_NAMESPACE

#endif //GRT_POST_PROCESSING_HEADER