Extension for AOD

[u/mrichter/AliRoot.git] / ANALYSIS / macros / AddTaskEventplane.C

diff --git a/ANALYSIS/macros/AddTaskEventplane.C b/ANALYSIS/macros/AddTaskEventplane.C
index fe35e6ab592482933eabc788aa9acd6f838da212..3d33454a527f4cce159d8a1a4b26e60206b6b356 100644 (file)
--- a/ANALYSIS/macros/AddTaskEventplane.C
+++ b/ANALYSIS/macros/AddTaskEventplane.C
@@ -13,13 +13,13 @@ AliEPSelectionTask *AddTaskEventplane()
     return NULL;
   }
   TString inputDataType = mgr->GetInputEventHandler()->GetDataType(); // can be "ESD" or "AOD"
-  if (inputDataType != "ESD") {
-    ::Error("AddTaskEventplane", "This task works only on ESD analysis");
-    return NULL;
-  }
+ 
   
   AliEPSelectionTask *eventplaneTask = new AliEPSelectionTask("EventplaneSelection");
   eventplaneTask->SelectCollisionCandidates(AliVEvent::kMB);
+  if (inputDataType == "AOD"){
+    eventplaneTask->SetInput("AOD");
+  }
   eventplaneTask->SetTrackType("TPC");
   eventplaneTask->SetUsePtWeight();
   eventplaneTask->SetUsePhiWeight();