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(const UINT fftWindowSize=512,const UINT numChannelsInFFTSignal=1,const bool computeMaxFreqFeature = true,const bool computeMaxFreqSpectrumRatio = true,const bool computeCentroidFeature = true,const bool computeTopNFreqFeatures = true,const 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 save( std::fstream &file ) const;
 
 virtual bool load( std::fstream &file );
 
 bool init(const UINT fftWindowSize,const UINT numChannelsInFFTSignal,const bool computeMaxFreqFeature,const bool computeMaxFreqSpectrumRatio,const bool computeCentroidFeature,const bool computeTopNFreqFeatures,const UINT N);
 
 //Tell the compiler we are using the following functions from the FeatureExtraction class to stop hidden virtual function warnings
 using MLBase::save;
 using MLBase::load;

 static std::string getId();
 
protected:
 UINT fftWindowSize;
 UINT numChannelsInFFTSignal;
 bool computeMaxFreqFeature;
 bool computeMaxFreqSpectrumRatio;
 bool computeCentroidFeature;
 bool computeTopNFreqFeatures;
 
 UINT N;
 Float maxFreqFeature;
 Float maxFreqSpectrumRatio;
 Float centroidFeature;
 VectorFloat topNFreqFeatures;
 
private:
 static RegisterFeatureExtractionModule< FFTFeatures > registerModule;
 static std::string id;
 
};

GRT_END_NAMESPACE

#endif //GRT_FFT_FEATURES_HEADER