Updated macros for D0 analysis (Massimo)

[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / AliRsnEvent.cxx

diff --git a/PWGLF/RESONANCES/AliRsnEvent.cxx b/PWGLF/RESONANCES/AliRsnEvent.cxx
index 20410c629d749ee4d0497921c68e697368ba59b3..51f343b79bda090126f9d6a6e6bfe1a77c88be46 100644 (file)
--- a/PWGLF/RESONANCES/AliRsnEvent.cxx
+++ b/PWGLF/RESONANCES/AliRsnEvent.cxx
@@ -252,6 +252,12 @@ void AliRsnEvent::SetDaughterESDv0(AliRsnDaughter &out, Int_t i)
                   // otherwise label remains '-1' --> fake V0
                   if (pp->GetFirstMother() == pn->GetFirstMother() && pp->GetFirstMother() >= 0) {
                      out.SetLabel(pp->GetFirstMother());
+                     //patch for k0s/k0l
+                     TParticle *mom = mc->Stack()->Particle(pn->GetFirstMother());
+                     if(mom->GetPdgCode() == 310) {
+                        //take the mother of the k0s which is a k0 (311)
+                        out.SetLabel(mom->GetFirstMother());
+                     }
                      SetMCInfoESD(out);
                   }
                }