Use aod track selection wrapper inside the task, when running on aods

[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / UserTasks / AliAnalysisTaskPtEMCalTrigger.cxx

diff --git a/PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskPtEMCalTrigger.cxx b/PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskPtEMCalTrigger.cxx
index a26ca34375ca2531ad5be283ab03d90052d11a2e..8cbeaa730a7c6551ad0fa0a2b89d20cb77e0aecf 100644 (file)
--- a/PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskPtEMCalTrigger.cxx
+++ b/PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskPtEMCalTrigger.cxx
@@ -49,6 +49,7 @@
 
 #include "AliEMCalHistoContainer.h"
 #include "AliEMCalPtTaskVTrackSelection.h"
+#include "AliEMCalPtTaskTrackSelectionAOD.h"
 #include "AliEMCalPtTaskTrackSelectionESD.h"
 #include "AliAnalysisTaskPtEMCalTrigger.h"
 
@@ -694,6 +695,17 @@ namespace EMCalTriggerPtAnalysis {
                fListTrackCuts->AddLast(new AliEMCalPtTaskTrackSelectionESD(trackCuts));
        }
 
+       //______________________________________________________________________________
+       void AliAnalysisTaskPtEMCalTrigger::AddCutsForAOD(AliESDtrackCuts* trackCuts, UInt_t filterbits) {
+               /*
+                * Add new track cuts to the task
+                *
+                * @param trackCuts: Object of type AliESDtrackCuts
+                */
+               fListTrackCuts->AddLast(new AliEMCalPtTaskTrackSelectionAOD(trackCuts, filterbits));
+       }
+
+
        //______________________________________________________________________________
        TString AliAnalysisTaskPtEMCalTrigger::BuildTriggerString() {
                /*