sort pixel tracks from SoA by pt

cms-sw · May 30, 2022 · 7efac99 · 7efac99
1 parent 3d2a0f9
commit 7efac99
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/RecoPixelVertexing/PixelTrackFitting/plugins/PixelTrackProducerFromSoA.cc b/RecoPixelVertexing/PixelTrackFitting/plugins/PixelTrackProducerFromSoA.cc
@@ -160,16 +160,23 @@ void PixelTrackProducerFromSoA::produce(edm::StreamID streamID,
 
   int32_t nt = 0;
 
-  for (int32_t it = 0; it < nTracks; ++it) {
+  //sort index by pt
+  std::vector<int32_t> sortIdxs(nTracks);
+  std::iota(sortIdxs.begin(), sortIdxs.end(), 0);
+  std::sort(
+      sortIdxs.begin(), sortIdxs.end(), [&](int32_t const i1, int32_t const i2) { return tsoa.pt(i1) > tsoa.pt(i2); });
+
+  //store the index of the SoA: indToEdm[index_SoAtrack] -> index_edmTrack (if it exists)
+  indToEdm.resize(sortIdxs.size(), -1);
+  for (const auto &it : sortIdxs) {
     auto nHits = tsoa.nHits(it);
     assert(nHits >= 3);
-    indToEdm.push_back(-1);
     auto q = quality[it];
     if (q < minQuality_)
       continue;
     if (nHits < minNumberOfHits_)
       continue;
-    indToEdm.back() = nt;
+    indToEdm[it] = nt;
     ++nt;
 
     hits.resize(nHits);