Some bug fixes, removal of some duplicates and clarified the logic of some pieces...

[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnInputHandler.cxx

diff --git a/PWG2/RESONANCES/AliRsnInputHandler.cxx b/PWG2/RESONANCES/AliRsnInputHandler.cxx
index 07379c7d840b05159cd4812d226eaebf7128c1d0..071d0f28ffd93d4567b899fb993151028e0905bc 100644 (file)
--- a/PWG2/RESONANCES/AliRsnInputHandler.cxx
+++ b/PWG2/RESONANCES/AliRsnInputHandler.cxx
@@ -1,3 +1,4 @@
+#include <Riostream.h>
 #include "AliLog.h"
 
 #include "AliRsnEvent.h"
@@ -97,8 +98,9 @@ Bool_t AliRsnInputHandler::BeginEvent(Long64_t entry)
                AliMCEventHandler *mcH =  multiIH->GetFirstMCEventHandler();
                if (mcH) fRsnEvent->SetRefMC(mcH->MCEvent());
             } else if (fRsnEvent->GetRefAOD()) {
-               // TODO AOD MC
-//                fRsnEvent->SetRefMC(mcH->MCEvent());
+               AliAODEvent *aod = fRsnEvent->GetRefAOD();
+               TClonesArray *listAOD = (TClonesArray*)(aod->GetList()->FindObject(AliAODMCParticle::StdBranchName()));
+               if (listAOD) fRsnEvent->SetRefMC(fRsnEvent->GetRefAOD());
             }
             if (fParentHandler->ParentHandler()) tmp = "MIX";
             // applying pid cuts