Make sure to also TrackerHits correctly from EDM4hep to LCIO

key4hep · Jan 19, 2024 · 073cbfa · 073cbfa
1 parent 75255da
commit 073cbfa
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/k4EDM4hep2LcioConv/include/k4EDM4hep2LcioConv/k4EDM4hep2LcioConv.ipp b/k4EDM4hep2LcioConv/include/k4EDM4hep2LcioConv/k4EDM4hep2LcioConv.ipp
@@ -735,6 +735,11 @@ namespace EDM4hep2LCIOConv {
           if (const auto lcio_trh = k4EDM4hep2LcioConv::detail::mapLookupFrom(edm_tr_trh, lookup_pairs.trackerHits)) {
             lcio_tr->addHit(lcio_trh.value());
           }
+          else if (
+            const auto lcio_trh =
+              k4EDM4hep2LcioConv::detail::mapLookupFrom(edm_tr_trh, lookup_pairs.trackerHitPlanes)) {
+            lcio_tr->addHit(lcio_trh.value());
+          }
         }
       }
     }

diff --git a/tests/src/EDM4hep2LCIOUtilities.cc b/tests/src/EDM4hep2LCIOUtilities.cc
@@ -188,6 +188,7 @@ edm4hep::TrackCollection createTracks(
   const int subdetectorhitnumbers,
   const int num_track_states,
   const edm4hep::TrackerHit3DCollection& trackerHits,
+  const edm4hep::TrackerHitPlaneCollection& trackerHitPlanes,
   const std::vector<std::size_t>& link_trackerhit_idcs,
   const std::vector<test_config::IdxPair>& track_link_tracks_idcs)
 {
@@ -213,6 +214,7 @@ edm4hep::TrackCollection createTracks(
 
     for (auto& idx : link_trackerhit_idcs) {
       elem.addToTrackerHits(trackerHits[idx]);
+      elem.addToTrackerHits(trackerHitPlanes[idx]);
     }
 
     for (int j = 0; j < num_track_states; ++j) {
@@ -335,6 +337,7 @@ podio::Frame createExampleEvent()
       test_config::nSubdetectorHitNumbers,
       test_config::nTrackStates,
       trackerHits,
+      trackerHitPlanes,
       test_config::trackTrackerHitIdcs,
       test_config::trackTrackIdcs),
     "tracks");