From Francesco: Now I pass the MC event handler with a method avoiding to call AliAna...
[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTaskPIDResponse.cxx
CommitLineData
66ab8ab2 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
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 **************************************************************************/
15
16/* $Id: AliAnalysisTaskPIDResponse.cxx 43811 2010-09-23 14:13:31Z wiechula $ */
66ab8ab2 17#include <TFile.h>
66ab8ab2 18#include <TChain.h>
19
20#include <AliAnalysisManager.h>
21#include <AliInputEventHandler.h>
22#include <AliVEventHandler.h>
23#include <AliVEvent.h>
24#include <AliVParticle.h>
25#include <AliVTrack.h>
26#include <AliLog.h>
66ab8ab2 27#include <AliPIDResponse.h>
a864479b 28#include <AliESDpid.h>
29
66ab8ab2 30
31#include "AliAnalysisTaskPIDResponse.h"
32
66ab8ab2 33ClassImp(AliAnalysisTaskPIDResponse)
34
35//______________________________________________________________________________
36AliAnalysisTaskPIDResponse::AliAnalysisTaskPIDResponse():
37AliAnalysisTaskSE(),
38fIsMC(kFALSE),
67ad8680 39fOADBPath(),
66ab8ab2 40fPIDResponse(0x0),
66ab8ab2 41fRun(0),
42fOldRun(0),
539a5a59 43fRecoPass(0),
44fIsTunedOnData(kFALSE),
45fRecoPassTuned(0)
66ab8ab2 46{
47 //
48 // Dummy constructor
49 //
50}
51
52//______________________________________________________________________________
53AliAnalysisTaskPIDResponse::AliAnalysisTaskPIDResponse(const char* name):
54AliAnalysisTaskSE(name),
55fIsMC(kFALSE),
67ad8680 56fOADBPath(),
66ab8ab2 57fPIDResponse(0x0),
66ab8ab2 58fRun(0),
59fOldRun(0),
539a5a59 60fRecoPass(0),
61fIsTunedOnData(kFALSE),
62fRecoPassTuned(0)
66ab8ab2 63{
64 //
65 // Default constructor
66 //
67 DefineInput(0,TChain::Class());
66ab8ab2 68}
69
70//______________________________________________________________________________
71AliAnalysisTaskPIDResponse::~AliAnalysisTaskPIDResponse()
72{
73 //
74 // Destructor
75 //
66ab8ab2 76}
77
78//______________________________________________________________________________
79void AliAnalysisTaskPIDResponse::UserCreateOutputObjects()
80{
81 //
82 // Create the output QA objects
83 //
a864479b 84
66ab8ab2 85 AliLog::SetClassDebugLevel("AliAnalysisTaskPIDResponse",10);
66ab8ab2 86 //input hander
87 AliAnalysisManager *man=AliAnalysisManager::GetAnalysisManager();
88 AliInputEventHandler *inputHandler=dynamic_cast<AliInputEventHandler*>(man->GetInputEventHandler());
89 if (!inputHandler) AliFatal("Input handler needed");
6a45d03f 90
66ab8ab2 91 //pid response object
92 inputHandler->CreatePIDResponse(fIsMC);
93 fPIDResponse=inputHandler->GetPIDResponse();
87977cf5 94 if (!fPIDResponse) AliFatal("PIDResponse object was not created");
66ab8ab2 95
6a45d03f 96 fPIDResponse->SetOADBPath(AliAnalysisManager::GetOADBPath());
67ad8680 97 if (!fOADBPath.IsNull()) fPIDResponse->SetOADBPath(fOADBPath.Data());
539a5a59 98
99 if(fIsTunedOnData) fPIDResponse->SetTunedOnData(kTRUE,fRecoPassTuned);
66ab8ab2 100}
101
66ab8ab2 102//______________________________________________________________________________
103void AliAnalysisTaskPIDResponse::UserExec(Option_t */*option*/)
104{
66ab8ab2 105 // Setup the PID response functions and fill the QA histograms
106 //
66ab8ab2 107 AliVEvent *event=InputEvent();
108 if (!event) return;
109 fRun=event->GetRunNumber();
6a45d03f 110
66ab8ab2 111 if (fRun!=fOldRun){
6a45d03f 112 SetRecoInfo();
66ab8ab2 113 fOldRun=fRun;
114 }
115
6a45d03f 116 fPIDResponse->InitialiseEvent(event,fRecoPass);
a864479b 117 AliESDpid *pidresp = dynamic_cast<AliESDpid*>(fPIDResponse);
118 if(pidresp && AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()){
119 pidresp->SetEventHandler(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
120 }
66ab8ab2 121}
122
66ab8ab2 123//______________________________________________________________________________
124void AliAnalysisTaskPIDResponse::SetRecoInfo()
125{
126 //
127 // Set reconstruction information
128 //
129
130 //reset information
131 fRecoPass=0;
66ab8ab2 132
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;
137
138 TTree *tree= (TTree*)inputHandler->GetTree();
139 TFile *file= (TFile*)tree->GetCurrentFile();
140
141 if (!file) {
142 AliError("Current file not found, cannot set reconstruction information");
143 return;
144 }
66ab8ab2 145
146 //find pass from file name (UGLY, but not stored in ESD... )
147 TString fileName(file->GetName());
5a0aa7cb 148 if (fileName.Contains("pass1") ) {
66ab8ab2 149 fRecoPass=1;
5a0aa7cb 150 } else if (fileName.Contains("pass2") ) {
66ab8ab2 151 fRecoPass=2;
5a0aa7cb 152 } else if (fileName.Contains("pass3") ) {
66ab8ab2 153 fRecoPass=3;
6ca0da23 154 } else if (fileName.Contains("pass4") ) {
155 fRecoPass=4;
156 } else if (fileName.Contains("pass5") ) {
157 fRecoPass=5;
66ab8ab2 158 }
67ad8680 159
ab4db4eb 160 fPIDResponse->SetCurrentFile(fileName.Data());
66ab8ab2 161}