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 GRT_API MedianFilter : public PreProcessing {
    public:
    MedianFilter(const UINT filterSize = 5,const 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 save( std::fstream &file) const;
    
    virtual bool load( std::fstream &file );
    
    bool init(const UINT filterSize,const UINT numDimensions);
    
    Float filter(const Float x);
    
    VectorFloat filter(const VectorFloat &x);
    
    UINT getFilterSize() const;
    
    VectorFloat getFilteredData() const;
    
    Vector< VectorFloat > getDataBuffer() const;

    static std::string getId();
    
    //Tell the compiler we are using the following functions from the MLBase class to stop hidden virtual function warnings
    using MLBase::save;
    using MLBase::load;
    
protected:
    UINT filterSize;                                        
    UINT inputSampleCounter;                                
    CircularBuffer< VectorFloat > dataBuffer;           
    
private:
    static const std::string id;   
    static RegisterPreProcessingModule< MedianFilter > registerModule;
};

GRT_END_NAMESPACE

#endif //GRT_MEDIAN_FILTER_HEADER