]>
Commit | Line | Data |
---|---|---|
7a3e34f4 | 1 | #ifndef ALIFMDMCTRACKINSPECTOR_H |
2 | #define ALIFMDMCTRACKINSPECTOR_H | |
3 | #include "AliFMDEnergyFitter.h" | |
4 | #include "AliFMDMCTrackELoss.h" | |
5 | #include <TArrayF.h> | |
6 | class AliMCEvent; | |
7 | class AliESDEvent; | |
8 | ||
9 | /** | |
10 | * Class to fit the simulated energy loss in the FMD | |
11 | * | |
12 | * @ingroup pwglf_forward_mc | |
13 | * @ingroup pwglf_forward_eloss | |
14 | */ | |
15 | class AliFMDMCTrackInspector : public AliFMDEnergyFitter | |
16 | { | |
17 | public: | |
18 | /** | |
19 | * Constructor - do not use | |
20 | */ | |
21 | AliFMDMCTrackInspector(); | |
22 | /** | |
23 | * Constructor | |
24 | * | |
25 | * @param title Title of object - not significant | |
26 | */ | |
27 | AliFMDMCTrackInspector(const char* title); | |
28 | /** | |
29 | * Destructor | |
30 | */ | |
31 | virtual ~AliFMDMCTrackInspector(); | |
32 | ||
33 | /** | |
34 | * Define the output histograms. These are put in a sub list of the | |
35 | * passed list. The histograms are merged before the parent task calls | |
36 | * AliAnalysisTaskSE::Terminate | |
37 | * | |
38 | * @param dir Directory to add to | |
39 | */ | |
40 | virtual void CreateOutputObjects(TList* dir); | |
41 | /** | |
42 | * Set-up before processing an event | |
43 | * | |
44 | * @param mcInput MC input | |
45 | * | |
46 | * @return true | |
47 | */ | |
48 | virtual Bool_t PreEvent(const AliMCEvent& mcInput); | |
49 | /** | |
50 | * Process a single event | |
51 | * | |
52 | * @param esdInput ESD input | |
53 | * @param mcInput MC input | |
54 | * @param cent Event centrality | |
55 | * | |
56 | * @return true | |
57 | */ | |
58 | virtual Bool_t Event(const AliESDEvent& esdInput, | |
59 | const AliMCEvent& mcInput, | |
60 | Double_t cent=-1); | |
61 | /** | |
62 | * Post-process accumulated signals | |
63 | * | |
64 | * @return true | |
65 | */ | |
66 | virtual Bool_t PostEvent(); | |
67 | ||
68 | /** | |
69 | * Get reference to the tracker of energy loss | |
70 | * | |
71 | * @return Reference to tracker of energy loss | |
72 | */ | |
73 | AliFMDMCTrackELoss& GetTracker() { return fTracker; } | |
74 | /** | |
75 | * Get constant reference to the tracker of energy loss | |
76 | * | |
77 | * @return Constant reference to tracker of energy loss | |
78 | */ | |
79 | const AliFMDMCTrackELoss& GetTracker() const { return fTracker; } | |
80 | protected: | |
81 | /** | |
82 | * copy constructor - not implemented | |
83 | */ | |
84 | AliFMDMCTrackInspector(const AliFMDMCTrackInspector&); | |
85 | /** | |
86 | * Assignment operator - not implemented | |
87 | * | |
88 | */ | |
89 | AliFMDMCTrackInspector& operator=(const AliFMDMCTrackInspector&); | |
90 | ||
91 | /** | |
92 | * Container of ring histograms | |
93 | * | |
94 | */ | |
cac751a4 | 95 | public: |
7a3e34f4 | 96 | struct RingHistos : public AliFMDEnergyFitter::RingHistos |
97 | { | |
98 | /** | |
99 | * Default CTOR - do not use | |
100 | */ | |
101 | RingHistos(); | |
102 | /** | |
103 | * User CTOR | |
104 | * | |
105 | * @param d Detector number | |
106 | * @param r Ring identifier | |
107 | */ | |
108 | RingHistos(UShort_t d, Char_t r); | |
109 | /** | |
110 | * DTOR | |
111 | */ | |
112 | ~RingHistos() {} | |
113 | /** | |
114 | * Copy constructor - not defined | |
115 | * | |
116 | * @param o Object to copy from | |
117 | */ | |
ee83b849 | 118 | RingHistos(const RingHistos& o){;} |
7a3e34f4 | 119 | /** |
120 | * Assignment operator - not defined | |
121 | * | |
122 | * @param o Object to assign from | |
123 | * | |
124 | * @return Reference to this | |
125 | */ | |
ee83b849 | 126 | RingHistos& operator=(const RingHistos& o){return *this;} |
7a3e34f4 | 127 | /** |
128 | * Create a bin array of increasing bins. This overload uses the | |
129 | * service AliFMDEncodedEdx::Spec::FillBinArray. | |
130 | * | |
131 | * @param nBins Number of bins - ignored | |
132 | * @param low Low cut - ignored | |
133 | * @param high High cut - ignored | |
134 | * | |
135 | * @return Array of bin boundaries | |
136 | */ | |
137 | TArrayD MakeIncreasingAxis(Int_t nBins, | |
138 | Double_t low, | |
139 | Double_t high) const; | |
140 | /** | |
141 | * Initialise object | |
142 | * | |
143 | * @param eAxis Eta axis | |
144 | * @param cAxis Centrality axis | |
145 | * @param maxDE Max energy loss to consider | |
146 | * @param nDEbins Number of bins | |
147 | * @param useIncrBin Whether to use an increasing bin size | |
148 | */ | |
149 | virtual void SetupForData(const TAxis& eAxis, | |
150 | const TAxis& cAxis, | |
151 | Double_t maxDE=10, | |
152 | Int_t nDEbins=300, | |
153 | Bool_t useIncrBin=true); | |
154 | /** | |
155 | * Fill in observation | |
156 | * | |
157 | * @param flag 0 - fill all, 1 - primary, 2 - secondary | |
158 | * @param eta Eta of particle observations | |
159 | * @param mult Scaled energy loss | |
160 | */ | |
161 | virtual void FillMC(UShort_t flag, Double_t eta, Double_t mult); | |
162 | /** | |
92fdfa31 | 163 | * Do scaling of histogram before fitting. This can be |
164 | * overwritten to do some smoothing or the like. By default, this | |
165 | * simply scales to the bin width. | |
7a3e34f4 | 166 | * |
92fdfa31 | 167 | * @param dist Histogram to scale. |
168 | */ | |
169 | virtual void Scale(TH1* dist) const; | |
7a3e34f4 | 170 | /** |
171 | * Fit the final distributions - called via Terminate | |
172 | * | |
173 | * @param dir Containing directory | |
174 | * @param lowCut Lower cut on @f$\Delta/\Delta_{mip}@f$ | |
175 | * @param nParticles Max. number of particle peaks to fit | |
176 | * @param minEntries Least number of entries required before fitting | |
177 | * @param minusBins Number of bins below the 1st peak we start fitting | |
178 | * @param relErrorCut Largest relative error on paramters | |
179 | * @param chi2nuCut Largest value of the @f$\chi^2/\nu@f$ | |
180 | * @param minWeight Least weight to consider | |
181 | * @param regCut When to regalurize | |
182 | * @param residuals How to do residuals - if at all | |
183 | * | |
184 | * @return List of histograms of parameters | |
185 | */ | |
186 | TObjArray* Fit(TList* dir, | |
187 | Double_t lowCut, | |
188 | UShort_t nParticles, | |
189 | UShort_t minEntries, | |
190 | UShort_t minusBins, | |
191 | Double_t relErrorCut, | |
192 | Double_t chi2nuCut, | |
193 | Double_t minWeight, | |
194 | Double_t regCut, | |
195 | EResidualMethod residuals) const; | |
196 | ||
197 | TH2* fPrimary; // @f$\Delta@f$ vs @f$\eta@f$ for primaries | |
198 | TH2* fSecondary; // @f$\Delta@f$ vs @f$\eta@f$ for second. | |
199 | public: | |
200 | TH2* fBetaGammadEdx; | |
201 | TH2* fBetaGammaEta; | |
202 | TH2* fDedxEta; | |
203 | ClassDef(RingHistos,1); // Cache of histograms per ring | |
204 | }; | |
cac751a4 | 205 | protected: |
7a3e34f4 | 206 | /** |
207 | * Create a container of histograms for a single ring | |
208 | * | |
209 | * @param d Detector | |
210 | * @param r Ring | |
211 | * | |
212 | * @return Newly allocated container | |
213 | */ | |
214 | AliFMDEnergyFitter::RingHistos* CreateRingHistos(UShort_t d, Char_t r) const; | |
215 | /** Our 'tracker' */ | |
216 | AliFMDMCTrackELoss fTracker; | |
217 | /** Cache of current MC IP */ | |
218 | TArrayF fIp; | |
219 | /** Cache of number of tracks */ | |
220 | Int_t fNTrack; | |
221 | /** Cache of numbr of primaries */ | |
222 | Int_t fNPrimary; | |
223 | ||
224 | ClassDef(AliFMDMCTrackInspector,1); | |
225 | }; | |
226 | ||
227 | #endif | |
228 | // Local Variables: | |
229 | // mode: C++ | |
230 | // End: |