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>
31 #include "AliAnalysisTaskPIDResponse.h"
33 ClassImp(AliAnalysisTaskPIDResponse)
35 //______________________________________________________________________________
36 AliAnalysisTaskPIDResponse::AliAnalysisTaskPIDResponse():
44 fIsTunedOnData(kFALSE),
52 //______________________________________________________________________________
53 AliAnalysisTaskPIDResponse::AliAnalysisTaskPIDResponse(const char* name):
54 AliAnalysisTaskSE(name),
61 fIsTunedOnData(kFALSE),
65 // Default constructor
67 DefineInput(0,TChain::Class());
70 //______________________________________________________________________________
71 AliAnalysisTaskPIDResponse::~AliAnalysisTaskPIDResponse()
78 //______________________________________________________________________________
79 void AliAnalysisTaskPIDResponse::UserCreateOutputObjects()
82 // Create the output QA objects
85 AliLog::SetClassDebugLevel("AliAnalysisTaskPIDResponse",10);
87 AliAnalysisManager *man=AliAnalysisManager::GetAnalysisManager();
88 AliInputEventHandler *inputHandler=dynamic_cast<AliInputEventHandler*>(man->GetInputEventHandler());
89 if (!inputHandler) AliFatal("Input handler needed");
92 inputHandler->CreatePIDResponse(fIsMC);
93 fPIDResponse=inputHandler->GetPIDResponse();
94 if (!fPIDResponse) AliFatal("PIDResponse object was not created");
96 fPIDResponse->SetOADBPath(AliAnalysisManager::GetOADBPath());
97 if (!fOADBPath.IsNull()) fPIDResponse->SetOADBPath(fOADBPath.Data());
99 if(fIsTunedOnData) fPIDResponse->SetTunedOnData(kTRUE,fRecoPassTuned);
102 //______________________________________________________________________________
103 void AliAnalysisTaskPIDResponse::UserExec(Option_t */*option*/)
105 // Setup the PID response functions and fill the QA histograms
107 AliVEvent *event=InputEvent();
109 fRun=event->GetRunNumber();
116 fPIDResponse->InitialiseEvent(event,fRecoPass);
117 AliESDpid *pidresp = dynamic_cast<AliESDpid*>(fPIDResponse);
118 if(pidresp && AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()){
119 pidresp->SetEventHandler(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
123 //______________________________________________________________________________
124 void AliAnalysisTaskPIDResponse::SetRecoInfo()
127 // Set reconstruction information
133 //Get the current file to check the reconstruction pass (UGLY, but not stored in ESD... )
134 AliAnalysisManager *mgr=AliAnalysisManager::GetAnalysisManager();
135 AliVEventHandler *inputHandler=mgr->GetInputEventHandler();
136 if (!inputHandler) return;
138 TTree *tree= (TTree*)inputHandler->GetTree();
139 TFile *file= (TFile*)tree->GetCurrentFile();
142 AliError("Current file not found, cannot set reconstruction information");
146 //find pass from file name (UGLY, but not stored in ESD... )
147 TString fileName(file->GetName());
148 if (fileName.Contains("pass1") ) {
150 } else if (fileName.Contains("pass2") ) {
152 } else if (fileName.Contains("pass3") ) {
154 } else if (fileName.Contains("pass4") ) {
156 } else if (fileName.Contains("pass5") ) {
160 fPIDResponse->SetCurrentFile(fileName.Data());