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