]>
Commit | Line | Data |
---|---|---|
152a39e6 | 1 | /**************************************************************************\r |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r | |
3 | * *\r | |
4 | * Author: The ALICE Off-line Project. *\r | |
5 | * Contributors are mentioned in the code where appropriate. *\r | |
6 | * *\r | |
7 | * Permission to use, copy, modify and distribute this software and its *\r | |
8 | * documentation strictly for non-commercial purposes is hereby granted *\r | |
9 | * without fee, provided that the above copyright notice appears in all *\r | |
10 | * copies and that both the copyright notice and this permission notice *\r | |
11 | * appear in the supporting documentation. The authors make no claims *\r | |
12 | * about the suitability of this software for any purpose. It is *\r | |
13 | * provided "as is" without express or implied warranty. *\r | |
14 | **************************************************************************/\r | |
15 | \r | |
16 | /* $Id: AliAnalysisTaskPIDqa.cxx 43811 2010-09-23 14:13:31Z wiechula $ */\r | |
17 | #include <TList.h>\r | |
18 | #include <TVectorD.h>\r | |
19 | #include <TObjArray.h>\r | |
20 | #include <TH2.h>\r | |
21 | #include <TFile.h>\r | |
22 | #include <TPRegexp.h>\r | |
23 | #include <TChain.h>\r | |
24 | #include <TF1.h>\r | |
25 | #include <TSpline.h>\r | |
26 | \r | |
27 | #include <AliAnalysisManager.h>\r | |
28 | #include <AliInputEventHandler.h>\r | |
29 | #include <AliVEventHandler.h>\r | |
30 | #include <AliVEvent.h>\r | |
31 | #include <AliVParticle.h>\r | |
32 | #include <AliVTrack.h>\r | |
33 | #include <AliLog.h>\r | |
34 | #include <AliPID.h>\r | |
35 | #include <AliPIDResponse.h>\r | |
36 | #include <AliITSPIDResponse.h>\r | |
37 | #include <AliTPCPIDResponse.h>\r | |
38 | #include <AliTRDPIDResponse.h>\r | |
39 | #include <AliTOFPIDResponse.h>\r | |
40 | \r | |
41 | #include "AliAnalysisTaskPIDqa.h"\r | |
42 | \r | |
43 | \r | |
44 | ClassImp(AliAnalysisTaskPIDqa)\r | |
45 | \r | |
46 | //______________________________________________________________________________\r | |
47 | AliAnalysisTaskPIDqa::AliAnalysisTaskPIDqa():\r | |
48 | AliAnalysisTaskSE(),\r | |
49 | fPIDResponse(0x0),\r | |
50 | fListQA(0x0),\r | |
51 | fListQAits(0x0),\r | |
52 | fListQAitsSA(0x0),\r | |
53 | fListQAitsPureSA(0x0),\r | |
54 | fListQAtpc(0x0),\r | |
55 | fListQAtrd(0x0),\r | |
56 | fListQAtof(0x0),\r | |
57 | fListQAemcal(0x0),\r | |
58 | fListQAtpctof(0x0)\r | |
59 | {\r | |
60 | //\r | |
61 | // Dummy constructor\r | |
62 | //\r | |
63 | }\r | |
64 | \r | |
65 | //______________________________________________________________________________\r | |
66 | AliAnalysisTaskPIDqa::AliAnalysisTaskPIDqa(const char* name):\r | |
67 | AliAnalysisTaskSE(name),\r | |
68 | fPIDResponse(0x0),\r | |
69 | fListQA(0x0),\r | |
70 | fListQAits(0x0),\r | |
71 | fListQAitsSA(0x0),\r | |
72 | fListQAitsPureSA(0x0),\r | |
73 | fListQAtpc(0x0),\r | |
74 | fListQAtrd(0x0),\r | |
75 | fListQAtof(0x0),\r | |
76 | fListQAemcal(0x0),\r | |
77 | fListQAtpctof(0x0)\r | |
78 | {\r | |
79 | //\r | |
80 | // Default constructor\r | |
81 | //\r | |
82 | DefineInput(0,TChain::Class());\r | |
83 | DefineOutput(1,TList::Class());\r | |
84 | }\r | |
85 | \r | |
86 | //______________________________________________________________________________\r | |
87 | AliAnalysisTaskPIDqa::~AliAnalysisTaskPIDqa()\r | |
88 | {\r | |
89 | //\r | |
90 | // Destructor\r | |
91 | //\r | |
92 | \r | |
93 | }\r | |
94 | \r | |
95 | //______________________________________________________________________________\r | |
96 | void AliAnalysisTaskPIDqa::UserCreateOutputObjects()\r | |
97 | {\r | |
98 | //\r | |
99 | // Create the output QA objects\r | |
100 | //\r | |
101 | \r | |
102 | AliLog::SetClassDebugLevel("AliAnalysisTaskPIDqa",10);\r | |
103 | \r | |
104 | //input hander\r | |
105 | AliAnalysisManager *man=AliAnalysisManager::GetAnalysisManager();\r | |
106 | AliInputEventHandler *inputHandler=dynamic_cast<AliInputEventHandler*>(man->GetInputEventHandler());\r | |
107 | if (!inputHandler) AliFatal("Input handler needed");\r | |
108 | \r | |
109 | //pid response object\r | |
110 | fPIDResponse=inputHandler->GetPIDResponse();\r | |
111 | if (!fPIDResponse) AliError("PIDResponse object was not created");\r | |
112 | \r | |
113 | //\r | |
114 | fListQA=new TList;\r | |
115 | fListQA->SetOwner();\r | |
116 | \r | |
117 | fListQAits=new TList;\r | |
118 | fListQAits->SetOwner();\r | |
119 | fListQAits->SetName("ITS");\r | |
120 | \r | |
121 | fListQAitsSA=new TList;\r | |
122 | fListQAitsSA->SetOwner();\r | |
123 | fListQAitsSA->SetName("ITS");\r | |
124 | \r | |
125 | fListQAitsPureSA=new TList;\r | |
126 | fListQAitsPureSA->SetOwner();\r | |
127 | fListQAitsPureSA->SetName("ITS");\r | |
128 | \r | |
129 | fListQAtpc=new TList;\r | |
130 | fListQAtpc->SetOwner();\r | |
131 | fListQAtpc->SetName("TPC");\r | |
132 | \r | |
133 | fListQAtrd=new TList;\r | |
134 | fListQAtrd->SetOwner();\r | |
135 | fListQAtrd->SetName("TRD");\r | |
136 | \r | |
137 | fListQAtof=new TList;\r | |
138 | fListQAtof->SetOwner();\r | |
139 | fListQAtof->SetName("TOF");\r | |
140 | \r | |
141 | fListQAemcal=new TList;\r | |
142 | fListQAemcal->SetOwner();\r | |
143 | fListQAemcal->SetName("EMCAL");\r | |
144 | \r | |
145 | fListQAtpctof=new TList;\r | |
146 | fListQAtpctof->SetOwner();\r | |
147 | fListQAtpctof->SetName("TPC_TOF");\r | |
148 | \r | |
149 | fListQA->Add(fListQAits);\r | |
150 | fListQA->Add(fListQAitsSA);\r | |
151 | fListQA->Add(fListQAitsPureSA);\r | |
152 | fListQA->Add(fListQAtpc);\r | |
153 | fListQA->Add(fListQAtrd);\r | |
154 | fListQA->Add(fListQAtof);\r | |
155 | fListQA->Add(fListQAemcal);\r | |
156 | fListQA->Add(fListQAtpctof);\r | |
157 | \r | |
158 | SetupITSqa();\r | |
159 | SetupTPCqa();\r | |
160 | SetupTRDqa();\r | |
161 | SetupTOFqa();\r | |
162 | SetupEMCALqa();\r | |
163 | SetupTPCTOFqa();\r | |
164 | \r | |
165 | PostData(1,fListQA);\r | |
166 | }\r | |
167 | \r | |
168 | \r | |
169 | //______________________________________________________________________________\r | |
170 | void AliAnalysisTaskPIDqa::UserExec(Option_t */*option*/)\r | |
171 | {\r | |
172 | //\r | |
173 | // Setup the PID response functions and fill the QA histograms\r | |
174 | //\r | |
175 | \r | |
176 | AliVEvent *event=InputEvent();\r | |
177 | if (!event||!fPIDResponse) return;\r | |
178 | \r | |
179 | \r | |
180 | FillITSqa();\r | |
181 | FillTPCqa();\r | |
182 | FillTRDqa();\r | |
183 | FillTOFqa();\r | |
184 | FillEMCALqa();\r | |
185 | FillTPCTOFqa();\r | |
186 | \r | |
187 | PostData(1,fListQA);\r | |
188 | }\r | |
189 | \r | |
190 | //______________________________________________________________________________\r | |
191 | void AliAnalysisTaskPIDqa::FillITSqa()\r | |
192 | {\r | |
193 | //\r | |
194 | // Fill PID qa histograms for the ITS\r | |
195 | //\r | |
196 | \r | |
197 | AliVEvent *event=InputEvent();\r | |
198 | \r | |
199 | Int_t ntracks=event->GetNumberOfTracks();\r | |
200 | for(Int_t itrack = 0; itrack < ntracks; itrack++){\r | |
201 | AliVTrack *track=(AliVTrack*)event->GetTrack(itrack);\r | |
202 | ULong_t status=track->GetStatus();\r | |
203 | // not that nice. status bits not in virtual interface\r | |
204 | // ITS refit + ITS pid\r | |
205 | if (!( ( (status & AliVTrack::kITSrefit)==AliVTrack::kITSrefit ) &&\r | |
206 | ( (status & AliVTrack::kITSpid )==AliVTrack::kITSpid ) )) continue;\r | |
207 | Double_t mom=track->P();\r | |
208 | \r | |
209 | for (Int_t ispecie=0; ispecie<AliPID::kSPECIES; ++ispecie){\r | |
210 | TH2 *h=(TH2*)fListQAits->At(ispecie);\r | |
211 | if (!h) continue;\r | |
212 | Double_t nSigma=fPIDResponse->NumberOfSigmasITS(track, (AliPID::EParticleType)ispecie);\r | |
213 | h->Fill(mom,nSigma);\r | |
214 | }\r | |
215 | \r | |
216 | TH2 *h=(TH2*)fListQAits->At(AliPID::kSPECIES);\r | |
217 | if (h) {\r | |
218 | Double_t sig=track->GetITSsignal();\r | |
219 | h->Fill(mom,sig);\r | |
220 | }\r | |
221 | \r | |
222 | }\r | |
223 | }\r | |
224 | \r | |
225 | //______________________________________________________________________________\r | |
226 | void AliAnalysisTaskPIDqa::FillTPCqa()\r | |
227 | {\r | |
228 | //\r | |
229 | // Fill PID qa histograms for the TPC\r | |
230 | //\r | |
231 | \r | |
232 | AliVEvent *event=InputEvent();\r | |
233 | \r | |
234 | Int_t ntracks=event->GetNumberOfTracks();\r | |
235 | for(Int_t itrack = 0; itrack < ntracks; itrack++){\r | |
236 | AliVTrack *track=(AliVTrack*)event->GetTrack(itrack);\r | |
237 | \r | |
238 | //\r | |
239 | //basic track cuts\r | |
240 | //\r | |
241 | ULong_t status=track->GetStatus();\r | |
242 | // not that nice. status bits not in virtual interface\r | |
243 | // TPC refit + ITS refit + TPC pid\r | |
244 | if (!( (status & AliVTrack::kTPCrefit) == AliVTrack::kTPCrefit) ||\r | |
245 | !( (status & AliVTrack::kITSrefit) == AliVTrack::kITSrefit) ||\r | |
246 | !( (status & AliVTrack::kTPCpid ) == AliVTrack::kTPCpid ) ) continue;\r | |
247 | \r | |
248 | Float_t nCrossedRowsTPC = track->GetTPCClusterInfo(2,1);\r | |
249 | Float_t ratioCrossedRowsOverFindableClustersTPC = 1.0;\r | |
250 | if (track->GetTPCNclsF()>0) {\r | |
251 | ratioCrossedRowsOverFindableClustersTPC = nCrossedRowsTPC/track->GetTPCNclsF();\r | |
252 | }\r | |
253 | \r | |
254 | if ( nCrossedRowsTPC<70 || ratioCrossedRowsOverFindableClustersTPC<.8 ) continue;\r | |
255 | \r | |
256 | Double_t mom=track->GetTPCmomentum();\r | |
257 | \r | |
258 | for (Int_t ispecie=0; ispecie<AliPID::kSPECIES; ++ispecie){\r | |
259 | TH2 *h=(TH2*)fListQAtpc->At(ispecie);\r | |
260 | if (!h) continue;\r | |
261 | Double_t nSigma=fPIDResponse->NumberOfSigmasTPC(track, (AliPID::EParticleType)ispecie);\r | |
262 | h->Fill(mom,nSigma);\r | |
263 | }\r | |
264 | \r | |
265 | TH2 *h=(TH2*)fListQAtpc->At(AliPID::kSPECIES);\r | |
266 | if (h) {\r | |
267 | Double_t sig=track->GetTPCsignal();\r | |
268 | h->Fill(mom,sig);\r | |
269 | }\r | |
270 | }\r | |
271 | }\r | |
272 | \r | |
273 | //______________________________________________________________________________\r | |
274 | void AliAnalysisTaskPIDqa::FillTRDqa()\r | |
275 | {\r | |
276 | //\r | |
277 | // Fill PID qa histograms for the TRD\r | |
278 | //\r | |
279 | \r | |
280 | }\r | |
281 | \r | |
282 | //______________________________________________________________________________\r | |
283 | void AliAnalysisTaskPIDqa::FillTOFqa()\r | |
284 | {\r | |
285 | //\r | |
286 | // Fill PID qa histograms for the TOF\r | |
287 | //\r | |
288 | \r | |
289 | AliVEvent *event=InputEvent();\r | |
290 | \r | |
291 | Int_t ntracks=event->GetNumberOfTracks();\r | |
292 | for(Int_t itrack = 0; itrack < ntracks; itrack++){\r | |
293 | AliVTrack *track=(AliVTrack*)event->GetTrack(itrack);\r | |
294 | \r | |
295 | //\r | |
296 | //basic track cuts\r | |
297 | //\r | |
298 | ULong_t status=track->GetStatus();\r | |
299 | // not that nice. status bits not in virtual interface\r | |
300 | // TPC refit + ITS refit +\r | |
301 | // TOF out + TOFpid +\r | |
302 | // kTIME\r | |
303 | if (!((status & AliVTrack::kTPCrefit) == AliVTrack::kTPCrefit) ||\r | |
304 | !((status & AliVTrack::kITSrefit) == AliVTrack::kITSrefit) ||\r | |
305 | !((status & AliVTrack::kTOFout ) == AliVTrack::kTOFout ) ||\r | |
306 | !((status & AliVTrack::kTOFpid ) == AliVTrack::kTOFpid ) ||\r | |
307 | !((status & AliVTrack::kTIME ) == AliVTrack::kTIME ) ) continue;\r | |
308 | \r | |
309 | Float_t nCrossedRowsTPC = track->GetTPCClusterInfo(2,1);\r | |
310 | Float_t ratioCrossedRowsOverFindableClustersTPC = 1.0;\r | |
311 | if (track->GetTPCNclsF()>0) {\r | |
312 | ratioCrossedRowsOverFindableClustersTPC = nCrossedRowsTPC/track->GetTPCNclsF();\r | |
313 | }\r | |
314 | \r | |
315 | if ( nCrossedRowsTPC<70 || ratioCrossedRowsOverFindableClustersTPC<.8 ) continue;\r | |
316 | \r | |
317 | \r | |
318 | Double_t mom=track->P();\r | |
319 | \r | |
320 | for (Int_t ispecie=0; ispecie<AliPID::kSPECIES; ++ispecie){\r | |
321 | //TOF nSigma\r | |
322 | TH2 *h=(TH2*)fListQAtof->At(ispecie);\r | |
323 | if (!h) continue;\r | |
324 | Double_t nSigma=fPIDResponse->NumberOfSigmasTOF(track, (AliPID::EParticleType)ispecie);\r | |
325 | h->Fill(mom,nSigma);\r | |
326 | }\r | |
327 | \r | |
328 | TH2 *h=(TH2*)fListQAtof->At(AliPID::kSPECIES);\r | |
329 | if (h) {\r | |
330 | Double_t sig=track->GetTOFsignal();\r | |
331 | h->Fill(mom,sig);\r | |
332 | }\r | |
333 | \r | |
334 | }\r | |
335 | }\r | |
336 | \r | |
337 | //______________________________________________________________________________\r | |
338 | void AliAnalysisTaskPIDqa::FillEMCALqa()\r | |
339 | {\r | |
340 | //\r | |
341 | // Fill PID qa histograms for the EMCAL\r | |
342 | //\r | |
343 | \r | |
a3851b68 | 344 | AliVEvent *event=InputEvent();\r |
345 | \r | |
346 | Int_t ntracks=event->GetNumberOfTracks();\r | |
347 | for(Int_t itrack = 0; itrack < ntracks; itrack++){\r | |
348 | AliVTrack *track=(AliVTrack*)event->GetTrack(itrack);\r | |
349 | \r | |
350 | //\r | |
351 | //basic track cuts\r | |
352 | //\r | |
353 | ULong_t status=track->GetStatus();\r | |
354 | // not that nice. status bits not in virtual interface\r | |
355 | // TPC refit + ITS refit +\r | |
356 | // TOF out + TOFpid +\r | |
357 | // kTIME\r | |
358 | if (!( (status & AliVTrack::kEMCALmatch) == AliVTrack::kEMCALmatch) ) continue;\r | |
359 | \r | |
360 | Double_t pt=track->Pt();\r | |
361 | \r | |
362 | //EMCAL nSigma (only for electrons at the moment)\r | |
363 | TH2 *h=(TH2*)fListQAemcal->At(0);\r | |
364 | if (!h) continue;\r | |
365 | Double_t nSigma=fPIDResponse->NumberOfSigmasEMCAL(track, (AliPID::EParticleType)0);\r | |
366 | h->Fill(pt,nSigma);\r | |
367 | \r | |
368 | //EMCAL signal (E/p vs. pT)\r | |
369 | h=(TH2*)fListQAemcal->At(1);\r | |
370 | if (h) {\r | |
371 | \r | |
372 | Int_t nMatchClus = track->GetEMCALcluster();\r | |
373 | Double_t mom = track->P();\r | |
374 | Double_t eop = -1.;\r | |
375 | \r | |
376 | if(nMatchClus > -1){\r | |
377 | \r | |
378 | AliVCluster *matchedClus = (AliVCluster*)event->GetCaloCluster(nMatchClus);\r | |
379 | \r | |
380 | if(matchedClus){\r | |
381 | \r | |
382 | // matched cluster is EMCAL\r | |
383 | if(matchedClus->IsEMCAL()){\r | |
384 | \r | |
385 | Double_t fClsE = matchedClus->E();\r | |
386 | eop = fClsE/mom;\r | |
387 | \r | |
388 | h->Fill(pt,eop);\r | |
389 | \r | |
390 | }\r | |
391 | }\r | |
392 | }\r | |
393 | else{\r | |
394 | Printf("status status = AliVTrack::kEMCALmatch, BUT no matched cluster!");\r | |
395 | }\r | |
396 | }\r | |
397 | }\r | |
152a39e6 | 398 | }\r |
399 | \r | |
400 | //______________________________________________________________________________\r | |
401 | void AliAnalysisTaskPIDqa::FillTPCTOFqa()\r | |
402 | {\r | |
403 | //\r | |
404 | // Fill PID qa histograms for the TOF\r | |
405 | // Here also the TPC histograms after TOF selection are filled\r | |
406 | //\r | |
407 | \r | |
408 | AliVEvent *event=InputEvent();\r | |
409 | \r | |
410 | Int_t ntracks=event->GetNumberOfTracks();\r | |
411 | for(Int_t itrack = 0; itrack < ntracks; itrack++){\r | |
412 | AliVTrack *track=(AliVTrack*)event->GetTrack(itrack);\r | |
413 | \r | |
414 | //\r | |
415 | //basic track cuts\r | |
416 | //\r | |
417 | ULong_t status=track->GetStatus();\r | |
418 | // not that nice. status bits not in virtual interface\r | |
419 | // TPC refit + ITS refit +\r | |
420 | // TOF out + TOFpid +\r | |
421 | // kTIME\r | |
422 | if (!((status & AliVTrack::kTPCrefit) == AliVTrack::kTPCrefit) ||\r | |
423 | !((status & AliVTrack::kITSrefit) == AliVTrack::kITSrefit) ||\r | |
424 | !( (status & AliVTrack::kTPCpid ) == AliVTrack::kTPCpid ) ||\r | |
425 | !((status & AliVTrack::kTOFout ) == AliVTrack::kTOFout ) ||\r | |
426 | !((status & AliVTrack::kTOFpid ) == AliVTrack::kTOFpid ) ||\r | |
427 | !((status & AliVTrack::kTIME ) == AliVTrack::kTIME ) ) continue;\r | |
428 | \r | |
429 | Float_t nCrossedRowsTPC = track->GetTPCClusterInfo(2,1);\r | |
430 | Float_t ratioCrossedRowsOverFindableClustersTPC = 1.0;\r | |
431 | if (track->GetTPCNclsF()>0) {\r | |
432 | ratioCrossedRowsOverFindableClustersTPC = nCrossedRowsTPC/track->GetTPCNclsF();\r | |
433 | }\r | |
434 | \r | |
435 | if ( nCrossedRowsTPC<70 || ratioCrossedRowsOverFindableClustersTPC<.8 ) continue;\r | |
436 | \r | |
437 | \r | |
438 | Double_t mom=track->P();\r | |
439 | Double_t momTPC=track->GetTPCmomentum();\r | |
440 | \r | |
441 | for (Int_t ispecie=0; ispecie<AliPID::kSPECIES; ++ispecie){\r | |
442 | //TOF nSigma\r | |
443 | Double_t nSigmaTOF=fPIDResponse->NumberOfSigmasTOF(track, (AliPID::EParticleType)ispecie);\r | |
444 | Double_t nSigmaTPC=fPIDResponse->NumberOfSigmasTPC(track, (AliPID::EParticleType)ispecie);\r | |
445 | \r | |
446 | //TPC after TOF cut\r | |
447 | TH2 *h=(TH2*)fListQAtpctof->At(ispecie);\r | |
448 | if (h && TMath::Abs(nSigmaTOF)<3.) h->Fill(momTPC,nSigmaTPC);\r | |
449 | \r | |
450 | //TOF after TPC cut\r | |
451 | h=(TH2*)fListQAtpctof->At(ispecie+AliPID::kSPECIES);\r | |
452 | if (h && TMath::Abs(nSigmaTPC)<3.) h->Fill(mom,nSigmaTOF);\r | |
453 | \r | |
a3851b68 | 454 | //EMCAL after TOF and TPC cut\r |
455 | h=(TH2*)fListQAtpctof->At(ispecie+2*AliPID::kSPECIES);\r | |
456 | if (h && TMath::Abs(nSigmaTOF)<3. && TMath::Abs(nSigmaTPC)<3. ){\r | |
457 | \r | |
458 | Int_t nMatchClus = track->GetEMCALcluster();\r | |
459 | Double_t pt = track->Pt();\r | |
460 | Double_t eop = -1.;\r | |
461 | \r | |
462 | if(nMatchClus > -1){\r | |
463 | \r | |
464 | AliVCluster *matchedClus = (AliVCluster*)event->GetCaloCluster(nMatchClus);\r | |
465 | \r | |
466 | if(matchedClus){\r | |
467 | \r | |
468 | // matched cluster is EMCAL\r | |
469 | if(matchedClus->IsEMCAL()){\r | |
470 | \r | |
471 | Double_t fClsE = matchedClus->E();\r | |
472 | eop = fClsE/mom;\r | |
473 | \r | |
474 | h->Fill(pt,eop);\r | |
475 | \r | |
476 | \r | |
477 | }\r | |
478 | }\r | |
479 | }\r | |
480 | }\r | |
152a39e6 | 481 | }\r |
482 | }\r | |
483 | }\r | |
484 | \r | |
485 | //______________________________________________________________________________\r | |
486 | void AliAnalysisTaskPIDqa::SetupITSqa()\r | |
487 | {\r | |
488 | //\r | |
489 | // Create the ITS qa objects\r | |
490 | //\r | |
491 | \r | |
492 | TVectorD *vX=MakeLogBinning(200,.1,30);\r | |
493 | \r | |
494 | for (Int_t ispecie=0; ispecie<AliPID::kSPECIES; ++ispecie){\r | |
495 | TH2F *hNsigmaP = new TH2F(Form("hNsigmaP_ITS_%s",AliPID::ParticleName(ispecie)),\r | |
496 | Form("ITS n#sigma %s vs. p;p [GeV]; n#sigma",AliPID::ParticleName(ispecie)),\r | |
497 | vX->GetNrows()-1,vX->GetMatrixArray(),\r | |
498 | 200,-10,10);\r | |
499 | fListQAits->Add(hNsigmaP);\r | |
500 | }\r | |
501 | \r | |
502 | \r | |
503 | TH2F *hSig = new TH2F("hSigP_ITS",\r | |
504 | "ITS signal vs. p;p [GeV]; ITS signal [arb. units]",\r | |
505 | vX->GetNrows()-1,vX->GetMatrixArray(),\r | |
506 | 300,0,300);\r | |
507 | fListQAits->Add(hSig);\r | |
508 | \r | |
509 | delete vX; \r | |
510 | }\r | |
511 | \r | |
512 | //______________________________________________________________________________\r | |
513 | void AliAnalysisTaskPIDqa::SetupTPCqa()\r | |
514 | {\r | |
515 | //\r | |
516 | // Create the TPC qa objects\r | |
517 | //\r | |
518 | \r | |
519 | TVectorD *vX=MakeLogBinning(200,.1,30);\r | |
520 | \r | |
521 | for (Int_t ispecie=0; ispecie<AliPID::kSPECIES; ++ispecie){\r | |
522 | TH2F *hNsigmaP = new TH2F(Form("hNsigmaP_TPC_%s",AliPID::ParticleName(ispecie)),\r | |
523 | Form("TPC n#sigma %s vs. p;p [GeV]; n#sigma",AliPID::ParticleName(ispecie)),\r | |
524 | vX->GetNrows()-1,vX->GetMatrixArray(),\r | |
525 | 200,-10,10);\r | |
526 | fListQAtpc->Add(hNsigmaP);\r | |
527 | }\r | |
528 | \r | |
529 | \r | |
530 | TH2F *hSig = new TH2F("hSigP_TPC",\r | |
531 | "TPC signal vs. p;p [GeV]; TPC signal [arb. units]",\r | |
532 | vX->GetNrows()-1,vX->GetMatrixArray(),\r | |
533 | 300,0,300);\r | |
534 | fListQAtpc->Add(hSig);\r | |
535 | \r | |
536 | delete vX; \r | |
537 | }\r | |
538 | \r | |
539 | //______________________________________________________________________________\r | |
540 | void AliAnalysisTaskPIDqa::SetupTRDqa()\r | |
541 | {\r | |
542 | //\r | |
543 | // Create the TRD qa objects\r | |
544 | //\r | |
545 | \r | |
546 | }\r | |
547 | \r | |
548 | //______________________________________________________________________________\r | |
549 | void AliAnalysisTaskPIDqa::SetupTOFqa()\r | |
550 | {\r | |
551 | //\r | |
552 | // Create the TOF qa objects\r | |
553 | //\r | |
554 | \r | |
555 | TVectorD *vX=MakeLogBinning(200,.1,30);\r | |
556 | \r | |
557 | for (Int_t ispecie=0; ispecie<AliPID::kSPECIES; ++ispecie){\r | |
558 | TH2F *hNsigmaP = new TH2F(Form("hNsigmaP_TOF_%s",AliPID::ParticleName(ispecie)),\r | |
559 | Form("TOF n#sigma %s vs. p;p [GeV]; n#sigma",AliPID::ParticleName(ispecie)),\r | |
560 | vX->GetNrows()-1,vX->GetMatrixArray(),\r | |
561 | 200,-10,10);\r | |
562 | fListQAtof->Add(hNsigmaP);\r | |
563 | }\r | |
564 | \r | |
565 | \r | |
566 | TH2F *hSig = new TH2F("hSigP_TOF",\r | |
567 | "TOF signal vs. p;p [GeV]; TOF signal [arb. units]",\r | |
568 | vX->GetNrows()-1,vX->GetMatrixArray(),\r | |
569 | 300,0,300);\r | |
570 | fListQAtof->Add(hSig);\r | |
571 | \r | |
572 | delete vX; \r | |
573 | }\r | |
574 | \r | |
575 | //______________________________________________________________________________\r | |
576 | void AliAnalysisTaskPIDqa::SetupEMCALqa()\r | |
577 | {\r | |
578 | //\r | |
579 | // Create the EMCAL qa objects\r | |
580 | //\r | |
581 | \r | |
a3851b68 | 582 | TVectorD *vX=MakeLogBinning(200,.1,30);\r |
583 | \r | |
584 | TH2F *hNsigmaPt = new TH2F(Form("hNsigmaPt_EMCAL_%s",AliPID::ParticleName(0)),\r | |
585 | Form("EMCAL n#sigma %s vs. p_{T};p_{T} [GeV]; n#sigma",AliPID::ParticleName(0)),\r | |
586 | vX->GetNrows()-1,vX->GetMatrixArray(),\r | |
587 | 200,-10,10);\r | |
588 | fListQAemcal->Add(hNsigmaPt); \r | |
589 | \r | |
590 | TH2F *hSigPt = new TH2F("hSigPt_EMCAL",\r | |
591 | "EMCAL signal (E/p) vs. p_{T};p_{T} [GeV]; EMCAL signal (E/p) [arb. units]",\r | |
592 | vX->GetNrows()-1,vX->GetMatrixArray(),\r | |
593 | 200,0,2);\r | |
594 | fListQAemcal->Add(hSigPt);\r | |
595 | \r | |
596 | delete vX; \r | |
152a39e6 | 597 | }\r |
598 | \r | |
599 | //______________________________________________________________________________\r | |
600 | void AliAnalysisTaskPIDqa::SetupTPCTOFqa()\r | |
601 | {\r | |
602 | //\r | |
603 | // Create the qa objects for TPC + TOF combination\r | |
604 | //\r | |
605 | \r | |
606 | TVectorD *vX=MakeLogBinning(200,.1,30);\r | |
607 | \r | |
608 | //TPC signals after TOF cut\r | |
609 | for (Int_t ispecie=0; ispecie<AliPID::kSPECIES; ++ispecie){\r | |
610 | TH2F *hNsigmaP = new TH2F(Form("hNsigmaP_TPC_TOF_%s",AliPID::ParticleName(ispecie)),\r | |
611 | Form("TPC n#sigma %s vs. p (after TOF 3#sigma cut);p_{TPC} [GeV]; n#sigma",AliPID::ParticleName(ispecie)),\r | |
612 | vX->GetNrows()-1,vX->GetMatrixArray(),\r | |
613 | 200,-10,10);\r | |
614 | fListQAtpctof->Add(hNsigmaP);\r | |
615 | }\r | |
616 | \r | |
617 | //TOF signals after TPC cut\r | |
618 | for (Int_t ispecie=0; ispecie<AliPID::kSPECIES; ++ispecie){\r | |
619 | TH2F *hNsigmaP = new TH2F(Form("hNsigmaP_TOF_TPC_%s",AliPID::ParticleName(ispecie)),\r | |
620 | Form("TOF n#sigma %s vs. p (after TPC n#sigma cut);p [GeV]; n#sigma",AliPID::ParticleName(ispecie)),\r | |
621 | vX->GetNrows()-1,vX->GetMatrixArray(),\r | |
622 | 200,-10,10);\r | |
623 | fListQAtpctof->Add(hNsigmaP);\r | |
624 | }\r | |
625 | \r | |
a3851b68 | 626 | //EMCAL signal after TOF and TPC cut\r |
627 | for (Int_t ispecie=0; ispecie<AliPID::kSPECIES; ++ispecie){\r | |
628 | TH2F *heopPt = new TH2F(Form("heopPt_TOF_TPC_%s",AliPID::ParticleName(ispecie)),\r | |
629 | Form("EMCAL signal (E/p) %s vs. p_{T};p_{T} [GeV]; EMCAL signal (E/p) [arb. units]",AliPID::ParticleName(ispecie)),\r | |
630 | vX->GetNrows()-1,vX->GetMatrixArray(),\r | |
631 | 200,0,2);\r | |
632 | fListQAtpctof->Add(heopPt);\r | |
633 | }\r | |
634 | \r | |
152a39e6 | 635 | delete vX;\r |
636 | }\r | |
637 | \r | |
638 | //______________________________________________________________________________\r | |
639 | TVectorD* AliAnalysisTaskPIDqa::MakeLogBinning(Int_t nbinsX, Double_t xmin, Double_t xmax)\r | |
640 | {\r | |
641 | //\r | |
642 | // Make logarithmic binning\r | |
643 | // the user has to delete the array afterwards!!!\r | |
644 | //\r | |
645 | \r | |
646 | //check limits\r | |
647 | if (xmin<1e-20 || xmax<1e-20){\r | |
648 | AliError("For Log binning xmin and xmax must be > 1e-20. Using linear binning instead!");\r | |
649 | return MakeLinBinning(nbinsX, xmin, xmax);\r | |
650 | }\r | |
651 | if (xmax<xmin){\r | |
652 | Double_t tmp=xmin;\r | |
653 | xmin=xmax;\r | |
654 | xmax=tmp;\r | |
655 | }\r | |
656 | TVectorD *binLim=new TVectorD(nbinsX+1);\r | |
657 | Double_t first=xmin;\r | |
658 | Double_t last=xmax;\r | |
659 | Double_t expMax=TMath::Log(last/first);\r | |
660 | for (Int_t i=0; i<nbinsX+1; ++i){\r | |
661 | (*binLim)[i]=first*TMath::Exp(expMax/nbinsX*(Double_t)i);\r | |
662 | }\r | |
663 | return binLim;\r | |
664 | }\r | |
665 | \r | |
666 | //______________________________________________________________________________\r | |
667 | TVectorD* AliAnalysisTaskPIDqa::MakeLinBinning(Int_t nbinsX, Double_t xmin, Double_t xmax)\r | |
668 | {\r | |
669 | //\r | |
670 | // Make linear binning\r | |
671 | // the user has to delete the array afterwards!!!\r | |
672 | //\r | |
673 | if (xmax<xmin){\r | |
674 | Double_t tmp=xmin;\r | |
675 | xmin=xmax;\r | |
676 | xmax=tmp;\r | |
677 | }\r | |
678 | TVectorD *binLim=new TVectorD(nbinsX+1);\r | |
679 | Double_t first=xmin;\r | |
680 | Double_t last=xmax;\r | |
681 | Double_t binWidth=(last-first)/nbinsX;\r | |
682 | for (Int_t i=0; i<nbinsX+1; ++i){\r | |
683 | (*binLim)[i]=first+binWidth*(Double_t)i;\r | |
684 | }\r | |
685 | return binLim;\r | |
686 | }\r | |
687 | \r | |
688 | //_____________________________________________________________________________\r | |
689 | TVectorD* AliAnalysisTaskPIDqa::MakeArbitraryBinning(const char* bins)\r | |
690 | {\r | |
691 | //\r | |
692 | // Make arbitrary binning, bins separated by a ','\r | |
693 | //\r | |
694 | TString limits(bins);\r | |
695 | if (limits.IsNull()){\r | |
696 | AliError("Bin Limit string is empty, cannot add the variable");\r | |
697 | return 0x0;\r | |
698 | }\r | |
699 | \r | |
700 | TObjArray *arr=limits.Tokenize(",");\r | |
701 | Int_t nLimits=arr->GetEntries();\r | |
702 | if (nLimits<2){\r | |
703 | AliError("Need at leas 2 bin limits, cannot add the variable");\r | |
704 | delete arr;\r | |
705 | return 0x0;\r | |
706 | }\r | |
707 | \r | |
708 | TVectorD *binLimits=new TVectorD(nLimits);\r | |
709 | for (Int_t iLim=0; iLim<nLimits; ++iLim){\r | |
710 | (*binLimits)[iLim]=(static_cast<TObjString*>(arr->At(iLim)))->GetString().Atof();\r | |
711 | }\r | |
712 | \r | |
713 | delete arr;\r | |
714 | return binLimits;\r | |
715 | }\r |