]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGPP/TRD/TRDPID/macros/AddTaskTRDPIDTree.C
Merge branch 'feature-movesplit'
[u/mrichter/AliRoot.git] / PWGPP / TRD / TRDPID / macros / AddTaskTRDPIDTree.C
1 AliAnalysisTask *AddTaskTRDPIDTree(Int_t trigger=0, Int_t system=0){
2   //get the current analysis manager
3   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
4   if (!mgr) {
5     Error("AddTask_pachmay_PIDTRDTree", "No analysis manager found.");
6     return 0;
7   }
8
9
10   //========= Add task to the ANALYSIS manager =====
11   AliTRDPIDTree *task = new AliTRDPIDTree("TRDPIDTree");
12   if(trigger==0) task->SelectCollisionCandidates(AliVEvent::kMB | AliVEvent::kINT7 | AliVEvent::kINT8);
13   else task->SelectCollisionCandidates(AliVEvent::kTRD);
14   if(system==1) task->SetppAnalysis();
15   if(system==2) task->SetpPbAnalysis();
16   if(system==3) task->SetPbPbAnalysis();
17 //  task->SetPbPbAnalysis();
18   mgr->AddTask(task);
19
20
21   //================================================
22   //              data containers
23   //================================================
24   //            find input container
25   //below the trunk version
26   AliAnalysisDataContainer *cinput  = mgr->GetCommonInputContainer();
27
28   //dumm output container
29   AliAnalysisDataContainer *coutput1 =
30       mgr->CreateContainer("TRDPIDTree_tree",
31                            TTree::Class(),
32                            AliAnalysisManager::kOutputContainer,
33                            "TRDPIDTree_tree.root");
34   AliAnalysisDataContainer *coutput2 =
35       mgr->CreateContainer("TRDPIDTree_list",
36                            TList::Class(),
37                            AliAnalysisManager::kOutputContainer,
38                            "TRDPIDTree_hists.root");
39
40   //connect containers
41   mgr->ConnectInput(task,  0, cinput );
42   mgr->ConnectOutput(task,  1, coutput1);
43   mgr->ConnectOutput(task,  2, coutput2);
44
45   return task;
46 }