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 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