/afs/in2p3.fr/throng/virgo/virgoDev/SoftFlorent/GWOLLUM/v2r2/src/Streams/Spectrum.h

Go to the documentation of this file.

//  Author : florent robinet (LAL - Orsay): robinet@lal.in2p3.fr
#ifndef __Spectrum__
#define __Spectrum__

#include "CUtils.h"
#include "FFT.h"
#include "TMath.h"
#include "TGraph.h"
#include "TH1.h"
#include "TFile.h"
#include <gsl/gsl_errno.h>
#include <gsl/gsl_spline.h>

using namespace std;

class Spectrum{

 public:

  Spectrum(const int aSpectrumSize,
       const int aDataDuration, 
       const int aDataSamplingFrequency,
       const int aVerbosity=0);

  virtual ~Spectrum(void);
  void Reset(void);
  
  bool AddData(const int aDataSize, const double *aData, const int aDataStart=0);

  bool LoadData(const int aDataSize, const double *aData, const int aDataStart=0);

  double GetPower(const double aFrequency);

  TGraph* GetPSD(void);

  TGraph* GetPSD(const double aFrequencyMin, const double aFrequencyMax);

  TGraph* GetASD(void);

  TGraph* GetASD(const double aFrequencyMin, const double aFrequencyMax);

  TGraph* GetSubPSD(const int aParity, const int aIndex);

  TGraph* GetSubASD(const int aParity, const int aIndex);

  bool WriteSubPSD(const string aOutFileName);

  bool WriteSubASD(const string aOutFileName);

  inline int GetSpectrumSize(void){ return fft_psd->GetSize_f(); };

  inline double GetSpectrumFrequency(const int aIndex){ return f_PSD[aIndex]; };

  inline double GetSpectrumResolution(void){ return (double)DataSamplingFrequency/2.0/(double)fft_psd->GetSize_f(); };

  inline int GetSpectrumNyquist(void){ return DataSamplingFrequency/2; };

  inline bool GetStatus(void){ return status_OK; };

  inline int GetNSubSegmentsMax(const int aParity){
    return nSubSegments[aParity%2];
  };

  inline int GetDataBufferLength(void){ return nSubSegments[0]*(fft_psd->GetSize_t()/DataSamplingFrequency); };

  inline bool IsBufferEmpty(void){
    if(stop_buffer[0]<0&&stop_buffer[1]<0) return true;
    return false;
  };


 private:

  bool status_OK;              
  int fVerbosity;              

  // input parameters
  int DataSamplingFrequency;   

  // derived parameters
  int nSubSegments[2];         

  // containers
  double *HannWindow;          
  double *PSD;                 
  double *f_PSD;               
  double **PSD_buffer[2];      
  bool   *flag_buffer[2];      
  int    stop_buffer[2];       

  fft *fft_psd;                

  gsl_interp_accel *acc;       
  gsl_spline *interp_psd;      
  
  double* GetHannWindow(const int aSize);
  void ComputeMedianMean(void);
  
  ClassDef(Spectrum,0)  
};

#endif

---