Merge pull request #29596 from dntaylor/deltaT-IP-1110pre6

Add time support to IPCutPFCandidateSelector
cms-sw · May 5, 2020 · aebe4fb · aebe4fb
2 parents cc0b33b + deef0a6
commit aebe4fb
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/CommonTools/ParticleFlow/interface/IPCutPFCandidateSelectorDefinition.h b/CommonTools/ParticleFlow/interface/IPCutPFCandidateSelectorDefinition.h
@@ -26,8 +26,10 @@ namespace pf2pat {
         : verticesToken_(iC.consumes<reco::VertexCollection>(cfg.getParameter<edm::InputTag>("vertices"))),
           d0Cut_(cfg.getParameter<double>("d0Cut")),
           dzCut_(cfg.getParameter<double>("dzCut")),
+          dtCut_(cfg.getParameter<double>("dtCut")),
           d0SigCut_(cfg.getParameter<double>("d0SigCut")),
-          dzSigCut_(cfg.getParameter<double>("dzSigCut")) {}
+          dzSigCut_(cfg.getParameter<double>("dzSigCut")),
+          dtSigCut_(cfg.getParameter<double>("dtSigCut")) {}
 
     void select(const HandleToCollection &hc, const edm::Event &e, const edm::EventSetup &s) {
       selected_.clear();
@@ -37,6 +39,8 @@ namespace pf2pat {
       if (vertices->empty())
         return;
       const reco::Vertex &vtx = (*vertices)[0];
+      double vt = vtx.t();
+      double vte = vtx.tError();
 
       unsigned key = 0;
       for (collection::const_iterator pfc = hc->begin(); pfc != hc->end(); ++pfc, ++key) {
@@ -61,6 +65,14 @@ namespace pf2pat {
           if (dzSigCut_ > 0 && dze > 0 && dz / dze > dzSigCut_)
             passing = false;
         }
+        double pfct = pfc->time();
+        double pfcte = pfc->timeError();
+        double dt = fabs(pfct - vt);
+        double dte = std::sqrt(pfcte * pfcte + vte * vte);
+        if (dtCut_ > 0 && pfcte > 0 && vte > 0 && dt > dtCut_)
+          passing = false;
+        if (dtSigCut_ > 0 && pfcte > 0 && vte > 0 && dt / dte > dtSigCut_)
+          passing = false;
 
         if (passing) {
           selected_.push_back(reco::PFCandidate(*pfc));
@@ -79,8 +91,10 @@ namespace pf2pat {
     edm::EDGetTokenT<reco::VertexCollection> verticesToken_;
     double d0Cut_;
     double dzCut_;
+    double dtCut_;
     double d0SigCut_;
     double dzSigCut_;
+    double dtSigCut_;
   };
 }  // namespace pf2pat
 

diff --git a/CommonTools/ParticleFlow/python/ParticleSelectors/pfElectronsFromVertex_cfi.py b/CommonTools/ParticleFlow/python/ParticleSelectors/pfElectronsFromVertex_cfi.py
@@ -6,6 +6,8 @@
     vertices = cms.InputTag("offlinePrimaryVertices"),  # vertices source
     d0Cut = cms.double(0.2),  # transverse IP
     dzCut = cms.double(0.5),  # longitudinal IP
+    dtCut = cms.double(-1.0), # time
     d0SigCut = cms.double(99.),  # transverse IP significance
     dzSigCut = cms.double(99.),  # longitudinal IP significance
+    dtSigCut = cms.double(-1.0), # time significance
     )
diff --git a/CommonTools/ParticleFlow/python/ParticleSelectors/pfMuonsFromVertex_cfi.py b/CommonTools/ParticleFlow/python/ParticleSelectors/pfMuonsFromVertex_cfi.py
@@ -6,6 +6,8 @@
     vertices = cms.InputTag("offlinePrimaryVertices"),  # vertices source
     d0Cut = cms.double(0.2),  # transverse IP
     dzCut = cms.double(0.5),  # longitudinal IP
+    dtCut = cms.double(-1.0), # time
     d0SigCut = cms.double(99.),  # transverse IP significance
     dzSigCut = cms.double(99.),  # longitudinal IP significance
+    dtSigCut = cms.double(-1.0), # time significance
     )