better ESD performance plot
[u/mrichter/AliRoot.git] / TRD / qaRec / macros / AddTRDcheckPID.C
1 #if ! defined (__CINT__) || defined (__MAKECINT__)
2 #include "TTree.h"
3 #include "AliLog.h"
4 #include "AliAnalysisManager.h"
5 #include "AliAnalysisDataContainer.h"
6 #include "TRD/qaRec/macros/AliTRDperformanceTrain.h"
7 #include "TRD/qaRec/AliTRDcheckPID.h"
8 #include "TRD/qaRec/AliTRDpidRefMaker.h"
9 #endif
10
11 #include "TRD/qaRec/macros/helper.C"
12 void AddTRDcheckPID(AliAnalysisManager *mgr, Char_t *trd, AliAnalysisDataContainer **ci/*, AliAnalysisDataContainer **co*/)
13 {
14   Int_t map = ParseOptions(trd);
15   if(TSTBIT(map, kCheckPID)){
16     AliTRDcheckPID *pid = 0x0;
17     mgr->AddTask(pid = new AliTRDcheckPID());
18     pid->SetDebugLevel(0);
19     pid->SetMCdata(mgr->GetMCtruthEventHandler());
20     mgr->ConnectInput(pid, 0, ci[0]);
21     mgr->ConnectOutput(pid, 0, mgr->CreateContainer(pid->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, "TRD.Performance.root"));
22   }
23
24   if(TSTBIT(map, kPIDRefMaker)){
25     // TRD pid reference maker 
26     AliTRDpidRefMaker *ref = new AliTRDpidRefMaker(); 
27     mgr->AddTask(ref);
28     ref->SetDebugLevel(3);
29     AliLog::SetClassDebugLevel("AliTRDpidRefMaker", 3);
30     ref->SetMCdata(mgr->GetMCtruthEventHandler());
31     ref->SetFriends(kTRUE);
32     mgr->ConnectInput( ref, 0, ci[0]);
33     mgr->ConnectInput( ref, 1, ci[2]);
34     mgr->ConnectOutput(ref, 0, mgr->CreateContainer(Form("Moni%s", ref->GetName()), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Calib%s.root", ref->GetName())));
35     mgr->ConnectOutput(ref, 1, mgr->CreateContainer(ref->GetName(), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Calib%s.root", ref->GetName())));
36   }
37 }
38