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