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