MCTrackNavigator: Access Daugther0 and Daughter1

sawenzel · sawenzel · commit ff68fdcc02b2 · 2021-12-02T21:47:36.000+01:00
diff --git a/DataFormats/simulation/include/SimulationDataFormat/MCUtils.h b/DataFormats/simulation/include/SimulationDataFormat/MCUtils.h
@@ -46,6 +46,11 @@ class MCTrackNavigator
   /// Given an MCTrack p; Return it's direct daughter or nullptr. (we follow only first daughter)
   static o2::MCTrack const* getDaughter(o2::MCTrack const& p, std::vector<o2::MCTrack> const& pcontainer);
 
+  /// Given an MCTrack p; Return it's first direct daughter or nullptr.
+  static o2::MCTrack const* getDaughter0(o2::MCTrack const& p, std::vector<o2::MCTrack> const& pcontainer);
+  /// Given an MCTrack p; Return it's second direct daughter or nullptr.
+  static o2::MCTrack const* getDaughter1(o2::MCTrack const& p, std::vector<o2::MCTrack> const& pcontainer);
+
   /// Given an MCTrack p; Fill the complete parent chain (ancestorchain) up to the most fundamental particle (follow only
   /// first mothers).
   // static void getParentChain(o2::MCTrack const& p, std::vector<o2::MCTrack> const& pcontainer, std::vector<o2::MCTrack> &ancestorchain);
diff --git a/DataFormats/simulation/src/MCUtils.cxx b/DataFormats/simulation/src/MCUtils.cxx
@@ -36,6 +36,24 @@ o2::MCTrack const* MCTrackNavigator::getDaughter(const o2::MCTrack& p, const std
   return &(pcontainer[did]);
 }
 
+o2::MCTrack const* MCTrackNavigator::getDaughter0(const o2::MCTrack& p, const std::vector<o2::MCTrack>& pcontainer)
+{
+  const auto did = p.getFirstDaughterTrackId();
+  if (did < 0 or did > pcontainer.size()) {
+    return nullptr;
+  }
+  return &(pcontainer[did]);
+}
+
+o2::MCTrack const* MCTrackNavigator::getDaughter1(const o2::MCTrack& p, const std::vector<o2::MCTrack>& pcontainer)
+{
+  const auto did = p.getLastDaughterTrackId();
+  if (did < 0 or did > pcontainer.size()) {
+    return nullptr;
+  }
+  return &(pcontainer[did]);
+}
+
 o2::MCTrack const& MCTrackNavigator::getFirstPrimary(const o2::MCTrack& p, const std::vector<o2::MCTrack>& pcontainer)
 {
   if (p.isPrimary()) {
