MedianFilter.h

Go to the documentation of this file.

#ifndef GRT_MEDIAN_FILTER_HEADER
#define GRT_MEDIAN_FILTER_HEADER

#include "../CoreModules/PreProcessing.h"

GRT_BEGIN_NAMESPACE

class MedianFilter : public PreProcessing {
public:
    MedianFilter(UINT filterSize = 5,UINT numDimensions = 1);
    
    MedianFilter(const MedianFilter &rhs);

    virtual ~MedianFilter();
    
    MedianFilter& operator=(const MedianFilter &rhs);
    
    virtual bool deepCopyFrom(const PreProcessing *preProcessing);
    
    virtual bool process(const VectorFloat &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(UINT filterSize,UINT numDimensions);
    
    Float filter(const Float x);
    
    VectorFloat filter(const VectorFloat &x);

 UINT getFilterSize() const { return filterSize; }
    
    VectorFloat getFilteredData() const { return processedData; }
    
    Vector< VectorFloat > getDataBuffer() const;
    
protected:
    UINT filterSize;                                        
    UINT inputSampleCounter;                                
    CircularBuffer< VectorFloat > dataBuffer;           
    
    static RegisterPreProcessingModule< MedianFilter > registerModule;
};

GRT_END_NAMESPACE

#endif //GRT_MEDIAN_FILTER_HEADER