FFTFeatures.h

Go to the documentation of this file.

#ifndef GRT_FFT_FEATURES_HEADER
#define GRT_FFT_FEATURES_HEADER

#include "../../CoreModules/FeatureExtraction.h"
#include "FFT.h"

GRT_BEGIN_NAMESPACE

class GRT_API FFTFeatures : public FeatureExtraction
{
public:
 FFTFeatures(UINT fftWindowSize=512,UINT numChannelsInFFTSignal=1,bool computeMaxFreqFeature = true,bool computeMaxFreqSpectrumRatio = true,bool computeCentroidFeature = true,bool computeTopNFreqFeatures = true,UINT N = 10);
 
 FFTFeatures(const FFTFeatures &rhs);
 
 virtual ~FFTFeatures(void);
 
 FFTFeatures& operator=(const FFTFeatures &rhs);

 virtual bool deepCopyFrom(const FeatureExtraction *featureExtraction);
 
 virtual bool computeFeatures(const VectorFloat &inputVector);
 
 virtual bool reset();
 
 virtual bool saveModelToFile( std::string filename ) const;
 
 virtual bool loadModelFromFile( std::string filename );
 
 virtual bool saveModelToFile( std::fstream &file ) const;
 
 virtual bool loadModelFromFile( std::fstream &file );
 
 bool init(UINT fftWindowSize,UINT numChannelsInFFTSignal,bool computeMaxFreqFeature,bool computeMaxFreqSpectrumRatio,bool computeCentroidFeature,bool computeTopNFreqFeatures,UINT N);
 
 //Tell the compiler we are using the following functions from the FeatureExtraction class to stop hidden virtual function warnings
 
protected:
 UINT fftWindowSize;
 UINT numChannelsInFFTSignal;
 bool computeMaxFreqFeature;
 bool computeMaxFreqSpectrumRatio;
 bool computeCentroidFeature;
 bool computeTopNFreqFeatures;
 
 UINT N;
 Float maxFreqFeature;
 Float maxFreqSpectrumRatio;
 Float centroidFeature;
 VectorFloat topNFreqFeatures;
 
 static RegisterFeatureExtractionModule< FFTFeatures > registerModule;
 
public:
};

GRT_END_NAMESPACE

#endif //GRT_FFT_FEATURES_HEADER