source: trunk/MagicSoft/Mars/mcalib/MHGausEvents.h@ 3642

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