1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
16 /* $Id: AliAnalysisTaskPIDResponse.cxx 43811 2010-09-23 14:13:31Z wiechula $ */
20 #include <AliAnalysisManager.h>
21 #include <AliInputEventHandler.h>
22 #include <AliVEventHandler.h>
23 #include <AliVEvent.h>
24 #include <AliVParticle.h>
25 #include <AliVTrack.h>
27 #include <AliPIDResponse.h>
28 #include <AliESDpid.h>
30 #include "AliAnalysisTaskPIDResponse.h"
32 ClassImp(AliAnalysisTaskPIDResponse)
34 //______________________________________________________________________________
35 AliAnalysisTaskPIDResponse::AliAnalysisTaskPIDResponse():
44 fIsTunedOnData(kFALSE),
52 //______________________________________________________________________________
53 AliAnalysisTaskPIDResponse::AliAnalysisTaskPIDResponse(const char* name):
54 AliAnalysisTaskSE(name),
62 fIsTunedOnData(kFALSE),
66 // Default constructor
68 DefineInput(0,TChain::Class());
71 //______________________________________________________________________________
72 AliAnalysisTaskPIDResponse::~AliAnalysisTaskPIDResponse()
79 //______________________________________________________________________________
80 void AliAnalysisTaskPIDResponse::UserCreateOutputObjects()
83 // Create the output QA objects
86 AliLog::SetClassDebugLevel("AliAnalysisTaskPIDResponse",10);
89 AliAnalysisManager *man=AliAnalysisManager::GetAnalysisManager();
90 AliInputEventHandler *inputHandler=dynamic_cast<AliInputEventHandler*>(man->GetInputEventHandler());
91 if (!inputHandler) AliFatal("Input handler needed");
94 inputHandler->CreatePIDResponse(fIsMC);
95 fPIDResponse=inputHandler->GetPIDResponse();
96 if (!fPIDResponse) AliFatal("PIDResponse object was not created");
98 fPIDResponse->SetOADBPath(AliAnalysisManager::GetOADBPath());
99 if (!fOADBPath.IsNull()) fPIDResponse->SetOADBPath(fOADBPath.Data());
101 if(fIsTunedOnData) fPIDResponse->SetTunedOnData(kTRUE,fRecoPassTuned);
104 //______________________________________________________________________________
105 void AliAnalysisTaskPIDResponse::UserExec(Option_t */*option*/)
107 // Setup the PID response functions and fill the QA histograms
109 AliVEvent *event=InputEvent();
111 fRun=event->GetRunNumber();
118 fPIDResponse->InitialiseEvent(event,fRecoPass);
119 AliESDpid *pidresp = dynamic_cast<AliESDpid*>(fPIDResponse);
120 if(pidresp && AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()){
121 pidresp->SetEventHandler(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
123 //create and attach transient PID object
125 fPIDResponse->FillTrackDetectorPID();
129 //______________________________________________________________________________
130 void AliAnalysisTaskPIDResponse::SetRecoInfo()
133 // Set reconstruction information
139 //Get the current file to check the reconstruction pass (UGLY, but not stored in ESD... )
140 AliAnalysisManager *mgr=AliAnalysisManager::GetAnalysisManager();
141 AliVEventHandler *inputHandler=mgr->GetInputEventHandler();
142 if (!inputHandler) return;
144 TTree *tree= (TTree*)inputHandler->GetTree();
145 TFile *file= (TFile*)tree->GetCurrentFile();
148 AliError("Current file not found, cannot set reconstruction information");
152 //find pass from file name (UGLY, but not stored in ESD... )
153 TString fileName(file->GetName());
154 if (fileName.Contains("pass1") ) {
156 } else if (fileName.Contains("pass2") ) {
158 } else if (fileName.Contains("pass3") ) {
160 } else if (fileName.Contains("pass4") ) {
162 } else if (fileName.Contains("pass5") ) {
166 fPIDResponse->SetCurrentFile(fileName.Data());