Added user-callable check if we have an MC input handler

diff --git a/PWGLF/FORWARD/trains/TrainSetup.C b/PWGLF/FORWARD/trains/TrainSetup.C
index cec7c270f22a10b847b9d538c6d1a1724a208a2f..cdbf3e7e75cc615fa98641f91f3c2074fd680007 100644 (file)
--- a/PWGLF/FORWARD/trains/TrainSetup.C
+++ b/PWGLF/FORWARD/trains/TrainSetup.C
@@ -636,7 +636,18 @@ protected:
     TString args;
     return AddSETask(macro, args);
   }
-
+  /** 
+   * Check if we have an MC handler attached 
+   * 
+   * @return True if MC handler is found in a valid manager.  False if
+   * manager is not defined, or has no MC handler.
+   */
+  virtual Bool_t HasMCHandler() const 
+  {
+    AliAnalysisManager* mgr = AliAnalysisManager::GetAnalysisManager();
+    if (!mgr) return false;
+    return mgr->GetMCtruthEventHandler() != 0;
+  }
   /** 
    * Set the name of the train - should be name of the class.  Must be
    * overloaded.