source: trunk/MagicSoft/Mars/mhcalib/MHGausEvents.h@ 5138

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