]> git.uio.no Git - u/mrichter/AliRoot.git/blob - PWGJE/EMCALJetTasks/macros/AddTaskEmcalTriggerTreeWriter.C
Add task writing EMCAL trigger information into a tree
[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / macros / AddTaskEmcalTriggerTreeWriter.C
1 #if !defined (__CINT__) || defined (__MAKECINT__)
2 #include "AliAnalysisManager.h"
3 #include "AliAnalysisTaskPtEMCalTrigger.h"
4 #include "AliESDtrackCuts.h"
5 #include <TList.h>
6 #include <TString.h>
7 #endif
8
9 AliAnalysisTask* AddTaskEmcalTriggerTreeWriter(){
10         AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
11          
12         if (!mgr) {
13              ::Error("AddTaskPtEMCalTrigger", "No analysis manager to connect to.");
14              return NULL;
15         }
16          
17         if (!mgr->GetInputEventHandler()) {
18              ::Error("AddTaskPtEMCalTrigger", "This task requires an input event handler");
19              return NULL;
20         }
21         
22         AliAnalysisTaskEmcalTriggerTreeWriter *treewriter = new AliAnalysisTaskEmcalTriggerTreeWriter("TriggerTreewriterTask");
23         //pttriggertask->SelectCollisionCandidates(AliVEvent::kINT7 | AliVEvent::kEMC7);                          // Select both INT7 or EMC7 triggered events
24         treewriter->SelectCollisionCandidates(AliVEvent::kAny);
25         mgr->AddTask(treewriter);
26
27         AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
28         AliAnalysisDataContainer *coutputTree = mgr->CreateContainer("EMCalTriggerTree", TTree::Class(),    AliAnalysisManager::kOutputContainer, "EMCalTriggerTree.root");
29    
30         //Connect input/output
31         mgr->ConnectInput(treewriter, 0, cinput);
32         mgr->ConnectOutput(treewriter, 1, coutputTree);
33            
34         return treewriter;
35 }