Updates by Ionut
[u/mrichter/AliRoot.git] / PWGPP / TRD / macros / AddTRDcheckESD.C
1
2 #if ! defined (__CINT__) || defined (__MAKECINT__)
3 #include "AliLog.h"
4 #include "AliAnalysisManager.h"
5 #include "AliAnalysisDataContainer.h"
6 #include "AliAnalysisCuts.h"
7 #include "AliESDtrackCuts.h"
8 #include "PWGPP/TRD/AliTRDcheckESD.h"
9 #endif
10
11 AliESDtrackCuts* SetupESDcuts();
12
13 void AddTRDcheckESD(AliAnalysisManager *mgr)
14 {
15   //AliLog::SetClassDebugLevel("AliTRDcheckESD", 5);
16   //  AliInfo("aaaaaa6666666666");
17   AliTRDcheckESD *checkESD = new AliTRDcheckESD((char*)"TRDcheckESD");
18   checkESD->SetRefTrackFilter(SetupESDcuts());
19   mgr->AddTask(checkESD);
20   Bool_t mc = mgr->GetMCtruthEventHandler();
21   checkESD->SetMC(mc);
22   checkESD->SetCollision(/*kFALSE*/);
23   checkESD->SetDebugLevel(0);
24   checkESD->AddUserTrigger("WU");
25   checkESD->AddUserTrigger("QU");
26   checkESD->AddUserTrigger("SE");
27   checkESD->AddUserTrigger("JT");
28   
29   mgr->ConnectInput(checkESD,  0, mgr->GetCommonInputContainer());  
30   mgr->ConnectOutput(checkESD, 1, mgr->CreateContainer(checkESD->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Performance",mgr->GetCommonFileName())));
31 }
32
33 AliESDtrackCuts* SetupESDcuts() {
34   // Setup ESD cuts for the TPC reference tracks
35   AliESDtrackCuts* esdCuts = new AliESDtrackCuts;
36   esdCuts->SetPtRange(0.5, 100.0);
37   esdCuts->SetEtaRange(-0.9, +0.9);
38   esdCuts->SetRequireTPCRefit(kTRUE);
39   esdCuts->SetAcceptKinkDaughters(kFALSE);
40   esdCuts->SetMaxDCAToVertexXY(1.);
41   esdCuts->SetMaxDCAToVertexZ(3.);
42   esdCuts->SetMinNClustersTPC(70);
43   esdCuts->SetRequireITSRefit(kTRUE);
44   //esdCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kAny);
45   return esdCuts;
46 }
47