1 /**************************************************************************
\r
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
\r
4 * Author: The ALICE Off-line Project. *
\r
5 * Contributors are mentioned in the code where appropriate. *
\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
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
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
25 // Author: J.Otwinowski 01/04/2009
\r
26 // Changes by M.Knichel 15/10/2010
\r
27 //------------------------------------------------------------------------------
\r
34 #include "TCanvas.h"
\r
37 #include "TSystem.h"
\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
52 #include "AliCentrality.h"
\r
53 #include "AliESDVZERO.h"
\r
54 #include "AliMultiplicity.h"
\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
68 using namespace std;
\r
70 ClassImp(AliPerformanceTask)
\r
72 //_____________________________________________________________________________
\r
73 AliPerformanceTask::AliPerformanceTask()
\r
74 : AliAnalysisTaskSE("Performance")
\r
82 , fUseMCInfo(kFALSE)
\r
83 , fUseESDfriend(kFALSE)
\r
85 , fUseTerminate(kTRUE)
\r
88 , fUseCentralityBin(0)
\r
90 // Dummy Constructor
\r
91 // should not be used
\r
94 //_____________________________________________________________________________
\r
95 AliPerformanceTask::AliPerformanceTask(const char *name, const char */*title*/)
\r
96 : AliAnalysisTaskSE(name)
\r
101 , fOutputSummary(0)
\r
104 , fUseMCInfo(kFALSE)
\r
105 , fUseESDfriend(kFALSE)
\r
107 , fUseTerminate(kTRUE)
\r
108 , fUseCentrality(0)
\r
110 , fUseCentralityBin(0)
\r
114 // Define input and output slots here
\r
115 DefineOutput(0, TTree::Class());
\r
116 DefineOutput(1, TList::Class());
\r
118 // create the list for comparison objects
\r
119 fCompList = new TList;
\r
122 //_____________________________________________________________________________
\r
123 AliPerformanceTask::~AliPerformanceTask()
\r
125 if (fOutput) delete fOutput; fOutput = 0;
\r
126 if (fOutputSummary) delete fOutputSummary; fOutputSummary = 0;
\r
127 if (fCompList) delete fCompList; fCompList = 0;
\r
130 //_____________________________________________________________________________
\r
131 Bool_t AliPerformanceTask::AddPerformanceObject(AliPerformanceObject *pObj)
\r
133 // add comparison object to the list
\r
135 Printf("ERROR: Could not add comparison object");
\r
139 // add object to the list
\r
140 fCompList->AddLast(pObj);
\r
145 //_____________________________________________________________________________
\r
146 void AliPerformanceTask::UserCreateOutputObjects()
\r
148 // Create histograms
\r
151 // create output list
\r
152 fOutput = new TList;
\r
153 fOutput->SetOwner();
\r
154 fPitList = fOutput->MakeIterator();
\r
156 // create output list
\r
157 //fOutputSummary = new TTree;
\r
159 // add comparison objects to the output
\r
160 AliPerformanceObject *pObj=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
168 Printf("UserCreateOutputObjects(): Number of output comparison objects: %d \n", count);
\r
170 PostData(1, fOutput);
\r
171 PostData(0, fOutputSummary);
\r
174 //_____________________________________________________________________________
\r
175 void AliPerformanceTask::UserExec(Option_t *)
\r
178 // Called for each event
\r
181 // Decide whether to use HLT or Offline ESD
\r
184 AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*>
\r
185 (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
\r
188 printf("ERROR: Could not get ESDInputHandler");
\r
191 fESD = esdH->GetHLTEvent();
\r
194 fESD = (AliESDEvent*) (InputEvent());
\r
198 fESDfriend = static_cast<AliESDfriend*>(fESD->FindListObject("AliESDfriend"));
\r
200 Printf("ERROR: ESD friends not available");
\r
210 Printf("ERROR: ESD event not available");
\r
214 if (fUseMCInfo && !fMC) {
\r
215 Printf("ERROR: MC event not available");
\r
222 Printf("ERROR: ESD friends not available");
\r
226 // Process analysis
\r
227 Bool_t process = kTRUE;
\r
229 // Check for centrality
\r
230 if (fUseCentrality) {
\r
231 if ( CalculateCentralityBin() != fUseCentralityBin ) {
\r
236 // Process comparison
\r
238 AliPerformanceObject *pObj=0;
\r
240 while(( pObj = (AliPerformanceObject *)fPitList->Next()) != NULL) {
\r
241 pObj->Exec(fMC,fESD,fESDfriend,fUseMCInfo,fUseESDfriend);
\r
245 // Post output data.
\r
246 PostData(1, fOutput);
\r
249 //_____________________________________________________________________________
\r
250 void AliPerformanceTask::Terminate(Option_t *)
\r
252 // Called once at the end
\r
254 if ( !fUseTerminate )
\r
257 // check output data
\r
258 fOutputSummary = dynamic_cast<TTree*> (GetOutputData(0));
\r
259 fOutput = dynamic_cast<TList*> (GetOutputData(1));
\r
261 Printf("ERROR: AliPerformanceTask::Terminate(): fOutput data not available ..." );
\r
264 if (fOutputSummary) { delete fOutputSummary; fOutputSummary=0; }
\r
265 AliPerformanceObject* pObj=0;
\r
266 AliPerformanceTPC* pTPC = 0;
\r
267 AliPerformanceDEdx* pDEdx = 0;
\r
268 AliPerformanceMatch* pMatch = 0;
\r
269 TIterator* itOut = fOutput->MakeIterator();
\r
271 while(( pObj = dynamic_cast<AliPerformanceObject*>(itOut->Next())) != NULL) {
\r
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
279 printf("DO NOT USE OCDB \n");
\r
283 if (! AliCDBManager::Instance()->GetDefaultStorage()) { AliCDBManager::Instance()->SetDefaultStorage("raw://"); }
\r
285 TString tmpFile = gSystem->TempDirectory() + TString("/TPCQASummary.") + uuid.AsString() + TString(".root");
\r
286 AliTPCPerformanceSummary::WriteToFile(pTPC, pDEdx, pMatch, tmpFile.Data());
\r
287 TChain* chain = new TChain("tpcQA");
\r
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
294 // Post output data.
\r
295 PostData(0, fOutputSummary);
\r
299 //_____________________________________________________________________________
\r
300 void AliPerformanceTask::FinishTaskOutput()
\r
302 // called once at the end of each job (on the workernode)
\r
304 // projects THnSparse to TH1,2,3
\r
306 fOutput = dynamic_cast<TList*> (GetOutputData(1));
\r
308 Printf("ERROR: AliPerformanceTask::FinishTaskOutput(): fOutput data not available ..." );
\r
312 AliPerformanceObject* pObj=0;
\r
313 TIterator* itOut = fOutput->MakeIterator();
\r
315 while(( pObj = dynamic_cast<AliPerformanceObject*>(itOut->Next())) != NULL) {
\r
316 pObj->SetRunNumber(fCurrentRunNumber);
\r
320 // Post output data.
\r
321 PostData(1, fOutput);
\r
324 //_____________________________________________________________________________
\r
325 Bool_t AliPerformanceTask::Notify()
\r
327 static Int_t count = 0;
\r
329 Printf("Processing %d. file", count);
\r
334 //________________________________________________________________________
\r
335 Int_t AliPerformanceTask::CalculateCentralityBin(){
\r
336 // Get Centrality bin
\r
338 Int_t centrality = -1;
\r
339 Float_t centralityF = -1;
\r
341 if (fUseCentrality == 0)
\r
344 AliCentrality *esdCentrality = fESD->GetCentrality();
\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
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