SavitzkyGolayFilter.h

Go to the documentation of this file.

#ifndef GRT_SAVITZKY_GOLAY_FILTER_HEADER
#define GRT_SAVITZKY_GOLAY_FILTER_HEADER

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

GRT_BEGIN_NAMESPACE
 
class SavitzkyGolayFilter : public PreProcessing{
public:
 SavitzkyGolayFilter(UINT numLeftHandPoints=10,UINT numRightHandPoints=10,UINT derivativeOrder=0,UINT smoothingPolynomialOrder=2,UINT numDimensions = 1);
 
 SavitzkyGolayFilter(const SavitzkyGolayFilter &rhs);
 
 virtual ~SavitzkyGolayFilter();
 
 SavitzkyGolayFilter& operator=(const SavitzkyGolayFilter &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 numLeftHandPoints,UINT numRightHandPoints,UINT derivativeOrder,UINT smoothingPolynomialOrder,UINT numDimensions);
 
 Float filter(const Float x);
 
 VectorFloat filter(const VectorFloat &x);
 
 VectorFloat getFilteredData() const { return processedData; }

protected:
 inline int min_(int a,int b) {return b < a ? (b) : (a);}
 inline Float min_(Float a,Float b) {return b < a ? (b) : (a);}
 bool calCoeff();
 
 UINT numPoints; //The physical length of the output array
 UINT numLeftHandPoints; //Num of leftward (past) points to use
 UINT numRightHandPoints; //Num of rightward (future) points to use
 UINT derivativeOrder; //Order of the derivative desired
 UINT smoothingPolynomialOrder; //Order of smoothing polynomial
 CircularBuffer< VectorFloat > data; //A buffer to hold the input data
 VectorFloat yy; //The filtered values
 VectorFloat coeff; //Buffer for the filter coefficients
 
 static RegisterPreProcessingModule< SavitzkyGolayFilter > registerModule;
 
};

GRT_END_NAMESPACE

#endif //GRT_SAVITZKY_GOLAY_FILTER_HEADER