Spectrum.h

Go to the documentation of this file.

 
#ifndef __Spectrum__
#define __Spectrum__
 
#include <TMath.h>
#include <TRandom.h>
#include <TH1.h>
#include <TGraph.h>
#include <TFile.h>
#include "FFT.h"
 
using namespace std;
 
class Spectrum{
 
 public:
 
  Spectrum(const unsigned int aSpectrumSize,
       const unsigned int aDataDuration, 
       const unsigned int aDataSamplingFrequency,
       const unsigned int aVerbosity=0);
 
  virtual ~Spectrum(void);
  void Reset(void);
  
  bool AddData(const unsigned int aDataSize,
               const double *aData,
               const unsigned int aDataStart=0);
 
  bool LoadData(const unsigned int aDataSize,
                const double *aData,
                const unsigned int aDataStart=0);
 
  double GetPower(const double aFrequency);
 
  inline double GetPower(const unsigned int aFrequencyIndex){
    return fft_psd->GetRe_f(aFrequencyIndex);
  };
 
  inline double GetAmplitude(const double aFrequency){
    return TMath::Sqrt(GetPower(aFrequency));
  };
 
  TGraph* GetPSD(void);
 
  TGraph* GetPSD(const double aFrequencyMin, const double aFrequencyMax);
 
  TGraph* GetASD(void);
 
  TGraph* GetASD(const double aFrequencyMin, const double aFrequencyMax);
 
  TGraph* GetPeriodogram(const unsigned int aParity, const unsigned int aIndex);
 
  TGraph* GetAmplitudePeriodogram(const unsigned int aParity, const unsigned int aIndex);
 
  bool WritePeriodogram(const string aOutFileName);
 
  inline unsigned int GetSpectrumSize(void){ return fft_psd->GetSize_f(); };
 
  inline double GetSpectrumFrequency(const unsigned int aIndex){
    return (double)aIndex*(double)sampling/(double)fft_psd->GetSize_t();
  };
 
  inline double GetSpectrumResolution(void){ return GetSpectrumFrequency(1); };
 
  inline unsigned int GetSpectrumNyquist(void){
    return (unsigned int)(GetSpectrumResolution()*(double)fft_psd->GetSize_f()); };
 
  inline unsigned int GetNSubSegmentsMax(const unsigned int aParity){
    return nSubSegments[aParity%2];
  };
 
  inline unsigned int GetDataBufferLength(void){ return (double)nSubSegments[0]/GetSpectrumResolution(); };
 
 private:
 
  unsigned int fVerbosity;     
  unsigned int sampling;       
 
  // derived parameters
  unsigned int nSubSegments[2];
 
  // containers
  double *HannWindow;          
  double **periodogram[2];     
  unsigned int write_index[2]; 
 
  fft *fft_psd;                
 
  void ComputeMedianMean(void);
 
  ClassDef(Spectrum,0)  
};
 
#endif