PreProcessing.h

Go to the documentation of this file.

#ifndef GRT_PRE_PROCESSING_HEADER
#define GRT_PRE_PROCESSING_HEADER

#include "MLBase.h"

GRT_BEGIN_NAMESPACE

class PreProcessing : public MLBase
{
public:
 PreProcessing(void);
 
 virtual ~PreProcessing(void);
    
    virtual bool deepCopyFrom(const PreProcessing *rhs){ return false; }
    
    bool copyBaseVariables(const PreProcessing *preProcessingModule);

    virtual bool process(const VectorFloat &inputVector){ return false; }
    
    virtual bool reset();
    
    virtual bool clear();
    
    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 getPreProcessingType() const;
    
    UINT getNumInputDimensions() const;
    
    UINT getNumOutputDimensions() const;
    
    bool getInitialized() const;

 VectorFloat getProcessedData() const;
    
 typedef std::map< std::string, PreProcessing*(*)() > StringPreProcessingMap;
    
    static PreProcessing* createInstanceFromString(std::string const &preProcessingType);
    
    PreProcessing* createNewInstance() const;
    
protected:
    bool init();
    
    bool savePreProcessingSettingsToFile(std::fstream &file) const;
    
    bool loadPreProcessingSettingsFromFile(std::fstream &file);

    std::string preProcessingType;
    bool initialized;
    VectorFloat processedData;

 static StringPreProcessingMap *getMap() {
        if( !stringPreProcessingMap ){ stringPreProcessingMap = new StringPreProcessingMap; } 
        return stringPreProcessingMap; 
    }

private:
    static StringPreProcessingMap *stringPreProcessingMap;
    static UINT numPreProcessingInstances;
};

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

template< typename T > 
class RegisterPreProcessingModule : PreProcessing { 
public:
    RegisterPreProcessingModule( const std::string &newPreProcessingModuleName ) { 
        getMap()->insert( std::pair< std::string, PreProcessing*(*)() >(newPreProcessingModuleName, &newPreProcessingModuleInstance< T > ) );
    }
};

GRT_END_NAMESPACE

#endif // GRT_POST_PROCESSING_HEADER