returning the handler in its own type and not its baseclass

[u/mrichter/AliRoot.git] / ANALYSIS / macros / train / AddMCHandler.C

AliMCEventHandler* AddMCHandler(Bool_t readTrackRef = kFALSE)
{
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) 
  {
    ::Error("AddMCHandler", "No analysis manager to connect to.");
    return NULL;
  }

  AliMCEventHandler* handler = new AliMCEventHandler();
  mgr->SetMCtruthEventHandler(handler);
  handler->SetReadTR(readTrackRef);

  return handler;
}