AveragePowerSpectrum.h

/* -*- C++ -*-.***************************************************************************************************
 Author: Ben Strutt
 Email: b.strutt.12@ucl.ac.uk

 Description: 
             A Class to manage power spectra, probably needs to do things like rolling averages.
*************************************************************************************************************** */

#ifndef AVERAGEPOWERSPECTRUM_H
#define AVERAGEPOWERSPECTRUM_H

#include "TGraph.h"
#include "TH1D.h"
#include "TH2D.h"
#include "TF1.h"
#include "TNamed.h"

#include "FancyFFTs.h"
#include "RootTools.h"
#include "CrossCorrelator.h"

#define NUM_AMPLITUDE_BINS 64
#define INITIAL_MAX_AMPLITUDE 4
#define INITIAL_MIN_AMPLITUDE 0

#define MAX_NUM_OUTLIERS 10

// NUM_SAMPLES is included from CrossCorrelator.h
#define NUM_FREQS ((NUM_SAMPLES/2)+1)


class AveragePowerSpectrum : public TNamed {

public:
  
  AveragePowerSpectrum();
  AveragePowerSpectrum(TString name, TString title);
  ~AveragePowerSpectrum();

  size_t add(TGraph* gr);
  
  TGraph* makeAvePowSpecTGraph();
  TGraph* makeAvePowSpecTGraph_dB();
  
  void deleteRayleighDistributions();
  void rebinAllRayleighHistograms(Int_t rebinFactor);

  
  void fitRayleighHistogramOverRange(Int_t freqInd, Double_t xLowVal, Double_t xHighVal,
                     Double_t* rAmplitudes,
                     Double_t* rChiSquares,
                     Int_t* rNdf,
                     Double_t* rChiSquaresFullRange,
                     Int_t* rNdfFullRange);


  void fitRayleighHistogram(Int_t freqInd);
  void fitRayleighHistogramRisingEdge(Int_t freqInd);
  void fitRayleighHistogramRisingEdgeAndHalfFallingEdge(Int_t freqInd);

  void fitAllRayleighHistograms();
  void fitAllRayleighHistogramsRisingEdge();
  void fitAllRayleighHistogramsRisingEdgeAndHalfFallingEdge();

  TH1D* getRayleighHistogram(Int_t freqInd);
  TH1D* getRayleighHistogramFromFrequencyMHz(Double_t freqMHz);
  TF1* getRayleighHistogramFit(Int_t freqInd);  

  TH2D* makeRayleigh2DHistogram();
  

  void getEventRayleighAmplitudes(TGraph* gr);
  
  static TF1* makeRayleighFunction(TString name, Double_t xMin, Double_t xMax);
  static TString getRayleighFunctionText();
  TF1* constructFitFromAmplitude(Int_t freqInd, Double_t amplitude);
  TF1* constructFitFromRayleighAmplitude(Int_t freqInd);
  TF1* constructFitFromRayleighAmplitudeRisingEdge(Int_t freqInd);
  TF1* constructFitFromRayleighAmplitudeRisingEdgeAndHalfFalling(Int_t freqInd);  

  Double_t deltaFMHz; 
  Double_t summedPowSpec[NUM_FREQS]; 
  TH1D* hRayleighs[NUM_FREQS]; 
  Int_t count; 
  
  Double_t rayleighFitChiSquares[NUM_FREQS]; 
  Double_t rayleighFitChiSquaresRisingEdge[NUM_FREQS]; 
  Double_t rayleighFitChiSquaresRisingEdgeAndHalfFalling[NUM_FREQS]; 

  Double_t rayleighAmplitudes[NUM_FREQS]; 
  Double_t rayleighAmplitudesRisingEdge[NUM_FREQS]; 
  Double_t rayleighAmplitudesRisingEdgeAndHalfFalling[NUM_FREQS]; 

  Int_t rayleighNdf[NUM_FREQS]; 
  Int_t rayleighNdfRisingEdge[NUM_FREQS]; 
  Int_t rayleighNdfRisingEdgeAndHalfFalling[NUM_FREQS]; 

  Double_t rayleighFitChiSquaresFullRange[NUM_FREQS]; 
  Double_t rayleighFitChiSquaresRisingEdgeFullRange[NUM_FREQS]; 
  Double_t rayleighFitChiSquaresRisingEdgeAndHalfFallingFullRange[NUM_FREQS]; 

  Int_t rayleighNdfFullRange[NUM_FREQS]; 
  Int_t rayleighNdfRisingEdgeFullRange[NUM_FREQS]; 
  Int_t rayleighNdfRisingEdgeAndHalfFallingFullRange[NUM_FREQS]; 

  Double_t xHigh[NUM_FREQS]; 
  Double_t xHighRisingEdge[NUM_FREQS]; 
  Double_t xHighRisingEdgeAndHalfFalling[NUM_FREQS]; 

  Double_t outliers[NUM_FREQS][MAX_NUM_OUTLIERS]; 
  Int_t numOutliers[NUM_FREQS]; 

  Double_t eventRayleighAmplitudes[NUM_FREQS]; 
  Double_t eventPowSpec[NUM_FREQS]; 

  ClassDef(AveragePowerSpectrum, 10); 
};

  



#endif