Match clusters/tracks in a common bin by pt

cms-pepr · Sep 30, 2020 · 94322db · 94322db
1 parent de7874c
commit 94322db
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/RecoHGCal/GraphReco/plugins/HGCClusterTrackLinker.cc b/RecoHGCal/GraphReco/plugins/HGCClusterTrackLinker.cc
@@ -80,7 +80,14 @@ void HGCClusterTrackLinker::produce(edm::Event& iEvent, const edm::EventSetup& i
       if (trackIndices.size()) {
         // Find best match (just set to the first one for now)
         // Can do closest in pt or mix of closest in pt and eta/phi
-        int tidx = 0;
+        std::vector<float> ptdiffs;
+        ptdiffs.reserve(trackIndices.size());
+        for (size_t entry : trackIndices) {
+          edm::RefToBase<reco::Track> t(tracks, entry);
+          ptdiffs.push_back(std::abs(t->pt()-cand.pt()));
+        }
+
+        size_t tidx = std::distance(ptdiffs.begin(), std::min_element(ptdiffs.begin(), ptdiffs.end()));
         edm::RefToBase<reco::Track> matchingTrack(tracks, trackIndices.at(tidx));
         trackIndices.erase(trackIndices.begin()+tidx);