Add PV sumPT2 of pf charged particles associated to PV in NanoAOD

PhysicsTools/NanoAOD/plugins/VertexTableProducer.cc

@@ -41,6 +41,9 @@
 #include "RecoVertex/VertexPrimitives/interface/VertexState.h"
 #include "DataFormats/Common/interface/ValueMap.h"
 
+#include "DataFormats/PatCandidates/interface/PackedCandidate.h"
+#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
+
 //
 // class declaration
 //
@@ -65,6 +68,7 @@ class VertexTableProducer : public edm::stream::EDProducer<> {
   // ----------member data ---------------------------
 
   const edm::EDGetTokenT<std::vector<reco::Vertex>> pvs_;
+  const edm::EDGetTokenT<pat::PackedCandidateCollection> pfc_;
   const edm::EDGetTokenT<edm::ValueMap<float>> pvsScore_;
   const edm::EDGetTokenT<edm::View<reco::VertexCompositePtrCandidate>> svs_;
   const StringCutObjectSelector<reco::Candidate> svCut_;
@@ -81,6 +85,7 @@ class VertexTableProducer : public edm::stream::EDProducer<> {
 //
 VertexTableProducer::VertexTableProducer(const edm::ParameterSet& params)
     : pvs_(consumes<std::vector<reco::Vertex>>(params.getParameter<edm::InputTag>("pvSrc"))),
+      pfc_(consumes<pat::PackedCandidateCollection>(params.getParameter<edm::InputTag>("pfcSrc"))),
       pvsScore_(consumes<edm::ValueMap<float>>(params.getParameter<edm::InputTag>("pvSrc"))),
       svs_(consumes<edm::View<reco::VertexCompositePtrCandidate>>(params.getParameter<edm::InputTag>("svSrc"))),
       svCut_(params.getParameter<std::string>("svCut"), true),
@@ -115,6 +120,9 @@ void VertexTableProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSe
   const auto& pvsScoreProd = iEvent.get(pvsScore_);
   auto pvsIn = iEvent.getHandle(pvs_);
 
+  //pf candidates collection
+  auto pfcIn = iEvent.getHandle(pfc_);
+
   auto pvTable = std::make_unique<nanoaod::FlatTable>(1, pvName_, true);
   pvTable->addColumnValue<float>("ndof", (*pvsIn)[0].ndof(), "main primary vertex number of degree of freedom", 8);
   pvTable->addColumnValue<float>("x", (*pvsIn)[0].position().x(), "main primary vertex position x coordinate", 10);
@@ -131,6 +139,31 @@ void VertexTableProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSe
   pvTable->addColumnValue<float>(
       "score", pvsScoreProd.get(pvsIn.id(), 0), "main primary vertex score, i.e. sum pt2 of clustered objects", 8);
 
+  float pv_sumpt2 = 0.0;
+  for (const auto& obj : *pfcIn) {
+    if (obj.charge() == 0) {
+      continue;
+    }  // skip neutrals
+    double dz = fabs(obj.dz((*pvsIn)[0].position()));
+    bool include_pfc = false;
+    if (dz < 0.2) {
+      include_pfc = true;
+      for (size_t j = 1; j < (*pvsIn).size(); j++) {
+        double newdz = fabs(obj.dz((*pvsIn)[j].position()));
+        if (newdz < dz) {
+          include_pfc = false;
+          break;
+        }
+      }  // this pf candidate belongs to other PV
+    }
+    if (include_pfc) {
+      float pfc_pt = obj.pt();
+      pv_sumpt2 += pfc_pt * pfc_pt;
+    }
+  }
+  pvTable->addColumnValue<float>(
+      "sumpt2", pv_sumpt2, "sum pt2 of pf charged candidates for the main primary vertex", 10);
+
   auto otherPVsTable =
       std::make_unique<nanoaod::FlatTable>((*pvsIn).size() > 4 ? 3 : (*pvsIn).size() - 1, "Other" + pvName_, false);
   std::vector<float> pvsz;
@@ -207,6 +240,7 @@ void VertexTableProducer::fillDescriptions(edm::ConfigurationDescriptions& descr
 
   desc.add<edm::InputTag>("pvSrc")->setComment(
       "std::vector<reco::Vertex> and ValueMap<float> primary vertex input collections");
+  desc.add<edm::InputTag>("pfcSrc")->setComment("packedPFCandidates input collections");
   desc.add<std::string>("goodPvCut")->setComment("selection on the primary vertex");
   desc.add<edm::InputTag>("svSrc")->setComment(
       "reco::VertexCompositePtrCandidate compatible secondary vertex input collection");

PhysicsTools/NanoAOD/python/nanoDQM_cfi.py

@@ -622,6 +622,7 @@
                 Plot1D('npvs', 'npvs', 20, 0, 60, 'total number of reconstructed primary vertices'),
                 Plot1D('npvsGood', 'npvsGood', 20, 0, 60, 'total number of Good primary vertices'),
                 Plot1D('score', 'score', 20, 0, 300000, 'main primary vertex score, i.e. sum pt2 of clustered objects'),
+                Plot1D('sumpt2', 'sumpt2', 100, 0, 300000, 'main primary vertex sum pt2 of the charged pf candidates'),
                 Plot1D('x', 'x', 20, -0.3, 0.3, 'main primary vertex position x coordinate'),
                 Plot1D('y', 'y', 20, -0.3, 0.3, 'main primary vertex position y coordinate'),
                 Plot1D('z', 'z', 20, -20, 20, 'main primary vertex position z coordinate'),

PhysicsTools/NanoAOD/python/vertices_cff.py

@@ -10,6 +10,7 @@
 vertexTable = cms.EDProducer("VertexTableProducer",
     pvSrc = cms.InputTag("offlineSlimmedPrimaryVertices"),
     goodPvCut = cms.string("!isFake && ndof > 4 && abs(z) <= 24 && position.Rho <= 2"), 
+    pfcSrc = cms.InputTag("packedPFCandidates"),
     svSrc = cms.InputTag("linkedObjects", "vertices"),
     svCut = cms.string(""),  # careful: adding a cut here would make the collection matching inconsistent with the SV table
     dlenMin = cms.double(0),