added also aod event selection

diff --git a/PWGGA/EMCALTasks/AliAnalysisTaskEmcal.cxx b/PWGGA/EMCALTasks/AliAnalysisTaskEmcal.cxx
index dc69d50a0f3f4cc582ef03290441fc4145562988..9c3002f9c7d25ba8495d60ced95d3abdd31bb138 100644 (file)
--- a/PWGGA/EMCALTasks/AliAnalysisTaskEmcal.cxx
+++ b/PWGGA/EMCALTasks/AliAnalysisTaskEmcal.cxx
@@ -11,13 +11,15 @@
 #include <TList.h>
 #include <TObject.h>
 
+#include "AliAODEvent.h"
 #include "AliAnalysisManager.h"
 #include "AliCentrality.h"
-#include "AliEventplane.h"
 #include "AliEMCALGeometry.h"
 #include "AliESDEvent.h"
 #include "AliEmcalJet.h"
 #include "AliEmcalParticle.h"
+#include "AliEventplane.h"
+#include "AliInputEventHandler.h"
 #include "AliLog.h"
 #include "AliMCParticle.h"
 #include "AliVCluster.h"
@@ -275,13 +277,22 @@ AliAnalysisTaskEmcal::BeamType AliAnalysisTaskEmcal::GetBeamType()
 }
 
 //________________________________________________________________________
-Bool_t AliAnalysisTaskEmcal::IsEventSelected() const
+Bool_t AliAnalysisTaskEmcal::IsEventSelected()
 {
   // Check if event is selected
 
   if (fOffTrigger != AliVEvent::kAny) {
-    UInt_t res = ((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected();
-    if (res & fOfftrigger == 0)
+    UInt_t res = 0;
+    const AliESDEvent *eev = dynamic_cast<const AliESDEvent*>(InputEvent());
+    if (eev) {
+      res = ((AliInputEventHandler*)(AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler()))->IsEventSelected();
+    } else {
+      const AliAODEvent *aev = dynamic_cast<const AliAODEvent*>(InputEvent());
+      if (aev) {
+        res = aev->GetHeader()->GetOfflineTrigger();
+      }
+    }
+    if (res & fOffTrigger == 0)
       return kFALSE;
   }
 

diff --git a/PWGGA/EMCALTasks/AliAnalysisTaskEmcal.h b/PWGGA/EMCALTasks/AliAnalysisTaskEmcal.h
index fcfe5086cd5d8dc040993836e23aac15b100910e..37ecec890473735a1c79ab7a9b8f90d6a94cd23a 100644 (file)
--- a/PWGGA/EMCALTasks/AliAnalysisTaskEmcal.h
+++ b/PWGGA/EMCALTasks/AliAnalysisTaskEmcal.h
@@ -56,7 +56,7 @@ class AliAnalysisTaskEmcal : public AliAnalysisTaskSE {
   virtual Bool_t              FillHistograms()                                     { return fCreateHisto; }
   BeamType                    GetBeamType();
   TClonesArray               *GetArrayFromEvent(const char *name, const char *clname=0);
-  virtual Bool_t              IsEventSelected() const;
+  virtual Bool_t              IsEventSelected();
   virtual Bool_t              RetrieveEventObjects();
   virtual Bool_t              Run()                                                { return kTRUE                 ; }
   void                        SetInitialized(Bool_t ini = kTRUE)                   { fInitialized    = ini        ; }