3c312fdaaa4634f3aa59b03721c21ffde6457f34
[u/mrichter/AliRoot.git] / PWGPP / TRD / macros / AddTRDefficiency.C
1 #if ! defined (__CINT__) || defined (__MAKECINT__)
2 #include "TError.h"
3 #include "AliAnalysisManager.h"
4 #include "AliAnalysisDataContainer.h"
5 #include "PWGPP/TRD/AliTRDpwgppHelper.h"
6 #include "PWGPP/TRD/AliTRDefficiency.h"
7 #include "PWGPP/TRD/AliTRDefficiencyMC.h"
8 #include "PWGPP/TRD/AliTRDmultiplicity.h"
9 #endif
10
11 void AddTRDefficiency(AliAnalysisManager *mgr, Int_t map, AliAnalysisDataContainer **ci/*, AliAnalysisDataContainer **co*/)
12 {
13   Info("AddTRDefficiency", Form("[0]=\"%s\" [1]=\"%s\" [2]=\"%s\" [3]=\"%s\" [4]=\"%s\"", ci[0]->GetName(), ci[1]->GetName(), ci[2]->GetName(), ci[3]->GetName(), ci[4]->GetName()));
14
15   AliAnalysisDataContainer *evInfoContainer = ci[3];
16   AliTRDrecoTask *eff(NULL);
17   mgr->AddTask(eff = new AliTRDefficiency((char*)"TRDefficiency"));
18   eff->SetDebugLevel(0);
19   //AliLog::SetClassDebugLevel("AliTRDefficiency", 5);  
20   Int_t trackStatus = 0; // barrel tracks
21 //                    = 1; // kink tracks
22 //                    = 2; // SA tracks
23   mgr->ConnectInput(eff, 0, mgr->GetCommonInputContainer());  // connect main (ESD) container
24   mgr->ConnectInput(eff, 1, ci[trackStatus]);                 // conect track info container
25   mgr->ConnectInput(eff, 2, evInfoContainer);                 // conect event info container
26   mgr->ConnectInput(eff, 3, ci[4]);                 // conect clusters container
27   mgr->ConnectOutput(eff,1, mgr->CreateContainer(eff->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Performance", mgr->GetCommonFileName())));
28     
29
30   // TRD combined tracking efficiency
31   if(mgr->GetMCtruthEventHandler() && TESTBIT(map, AliTRDpwgppHelper::kEfficiencyMC)) {
32     mgr->AddTask(eff = new AliTRDefficiencyMC((char*)"TRDefficiencyMC"));
33     eff->SetDebugLevel(0);
34     //AliLog::SetClassDebugLevel("AliTRDefficiencyMC", 5);  
35
36     // Create containers for input/output
37     mgr->ConnectInput(eff, 0, mgr->GetCommonInputContainer());  
38     mgr->ConnectInput(eff, 1, ci[trackStatus]);
39     mgr->ConnectInput(eff, 2, evInfoContainer);
40     mgr->ConnectOutput(eff, 1, mgr->CreateContainer(eff->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Performance", mgr->GetCommonFileName())));
41   }
42
43   // TRD single track selection
44   if(!(TESTBIT(map, AliTRDpwgppHelper::kMultiplicity))) return;
45
46   mgr->AddTask(eff = new AliTRDmultiplicity((char*)"TRDmultiplicity"));
47   eff->SetDebugLevel(0);
48   //AliLog::SetClassDebugLevel("AliTRDmultiplicity", 5);  
49   mgr->ConnectInput(eff, 0, mgr->GetCommonInputContainer());  
50   mgr->ConnectInput(eff, 1, ci[0]);
51   mgr->ConnectOutput(eff, 1, mgr->CreateContainer(eff->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("%s:TRD_Calibration", mgr->GetCommonFileName())));
52 }
53