QFFT.hh

Go to the documentation of this file.

 
#ifndef __QFFT_HH_
#define __QFFT_HH_
 
#include "QVector.hh"
#include "QVectorC.hh"
#include "QComplex.hh"
 
#include <gsl/gsl_fft_halfcomplex.h>
#include <gsl/gsl_fft_complex.h>
#include <gsl/gsl_errno.h>
#include <string>
#include <gsl/gsl_sf.h>
 
Q_BEGIN_NAMESPACE
 
class QFFT {
 
public:
  enum WindowType {
    WT_None = 0,
    WT_Welch = 1,
    WT_Rect = 2,
    WT_Hann = 3,
    WT_Hamming = 4,
    WT_Cosinus = 5,
    WT_BlackmanHarris = 6,
    WT_Kaiser3 = 7,
  }; 
  enum ZeroPadSide {
    kMiddle = 0,
    kSymmetric = 1,
    kLeft = 2,
    kRight = 3
  };
  enum FFTDirection {
    kForward = 1,
    kBackward = 0,
  };
  
  
  static QVector FFTAntiSym(const QVector& input);
  static QVector FFTSym(const QVector& input);
  static QVector ZeroPad(const QVector& input,size_t n_zeros, int Side = kMiddle, double zeroVal = 0.);
 
  static QVectorC ZeroPad(const QVectorC& input,size_t n_zeros, int Side = kMiddle, Diana::QComplex zeroVal = Diana::QComplex(0,0));
 
  static WindowType StrToWindowType(const std::string& winName);
  static QVector GetWindow(WindowType wt, size_t size, size_t zeros = 0,int coherent = 0, int param = 0);
  static QVector CutSides(const QVector& input, size_t ncut, bool isSym);
  
  QFFT( const size_t size);
  QFFT();
  virtual ~QFFT();
  virtual int Transform(const QVector& data, QVector& result) = 0;
  virtual bool Resize(size_t s) = 0;
  virtual void SetWindowType(WindowType wt, int coherent = false) = 0;
  int GetWindowType() const { return fWindowType; }
  
  void SetForward(bool isForward)
  { fDirection = (isForward ? kForward : kBackward); }
  void SetNormalized(bool isNormalized){fNormalized = isNormalized;}
  
  static bool IsPowerOf2(UInt_t Value);
  static bool IsPowerOf2(UInt_t Value,UInt_t &Power,
                         UInt_t &Remainder);
  
protected:
  
  void IsPowerOf2();
  size_t fSize; 
  FFTDirection fDirection;
  bool fNormalized;
  bool fIsPowerOf2;
  double* fData;
  double* fWindow;
  size_t fWindowSize;
  WindowType fWindowType;
  
  ClassDef(QFFT,0);
  
};
 
Q_END_NAMESPACE
#endif