Patch for jet analysis
[u/mrichter/AliRoot.git] / PWG / EMCAL / macros / AddTaskEmcalEsdTpcTrack.C
index e022a08..2c927d4 100644 (file)
@@ -1,7 +1,6 @@
 // $Id$
 
-AliEmcalEsdTpcTrackTask* AddTaskEmcalEsdTpcTrack(
-                                                const char *name       = "TpcSpdVertexConstrainedTracks",
+AliEmcalEsdTpcTrackTask* AddTaskEmcalEsdTpcTrack(const char *name       = "TpcSpdVertexConstrainedTracks",
                                                 const char *trackCuts  = "Hybrid_LHC11h"
                                                 )
 { 
@@ -48,20 +47,24 @@ AliEmcalEsdTpcTrackTask* AddTaskEmcalEsdTpcTrack(
   // Get the pointer to the existing analysis manager via the static access method.
   //==============================================================================
   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
-  if (!mgr)
-  {
+  if (!mgr) {
     ::Error("AddTaskEmcalEsdTpcTrack", "No analysis manager to connect to.");
     return NULL;
   }  
   
   // Check the analysis type using the event handlers connected to the analysis manager.
   //==============================================================================
-  if (!mgr->GetInputEventHandler())
-  {
+  AliVEventHandler *evhand = mgr->GetInputEventHandler();
+  if (!evhand) {
     ::Error("AddTaskEmcalEsdTpcTrack", "This task requires an input event handler");
     return NULL;
   }
   
+  if (!evhand->InheritsFrom("AliESDInputHandler")) {
+    ::Info("AddTaskEmcalEsdTpcTrack", "This task is only needed for ESD analysis. No task added.");
+    return NULL;
+  }
+  
   //-------------------------------------------------------
   // Init the task and do settings
   //-------------------------------------------------------