DoubleMovingAverageFilter.cpp

Go to the documentation of this file.

#define GRT_DLL_EXPORTS
#include "DoubleMovingAverageFilter.h"

GRT_BEGIN_NAMESPACE

//Define the string that will be used to identify the object
const std::string DoubleMovingAverageFilter::id = "DoubleMovingAverageFilter";
std::string DoubleMovingAverageFilter::getId() { return DoubleMovingAverageFilter::id; }

//Register the DoubleMovingAverageFilter module with the PreProcessing base class
RegisterPreProcessingModule< DoubleMovingAverageFilter > DoubleMovingAverageFilter::registerModule( DoubleMovingAverageFilter::getId() );

DoubleMovingAverageFilter::DoubleMovingAverageFilter(UINT filterSize,UINT numDimensions) : PreProcessing( DoubleMovingAverageFilter::getId() )
{
 init(filterSize,numDimensions);
}

DoubleMovingAverageFilter::DoubleMovingAverageFilter(const DoubleMovingAverageFilter &rhs) : PreProcessing( DoubleMovingAverageFilter::getId() )
{
 *this = rhs;
}

DoubleMovingAverageFilter::~DoubleMovingAverageFilter(){
 
}

DoubleMovingAverageFilter& DoubleMovingAverageFilter::operator=(const DoubleMovingAverageFilter &rhs){
 if(this!=&rhs){
 this->filterSize = rhs.filterSize;
 this->filter1 = rhs.filter1;
 this->filter2 = rhs.filter2;
 
 //Copy the base variables
 copyBaseVariables( (PreProcessing*)&rhs );
 }
 return *this;
}

bool DoubleMovingAverageFilter::deepCopyFrom(const PreProcessing *preProcessing){
 
 if( preProcessing == NULL ) return false;
 
 if( this->getId() == preProcessing->getId() ){
 
 const DoubleMovingAverageFilter *ptr = dynamic_cast<const DoubleMovingAverageFilter*>(preProcessing);
 
 //Clone the classLabelTimeoutFilter values
 this->filterSize = ptr->filterSize;
 this->filter1 = ptr->filter1;
 this->filter2 = ptr->filter2;
 
 //Clone the preprocessing base variables
 return copyBaseVariables( preProcessing );
 }
 
 errorLog << "deepCopyFrom(const PreProcessing *preProcessing) - PreProcessing Types Do Not Match!" << std::endl;
 
 return false;
}


bool DoubleMovingAverageFilter::process(const VectorFloat &inputVector){
 
 if( !initialized ){
 errorLog << "process(const VectorFloat &inputVector) - The filter has not been initialized!" << std::endl;
 return false;
 }
 
 if( inputVector.size() != numInputDimensions ){
 errorLog << "process(const VectorFloat &inputVector) - The size of the inputVector (" << inputVector.getSize() << ") does not match that of the filter (" << numInputDimensions << ")!" << std::endl;
 return false;
 }
 
 processedData = filter( inputVector );
 
 if( processedData.size() == numOutputDimensions ) return true;
 return false;
}

bool DoubleMovingAverageFilter::reset(){
 if( initialized ) return init(filterSize,numInputDimensions);
 return false;
}

bool DoubleMovingAverageFilter::save(std::fstream &file) const{
 
 if( !file.is_open() ){
 errorLog << "save(std::fstream &file) - The file is not open!" << std::endl;
 return false;
 }
 
 file << "GRT_DOUBLE_MOVING_AVERAGE_FILTER_FILE_V1.0" << std::endl;
 
 file << "NumInputDimensions: " << numInputDimensions << std::endl;
 file << "NumOutputDimensions: " << numOutputDimensions << std::endl;
 file << "FilterSize: " << filterSize << std::endl;
 
 return true;
}

