| 1 | #ifndef MARS_MHGausEvents
|
|---|
| 2 | #define MARS_MHGausEvents
|
|---|
| 3 |
|
|---|
| 4 | #ifndef ROOT_TH1
|
|---|
| 5 | #include <TH1.h>
|
|---|
| 6 | #endif
|
|---|
| 7 |
|
|---|
| 8 | #ifndef MARS_MH
|
|---|
| 9 | #include "MH.h"
|
|---|
| 10 | #endif
|
|---|
| 11 |
|
|---|
| 12 | class TVirtualPad;
|
|---|
| 13 | class TGraph;
|
|---|
| 14 | class TArrayF;
|
|---|
| 15 | class TH1F;
|
|---|
| 16 | class TH1I;
|
|---|
| 17 | class TF1;
|
|---|
| 18 | class MHGausEvents : public MH
|
|---|
| 19 | {
|
|---|
| 20 | private:
|
|---|
| 21 |
|
|---|
| 22 | const static Int_t fgBinsAfterStripping; //! Default for fBinsAfterStripping (now set to: 40)
|
|---|
| 23 | const static Float_t fgBlackoutLimit; //! Default for fBlackoutLimit (now set to: 5. )
|
|---|
| 24 | const static Int_t fgNDFLimit; //! Default for fNDFLimit (now set to: 2)
|
|---|
| 25 | const static Float_t fgProbLimit; //! Default for fProbLimit (now set to: 0.001)
|
|---|
| 26 | const static Float_t fgPickupLimit; //! Default for fPickupLimit (now set to: 5. )
|
|---|
| 27 | const static Int_t fgPowerProbabilityBins; //! Default for fPowerProbabilityBins (now set to: 20)
|
|---|
| 28 |
|
|---|
| 29 | Int_t fBinsAfterStripping; // Bins for the Gauss Histogram after stripping off the zeros at both ends
|
|---|
| 30 | Int_t fCurrentSize; // Current size of the array fEvents
|
|---|
| 31 | Float_t fEventFrequency; // Event frequency in Hertz (to be set)
|
|---|
| 32 | Byte_t fFlags; // Bit field for the fit result bits
|
|---|
| 33 | Int_t fPowerProbabilityBins; // Bins for the projected power spectrum
|
|---|
| 34 |
|
|---|
| 35 | TH1I *fHPowerProbability; // Fourier transform of fEvents projected on y-axis
|
|---|
| 36 | TArrayF *fPowerSpectrum; // Fourier transform of fEvents
|
|---|
| 37 | TGraph *fGraphEvents; //! TGraph to display the event array (will not be cloned!!)
|
|---|
| 38 | TGraph *fGraphPowerSpectrum; //! TGraph to display the power spectrum array (will not be cloned!!)
|
|---|
| 39 |
|
|---|
| 40 | enum { kGausFitOK, kExpFitOK, kFourierSpectrumOK, kExcluded }; // Bits for information about fit results
|
|---|
| 41 |
|
|---|
| 42 | protected:
|
|---|
| 43 |
|
|---|
| 44 | Float_t fBlackoutLimit; // Lower number sigmas from mean until event is considered blackout
|
|---|
| 45 | TArrayF fEvents; // Array which holds the entries of GausHist
|
|---|
| 46 | TF1 *fFGausFit; // Gauss fit for fHGausHist
|
|---|
| 47 | TF1 *fFExpFit; // Exponential fit for FHPowerProbability
|
|---|
| 48 | Axis_t fFirst; // Lower histogram edge for fHGausHist (used by InitBins())
|
|---|
| 49 | TH1F fHGausHist; // Histogram to hold the Gaussian distribution
|
|---|
| 50 | Axis_t fLast; // Upper histogram edge for fHGausHist (used by InitBins())
|
|---|
| 51 | Double_t fMean; // Mean of the Gauss fit
|
|---|
| 52 | Double_t fMeanErr; // Error of the mean of the Gauss fit
|
|---|
| 53 | Int_t fNbins; // Number histogram bins for fHGausHist (used by InitBins())
|
|---|
| 54 | Int_t fNDFLimit; // NDF limit for judgement if fit is OK
|
|---|
| 55 | Int_t fSaturated; // Number of events classified as saturated
|
|---|
| 56 | Double_t fSigma; // Sigma of the Gauss fit
|
|---|
| 57 | Double_t fSigmaErr; // Error of the sigma of the Gauss fit
|
|---|
| 58 | Float_t fPickupLimit; // Upper number sigmas from mean until event is considered pickup
|
|---|
| 59 | Int_t fPixId; // Pixel ID
|
|---|
| 60 | Double_t fProb; // Probability of the Gauss fit
|
|---|
| 61 | Float_t fProbLimit; // Probability limit for judgement if fit is OK
|
|---|
| 62 |
|
|---|
| 63 | Float_t *CreateEventXaxis(Int_t n); // Create an x-axis for the Event TGraphs
|
|---|
| 64 | Float_t *CreatePSDXaxis(Int_t n); // Create an x-axis for the PSD TGraphs
|
|---|
| 65 |
|
|---|
| 66 | void DrawPowerSpectrum(TVirtualPad &pad, Int_t i); // Draw graph of fPowerSpectrum and fHPowerProbability
|
|---|
| 67 |
|
|---|
| 68 | // Setters
|
|---|
| 69 | void SetBinsAfterStripping ( const Int_t nbins=fgBinsAfterStripping ) { fBinsAfterStripping =nbins; }
|
|---|
| 70 | void SetPowerProbabilityBins ( const Int_t nbins=fgPowerProbabilityBins ) { fPowerProbabilityBins=nbins; }
|
|---|
| 71 |
|
|---|
| 72 | public:
|
|---|
| 73 |
|
|---|
| 74 | MHGausEvents(const char* name=NULL, const char* title=NULL);
|
|---|
| 75 | ~MHGausEvents();
|
|---|
| 76 |
|
|---|
| 77 | virtual void Clear(Option_t *o="");
|
|---|
| 78 | virtual void Reset();
|
|---|
| 79 | virtual void InitBins();
|
|---|
| 80 |
|
|---|
| 81 | // Draws
|
|---|
| 82 | virtual void Draw(Option_t *option=""); // Default Draw
|
|---|
| 83 | void DrawEvents(); // Draw graph of fEvents
|
|---|
| 84 |
|
|---|
| 85 | // Getters
|
|---|
| 86 | const Double_t GetBlackout() const;
|
|---|
| 87 | const Double_t GetChiSquare() const;
|
|---|
| 88 | const Double_t GetExpChiSquare() const;
|
|---|
| 89 | const Int_t GetExpNdf() const;
|
|---|
| 90 | const Double_t GetExpProb() const;
|
|---|
| 91 | TArrayF *GetEvents() { return &fEvents; }
|
|---|
| 92 | const TArrayF *GetEvents() const { return &fEvents; }
|
|---|
| 93 | TF1 *GetFExpFit() { return fFExpFit; }
|
|---|
| 94 | const TF1 *GetFExpFit() const { return fFExpFit; }
|
|---|
| 95 | TF1 *GetFGausFit() { return fFGausFit; }
|
|---|
| 96 | const TF1 *GetFGausFit() const { return fFGausFit; }
|
|---|
| 97 | TGraph *GetGraphEvents() { return fGraphEvents; }
|
|---|
| 98 | const Double_t GetFirst() const { return fFirst; }
|
|---|
| 99 | const Double_t GetLast () const { return fLast ; }
|
|---|
| 100 | const TGraph *GetGraphEvents() const { return fGraphEvents; }
|
|---|
| 101 | TGraph *GetGraphPowerSpectrum() { return fGraphPowerSpectrum; }
|
|---|
| 102 | const TGraph *GetGraphPowerSpectrum() const { return fGraphPowerSpectrum; }
|
|---|
| 103 | TH1F *GetHGausHist() { return &fHGausHist; }
|
|---|
| 104 | const TH1F *GetHGausHist() const { return &fHGausHist; }
|
|---|
| 105 | TH1I *GetHPowerProbability() { return fHPowerProbability; }
|
|---|
| 106 | const TH1I *GetHPowerProbability() const { return fHPowerProbability; }
|
|---|
| 107 | const Double_t GetMean() const { return fMean; }
|
|---|
| 108 | const Double_t GetMeanErr() const { return fMeanErr; }
|
|---|
| 109 | const Int_t GetNdf() const;
|
|---|
| 110 | const Double_t GetOffset() const;
|
|---|
| 111 | const Double_t GetPickup() const;
|
|---|
| 112 | const Int_t GetPixId() const { return fPixId; }
|
|---|
| 113 | TArrayF *GetPowerSpectrum() { return fPowerSpectrum; }
|
|---|
| 114 | const TArrayF *GetPowerSpectrum() const { return fPowerSpectrum; }
|
|---|
| 115 | const Double_t GetProb() const { return fProb; }
|
|---|
| 116 | const Float_t GetSaturated() const { return fSaturated; }
|
|---|
| 117 | const Double_t GetSigma() const { return fSigma; }
|
|---|
| 118 | const Double_t GetSigmaErr() const { return fSigmaErr; }
|
|---|
| 119 | const Double_t GetSlope() const;
|
|---|
| 120 |
|
|---|
| 121 | const Bool_t IsExcluded() const;
|
|---|
| 122 | const Bool_t IsExpFitOK() const;
|
|---|
| 123 | const Bool_t IsEmpty() const;
|
|---|
| 124 | const Bool_t IsFourierSpectrumOK() const;
|
|---|
| 125 | const Bool_t IsGausFitOK() const;
|
|---|
| 126 | const Bool_t IsOnlyOverflow() const;
|
|---|
| 127 |
|
|---|
| 128 | // Fill
|
|---|
| 129 | void FillArray ( const Float_t f ); // Fill only the array fEvents
|
|---|
| 130 | Bool_t FillHist ( const Float_t f ); // Fill only the histogram HGausHist
|
|---|
| 131 | Bool_t FillHistAndArray( const Float_t f ); // Fill bothe the array fEvents and the histogram HGausHist
|
|---|
| 132 |
|
|---|
| 133 | // Fits
|
|---|
| 134 | Bool_t FitGaus( Option_t *option="RQ0",
|
|---|
| 135 | const Double_t xmin=0.,
|
|---|
| 136 | const Double_t xmax=0.); // Fit the histogram HGausHist with a Gaussian
|
|---|
| 137 | Bool_t RepeatFit(const Option_t *option="RQ0"); // Repeat fit within limits defined by fPickupLimit
|
|---|
| 138 | void BypassFit(); // Take mean and RMS from the histogram
|
|---|
| 139 |
|
|---|
| 140 | // Prints
|
|---|
| 141 | virtual void Print(const Option_t *o="") const; // Default Print
|
|---|
| 142 |
|
|---|
| 143 | // Setters
|
|---|
| 144 | void SetBlackoutLimit ( const Float_t lim=fgBlackoutLimit ) { fBlackoutLimit = lim; }
|
|---|
| 145 | void SetEventFrequency ( const Float_t f ) { fEventFrequency = f; }
|
|---|
| 146 | void SetExcluded ( const Bool_t b=kTRUE );
|
|---|
| 147 | void SetExpFitOK ( const Bool_t b=kTRUE );
|
|---|
| 148 | void SetFourierSpectrumOK( const Bool_t b=kTRUE );
|
|---|
| 149 | void SetGausFitOK ( const Bool_t b=kTRUE );
|
|---|
| 150 | void SetLast ( const Double_t d ) { fLast = d; }
|
|---|
| 151 | void SetFirst ( const Double_t d ) { fFirst = d; }
|
|---|
| 152 | void SetMean ( const Double_t d ) { fMean = d; }
|
|---|
| 153 | void SetMeanErr ( const Double_t d ) { fMeanErr = d; }
|
|---|
| 154 | void SetNbins ( const Int_t i ) { fNbins = i; }
|
|---|
| 155 | void SetNDFLimit ( const Int_t lim=fgNDFLimit ) { fNDFLimit = lim; }
|
|---|
| 156 | void SetPickupLimit ( const Float_t lim=fgPickupLimit ) { fPickupLimit = lim; }
|
|---|
| 157 | void SetPixId ( const Int_t i ) { fPixId = i; }
|
|---|
| 158 | void SetProb ( const Double_t d ) { fProb = d; }
|
|---|
| 159 | void SetProbLimit ( const Float_t lim=fgProbLimit ) { fProbLimit = lim; }
|
|---|
| 160 | void SetSaturated ( const Int_t i ) { fSaturated += i; }
|
|---|
| 161 | void SetSigma ( const Double_t d ) { fSigma = d; }
|
|---|
| 162 | void SetSigmaErr ( const Double_t d ) { fSigmaErr = d; }
|
|---|
| 163 |
|
|---|
| 164 | // Miscelleaneous
|
|---|
| 165 | virtual void ChangeHistId(const Int_t id); // Changes names and titles of the histogram
|
|---|
| 166 | virtual void Renorm(); // Re-normalize the results
|
|---|
| 167 |
|
|---|
| 168 | void CreateFourierSpectrum(); // Create the fourier spectrum out of fEvents
|
|---|
| 169 | void CreateGraphEvents(); // Create the TGraph fGraphEvents of fEvents
|
|---|
| 170 | void CreateGraphPowerSpectrum(); // Create the TGraph fGraphPowerSpectrum out of fPowerSpectrum
|
|---|
| 171 |
|
|---|
| 172 | ClassDef(MHGausEvents, 1) // Base class for events with Gaussian distributed values
|
|---|
| 173 | };
|
|---|
| 174 |
|
|---|
| 175 | #endif
|
|---|