]>
Commit | Line | Data |
---|---|---|
600b313f | 1 | // |
2 | // Calculate the corrections in the central regions | |
3 | // | |
4 | #ifndef ALICENTRALMCCORRECTIONS_H | |
5 | #define ALICENTRALMCCORRECTIONS_H | |
6 | /** | |
7 | * @file AliCentralMCCorrectionsTask.h | |
8 | * @author Christian Holm Christensen <cholm@dalsgaard.hehi.nbi.dk> | |
9 | * @date Wed Mar 23 14:05:51 2011 | |
10 | * | |
11 | * @brief | |
12 | * | |
13 | * | |
bd6f5206 | 14 | * @ingroup pwglf_central_aod |
600b313f | 15 | */ |
16 | #include <AliAnalysisTaskSE.h> | |
17 | #include "AliFMDMCEventInspector.h" | |
18 | #include "AliSPDMCTrackDensity.h" | |
19 | #include <TH1I.h> | |
20 | class AliCentralCorrSecondaryMap; | |
21 | class AliESDEvent; | |
22 | class TH2D; | |
23 | class TH1D; | |
24 | class TList; | |
25 | ||
26 | ||
27 | /** | |
28 | * Calculate the corrections in the central regions | |
29 | * | |
30 | * @par Inputs: | |
31 | * - AliESDEvent | |
32 | * | |
33 | * @par Outputs: | |
34 | * - AliAODCentralMult | |
35 | * | |
36 | * @par Histograms | |
37 | * | |
38 | * @par Corrections used | |
39 | * | |
bd6f5206 | 40 | * @ingroup pwglf_central_tasks |
41 | * @ingroup pwglf_central_mc | |
42 | * @ingroup pwglf_central_aod | |
600b313f | 43 | * |
44 | */ | |
45 | class AliCentralMCCorrectionsTask : public AliAnalysisTaskSE | |
46 | { | |
47 | public: | |
48 | /** | |
49 | * Constructor | |
50 | * | |
51 | * @param name Name of task | |
52 | */ | |
53 | AliCentralMCCorrectionsTask(const char* name); | |
54 | /** | |
55 | * Constructor | |
56 | */ | |
57 | AliCentralMCCorrectionsTask(); | |
58 | /** | |
59 | * Copy constructor | |
60 | * | |
61 | * @param o Object to copy from | |
62 | */ | |
63 | AliCentralMCCorrectionsTask(const AliCentralMCCorrectionsTask& o); | |
64 | /** | |
65 | * Assignment operator | |
66 | * | |
67 | * @param o Object to assign from | |
68 | * | |
69 | * @return Reference to this object | |
70 | */ | |
71 | AliCentralMCCorrectionsTask& operator=(const AliCentralMCCorrectionsTask& o); | |
72 | /** | |
73 | * @{ | |
74 | * @name Interface methods | |
75 | */ | |
76 | /** | |
77 | * Initialize the task | |
78 | * | |
79 | */ | |
80 | virtual void Init(); | |
81 | /** | |
82 | * Create output objects | |
83 | * | |
84 | */ | |
85 | virtual void UserCreateOutputObjects(); | |
86 | /** | |
87 | * Process each event | |
88 | * | |
89 | * @param option Not used | |
90 | */ | |
91 | virtual void UserExec(Option_t* option); | |
92 | /** | |
93 | * End of job | |
94 | * | |
95 | * @param option Not used | |
96 | */ | |
97 | virtual void Terminate(Option_t* option); | |
98 | /** | |
99 | * @} | |
100 | */ | |
101 | /** | |
102 | * Print this object | |
103 | * | |
104 | * @param option Not used | |
105 | */ | |
106 | void Print(Option_t* option="") const; | |
107 | ||
108 | /** | |
109 | * Set the vertex axis to use | |
110 | * | |
111 | * @param nBins Number of bins | |
112 | * @param vzMin Least @f$z@f$ coordinate of interation point | |
113 | * @param vzMax Largest @f$z@f$ coordinate of interation point | |
114 | */ | |
115 | void SetVertexAxis(Int_t nBins, Double_t vzMin, Double_t vzMax=-1000000); | |
116 | /** | |
117 | * Set the vertex axis to use | |
118 | * | |
119 | * @param axis Axis | |
120 | */ | |
121 | void SetVertexAxis(const TAxis& axis); | |
122 | /** | |
123 | * Set the eta axis to use | |
124 | * | |
125 | * @param nBins Number of bins | |
126 | * @param etaMin Least @f$\eta@f$ | |
127 | * @param etaMax Largest @f$\eta@f$ | |
128 | */ | |
129 | void SetEtaAxis(Int_t nBins, Double_t etaMin, Double_t etaMax=-1000000); | |
130 | /** | |
131 | * Set the eta axis to use | |
132 | * | |
133 | * @param axis Axis | |
134 | */ | |
135 | void SetEtaAxis(const TAxis& axis); | |
136 | /** | |
137 | * Set the number of phi bins to use | |
138 | * | |
139 | * @param nBins | |
140 | */ | |
141 | void SetNPhiBins(UShort_t nBins) { fNPhiBins = nBins; } | |
142 | /** | |
143 | * Get a reference to the track density calculator | |
144 | * | |
145 | * @return Reference to the track density calculator | |
146 | */ | |
147 | AliSPDMCTrackDensity& GetTrackDensity() { return fTrackDensity; } | |
148 | /** | |
149 | * Get a reference to the track density calculator | |
150 | * | |
151 | * @return Reference to the track density calculator | |
152 | */ | |
153 | const AliSPDMCTrackDensity& GetTrackDensity() const { return fTrackDensity; } | |
154 | /** | |
155 | * Get a reference to the event inspector | |
156 | * | |
157 | * @return Reference to the event inspector | |
158 | */ | |
159 | AliFMDMCEventInspector& GetEventInspector() { return fInspector; } | |
160 | /** | |
161 | * Get a reference to the event inspector | |
162 | * | |
163 | * @return Reference to the event inspector | |
164 | */ | |
165 | const AliFMDMCEventInspector& GetEventInspector() const { return fInspector;} | |
166 | protected: | |
167 | /** | |
168 | * A vertex bin | |
169 | * | |
170 | */ | |
171 | struct VtxBin : public TNamed | |
172 | { | |
173 | /** | |
174 | * Constructor | |
175 | */ | |
176 | VtxBin(); | |
177 | /** | |
178 | * Constructor | |
179 | * | |
180 | * @param low Lower @f$v_z@f$ bound | |
181 | * @param high Upper @f$v_z@f$ bound | |
182 | * @param etaAxis @f$\eta@f$ axis to use | |
c6115ede | 183 | * @param nPhi Number of phi bins |
600b313f | 184 | */ |
185 | VtxBin(Double_t low, Double_t high, const TAxis& etaAxis, UShort_t nPhi); | |
186 | /** | |
187 | * Copy constructor | |
188 | * | |
189 | * @param o Object to copy from | |
190 | */ | |
191 | VtxBin(const VtxBin& o); | |
192 | /** | |
193 | * Assignment operator | |
194 | * | |
195 | * @param o Object to assign from | |
196 | * | |
197 | * @return Reference to this object | |
198 | */ | |
199 | VtxBin& operator=(const VtxBin& o); | |
200 | /** | |
201 | * Get bin name | |
202 | * | |
203 | * @param low Lower @f$v_z@f$ bound | |
204 | * @param high Upper @f$v_z@f$ bound | |
205 | * | |
206 | * @return Bin name | |
207 | */ | |
208 | static const char* BinName(Double_t low, Double_t high); | |
209 | /** | |
210 | * Declare output in passed list | |
211 | * | |
212 | * @param list List to put output in | |
213 | */ | |
214 | void DefineOutput(TList* list); | |
215 | /** | |
216 | * End of job process | |
217 | * | |
c6115ede | 218 | * @param o List to add output to |
219 | * @param i Input list | |
220 | * @param iVz Vertex bin | |
221 | * @param map Correctons map | |
600b313f | 222 | */ |
223 | void Finish(const TList* i, | |
224 | TList* o, | |
225 | UShort_t iVz, | |
226 | AliCentralCorrSecondaryMap* map); | |
227 | ||
228 | TH2D* fHits; // Cache of per-ring histograms | |
229 | TH2D* fPrimary; // Cache or primary | |
230 | TH1D* fCounts; // Event count | |
231 | ||
232 | ClassDef(VtxBin,1); // Vertex bin | |
233 | }; | |
234 | /** | |
235 | * Define our vertex bins | |
236 | * | |
237 | * @param list List to read or add binst from/to | |
238 | */ | |
239 | void DefineBins(TList* list); | |
240 | ||
241 | AliFMDMCEventInspector fInspector; // Event inspector | |
242 | AliSPDMCTrackDensity fTrackDensity; // Get the track density | |
243 | ||
244 | TObjArray* fVtxBins; // Vertex bins | |
245 | Bool_t fFirstEvent; // First event flag | |
246 | TH1I* fHEvents; // All Events | |
247 | TH1I* fHEventsTr; // Histogram of events w/trigger | |
248 | TH1I* fHEventsTrVtx; // Events w/trigger and vertex | |
249 | TAxis fVtxAxis; // Vertex axis | |
250 | TAxis fEtaAxis; // Eta axis | |
251 | TList* fList; // Output list | |
252 | UShort_t fNPhiBins; // Nunber of phi bins | |
253 | ClassDef(AliCentralMCCorrectionsTask,1) // Central corrections class | |
254 | }; | |
255 | ||
256 | #endif | |
257 | // Local Variables: | |
258 | // mode: C++ | |
259 | // End: | |
260 |