]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWG4/macros/AddTaskJetCorrel.C
Adding the possibility to read additional tracks or MC information to the AOD momentu...
[u/mrichter/AliRoot.git] / PWG4 / macros / AddTaskJetCorrel.C
1 AliAnalysisTaskJetCorrel *AddTaskJetCorrel(){
2   //get the current analysis manager
3   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
4   if(!mgr) {
5     ::Error("AddTaskJetCorrel", "No analysis manager found.");
6     return NULL;
7   }
8   if(!mgr->GetInputEventHandler()) {
9     ::Error("AddTaskJetCorrel", "This task requires an input event handler.");
10     return NULL;
11   }
12   TString inputDataType = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
13   if(inputDataType.CompareTo("ESD")!=0) {
14     ::Error("AddTaskJetCorrel", "This task uses ESD as input.");
15     return NULL;
16   }
17
18   gROOT->LoadMacro("$ALICE_ROOT/PWG4/macros/ConfigJetCorrel.C");
19   AliJetCorrelSelector* Selector = ConfigJetCorrel();
20   AliAnalysisTaskJetCorrel *jetcorrel = new AliAnalysisTaskJetCorrel(Selector);
21   mgr->AddTask(jetcorrel);
22   
23   gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
24   AliPhysicsSelectionTask* physSelTask = AddTaskPhysicsSelection();
25   AliPhysicsSelection* physSele = physSelTask->GetPhysicsSelection();
26   jetcorrel->SelectCollisionCandidates(); 
27
28   //create data containers
29   AliAnalysisDataContainer *cinput = (AliAnalysisDataContainer*)mgr->GetContainers()->FindObject("cAUTO_INPUT");
30   mgr->ConnectInput(jetcorrel,0,cinput);
31   AliAnalysisDataContainer *coutput_JetCorrel = mgr->CreateContainer("JetCorrelHistos", TList::Class(),
32         AliAnalysisManager::kOutputContainer, Form("%s:JetCorrel",AliAnalysisManager::GetCommonFileName()));  
33   mgr->ConnectOutput(jetcorrel,0,coutput_JetCorrel);
34   
35   return jetcorrel;
36 }