Moving PWG1 to PWGPP
[u/mrichter/AliRoot.git] / PWGPP / TPC / AliPerformanceTask.cxx
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 //------------------------------------------------------------------------------\r
17 // Implementation of the AliPerformanceTask class. It checks reconstruction performance \r
18 // for the reconstructed vs MC particle tracks under several conditions. For real data \r
19 // the control QA histograms are filled.\r
20 //\r
21 // The comparison output objects deriving from AliPerformanceObject \r
22 // (e.g. AliPerformanceRes, AliPerformanceEff, AliPerformanceDEdx, AliPerformanceDCA ...) \r
23 // are stored in the output file (details in description of these classes).\r
24 // \r
25 // Author: J.Otwinowski 01/04/2009 \r
26 // Changes by M.Knichel 15/10/2010\r
27 //------------------------------------------------------------------------------\r
28 \r
29 #include "iostream"\r
30 \r
31 #include "TChain.h"\r
32 #include "TTree.h"\r
33 #include "TH1F.h"\r
34 #include "TCanvas.h"\r
35 #include "TList.h"\r
36 #include "TFile.h"\r
37 #include "TSystem.h"\r
38 \r
39 #include "AliAnalysisTask.h"\r
40 #include "AliAnalysisManager.h"\r
41 #include "AliESDEvent.h"\r
42 #include "AliESDfriend.h"\r
43 #include "AliMCEvent.h"\r
44 #include "AliESDInputHandler.h"\r
45 #include "AliMCEventHandler.h"\r
46 #include "AliESDVertex.h"\r
47 #include "AliMagF.h"\r
48 #include "AliTracker.h"\r
49 #include "AliGeomManager.h"\r
50 #include "AliCDBManager.h"\r
51 \r
52 #include "AliCentrality.h"\r
53 #include "AliESDVZERO.h"\r
54 #include "AliMultiplicity.h"\r
55 \r
56 #include "AliMCInfo.h"\r
57 #include "AliESDRecInfo.h"\r
58 #include "AliMCInfoCuts.h"\r
59 #include "AliRecInfoCuts.h"\r
60 #include "AliPerformanceObject.h"\r
61 #include "AliTPCPerformanceSummary.h"\r
62 #include "AliPerformanceTPC.h"\r
63 #include "AliPerformanceDEdx.h"\r
64 #include "AliPerformanceMatch.h"\r
65 #include "AliPerformanceTask.h"\r
66 \r
67 \r
68 using namespace std;\r
69 \r
70 ClassImp(AliPerformanceTask)\r
71 \r
72 //_____________________________________________________________________________\r
73 AliPerformanceTask::AliPerformanceTask() \r
74   : AliAnalysisTaskSE("Performance")\r
75   , fESD(0)\r
76   , fESDfriend(0)\r
77   , fMC(0)\r
78   , fOutput(0)\r
79   , fOutputSummary(0)\r
80   , fPitList(0)\r
81   , fCompList(0)\r
82   , fUseMCInfo(kFALSE)\r
83   , fUseESDfriend(kFALSE)\r
84   , fUseHLT(kFALSE)\r
85   , fUseTerminate(kTRUE)\r
86   , fUseCentrality(0)\r
87   , fUseOCDB(kTRUE)\r
88   , fUseCentralityBin(0)\r
89 {\r
90   // Dummy Constructor\r
91   // should not be used\r
92 }\r
93 \r
94 //_____________________________________________________________________________\r
95 AliPerformanceTask::AliPerformanceTask(const char *name, const char */*title*/) \r
96   : AliAnalysisTaskSE(name)\r
97   , fESD(0)\r
98   , fESDfriend(0)\r
99   , fMC(0)\r
100   , fOutput(0)\r
101   , fOutputSummary(0)\r
102   , fPitList(0)\r
103   , fCompList(0)\r
104   , fUseMCInfo(kFALSE)\r
105   , fUseESDfriend(kFALSE)\r
106   , fUseHLT(kFALSE)\r
107   , fUseTerminate(kTRUE)\r
108   , fUseCentrality(0)\r
109   , fUseOCDB(kTRUE)\r
110   , fUseCentralityBin(0)\r
111 {\r
112   // Constructor\r
113 \r
114   // Define input and output slots here\r
115   DefineOutput(0, TTree::Class());\r
116   DefineOutput(1, TList::Class());\r
117 \r
118   // create the list for comparison objects\r
119   fCompList = new TList;\r
120 }\r
121 \r
122 //_____________________________________________________________________________\r
123 AliPerformanceTask::~AliPerformanceTask()\r
124 {\r
125   if (!AliAnalysisManager::GetAnalysisManager()->IsProofMode()) {\r
126     if (fOutput)     delete fOutput;    fOutput   = 0; \r
127     if (fOutputSummary) delete fOutputSummary; fOutputSummary = 0;\r
128     if (fCompList)   delete fCompList;  fCompList = 0; \r
129   }\r
130 }\r
131 \r
132 //_____________________________________________________________________________\r
133 Bool_t AliPerformanceTask::AddPerformanceObject(AliPerformanceObject *pObj) \r
134 {\r
135   // add comparison object to the list\r
136   if(pObj == 0) {\r
137     Printf("ERROR: Could not add comparison object");\r
138     return kFALSE;\r
139   }\r
140 \r
141   // add object to the list\r
142   fCompList->AddLast(pObj);\r
143        \r
144 return kTRUE;\r
145 }\r
146 \r
147 //_____________________________________________________________________________\r
148 void AliPerformanceTask::UserCreateOutputObjects()\r
149 {\r
150   // Create histograms\r
151   // Called once\r
152 \r
153   // create output list\r
154   fOutput = new TList;\r
155   fOutput->SetOwner();\r
156   fPitList = fOutput->MakeIterator();\r
157   \r
158   // create output list\r
159   //fOutputSummary = new TTree;\r
160   \r
161   // add comparison objects to the output\r
162   AliPerformanceObject *pObj=0;\r
163   Int_t count=0;\r
164   TIterator *pitCompList = fCompList->MakeIterator();\r
165   pitCompList->Reset();\r
166   while(( pObj = (AliPerformanceObject *)pitCompList->Next()) != NULL) {\r
167     fOutput->Add(pObj);\r
168     count++;\r
169   }\r
170   Printf("UserCreateOutputObjects(): Number of output comparison objects: %d \n", count);\r
171   \r
172   PostData(1, fOutput);  \r
173   PostData(0, fOutputSummary);  \r
174 }\r
175 \r
176 //_____________________________________________________________________________\r
177 void AliPerformanceTask::UserExec(Option_t *) \r
178 {\r
179   // Main loop\r
180   // Called for each event\r
181 \r
182 \r
183   // Decide whether to use HLT or Offline ESD\r
184   if(fUseHLT){\r
185 \r
186     AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> \r
187       (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());\r
188     \r
189     if (!esdH) {\r
190       printf("ERROR: Could not get ESDInputHandler");\r
191       return;\r
192     } \r
193     fESD = esdH->GetHLTEvent();\r
194   }// end if fUseHLT\r
195   else  \r
196     fESD = (AliESDEvent*) (InputEvent());\r
197 \r
198   if(fUseESDfriend)\r
199     {\r
200       fESDfriend = static_cast<AliESDfriend*>(fESD->FindListObject("AliESDfriend"));\r
201       if(!fESDfriend) {\r
202         Printf("ERROR: ESD friends not available");\r
203       }\r
204     }\r
205   \r
206   if(fUseMCInfo) {\r
207       fMC = MCEvent();\r
208   }  \r
209 \r
210 \r
211   if (!fESD) {\r
212     Printf("ERROR: ESD event not available");\r
213     return;\r
214   }\r
215   \r
216   if (fUseMCInfo && !fMC) {\r
217     Printf("ERROR: MC event not available");\r
218     return;\r
219   }\r
220 \r
221   if(fUseESDfriend)\r
222   {\r
223     if(!fESDfriend) {\r
224     Printf("ERROR: ESD friends not available");\r
225     }\r
226   }\r
227 \r
228   // Process analysis\r
229   Bool_t process = kTRUE;\r
230 \r
231   // Check for centrality\r
232   if (fUseCentrality) {\r
233     if ( CalculateCentralityBin() != fUseCentralityBin ) {\r
234       process = kFALSE;\r
235     }\r
236   }\r
237 \r
238   // Process comparison\r
239   if (process) {\r
240     AliPerformanceObject *pObj=0;\r
241     fPitList->Reset();\r
242     while(( pObj = (AliPerformanceObject *)fPitList->Next()) != NULL) {\r
243           pObj->Exec(fMC,fESD,fESDfriend,fUseMCInfo,fUseESDfriend);\r
244     }\r
245   }\r
246 \r
247   // Post output data.\r
248   PostData(1, fOutput);\r
249 }\r
250 \r
251 //_____________________________________________________________________________\r
252 void AliPerformanceTask::Terminate(Option_t *) \r
253 {\r
254   // Called once at the end \r
255 \r
256   if ( !fUseTerminate )\r
257     return;\r
258   \r
259   // check output data\r
260     fOutputSummary = dynamic_cast<TTree*> (GetOutputData(0));\r
261     fOutput = dynamic_cast<TList*> (GetOutputData(1));\r
262     if (!fOutput) {\r
263         Printf("ERROR: AliPerformanceTask::Terminate(): fOutput data not available  ..." );\r
264         return;\r
265    }\r
266     if (fOutputSummary) { delete fOutputSummary; fOutputSummary=0; }      \r
267     AliPerformanceObject* pObj=0;\r
268     AliPerformanceTPC*  pTPC = 0;\r
269     AliPerformanceDEdx* pDEdx = 0;\r
270     AliPerformanceMatch* pMatch = 0;\r
271     AliPerformanceMatch* pPull = 0;\r
272     AliPerformanceMatch* pConstrain = 0;\r
273     TIterator* itOut = fOutput->MakeIterator();\r
274     itOut->Reset();\r
275     while(( pObj = dynamic_cast<AliPerformanceObject*>(itOut->Next())) != NULL) { \r
276       pObj->AnalyseFinal();\r
277       /*      if (!  pTPC)  {    pTPC = dynamic_cast<AliPerformanceTPC*>(pObj); }\r
278         if (! pDEdx)  {   pDEdx = dynamic_cast<AliPerformanceDEdx*>(pObj); }\r
279         if (! pMatch) {  pMatch = dynamic_cast<AliPerformanceMatch*>(pObj); }\r
280         if ((! pPull) && pMatch ) {  pPull = dynamic_cast<AliPerformanceMatch*>(pObj);*/\r
281 \r
282         if (!strcmp(pObj->GetName(),"AliPerformanceTPC"))  {    pTPC = dynamic_cast<AliPerformanceTPC*>(pObj); }\r
283         if (!strcmp(pObj->GetName(),"AliPerformanceDEdxTPCInner"))  {   pDEdx = dynamic_cast<AliPerformanceDEdx*>(pObj); }\r
284         if (!strcmp(pObj->GetName(),"AliPerformanceMatchTPCITS")) {  pMatch = dynamic_cast<AliPerformanceMatch*>(pObj); }\r
285         if (!strcmp(pObj->GetName(),"AliPerformanceMatchITSTPC")) {  pPull = dynamic_cast<AliPerformanceMatch*>(pObj);}\r
286         if (!strcmp(pObj->GetName(),"AliPerformanceMatchTPCConstrain")) {  pConstrain = dynamic_cast<AliPerformanceMatch*>(pObj);}\r
287     }\r
288   \r
289    \r
290     if(!fUseOCDB)  { \r
291       printf("DO NOT USE OCDB \n");\r
292       return;\r
293     }\r
294   \r
295     if (! AliCDBManager::Instance()->GetDefaultStorage()) { AliCDBManager::Instance()->SetDefaultStorage("raw://"); }\r
296     TUUID uuid;\r
297     TString tmpFile = gSystem->TempDirectory() + TString("/TPCQASummary.") + uuid.AsString() + TString(".root");\r
298     AliTPCPerformanceSummary::WriteToFile(pTPC, pDEdx, pMatch, pPull, pConstrain, tmpFile.Data());\r
299     TChain* chain = new TChain("tpcQA");\r
300     if(!chain) return;\r
301     chain->Add(tmpFile.Data());\r
302     TTree *tree = chain->CopyTree("1");\r
303     if (chain) { delete chain; chain=0; }\r
304     fOutputSummary = tree;\r
305       \r
306      // Post output data.\r
307      PostData(0, fOutputSummary);\r
308 \r
309 }\r
310 \r
311 //_____________________________________________________________________________\r
312 void AliPerformanceTask::FinishTaskOutput()\r
313 {\r
314     // called once at the end of each job (on the workernode)\r
315     //\r
316     // projects THnSparse to TH1,2,3\r
317     \r
318     fOutput = dynamic_cast<TList*> (GetOutputData(1));\r
319     if (!fOutput) {\r
320         Printf("ERROR: AliPerformanceTask::FinishTaskOutput(): fOutput data not available  ..." );\r
321         return;\r
322    }\r
323 \r
324       AliPerformanceObject* pObj=0;\r
325       TIterator* itOut = fOutput->MakeIterator();  \r
326       itOut->Reset();\r
327       while(( pObj = dynamic_cast<AliPerformanceObject*>(itOut->Next())) != NULL) {\r
328           pObj->SetRunNumber(fCurrentRunNumber);\r
329           pObj->Analyse();\r
330       }\r
331       \r
332      // Post output data.\r
333      PostData(1, fOutput);\r
334 }\r
335 \r
336 //_____________________________________________________________________________\r
337 Bool_t AliPerformanceTask::Notify()\r
338 {\r
339   static Int_t count = 0;\r
340   count++;\r
341   Printf("Processing %d. file", count);\r
342 \r
343   return kTRUE;\r
344 }\r
345 \r
346 //________________________________________________________________________\r
347 Int_t AliPerformanceTask::CalculateCentralityBin(){\r
348   // Get Centrality bin\r
349 \r
350   Int_t centrality = -1;\r
351   Float_t centralityF = -1;\r
352 \r
353   if (fUseCentrality == 0)\r
354     return centrality;\r
355 \r
356   AliCentrality *esdCentrality = fESD->GetCentrality();\r
357     \r
358   // New : 2010-11-18 JMT \r
359   if ( fUseCentrality == 1 )\r
360     centralityF = esdCentrality->GetCentralityPercentile("V0M");\r
361   else if ( fUseCentrality == 2 )\r
362     centralityF = esdCentrality->GetCentralityPercentile("CL1");\r
363   else if ( fUseCentrality == 3 )\r
364     centralityF = esdCentrality->GetCentralityPercentile("TRK"); \r
365   if (centralityF == 0.)\r
366     centralityF = 100.;\r
367 \r
368   if      ( centralityF >=  0. && centralityF <   5.) centrality =  0;\r
369   else if ( centralityF >=  5. && centralityF <  10.) centrality =  5;\r
370   else if ( centralityF >= 10. && centralityF <  20.) centrality = 10;\r
371   else if ( centralityF >= 20. && centralityF <  30.) centrality = 20;\r
372   else if ( centralityF >= 30. && centralityF <  40.) centrality = 30;\r
373   else if ( centralityF >= 40. && centralityF <  50.) centrality = 40;\r
374   else if ( centralityF >= 50. && centralityF <  60.) centrality = 50;\r
375   else if ( centralityF >= 60. && centralityF <  70.) centrality = 60;\r
376   else if ( centralityF >= 70. && centralityF <  80.) centrality = 70;\r
377   else if ( centralityF >= 80. && centralityF <  90.) centrality = 80;\r
378   else if ( centralityF >= 90. && centralityF <=100.) centrality = 90;\r
379   \r
380   return centrality;\r
381 }\r