refactored version of Dipohoton MVA filter by M. Musich

Author: smorovic

HLTrigger/Egamma/plugins/HLTEgammaDoubleXGBoostCombFilter.cc

@@ -73,48 +73,47 @@ void HLTEgammaDoubleXGBoostCombFilter::fillDescriptions(edm::ConfigurationDescri
 bool HLTEgammaDoubleXGBoostCombFilter::hltFilter(edm::Event& event,
                                                  const edm::EventSetup& setup,
                                                  trigger::TriggerFilterObjectWithRefs& filterproduct) const {
-  //producer collection (hltEgammaCandidates(Unseeded))
   const auto& recCollection = event.getHandle(candToken_);
-
-  //get hold of photon MVA association map
   const auto& mvaMap = event.getHandle(mvaToken_);
 
-  std::vector<math::XYZTLorentzVector> p4s(recCollection->size());
-  std::vector<bool> isTight(recCollection->size());
+  // Lambda to evaluate pair cuts
+  auto passesHighMassCuts = [&](float leadScore, float subScore, int leadEta, int subEta) {
+    return (leadScore > leadCutHighMass1_[leadEta] && subScore > subCutHighMass1_[subEta]) ||
+           (leadScore > leadCutHighMass2_[leadEta] && subScore > subCutHighMass2_[subEta]) ||
+           (leadScore > leadCutHighMass3_[leadEta] && subScore > subCutHighMass3_[subEta]);
+  };
+
+  // Lambda to evaluate a candidate pair
+  auto evaluatePair = [&](const edm::Ref<reco::RecoEcalCandidateCollection>& refi,
+                          const edm::Ref<reco::RecoEcalCandidateCollection>& refj) {
+    float mvaScorei = (*mvaMap).find(refi)->val;
+    float mvaScorej = (*mvaMap).find(refj)->val;
+
+    int etai = (std::abs(refi->eta()) < 1.5) ? 0 : 1;
+    int etaj = (std::abs(refj->eta()) < 1.5) ? 0 : 1;
 
-  bool accept = false;
+    double mass = (refi->p4() + refj->p4()).M();
+    if (mass < highMassCut_) return false;
 
-  for (size_t i = 0; i < recCollection->size(); i++) {
+    if (mvaScorei >= mvaScorej) {
+      return passesHighMassCuts(mvaScorei, mvaScorej, etai, etaj);
+    } else {
+      return passesHighMassCuts(mvaScorej, mvaScorei, etaj, etai);
+    }
+  };
+
+  // Loop through candidates
+  for (size_t i = 0; i < recCollection->size(); ++i) {
     edm::Ref<reco::RecoEcalCandidateCollection> refi(recCollection, i);
-    float EtaSCi = refi->eta();
-    int etai = (std::abs(EtaSCi) < 1.5) ? 0 : 1;
-    float mvaScorei = (*mvaMap).find(refi)->val;
-    math::XYZTLorentzVector p4i = refi->p4();
-    for (size_t j = i + 1; j < recCollection->size(); j++) {
+    for (size_t j = i + 1; j < recCollection->size(); ++j) {
       edm::Ref<reco::RecoEcalCandidateCollection> refj(recCollection, j);
-      float EtaSCj = refj->eta();
-      int etaj = (std::abs(EtaSCj) < 1.5) ? 0 : 1;
-      float mvaScorej = (*mvaMap).find(refj)->val;
-      math::XYZTLorentzVector p4j = refj->p4();
-      math::XYZTLorentzVector pairP4 = p4i + p4j;
-      double mass = pairP4.M();
-      if (mass >= highMassCut_) {
-        if (mvaScorei >= mvaScorej && ((mvaScorei > leadCutHighMass1_[etai] && mvaScorej > subCutHighMass1_[etaj]) ||
-                                       (mvaScorei > leadCutHighMass2_[etai] && mvaScorej > subCutHighMass2_[etaj]) ||
-                                       (mvaScorei > leadCutHighMass3_[etai] && mvaScorej > subCutHighMass3_[etaj]))) {
-          accept = true;
-        }  //if scoreI > scoreJ
-        else if (mvaScorej > mvaScorei &&
-                 ((mvaScorej > leadCutHighMass1_[etaj] && mvaScorei > subCutHighMass1_[etai]) ||
-                  (mvaScorej > leadCutHighMass2_[etaj] && mvaScorei > subCutHighMass2_[etai]) ||
-                  (mvaScorej > leadCutHighMass3_[etaj] && mvaScorei > subCutHighMass3_[etai]))) {
-          accept = true;
-        }  // if scoreJ > scoreI
-      }  //If high mass
-    }  //j loop
-  }  //i loop
-  return accept;
-}  //Definition
+      if (evaluatePair(refi, refj)) {
+        return true;
+      }
+    }
+  }
+  return false;
+}
 
 #include "FWCore/Framework/interface/MakerMacros.h"
 DEFINE_FWK_MODULE(HLTEgammaDoubleXGBoostCombFilter);