Merge pull request cms-sw#10 from gvonsem/pepr_CMSSW_11_1_0_pre7_ener…

…gyCut Energy threshold on reconstructed candidates
tklijnsma · Sep 30, 2020 · 250b6f1 · 250b6f1
2 parents cd54bad + 4b631fc
commit 250b6f1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/RecoHGCal/GraphReco/plugins/peprCandidateFromHitProducer.cc b/RecoHGCal/GraphReco/plugins/peprCandidateFromHitProducer.cc
@@ -82,6 +82,8 @@ class peprCandidateFromHitProducer: public edm::stream::EDProducer<> {
     // rechit tools
     hgcal::RecHitTools recHitTools_;
 
+    double minCandEnergy_;
+
     // windows
     std::vector<InferenceWindow> windows_;
 
@@ -101,6 +103,7 @@ peprCandidateFromHitProducer::peprCandidateFromHitProducer(const edm::ParameterS
         tritonScript_(config.getParameter<edm::FileInPath>("tritonScript").fullPath()), 
         inpipeName_(config.getParameter<std::string>("inpipeName")),
         outpipeName_(config.getParameter<std::string>("outpipeName")),
+        minCandEnergy_(config.getParameter<double>("minCandEnergy")),
         //FIXME: actually these are all not needed if windows are created in the constructor!
         minEta_(config.getParameter<double>("minEta")),
         maxEta_(config.getParameter<double>("maxEta"))
@@ -220,6 +223,10 @@ void peprCandidateFromHitProducer::produce(edm::Event& event, const edm::EventSe
 
             //inner index as filled in readOutputArrays
             E = windowoutputs[i][j][0];
+
+            //temporary lower threshold on energy of candidates
+            if(E < minCandEnergy_) continue;
+
             X = windowoutputs[i][j][1];
             Y = windowoutputs[i][j][2];
             Z = windowoutputs[i][j][3];

diff --git a/RecoHGCal/GraphReco/python/peprCandidateFromHitProducer_cfi.py b/RecoHGCal/GraphReco/python/peprCandidateFromHitProducer_cfi.py
@@ -24,6 +24,7 @@
     tritonScript=cms.FileInPath("RecoHGCal/GraphReco/test/cmssw_oc_forward_client.sh"),
     inpipeName=cms.string("arrayspipe"),
     outpipeName=cms.string("arrayspipe_pred"),
+    minCandEnergy=cms.double(1.0),
     minEta=cms.double(1.6),
     maxEta=cms.double(3.0)  
 )