bool DoubleMovingAverageFilter::load( std::fstream &file ){
 
 if( !file.is_open() ){
 errorLog << "load(std::fstream &file) - The file is not open!" << std::endl;
 return false;
 }
 
 std::string word;
 
 //Load the header
 file >> word;
 
 if( word != "GRT_DOUBLE_MOVING_AVERAGE_FILTER_FILE_V1.0" ){
 errorLog << "load(std::fstream &file) - Invalid file format!" << std::endl;
 return false;
 }
 
 //Load the number of input dimensions
 file >> word;
 if( word != "NumInputDimensions:" ){
 errorLog << "load(std::fstream &file) - Failed to read NumInputDimensions header!" << std::endl;
 return false;
 }
 file >> numInputDimensions;
 
 //Load the number of output dimensions
 file >> word;
 if( word != "NumOutputDimensions:" ){
 errorLog << "load(std::fstream &file) - Failed to read NumOutputDimensions header!" << std::endl;
 return false;
 }
 file >> numOutputDimensions;
 
 //Load the filter size
 file >> word;
 if( word != "FilterSize:" ){
 errorLog << "load(std::fstream &file) - Failed to read FilterSize header!" << std::endl;
 return false;
 }
 file >> filterSize;
 
 //Init the filter module to ensure everything is initialized correctly
 return init(filterSize,numInputDimensions);
}

bool DoubleMovingAverageFilter::init(const UINT filterSize,const UINT numDimensions){
 
 //Cleanup the old memory
 initialized = false;
 
 if( filterSize == 0 ){
 errorLog << "init(UINT filterSize,UINT numDimensions) - Filter size can not be zero!" << std::endl;
 return false;
 }
 
 if( numDimensions == 0 ){
 errorLog << "init(UINT filterSize,UINT numDimensions) - The number of dimensions must be greater than zero!" << std::endl;
 return false;
 }
 
 //Resize the filter
 this->filterSize = filterSize;
 numInputDimensions = numDimensions;
 numOutputDimensions = numDimensions;
 processedData.clear();
 processedData.resize(numDimensions,0 );
 
 if( !filter1.init(filterSize, numDimensions) ){
 errorLog << "init(UINT filterSize,UINT numDimensions) - Failed to initialize filter 1!" << std::endl;
 return false;
 }
 
 if( !filter2.init(filterSize, numDimensions) ){
 errorLog << "init(UINT filterSize,UINT numDimensions) - Failed to initialize filter 1!" << std::endl;
 return false;
 }
 
 //Flag that the filter has been initialized
 initialized = true;
 
 return true;
}

Float DoubleMovingAverageFilter::filter(const Float x){
 
 //If the filter has not been initialised then return 0, otherwise filter x and return y
 if( !initialized ){
 errorLog << "filter(const Float x) - The filter has not been initialized!" << std::endl;
 return 0;
 }
 
 VectorFloat y = filter(VectorFloat(1,x));
 
 if( y.getSize() == 0 ) return 0;
 return y[0];
}

VectorFloat DoubleMovingAverageFilter::filter(const VectorFloat &x){
 
 //If the filter has not been initialised then return 0, otherwise filter x and return y
 if( !initialized ){
 errorLog << "filter(const VectorFloat &x) - The filter has not been initialized!" << std::endl;
 return VectorFloat();
 }
 
 if( x.getSize() != numInputDimensions ){
 errorLog << "filter(const VectorFloat &x) - The size of the input vector (" << x.getSize() << ") does not match that of the number of dimensions of the filter (" << numInputDimensions << ")!" << std::endl;
 return VectorFloat();
 }
 
 //Perform the first filter
 VectorFloat y = filter1.filter( x );
 
 if( y.size() == 0 ) return y;
 
 //Perform the second filter
 VectorFloat yy = filter2.filter( y );
 
 if( yy.size() == 0 ) return y;
 
 //Account for the filter lag
 const UINT N = y.getSize();
 for(UINT i=0; i<N; i++){
 yy[i] = y[i] + (y[i] - yy[i]);
 processedData[i] = yy[i];
 }
 
 return yy;
}

VectorFloat DoubleMovingAverageFilter::getFilteredData() const { return processedData; }

GRT_END_NAMESPACE