check both algo and originalAlgo in electron track isolation calculators

cms-sw · Dec 4, 2021 · c0aec63 · c0aec63
1 parent f496aeb
commit c0aec63
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/RecoEgamma/EgammaIsolationAlgos/src/EleTkIsolFromCands.cc b/RecoEgamma/EgammaIsolationAlgos/src/EleTkIsolFromCands.cc
@@ -180,7 +180,11 @@ bool EleTkIsolFromCands::passQual(const reco::TrackBase& trk, const std::vector<
 
 bool EleTkIsolFromCands::passAlgo(const reco::TrackBase& trk,
                                   const std::vector<reco::TrackBase::TrackAlgorithm>& algosToRej) {
-  return algosToRej.empty() || !std::binary_search(algosToRej.begin(), algosToRej.end(), trk.algo());
+  return algosToRej.empty() ||
+         //check also the originalAlgo in case the track is reconstructed by more than one
+         //reject only if both algo and originalAlgo are not acceptable
+         !(std::binary_search(algosToRej.begin(), algosToRej.end(), trk.algo()) &&
+           std::binary_search(algosToRej.begin(), algosToRej.end(), trk.originalAlgo()));
 }
 
 EleTkIsolFromCands::TrackTable const& EleTkIsolFromCands::getPreselectedTracks(bool isBarrel) {

diff --git a/RecoEgamma/EgammaIsolationAlgos/src/ElectronTkIsolation.cc b/RecoEgamma/EgammaIsolationAlgos/src/ElectronTkIsolation.cc
@@ -107,7 +107,11 @@ double ElectronTkIsolation::getPtTracks(const reco::GsfElectron* electron) const
 bool ElectronTkIsolation::passAlgo(const reco::TrackBase& trk) const {
   int algo = trk.algo();
   bool rejAlgo = std::binary_search(algosToReject_.begin(), algosToReject_.end(), algo);
-  return rejAlgo == false;
+  //check also the originalAlgo in case the track is reconstructed by more than one
+  algo = trk.originalAlgo();
+  //reject only if both algo and originalAlgo are not acceptable
+  rejAlgo &= std::binary_search(algosToReject_.begin(), algosToReject_.end(), algo);
+  return !rejAlgo;
 }
 
 void ElectronTkIsolation::setAlgosToReject() {