Sample.h

Go to the documentation of this file.

 
#ifndef __Sample__
#define __Sample__
 
#include "Monitor.h"
#include <TMath.h>
#include <TObject.h>
 
#define NORDER 20
#define NORDERHP 12
 
using namespace std;
 
class ZPGFilter {
 
  friend class IIRFilter;
 
 public:
 
  ZPGFilter(const unsigned int aNumZeros, const unsigned int aNumPoles);
  
  virtual ~ZPGFilter(void);
  void WToZ(void);
 
  inline void SetPole(const unsigned int aPoleIndex,
                      const double aPoleRe, const double aPoleIm){
    poles[0][aPoleIndex] = aPoleRe;
    poles[1][aPoleIndex] = aPoleIm;
  }
 
  inline void SetZero(const unsigned int aZeroIndex,
                      const double aZeroRe, const double aZeroIm){
    zeros[0][aZeroIndex] = aZeroRe;
    zeros[1][aZeroIndex] = aZeroIm;
  }
 
  inline void SetGain(const double aGainRe, const double aGainIm){
    gain[0] = aGainRe;
    gain[1] = aGainIm;
  }
 
 
private:
  
  unsigned int nzeros; 
  unsigned int npoles; 
  double *zeros[2];    
  double *poles[2];    
  double gain[2];      
};
 
 
class IIRFilter{
 
 
public:
  
  IIRFilter(ZPGFilter *aZpgFilter);
 
  virtual ~IIRFilter(void);
  void Apply(const unsigned int aSize, double *aData);
 
 
private:
 
  unsigned int nzeros;            
  unsigned int npoles;            
  double *zeros[2];               
  double *poles[2];               
  double gain[2];                 
  unsigned int numDirect;         
  unsigned int numRecurs;         
  double *directCoef;             
  double *recursCoef;             
  double *history;                
  double *history_init;           
 
};
 
class Sample{
 
 public:
 
  Sample(const unsigned int aVerbosity=0);
 
  virtual ~Sample(void);
  inline bool GetStatus(void){ return mon->GetStatus(); };
 
  bool Transform(const unsigned int aInSize, double *aInData, const unsigned int aOutSize, double *aOutData);
 
  inline void SetDCRemoval(const bool aDCremove=true){ fRemoveDC=aDCremove; };
 
  inline void SetWindow(const unsigned int aWindowSize, double *aWindow){
    fWindowSize = aWindowSize;
    fWindow = aWindow;
  };
 
  inline unsigned int GetWindowSize(void){ return fWindowSize; };
  
  bool SetFrequencies(const unsigned int aNativeFrequency,
                      const unsigned int aWorkingFrequency,
                      const double aHighPassFrequency=0.0);
 
  inline bool SetNativeFrequency(const int aNativeFrequency){
    return SetFrequencies(aNativeFrequency,fWorkingFrequency,fHighPassFrequency);
  };
 
  inline bool SetWorkingFrequency(const unsigned int aWorkingFrequency){
    return SetFrequencies(fNativeFrequency,aWorkingFrequency,fHighPassFrequency);
  };
 
  inline bool SetHighPassFrequency(const double aHighPassFrequency){
      return SetFrequencies(fNativeFrequency,fWorkingFrequency,aHighPassFrequency);
  };
 
  inline unsigned int GetNativeFrequency(void){ return fNativeFrequency; };
 
  inline unsigned int GetWorkingFrequency(void){ return fWorkingFrequency; };
 
  inline double GetHighPassFrequency(void){ return fHighPassFrequency; };
 
 protected:
 
  Monitor *mon;                  
  bool fRemoveDC;                
  unsigned int fWorkingFrequency;
  unsigned int fNativeFrequency; 
  double fHighPassFrequency;     
  double *fWindow;               
  unsigned int fWindowSize;      
 
 private:
 
  bool presample;                     
  unsigned int fResampleFactor;       
  unsigned int fIntermediateFrequency;
  IIRFilter *iirFilter[NORDER];       
  IIRFilter *iirFilterHP[NORDERHP];   
 
  void MakeSamplingFilters(void);
 
  void MakeHighPassFilters(void);
 
  void RemoveDC(const unsigned int aSize, double *aData);
 
  void HighPass(const unsigned int aSize, double *aData);
 
  void PreSample(unsigned int &aSize, double *aData);
 
  void Resample(unsigned int aSize, double *aInData, double *aOutData);
  
  ClassDef(Sample,0)
};
 
#endif