MCEvent() is available when running with the AODInputHandler

[u/mrichter/AliRoot.git] / STEER / AliAODInputHandler.cxx

diff --git a/STEER/AliAODInputHandler.cxx b/STEER/AliAODInputHandler.cxx
index 1c0cda99cfd2bf1e514002f165a89940a0d6711c..6729a318a20c5c0b98dc1a59723e86e2bb8c5611 100644 (file)
--- a/STEER/AliAODInputHandler.cxx
+++ b/STEER/AliAODInputHandler.cxx
@@ -36,6 +36,7 @@ static Option_t *gAODDataType = "AOD";
 AliAODInputHandler::AliAODInputHandler() :
     AliInputEventHandler(),
     fEvent(0),
+    fMCEvent(new AliMCEvent()),
     fFriends(new TList())
 {
   // Default constructor
@@ -45,6 +46,7 @@ AliAODInputHandler::AliAODInputHandler() :
 AliAODInputHandler::AliAODInputHandler(const char* name, const char* title):
   AliInputEventHandler(name, title),
   fEvent(0),
+  fMCEvent(new AliMCEvent()),
   fFriends(new TList())
 {
     // Constructor
@@ -93,13 +95,15 @@ Bool_t AliAODInputHandler::Init(TTree* tree, Option_t* /*opt*/)
     if (!fEvent) fEvent = new AliAODEvent();
 
     fEvent->ReadFromTree(fTree);
+
     return kTRUE;
 }
 
 Bool_t AliAODInputHandler::BeginEvent(Long64_t /*entry*/)
 {
     //
-    //if (fTree) fTree->BranchRef();
+    TClonesArray* mcParticles = (TClonesArray*) (fEvent->FindListObject("mcparticles"));
+    if (mcParticles) fMCEvent->SetParticleArray(mcParticles);
     return kTRUE;
 }