]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWG3/muon/AliAnalysisTaskMuonQA.cxx
Coverity fix (an obsolete constructor removed)
[u/mrichter/AliRoot.git] / PWG3 / muon / AliAnalysisTaskMuonQA.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 /* $Id$ */
17
18 #include <Riostream.h>
19
20 // ROOT includes
21 #include "TH1F.h"
22 #include "TH2F.h"
23 #include "TCanvas.h"
24 #include "TROOT.h"
25 #include "TString.h"
26 #include "TObjArray.h"
27 #include "TMath.h"
28 #include "TFile.h"
29 #include "TRegexp.h"
30 #include "TMap.h"
31 #include "TList.h"
32 #include "TObjString.h"
33
34 // STEER includes
35 #include "AliESDEvent.h"
36 #include "AliESDMuonTrack.h"
37 #include "AliESDMuonCluster.h"
38 #include "AliESDInputHandler.h"
39 #include "AliESDVZERO.h"
40
41 // ANALYSIS includes
42 #include "AliAnalysisTaskSE.h"
43 #include "AliAnalysisDataSlot.h"
44 #include "AliAnalysisManager.h"
45 #include "AliAnalysisTaskMuonQA.h"
46 #include "AliCounterCollection.h"
47
48 ClassImp(AliAnalysisTaskMuonQA)
49
50 const Int_t AliAnalysisTaskMuonQA::nCh = 10;
51
52 const Int_t AliAnalysisTaskMuonQA::nDE = 1100;
53
54 const Float_t AliAnalysisTaskMuonQA::dMax[5] = {176.6, 229.0, 308.84, 418.2,  522.0}; // cm
55
56 //________________________________________________________________________
57 AliAnalysisTaskMuonQA::AliAnalysisTaskMuonQA() :
58   AliAnalysisTaskSE(), 
59   fList(0x0),
60   fListExpert(0x0),
61   fListNorm(0x0),
62   fTrackCounters(0x0),
63   fEventCounters(0x0),
64   fSelectCharge(0),
65   fSelectPhysics(kFALSE),
66   fSelectTrigger(kFALSE),
67   fTriggerMask(0),
68   fSelectMatched(kFALSE),
69   fApplyAccCut(kFALSE),
70   fTriggerClass(0x0),
71   fSelectTriggerClass(0x0)
72 {
73 // Dummy constructor
74 }
75
76 //________________________________________________________________________
77 AliAnalysisTaskMuonQA::AliAnalysisTaskMuonQA(const char *name) :
78   AliAnalysisTaskSE(name), 
79   fList(0x0),
80   fListExpert(0x0),
81   fListNorm(0x0),
82   fTrackCounters(0x0),
83   fEventCounters(0x0),
84   fSelectCharge(0),
85   fSelectPhysics(kFALSE),
86   fSelectTrigger(kFALSE),
87   fTriggerMask(0),
88   fSelectMatched(kFALSE),
89   fApplyAccCut(kFALSE),
90   fTriggerClass(0x0),
91   fSelectTriggerClass(0x0)
92 {
93   /// Constructor
94   
95   // Output slot #1 writes into a TObjArray container
96   DefineOutput(1,TObjArray::Class());
97   // Output slot #2 writes into a TObjArray container
98   DefineOutput(2,TObjArray::Class());
99   // Output slot #3 writes track counters
100   DefineOutput(3,AliCounterCollection::Class());
101   // Output slot #4 writes event counters
102   DefineOutput(4,AliCounterCollection::Class());
103   // Output slot #5 writes normalized histograms
104   DefineOutput(5,TObjArray::Class());
105 }
106
107 //________________________________________________________________________
108 AliAnalysisTaskMuonQA::~AliAnalysisTaskMuonQA()
109 {
110   /// Destructor
111   if (!AliAnalysisManager::GetAnalysisManager()->IsProofMode()) {
112     delete fList;
113     delete fListExpert;
114     delete fTrackCounters;
115     delete fEventCounters;
116   }
117   delete fListNorm;
118   delete fTriggerClass;
119   delete fSelectTriggerClass;
120 }
121
122 //___________________________________________________________________________
123 void AliAnalysisTaskMuonQA::UserCreateOutputObjects()
124 {
125   /// Create histograms and counters
126   
127   // set the list of trigger classes with corresponding short names
128   fTriggerClass = new TMap(20);
129   fTriggerClass->SetOwnerKeyValue();
130   // p-p trigger classes
131   fTriggerClass->Add(new TObjString("CBEAMB"), new TObjString("CBEAMB"));
132   fTriggerClass->Add(new TObjString("CINT1B-ABCE-NOPF-ALL"), new TObjString("CINT1B"));
133   fTriggerClass->Add(new TObjString("CMUS1B-ABCE-NOPF-MUON"), new TObjString("CMUS1B"));
134   fTriggerClass->Add(new TObjString("CINT1[AC]-"), new TObjString("CINT1AC"));
135   fTriggerClass->Add(new TObjString("CMUS1[AC]-"), new TObjString("CMUS1AC"));
136   fTriggerClass->Add(new TObjString("CINT1-E-"), new TObjString("CINT1E"));
137   fTriggerClass->Add(new TObjString("CINT5-E-"), new TObjString("CINT5E"));
138   fTriggerClass->Add(new TObjString("CMUS1-E-"), new TObjString("CMUS1E"));
139   fTriggerClass->Add(new TObjString("CMUS5-E-"), new TObjString("CMUS5E"));
140   fTriggerClass->Add(new TObjString("CINT1-B-"), new TObjString("CINT1B"));
141   fTriggerClass->Add(new TObjString("CINT5-B-"), new TObjString("CINT5B"));
142   fTriggerClass->Add(new TObjString("CMUS1-B-"), new TObjString("CMUS1B"));
143   fTriggerClass->Add(new TObjString("CMUS5-B-"), new TObjString("CMUS5B"));
144   fTriggerClass->Add(new TObjString("CINT1-AC-"), new TObjString("CINT1AC"));
145   fTriggerClass->Add(new TObjString("CINT5-AC-"), new TObjString("CINT5AC"));
146   fTriggerClass->Add(new TObjString("CMUS1-AC-"), new TObjString("CMUS1AC"));
147   fTriggerClass->Add(new TObjString("CMUS5-AC-"), new TObjString("CMUS5AC"));
148   fTriggerClass->Add(new TObjString("CSH1-B-"), new TObjString("CSH1B"));
149
150   TString side_pp[3] = {"B", "AC", "E"};
151         for(Int_t i = 0; i< 3; i++){
152                 fTriggerClass->Add(new TObjString(Form("CINT7-%s-", side_pp[i].Data())), new TObjString(Form("CINT7%s",side_pp[i].Data())));
153                 fTriggerClass->Add(new TObjString(Form("CMUSH7-%s-",side_pp[i].Data())), new TObjString(Form("CMUSH7%s",side_pp[i].Data())));
154                 fTriggerClass->Add(new TObjString(Form("CMUL7-%s-",side_pp[i].Data())), new TObjString(Form("CMUL7%s",side_pp[i].Data())));
155                 fTriggerClass->Add(new TObjString(Form("CMUU7-%s-",side_pp[i].Data())), new TObjString(Form("CMUU7%s",side_pp[i].Data())));
156         }
157   // Pb-Pb trigger classes
158   TString side[4] = {"B", "A", "C", "E"};
159   for (Int_t i = 0; i < 4; i++) {
160     fTriggerClass->Add(new TObjString(Form("CMBACS2-%s-", side[i].Data())), new TObjString(Form("CMBACS2-%s", side[i].Data())));
161     fTriggerClass->Add(new TObjString(Form("CMBS2A-%s-", side[i].Data())), new TObjString(Form("CMBS2A-%s", side[i].Data())));
162     fTriggerClass->Add(new TObjString(Form("CMBS2C-%s-", side[i].Data())), new TObjString(Form("CMBS2C-%s", side[i].Data())));
163     fTriggerClass->Add(new TObjString(Form("CMBAC-%s-", side[i].Data())), new TObjString(Form("CMBAC-%s", side[i].Data())));
164     fTriggerClass->Add(new TObjString(Form("C0SMH-%s-", side[i].Data())), new TObjString(Form("C0SMH-%s", side[i].Data())));
165   }
166   
167   // set the list of trigger classes that can be selected to fill histograms (in case the physics selection is not used)
168   fSelectTriggerClass = new TList();
169   fSelectTriggerClass->SetOwner();
170   // p-p trigger classes
171   fSelectTriggerClass->AddLast(new TObjString("CINT1B-ABCE-NOPF-ALL")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kMB);
172   fSelectTriggerClass->AddLast(new TObjString("CMUS1B-ABCE-NOPF-MUON")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kMUON);
173   fSelectTriggerClass->AddLast(new TObjString("CINT1-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kMB);
174   fSelectTriggerClass->AddLast(new TObjString("CINT5-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kCINT5);
175   fSelectTriggerClass->AddLast(new TObjString("CMUS1-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kMUON);
176   fSelectTriggerClass->AddLast(new TObjString("CMUS5-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kCMUS5);
177         fSelectTriggerClass->AddLast(new TObjString("CINT7-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kINT7);
178   fSelectTriggerClass->AddLast(new TObjString("CMUSH7-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kMUSH7);
179   fSelectTriggerClass->AddLast(new TObjString("CMUU7-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kMUU7);
180   fSelectTriggerClass->AddLast(new TObjString("CMUL7-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kMUL7);
181   fSelectTriggerClass->AddLast(new TObjString("CSH1-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kHighMult);
182         
183   // Pb-Pb trigger classes
184   fSelectTriggerClass->AddLast(new TObjString("CMBACS2-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kMB);
185   fSelectTriggerClass->AddLast(new TObjString("CMBS2A-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kMB);
186   fSelectTriggerClass->AddLast(new TObjString("CMBS2C-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kMB);
187   fSelectTriggerClass->AddLast(new TObjString("CMBAC-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kMB);
188   fSelectTriggerClass->AddLast(new TObjString("C0SMH-B-")); fSelectTriggerClass->Last()->SetUniqueID(AliVEvent::kHighMult);
189   
190   // create histograms
191   fList = new TObjArray(2000);
192   fList->SetOwner();
193   fListExpert = new TObjArray(2000);
194   fListExpert->SetOwner();
195   
196   // track info
197   TH1F* hNTracks = new TH1F("hNTracks", "number of tracks;n_{tracks}", 20, 0., 20.);
198   fList->AddAtAndExpand(hNTracks, kNTracks);
199   
200   TH1F* hMatchTrig = new TH1F("hMatchTrig", "number of tracks matched with trigger;n_{tracks}", 20, 0., 20.);
201   fList->AddAtAndExpand(hMatchTrig, kMatchTrig);
202   
203   TH1F* hSign = new TH1F("hSign", "track sign;sign", 3, -1.5, 1.5);
204   fList->AddAtAndExpand(hSign, kSign);
205   
206   TH1F* hDCA = new TH1F("hDCA", "DCA distribution;DCA (cm)", 500, 0., 500.);
207   fList->AddAtAndExpand(hDCA, kDCA);
208   
209   TH1F* hP = new TH1F("hP", "momentum distribution;p (GeV/c)", 300, 0., 300.);
210   fList->AddAtAndExpand(hP, kP);
211   
212   TH1F* hPMuPlus = new TH1F("hPMuPlus", "momentum distribution of #mu^{+};p (GeV/c)", 300, 0., 300.);
213   fList->AddAtAndExpand(hPMuPlus, kPMuPlus);
214   
215   TH1F* hPMuMinus = new TH1F("hPMuMinus", "momentum distribution of #mu^{-};p (GeV/c)", 300, 0., 300.);
216   fList->AddAtAndExpand(hPMuMinus, kPMuMinus);
217   
218   TH1F* hPt = new TH1F("hPt", "transverse momentum distribution;p_{t} (GeV/c)", 300, 0., 30);
219   fList->AddAtAndExpand(hPt, kPt);
220   
221   TH1F* hPtMuPlus = new TH1F("hPtMuPlus", "transverse momentum distribution of #mu^{+};p_{t} (GeV/c)", 300, 0., 30);
222   fList->AddAtAndExpand(hPtMuPlus, kPtMuPlus);
223   
224   TH1F* hPtMuMinus = new TH1F("hPtMuMinus", "transverse momentum distribution of #mu^{-};p_{t} (GeV/c)", 300, 0., 30);
225   fList->AddAtAndExpand(hPtMuMinus, kPtMuMinus);
226   
227   TH1F* hRapidity = new TH1F("hRapidity", "rapidity distribution;rapidity", 200, -4.5, -2.);
228   fList->AddAtAndExpand(hRapidity, kRapidity);
229   
230   TH1F* hThetaX = new TH1F("hThetaX", "#theta_{X} distribution;#theta_{X} (degree)", 360, -180., 180);
231   fList->AddAtAndExpand(hThetaX, kThetaX);
232   
233   TH1F* hThetaY = new TH1F("hThetaY", "#theta_{Y} distribution;#theta_{Y} (degree)", 360, -180., 180);
234   fList->AddAtAndExpand(hThetaY, kThetaY);
235   
236   TH1F* hChi2 = new TH1F("hChi2", "normalized #chi^{2} distribution;#chi^{2} / ndf", 500, 0., 50.);
237   fList->AddAtAndExpand(hChi2, kChi2);
238   
239   TH1F* hProbChi2 = new TH1F("hProbChi2", "distribution of probability of #chi^{2};prob(#chi^{2})", 100, 0., 1.);
240   fList->AddAtAndExpand(hProbChi2, kProbChi2);
241   
242   // cluster info
243   TH1F* hNClustersPerTrack = new TH1F("hNClustersPerTrack", "number of associated clusters per track;n_{clusters}", 20, 0., 20.);
244   fList->AddAtAndExpand(hNClustersPerTrack, kNClustersPerTrack);
245   
246   TH1F* hNChamberHitPerTrack = new TH1F("hNChamberHitPerTrack", "number of chambers hit per track;n_{chamber hit}", 15, 0., 15.);
247   fList->AddAtAndExpand(hNChamberHitPerTrack, kNChamberHitPerTrack);
248   
249   TH1F* hNClustersPerCh = new TH1F("hNClustersPerCh", "averaged number of clusters per chamber per track;chamber ID;<n_{clusters}>", nCh, -0.5, nCh-0.5);
250   hNClustersPerCh->Sumw2();
251   hNClustersPerCh->SetOption("P");
252   hNClustersPerCh->SetMarkerStyle(kFullDotMedium);
253   hNClustersPerCh->SetMarkerColor(kBlue);
254   fListExpert->AddAtAndExpand(hNClustersPerCh, kNClustersPerCh);
255   
256   TH1F* hNClustersPerDE = new TH1F("hNClustersPerDE", "averaged number of clusters per DE per track;DetElem ID;<n_{clusters}>", nDE+1, -0.5, nDE+0.5);
257   hNClustersPerDE->Sumw2();
258   hNClustersPerDE->SetOption("P");
259   hNClustersPerDE->SetMarkerStyle(kFullDotMedium);
260   hNClustersPerDE->SetMarkerColor(kBlue);
261   fListExpert->AddAtAndExpand(hNClustersPerDE, kNClustersPerDE);
262   
263   for (Int_t i = 0; i < nCh; i++) {
264     Float_t rMax = 0.5*dMax[i/2];
265     TH2F* hClusterHitMapInCh = new TH2F(Form("hClusterHitMapInCh%d",i+1), Form("cluster position distribution in chamber %d;X (cm);Y (cm)",i+1),
266                                        100, -rMax, rMax, 100, -rMax, rMax);
267     fListExpert->AddAtAndExpand(hClusterHitMapInCh, kClusterHitMapInCh+i);
268     
269     TH1F* hClusterChargeInCh = new TH1F(Form("hClusterChargeInCh%d",i+1), Form("cluster charge distribution in chamber %d;charge (fC)",i+1), 100, 0., 1000.);
270     fListExpert->AddAtAndExpand(hClusterChargeInCh, kClusterChargeInCh+i);
271     
272     TH1F* hClusterSizeInCh = new TH1F(Form("hClusterSizeInCh%d",i+1), Form("cluster size distribution in chamber %d;size (n_{pads})",i+1), 200, 0., 200.);
273     fListExpert->AddAtAndExpand(hClusterSizeInCh, kClusterSizeInCh+i);
274   }
275   
276   TH2F* hClusterChargePerDE = new TH2F("hClusterChargePerDE", "cluster charge distribution per DE;DetElem ID;charge (fC)", nDE+1, -0.5, nDE+0.5, 100, 0., 1000.);
277   fListExpert->AddAtAndExpand(hClusterChargePerDE, kClusterChargePerDE);
278   
279   TH2F* hClusterSizePerDE = new TH2F("hClusterSizePerDE", "cluster size distribution per DE;DetElem ID;size (n_{pads})", nDE+1, -0.5, nDE+0.5, 200, 0., 200.);
280   fListExpert->AddAtAndExpand(hClusterSizePerDE, kClusterSizePerDE);
281   
282   // initialize track counters
283   fTrackCounters = new AliCounterCollection("trackCounters");
284   fTrackCounters->AddRubric("track", "trackeronly/triggeronly/matched");
285   fTrackCounters->AddRubric("trigger", 1000000);
286   fTrackCounters->AddRubric("run", 1000000);
287   fTrackCounters->AddRubric("selected", "yes/no");
288   fTrackCounters->AddRubric("triggerRO", "good/bad");
289   fTrackCounters->AddRubric("v0mult", "low/int/high/any");
290   fTrackCounters->AddRubric("charge", "pos/neg/any");
291   fTrackCounters->AddRubric("pt", "low/high/any");
292   fTrackCounters->AddRubric("acc", "in/out");
293   fTrackCounters->Init();
294   
295   // initialize event counters
296   fEventCounters = new AliCounterCollection("eventCounters");
297   fEventCounters->AddRubric("event", "muon/any");
298   fEventCounters->AddRubric("trigger", 1000000);
299   fEventCounters->AddRubric("run", 1000000);
300   fEventCounters->AddRubric("selected", "yes/no");
301   fEventCounters->AddRubric("triggerRO", "good/bad");
302   fEventCounters->AddRubric("v0mult", "low/int/high/any");
303   fEventCounters->Init();
304   
305   // Post data at least once per task to ensure data synchronisation (required for merging)
306   PostData(1, fList);
307   PostData(2, fListExpert);
308   PostData(3, fTrackCounters);
309   PostData(4, fEventCounters);
310 }
311
312 //________________________________________________________________________
313 void AliAnalysisTaskMuonQA::UserExec(Option_t *)
314 {
315   /// Called for each event
316   
317   AliESDEvent* fESD = dynamic_cast<AliESDEvent*>(InputEvent());
318   if (!fESD) {
319     Printf("ERROR: fESD not available");
320     return;
321   }
322   
323   // check physics selection
324   UInt_t triggerWord = (fInputHandler) ? fInputHandler->IsEventSelected() : 0;
325   Bool_t isPhysicsSelected = (triggerWord != 0);
326   TString selected = isPhysicsSelected ? "selected:yes" : "selected:no";
327   
328   // check trigger selection 
329   TString FiredTriggerClasses = fESD->GetFiredTriggerClasses();
330   if (!fSelectPhysics) triggerWord = BuildTriggerWord(FiredTriggerClasses);
331   Bool_t isTriggerSelected = ((triggerWord & fTriggerMask) != 0);
332   
333   // get the V0 multiplicity (except for p-p)
334   AliESDVZERO* v0Data = fESD->GetVZEROData();
335   Float_t v0Mult = 0.;
336   if (v0Data && strcmp(fESD->GetBeamType(),"p-p"))
337     for (Int_t i = 0 ; i < 64 ; i++) v0Mult += v0Data->GetMultiplicity(i);
338   TList listV0MultKey;
339   listV0MultKey.SetOwner();
340   listV0MultKey.AddLast(new TObjString("v0mult:any"));
341   //if (v0Mult > 239. && v0Mult < 559.) listV0MultKey.AddLast(new TObjString("v0mult:low"));
342   if (v0Mult >= 239. && v0Mult < 1165.) listV0MultKey.AddLast(new TObjString("v0mult:low"));
343   else if (v0Mult >= 1165. && v0Mult < 12191.) listV0MultKey.AddLast(new TObjString("v0mult:int"));
344   else if (v0Mult >= 12191. && v0Mult < 20633.) listV0MultKey.AddLast(new TObjString("v0mult:high"));
345   TIter nextV0MultKey(&listV0MultKey);
346   
347   // first loop over tracks to check for trigger readout problem
348   Int_t maxTriggerRO = (!strcmp(fESD->GetBeamType(),"p-p")) ? 10 : 1000;
349   Int_t nTracks = (Int_t) fESD->GetNumberOfMuonTracks();
350   Int_t nTriggerTracks = 0;
351   for (Int_t iTrack = 0; iTrack < nTracks; ++iTrack)
352     if (fESD->GetMuonTrack(iTrack)->ContainTriggerData()) nTriggerTracks++;
353   TString triggerRO = (nTriggerTracks < maxTriggerRO) ? "triggerRO:good" : "triggerRO:bad";
354   
355   // --- fill event counters ---
356   
357   // build the list of trigger cases
358   TList* triggerCases = BuildListOfTriggerCases(FiredTriggerClasses);
359   
360   // loop over trigger cases
361   TObjString* triggerKey = 0x0;
362   TIter nextTriggerCase(triggerCases);
363   while ((triggerKey = static_cast<TObjString*>(nextTriggerCase()))) {
364     
365     // loop over V0Mult cases
366     TObjString* v0MultKey = 0x0;
367     nextV0MultKey.Reset();
368     while ((v0MultKey = static_cast<TObjString*>(nextV0MultKey()))) {
369       
370       // any event
371       fEventCounters->Count(Form("event:any/%s/run:%d/%s/%s/%s", triggerKey->GetName(), fCurrentRunNumber, selected.Data(), triggerRO.Data(), v0MultKey->GetName()));
372       
373       // muon event
374       if (nTracks > 0) fEventCounters->Count(Form("event:muon/%s/run:%d/%s/%s/%s", triggerKey->GetName(), fCurrentRunNumber, selected.Data(), triggerRO.Data(), v0MultKey->GetName()));
375       
376     }
377       
378   }
379   
380   // second loop over tracks to fill histograms and track counters
381   Int_t nSelectedTrackerTracks = 0;
382   Int_t nSelectedTrackMatchTrig = 0;
383   Int_t nPVTracks = fESD->GetPrimaryVertex()->GetNContributors();
384   for (Int_t iTrack = 0; iTrack < nTracks; ++iTrack) {
385     
386     // get the ESD track
387     AliESDMuonTrack* esdTrack = fESD->GetMuonTrack(iTrack);
388     
389     // --- fill track counters ---
390     
391     // define the key words
392     TString trackKey = "track:";
393     TString chargeKey = "charge:";
394     TString accKey = "acc:";
395     Bool_t lowPt = kFALSE;
396     Bool_t highPt = kFALSE;
397     if (esdTrack->ContainTrackerData()) {
398       
399       if (esdTrack->ContainTriggerData()) trackKey += "matched";
400       else  trackKey += "trackeronly";
401       
402       Short_t trackCharge = esdTrack->Charge();
403       if (trackCharge < 0) chargeKey += "neg";
404       else chargeKey += "pos";
405       
406       Double_t thetaTrackAbsEnd = TMath::ATan(esdTrack->GetRAtAbsorberEnd()/505.) * TMath::RadToDeg();
407       Double_t trackPt = esdTrack->Pt();
408       if (trackPt > 1. && nPVTracks>0 && thetaTrackAbsEnd>2. ) lowPt = kTRUE;
409       if (trackPt > 2. && nPVTracks>0 && thetaTrackAbsEnd>2. ) highPt = kTRUE;
410       
411       Double_t eta = esdTrack->Eta();
412       if (thetaTrackAbsEnd < 2. || thetaTrackAbsEnd > 9. || eta < -4. || eta > -2.5) accKey += "out";
413       else accKey += "in";
414       
415     } else {
416       
417       trackKey += "triggeronly";
418       chargeKey = ""; // ghost have no charge specified
419       accKey += "out"; // ghost are labelled out of the acceptance
420       
421     }
422     
423     // loop over trigger cases and fill counters
424     nextTriggerCase.Reset();
425     while ((triggerKey = static_cast<TObjString*>(nextTriggerCase()))) {
426       
427       // loop over V0Mult cases
428       TObjString* v0MultKey = 0x0;
429       nextV0MultKey.Reset();
430       while ((v0MultKey = static_cast<TObjString*>(nextV0MultKey()))) {
431         
432         // any charge / any pt
433         fTrackCounters->Count(Form("%s/%s/run:%d/%s/%s/charge:any/pt:any/%s/%s", trackKey.Data(), triggerKey->GetName(), fCurrentRunNumber, selected.Data(), triggerRO.Data(), v0MultKey->GetName(), accKey.Data()));
434         
435         // any charge / specific pt
436         if (lowPt) fTrackCounters->Count(Form("%s/%s/run:%d/%s/%s/charge:any/pt:low/%s/%s", trackKey.Data(), triggerKey->GetName(), fCurrentRunNumber, selected.Data(), triggerRO.Data(), v0MultKey->GetName(), accKey.Data()));
437         if (highPt) fTrackCounters->Count(Form("%s/%s/run:%d/%s/%s/charge:any/pt:high/%s/%s", trackKey.Data(), triggerKey->GetName(), fCurrentRunNumber, selected.Data(), triggerRO.Data(), v0MultKey->GetName(), accKey.Data()));
438         
439         if (!chargeKey.IsNull()) {
440           
441           // specific charge / any pt
442           fTrackCounters->Count(Form("%s/%s/run:%d/%s/%s/%s/pt:any/%s/%s", trackKey.Data(), triggerKey->GetName(), fCurrentRunNumber, selected.Data(), triggerRO.Data(), chargeKey.Data(), v0MultKey->GetName(), accKey.Data()));
443           
444           // specific charge / specific pt
445           if (lowPt) fTrackCounters->Count(Form("%s/%s/run:%d/%s/%s/%s/pt:low/%s/%s", trackKey.Data(), triggerKey->GetName(), fCurrentRunNumber, selected.Data(), triggerRO.Data(), chargeKey.Data(), v0MultKey->GetName(), accKey.Data()));
446           if (highPt) fTrackCounters->Count(Form("%s/%s/run:%d/%s/%s/%s/pt:high/%s/%s", trackKey.Data(), triggerKey->GetName(), fCurrentRunNumber, selected.Data(), triggerRO.Data(), chargeKey.Data(), v0MultKey->GetName(), accKey.Data()));
447           
448         }
449         
450       }
451       
452     }
453     
454     // --- apply selections and fill histograms with selected tracks ---
455     
456     // remove "ghost"
457     if (!esdTrack->ContainTrackerData()) continue;
458     
459     // select on "physics" before filling histograms
460     if (fSelectPhysics && !isPhysicsSelected) continue;
461     
462     // select on trigger before filling histograms
463     if (fSelectTrigger && !isTriggerSelected) continue;
464     
465     // select on track charge
466     if (fSelectCharge*esdTrack->Charge() < 0) continue;
467     
468     // select on track matching
469     if (fSelectMatched && !esdTrack->ContainTriggerData()) continue;
470     
471     // skip tracks that do not pass the acceptance cuts if required
472     if (fApplyAccCut && accKey.EndsWith("out")) continue;
473     
474     nSelectedTrackerTracks++;
475     if (esdTrack->ContainTriggerData()) nSelectedTrackMatchTrig++;
476     
477     Double_t trackP = esdTrack->P();
478     Double_t trackPt = esdTrack->Pt();
479     Short_t trackCharge = esdTrack->Charge();
480     ((TH1F*)fList->UncheckedAt(kP))->Fill(trackP);
481     ((TH1F*)fList->UncheckedAt(kPt))->Fill(trackPt);
482     if (trackCharge < 0) {
483       ((TH1F*)fList->UncheckedAt(kPMuMinus))->Fill(trackP);
484       ((TH1F*)fList->UncheckedAt(kPtMuMinus))->Fill(trackPt);
485     } else {
486       ((TH1F*)fList->UncheckedAt(kPMuPlus))->Fill(trackP);
487       ((TH1F*)fList->UncheckedAt(kPtMuPlus))->Fill(trackPt);
488     }
489     ((TH1F*)fList->UncheckedAt(kRapidity))->Fill(esdTrack->Y());
490     Int_t ndf = 2 * esdTrack->GetNHit() - 5;
491     ((TH1F*)fList->UncheckedAt(kChi2))->Fill(esdTrack->GetChi2()/ndf);
492     ((TH1F*)fList->UncheckedAt(kProbChi2))->Fill(TMath::Prob(esdTrack->GetChi2(),ndf));
493     ((TH1F*)fList->UncheckedAt(kThetaX))->Fill(ChangeThetaRange(esdTrack->GetThetaXUncorrected()));
494     ((TH1F*)fList->UncheckedAt(kThetaY))->Fill(ChangeThetaRange(esdTrack->GetThetaYUncorrected()));
495     ((TH1F*)fList->UncheckedAt(kNClustersPerTrack))->Fill(esdTrack->GetNHit());
496     ((TH1F*)fList->UncheckedAt(kSign))->Fill(trackCharge);
497     ((TH1F*)fList->UncheckedAt(kDCA))->Fill(esdTrack->GetDCA());
498     
499     Int_t nChamberHit = 0;
500     for (Int_t ich=0; ich<10; ich++) if (esdTrack->IsInMuonClusterMap(ich)) nChamberHit++;
501     ((TH1F*)fList->UncheckedAt(kNChamberHitPerTrack))->Fill(nChamberHit);
502     
503     // what follows concern clusters
504     if(!esdTrack->ClustersStored()) continue;
505     
506     AliESDMuonCluster *esdCluster = (AliESDMuonCluster*) esdTrack->GetClusters().First();
507     while (esdCluster) {
508       
509       Int_t chId = esdCluster->GetChamberId();
510       Int_t deId = esdCluster->GetDetElemId();
511       
512       ((TH1F*)fListExpert->UncheckedAt(kNClustersPerCh))->Fill(chId);
513       ((TH1F*)fListExpert->UncheckedAt(kNClustersPerDE))->Fill(deId);
514       
515       ((TH1F*)fListExpert->UncheckedAt(kClusterHitMapInCh+chId))->Fill(esdCluster->GetX(), esdCluster->GetY());
516       
517       ((TH1F*)fListExpert->UncheckedAt(kClusterChargeInCh+chId))->Fill(esdCluster->GetCharge());
518       ((TH1F*)fListExpert->UncheckedAt(kClusterChargePerDE))->Fill(deId, esdCluster->GetCharge());
519       
520       if (esdCluster->PadsStored()) { // discard clusters with pad not stored in ESD
521         ((TH1F*)fListExpert->UncheckedAt(kClusterSizeInCh+chId))->Fill(esdCluster->GetNPads());
522         ((TH1F*)fListExpert->UncheckedAt(kClusterSizePerDE))->Fill(deId, esdCluster->GetNPads());
523       }
524       
525       esdCluster = (AliESDMuonCluster*) esdTrack->GetClusters().After(esdCluster);
526     }
527     
528   }
529   
530   if ((!fSelectPhysics || isPhysicsSelected) && (!fSelectTrigger || isTriggerSelected)) {
531     ((TH1F*)fList->UncheckedAt(kNTracks))->Fill(nSelectedTrackerTracks);
532     ((TH1F*)fList->UncheckedAt(kMatchTrig))->Fill(nSelectedTrackMatchTrig);
533   }
534   
535   // clean memory
536   delete triggerCases;
537   
538   // Post final data. It will be written to a file with option "RECREATE"
539   PostData(1, fList);
540   PostData(2, fListExpert);
541   PostData(3, fTrackCounters);
542   PostData(4, fEventCounters);
543 }
544
545 //________________________________________________________________________
546 void AliAnalysisTaskMuonQA::Terminate(Option_t *)
547 {
548   /// Normalize histograms
549   /// Draw result to the screen
550   /// Print statistics
551   
552   // global statistic
553   fTrackCounters = static_cast<AliCounterCollection*>(GetOutputData(3));
554   fEventCounters = static_cast<AliCounterCollection*>(GetOutputData(4));
555   if (fTrackCounters && fEventCounters) {
556     if (!gROOT->IsBatch()) {
557       cout<<"whole statistics without selection:"<<endl;
558       fEventCounters->Print("trigger/event");
559       fTrackCounters->Print("trigger/track");
560       cout<<"whole statistics of selected events:"<<endl;
561       fEventCounters->Print("trigger/event","selected:yes");
562       fTrackCounters->Print("trigger/track","selected:yes");
563       new TCanvas();
564       fEventCounters->Draw("event","trigger","");
565       new TCanvas();
566       fTrackCounters->Draw("track","trigger","");
567       new TCanvas();
568       fEventCounters->Draw("event","trigger","selected:yes");
569       new TCanvas();
570       fTrackCounters->Draw("track","trigger","selected:yes");
571     }
572   }
573   
574   // recover output histograms
575   fList = static_cast<TObjArray*>(GetOutputData(1));
576   fListExpert = static_cast<TObjArray*>(GetOutputData(2));
577   if (!fList || !fListExpert) return;
578   
579   // create summary plots
580   fListNorm = new TObjArray(1000);
581   fListNorm->SetOwner();
582   
583   // mean/dispersion of cluster charge per chamber/DE
584   TH1F* hClusterChargePerChMean = new TH1F("hClusterChargePerChMean", "cluster mean charge per chamber;chamber ID;<charge> (fC)", nCh, -0.5, nCh-0.5);
585   hClusterChargePerChMean->SetOption("P");
586   hClusterChargePerChMean->SetMarkerStyle(kFullDotMedium);
587   hClusterChargePerChMean->SetMarkerColor(kBlue);
588   fListNorm->AddAtAndExpand(hClusterChargePerChMean, kClusterChargePerChMean);
589   
590   TH1F* hClusterChargePerChSigma = new TH1F("hClusterChargePerChSigma", "cluster charge dispersion per chamber;chamber ID;#sigma_{charge} (fC)", nCh, -0.5, nCh-0.5);
591   hClusterChargePerChSigma->SetOption("P");
592   hClusterChargePerChSigma->SetMarkerStyle(kFullDotMedium);
593   hClusterChargePerChSigma->SetMarkerColor(kBlue);
594   fListNorm->AddAtAndExpand(hClusterChargePerChSigma, kClusterChargePerChSigma);
595   
596   TH1F* hClusterChargePerDEMean = new TH1F("hClusterChargePerDEMean", "cluster mean charge per DE;DetElem ID;<charge> (fC)", nDE+1, -0.5, nDE+0.5);
597   hClusterChargePerDEMean->SetOption("P");
598   hClusterChargePerDEMean->SetMarkerStyle(kFullDotMedium);
599   hClusterChargePerDEMean->SetMarkerColor(kBlue);
600   fListNorm->AddAtAndExpand(hClusterChargePerDEMean, kClusterChargePerDEMean);
601   
602   TH1F* hClusterChargePerDESigma = new TH1F("hClusterChargePerDESigma", "cluster charge dispersion per DE;DetElem ID;#sigma_{charge} (fC)", nDE+1, -0.5, nDE+0.5);
603   hClusterChargePerDESigma->SetOption("P");
604   hClusterChargePerDESigma->SetMarkerStyle(kFullDotMedium);
605   hClusterChargePerDESigma->SetMarkerColor(kBlue);
606   fListNorm->AddAtAndExpand(hClusterChargePerDESigma, kClusterChargePerDESigma);
607   
608   // mean/dispersion of cluster size per chamber/DE
609   TH1F* hClusterSizePerChMean = new TH1F("hClusterSizePerChMean", "cluster mean size per chamber;chamber ID;<size> (n_{pads})", nCh, -0.5, nCh-0.5);
610   hClusterSizePerChMean->SetOption("P");
611   hClusterSizePerChMean->SetMarkerStyle(kFullDotMedium);
612   hClusterSizePerChMean->SetMarkerColor(kBlue);
613   fListNorm->AddAtAndExpand(hClusterSizePerChMean, kClusterSizePerChMean);
614   
615   TH1F* hClusterSizePerChSigma = new TH1F("hClusterSizePerChSigma", "cluster size dispersion per chamber;chamber ID;#sigma_{size} (n_{pads})", nCh, -0.5, nCh-0.5);
616   hClusterSizePerChSigma->SetOption("P");
617   hClusterSizePerChSigma->SetMarkerStyle(kFullDotMedium);
618   hClusterSizePerChSigma->SetMarkerColor(kBlue);
619   fListNorm->AddAtAndExpand(hClusterSizePerChSigma, kClusterSizePerChSigma);
620   
621   TH1F* hClusterSizePerDEMean = new TH1F("hClusterSizePerDEMean", "cluster mean size per DE;DetElem ID;<size> (n_{pads})", nDE+1, -0.5, nDE+0.5);
622   hClusterSizePerDEMean->SetOption("P");
623   hClusterSizePerDEMean->SetMarkerStyle(kFullDotMedium);
624   hClusterSizePerDEMean->SetMarkerColor(kBlue);
625   fListNorm->AddAtAndExpand(hClusterSizePerDEMean, kClusterSizePerDEMean);
626   
627   TH1F* hClusterSizePerDESigma = new TH1F("hClusterSizePerDESigma", "cluster size dispersion per DE;DetElem ID;#sigma_{size} (n_{pads})", nDE+1, -0.5, nDE+0.5);
628   hClusterSizePerDESigma->SetOption("P");
629   hClusterSizePerDESigma->SetMarkerStyle(kFullDotMedium);
630   hClusterSizePerDESigma->SetMarkerColor(kBlue);
631   fListNorm->AddAtAndExpand(hClusterSizePerDESigma, kClusterSizePerDESigma);
632   
633   // normalize histograms
634   Float_t nTracks = ((TH1F*)fList->UncheckedAt(kNClustersPerTrack))->GetEntries();
635   if (nTracks > 0.) {
636     ((TH1F*)fListExpert->UncheckedAt(kNClustersPerCh))->Scale(1./nTracks);
637     ((TH1F*)fListExpert->UncheckedAt(kNClustersPerDE))->Scale(1./nTracks);
638   }
639   fListNorm->AddAtAndExpand(((TH1F*)fListExpert->UncheckedAt(kNClustersPerCh))->Clone(), kNClustersPerChPerTrack);
640   fListNorm->AddAtAndExpand(((TH1F*)fListExpert->UncheckedAt(kNClustersPerDE))->Clone(), kNClustersPerDEPerTrack);
641   
642   // fill summary plots per chamber
643   for (Int_t iCh = 0; iCh < nCh; iCh++) {
644     
645     TH1* hClusterChargeInCh = ((TH1F*)fListExpert->UncheckedAt(kClusterChargeInCh+iCh));
646     hClusterChargePerChMean->SetBinContent(iCh+1, hClusterChargeInCh->GetMean());
647     hClusterChargePerChMean->SetBinError(iCh+1, hClusterChargeInCh->GetMeanError());
648     hClusterChargePerChSigma->SetBinContent(iCh+1, hClusterChargeInCh->GetRMS());
649     hClusterChargePerChSigma->SetBinError(iCh+1, hClusterChargeInCh->GetRMSError());
650     
651     TH1* hClusterSizeInCh = ((TH1F*)fListExpert->UncheckedAt(kClusterSizeInCh+iCh));
652     hClusterSizePerChMean->SetBinContent(iCh+1, hClusterSizeInCh->GetMean());
653     hClusterSizePerChMean->SetBinError(iCh+1, hClusterSizeInCh->GetMeanError());
654     hClusterSizePerChSigma->SetBinContent(iCh+1, hClusterSizeInCh->GetRMS());
655     hClusterSizePerChSigma->SetBinError(iCh+1, hClusterSizeInCh->GetRMSError());
656     
657   }
658   
659   // fill summary plots per DE
660   TH2F* hClusterChargePerDE = ((TH2F*)fListExpert->UncheckedAt(kClusterChargePerDE));
661   TH2F* hClusterSizePerDE = ((TH2F*)fListExpert->UncheckedAt(kClusterSizePerDE));
662   for (Int_t iDE = 1; iDE < nDE+1; iDE++) {
663     
664     TH1D *tmp = hClusterChargePerDE->ProjectionY("tmp",iDE,iDE,"e");
665     if (tmp->GetEntries() > 10.) {
666       hClusterChargePerDEMean->SetBinContent(iDE, tmp->GetMean());
667       hClusterChargePerDEMean->SetBinError(iDE, tmp->GetMeanError());
668       hClusterChargePerDESigma->SetBinContent(iDE, tmp->GetRMS());
669       hClusterChargePerDESigma->SetBinError(iDE, tmp->GetRMSError());
670     }
671     delete tmp;
672     
673     tmp = hClusterSizePerDE->ProjectionY("tmp",iDE,iDE,"e");
674     if (tmp->GetEntries() > 10.) {
675       hClusterSizePerDEMean->SetBinContent(iDE, tmp->GetMean());
676       hClusterSizePerDEMean->SetBinError(iDE, tmp->GetMeanError());
677       hClusterSizePerDESigma->SetBinContent(iDE, tmp->GetRMS());
678       hClusterSizePerDESigma->SetBinError(iDE, tmp->GetRMSError());
679     }
680     delete tmp;
681     
682   }
683   
684   // Post summary data.
685   PostData(5, fListNorm);
686 }
687
688 //________________________________________________________________________
689 Double_t AliAnalysisTaskMuonQA::ChangeThetaRange(Double_t theta)
690 {
691   /// set theta range from -180 to +180 degrees
692   if(theta < -2.5) return (theta / TMath::Pi() + 1.) * 180.;
693   else if(theta > 2.5) return (theta / TMath::Pi() - 1.) * 180.;
694   else return theta / TMath::Pi() * 180.;
695 }
696
697 //________________________________________________________________________
698 UInt_t AliAnalysisTaskMuonQA::BuildTriggerWord(TString& FiredTriggerClasses)
699 {
700   /// build the trigger word from the fired trigger classes and the list of selectable trigger
701   
702   UInt_t word = 0;
703   
704   TObjString* trigClasseName = 0x0;
705   TIter nextTrigger(fSelectTriggerClass);
706   while ((trigClasseName = static_cast<TObjString*>(nextTrigger()))) {
707     
708     TRegexp GenericTriggerClasseName(trigClasseName->String());
709     if (FiredTriggerClasses.Contains(GenericTriggerClasseName)) word |= trigClasseName->GetUniqueID();
710     
711   }
712   
713   return word;
714 }
715
716 //________________________________________________________________________
717 TList* AliAnalysisTaskMuonQA::BuildListOfTriggerCases(TString& FiredTriggerClasses)
718 {
719   /// build the list of trigger for the counters from the fired trigger classes and the list of trigger classes
720   /// returned TList must be deleted by user
721   
722   TList* list = new TList();
723   list->SetOwner();
724   Bool_t foundCINT1B = kFALSE;
725   Bool_t foundCMUS1B = kFALSE;
726   
727   // add case any
728   list->AddLast(new TObjString("trigger:any"));
729   
730   TObjString* trigClasseName = 0x0;
731   TIter nextTrigger(fTriggerClass);
732   while ((trigClasseName = static_cast<TObjString*>(nextTrigger()))) {
733     
734     TRegexp GenericTriggerClasseName(trigClasseName->String());
735     if (FiredTriggerClasses.Contains(GenericTriggerClasseName)) {
736       
737       // add specific trigger case
738       TObjString* trigShortName = static_cast<TObjString*>(fTriggerClass->GetValue(trigClasseName));
739       list->AddLast(new TObjString(Form("trigger:%s",trigShortName->GetName())));
740       
741       // check for CINT1B and CMUS1B trigger
742       if (trigShortName->String() == "CINT1B") foundCINT1B = kTRUE;
743       else if (trigShortName->String() == "CMUS1B") foundCMUS1B = kTRUE;
744       
745     }
746     
747   }
748   
749   // add the special case CINT1B+CMUS1B
750   if (foundCINT1B && foundCMUS1B) list->AddLast(new TObjString("trigger:CINT1B+CMUS1B"));
751   
752   // add case other if no specific trigger was found
753   if (list->GetSize() == 1) list->AddLast(new TObjString("trigger:other"));
754   
755   return list;
756 }
757