CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Groups Pages
List of all members | Public Member Functions | Static Public Member Functions | Private Member Functions | Private Attributes
VertexTableProducer Class Reference

#include <PhysicsTools/VertexTableProducer/plugins/VertexTableProducer.cc>

Inheritance diagram for VertexTableProducer:
edm::stream::EDProducer<>

Public Member Functions

 VertexTableProducer (const edm::ParameterSet &)
 
 ~VertexTableProducer () override
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
 EDProducer (const EDProducer &)=delete
 
bool hasAbilityToProduceInBeginLumis () const final
 
bool hasAbilityToProduceInBeginProcessBlocks () const final
 
bool hasAbilityToProduceInBeginRuns () const final
 
bool hasAbilityToProduceInEndLumis () const final
 
bool hasAbilityToProduceInEndProcessBlocks () const final
 
bool hasAbilityToProduceInEndRuns () const final
 
const EDProduceroperator= (const EDProducer &)=delete
 

Static Public Member Functions

static void fillDescriptions (edm::ConfigurationDescriptions &descriptions)
 

Private Member Functions

void beginStream (edm::StreamID) override
 
void endStream () override
 
void produce (edm::Event &, const edm::EventSetup &) override
 

Private Attributes

const double dlenMin_
 
const double dlenSigMin_
 
const StringCutObjectSelector
< reco::Vertex
goodPvCut_
 
const std::string goodPvCutString_
 
const std::string pvName_
 
const edm::EDGetTokenT
< std::vector< reco::Vertex > > 
pvs_
 
const edm::EDGetTokenT
< edm::ValueMap< float > > 
pvsScore_
 
const StringCutObjectSelector
< reco::Candidate
svCut_
 
const std::string svDoc_
 
const std::string svName_
 
const edm::EDGetTokenT
< edm::View
< reco::VertexCompositePtrCandidate > > 
svs_
 

Additional Inherited Members

- Public Types inherited from edm::stream::EDProducer<>
using CacheTypes = CacheContexts< T...>
 
using GlobalCache = typename CacheTypes::GlobalCache
 
using HasAbility = AbilityChecker< T...>
 
using InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
 
using LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
 
using LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
 
using LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
 
using RunCache = typename CacheTypes::RunCache
 
using RunContext = RunContextT< RunCache, GlobalCache >
 
using RunSummaryCache = typename CacheTypes::RunSummaryCache
 

Detailed Description

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 48 of file VertexTableProducer.cc.

Constructor & Destructor Documentation

VertexTableProducer::VertexTableProducer ( const edm::ParameterSet params)
explicit

Definition at line 82 of file VertexTableProducer.cc.

83  : pvs_(consumes<std::vector<reco::Vertex>>(params.getParameter<edm::InputTag>("pvSrc"))),
84  pvsScore_(consumes<edm::ValueMap<float>>(params.getParameter<edm::InputTag>("pvSrc"))),
86  svCut_(params.getParameter<std::string>("svCut"), true),
87  goodPvCut_(params.getParameter<std::string>("goodPvCut"), true),
88  goodPvCutString_(params.getParameter<std::string>("goodPvCut")),
89  pvName_(params.getParameter<std::string>("pvName")),
90  svName_(params.getParameter<std::string>("svName")),
91  svDoc_(params.getParameter<std::string>("svDoc")),
92  dlenMin_(params.getParameter<double>("dlenMin")),
93  dlenSigMin_(params.getParameter<double>("dlenSigMin"))
94 
95 {
96  produces<nanoaod::FlatTable>("pv");
97  produces<nanoaod::FlatTable>("otherPVs");
98  produces<nanoaod::FlatTable>("svs");
99  produces<edm::PtrVector<reco::Candidate>>();
100 }
const edm::EDGetTokenT< std::vector< reco::Vertex > > pvs_
const std::string pvName_
const std::string svDoc_
const edm::EDGetTokenT< edm::ValueMap< float > > pvsScore_
const std::string goodPvCutString_
const edm::EDGetTokenT< edm::View< reco::VertexCompositePtrCandidate > > svs_
T getParameter(std::string const &) const
Definition: ParameterSet.h:303
const std::string svName_
const StringCutObjectSelector< reco::Vertex > goodPvCut_
const StringCutObjectSelector< reco::Candidate > svCut_
VertexTableProducer::~VertexTableProducer ( )
override

Definition at line 102 of file VertexTableProducer.cc.

102  {
103  // do anything here that needs to be done at destruction time
104  // (e.g. close files, deallocate resources etc.)
105 }

Member Function Documentation

void VertexTableProducer::beginStream ( edm::StreamID  )
overrideprivate

Definition at line 200 of file VertexTableProducer.cc.

200 {}
void VertexTableProducer::endStream ( )
overrideprivate

Definition at line 203 of file VertexTableProducer.cc.

203 {}
void VertexTableProducer::fillDescriptions ( edm::ConfigurationDescriptions descriptions)
static

Definition at line 206 of file VertexTableProducer.cc.

References edm::ConfigurationDescriptions::addDefault(), submitPVResolutionJobs::desc, and edm::ParameterSetDescription::setUnknown().

206  {
207  //The following says we do not know what parameters are allowed so do no validation
208  // Please change this to state exactly what you do use, even if it is no parameters
210  desc.setUnknown();
211  descriptions.addDefault(desc);
212 }
void addDefault(ParameterSetDescription const &psetDescription)
void VertexTableProducer::produce ( edm::Event iEvent,
const edm::EventSetup iSetup 
)
overrideprivate

Definition at line 113 of file VertexTableProducer.cc.

References c, RecoTauCleanerPlugins::charge, reco::Candidate::charge(), RecoVertex::convertError(), RecoVertex::convertPos(), reco::Candidate::daughter(), dlenMin_, dlenSigMin_, PVValHelper::dx, PVValHelper::dxy, PVValHelper::dy, PVValHelper::dz, edm::Event::getByToken(), goodPvCut_, goodPvCutString_, AlcaRecoSelection_cff::goodPVs, mps_fire::i, gpuClustering::id, eostools::move(), position, edm::Event::put(), MetAnalyzer::pv(), pvName_, pvs_, pvsScore_, Measurement1D::significance(), mathSSE::sqrt(), svCut_, svName_, svs_, Measurement1D::value(), and z.

113  {
114  using namespace edm;
117  iEvent.getByToken(pvs_, pvsIn);
118  iEvent.getByToken(pvsScore_, pvsScoreIn);
119  auto pvTable = std::make_unique<nanoaod::FlatTable>(1, pvName_, true);
120  pvTable->addColumnValue<float>("ndof", (*pvsIn)[0].ndof(), "main primary vertex number of degree of freedom", 8);
121  pvTable->addColumnValue<float>("x", (*pvsIn)[0].position().x(), "main primary vertex position x coordinate", 10);
122  pvTable->addColumnValue<float>("y", (*pvsIn)[0].position().y(), "main primary vertex position y coordinate", 10);
123  pvTable->addColumnValue<float>("z", (*pvsIn)[0].position().z(), "main primary vertex position z coordinate", 16);
124  pvTable->addColumnValue<float>("chi2", (*pvsIn)[0].normalizedChi2(), "main primary vertex reduced chi2", 8);
125  int goodPVs = 0;
126  for (const auto& pv : *pvsIn)
127  if (goodPvCut_(pv))
128  goodPVs++;
129  pvTable->addColumnValue<int>("npvs", pvsIn->size(), "total number of reconstructed primary vertices");
130  pvTable->addColumnValue<int>(
131  "npvsGood", goodPVs, "number of good reconstructed primary vertices. selection:" + goodPvCutString_);
132  pvTable->addColumnValue<float>(
133  "score", pvsScoreIn->get(pvsIn.id(), 0), "main primary vertex score, i.e. sum pt2 of clustered objects", 8);
134 
135  auto otherPVsTable =
136  std::make_unique<nanoaod::FlatTable>((*pvsIn).size() > 4 ? 3 : (*pvsIn).size() - 1, "Other" + pvName_, false);
137  std::vector<float> pvsz;
138  std::vector<float> pvscores;
139  for (size_t i = 1; i < (*pvsIn).size() && i < 4; i++) {
140  pvsz.push_back((*pvsIn)[i].position().z());
141  pvscores.push_back((*pvsScoreIn).get(pvsIn.id(), i));
142  }
143  otherPVsTable->addColumn<float>("z", pvsz, "Z position of other primary vertices, excluding the main PV", 8);
144  otherPVsTable->addColumn<float>("score", pvscores, "scores of other primary vertices, excluding the main PV", 8);
145 
147  iEvent.getByToken(svs_, svsIn);
148  auto selCandSv = std::make_unique<PtrVector<reco::Candidate>>();
149  std::vector<float> dlen, dlenSig, pAngle, dxy, dxySig;
150  std::vector<int> charge;
151  VertexDistance3D vdist;
152  VertexDistanceXY vdistXY;
153 
154  size_t i = 0;
155  const auto& PV0 = pvsIn->front();
156  for (const auto& sv : *svsIn) {
157  if (svCut_(sv)) {
158  Measurement1D dl =
159  vdist.distance(PV0, VertexState(RecoVertex::convertPos(sv.position()), RecoVertex::convertError(sv.error())));
160  if (dl.value() > dlenMin_ and dl.significance() > dlenSigMin_) {
161  dlen.push_back(dl.value());
162  dlenSig.push_back(dl.significance());
163  edm::Ptr<reco::Candidate> c = svsIn->ptrAt(i);
164  selCandSv->push_back(c);
165  double dx = (PV0.x() - sv.vx()), dy = (PV0.y() - sv.vy()), dz = (PV0.z() - sv.vz());
166  double pdotv = (dx * sv.px() + dy * sv.py() + dz * sv.pz()) / sv.p() / sqrt(dx * dx + dy * dy + dz * dz);
167  pAngle.push_back(std::acos(pdotv));
168  Measurement1D d2d = vdistXY.distance(
169  PV0, VertexState(RecoVertex::convertPos(sv.position()), RecoVertex::convertError(sv.error())));
170  dxy.push_back(d2d.value());
171  dxySig.push_back(d2d.significance());
172 
173  int sum_charge = 0;
174  for (unsigned int id = 0; id < sv.numberOfDaughters(); ++id) {
175  const reco::Candidate* daughter = sv.daughter(id);
176  sum_charge += daughter->charge();
177  }
178  charge.push_back(sum_charge);
179  }
180  }
181  i++;
182  }
183 
184  auto svsTable = std::make_unique<nanoaod::FlatTable>(selCandSv->size(), svName_, false);
185  // For SV we fill from here only stuff that cannot be created with the SimpleFlatTableProducer
186  svsTable->addColumn<float>("dlen", dlen, "decay length in cm", 10);
187  svsTable->addColumn<float>("dlenSig", dlenSig, "decay length significance", 10);
188  svsTable->addColumn<float>("dxy", dxy, "2D decay length in cm", 10);
189  svsTable->addColumn<float>("dxySig", dxySig, "2D decay length significance", 10);
190  svsTable->addColumn<float>("pAngle", pAngle, "pointing angle, i.e. acos(p_SV * (SV - PV)) ", 10);
191  svsTable->addColumn<int>("charge", charge, "sum of the charge of the SV tracks", 10);
192 
193  iEvent.put(std::move(pvTable), "pv");
194  iEvent.put(std::move(otherPVsTable), "otherPVs");
195  iEvent.put(std::move(svsTable), "svs");
196  iEvent.put(std::move(selCandSv));
197 }
reco::Vertex::Point convertPos(const GlobalPoint &p)
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:133
virtual const Candidate * daughter(size_type i) const =0
return daughter at a given position, i = 0, ... numberOfDaughters() - 1 (read only mode) ...
const edm::EventSetup & c
const edm::EDGetTokenT< std::vector< reco::Vertex > > pvs_
const std::string pvName_
uint16_t *__restrict__ id
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:539
reco::Vertex::Error convertError(const GlobalError &ge)
Definition: ConvertError.h:8
const edm::EDGetTokenT< edm::ValueMap< float > > pvsScore_
const std::string goodPvCutString_
T sqrt(T t)
Definition: SSEVec.h:19
def move
Definition: eostools.py:511
const edm::EDGetTokenT< edm::View< reco::VertexCompositePtrCandidate > > svs_
virtual int charge() const =0
electric charge
double significance() const
Definition: Measurement1D.h:29
double value() const
Definition: Measurement1D.h:25
const std::string svName_
static int position[264][3]
Definition: ReadPGInfo.cc:289
tuple goodPVs
Good Primary Vertex Selection.
const StringCutObjectSelector< reco::Vertex > goodPvCut_
const StringCutObjectSelector< reco::Candidate > svCut_

Member Data Documentation

const double VertexTableProducer::dlenMin_
private

Definition at line 76 of file VertexTableProducer.cc.

Referenced by produce().

const double VertexTableProducer::dlenSigMin_
private

Definition at line 76 of file VertexTableProducer.cc.

Referenced by produce().

const StringCutObjectSelector<reco::Vertex> VertexTableProducer::goodPvCut_
private

Definition at line 71 of file VertexTableProducer.cc.

Referenced by produce().

const std::string VertexTableProducer::goodPvCutString_
private

Definition at line 72 of file VertexTableProducer.cc.

Referenced by produce().

const std::string VertexTableProducer::pvName_
private

Definition at line 73 of file VertexTableProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<std::vector<reco::Vertex> > VertexTableProducer::pvs_
private

Definition at line 67 of file VertexTableProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<edm::ValueMap<float> > VertexTableProducer::pvsScore_
private

Definition at line 68 of file VertexTableProducer.cc.

Referenced by produce().

const StringCutObjectSelector<reco::Candidate> VertexTableProducer::svCut_
private

Definition at line 70 of file VertexTableProducer.cc.

Referenced by produce().

const std::string VertexTableProducer::svDoc_
private

Definition at line 75 of file VertexTableProducer.cc.

const std::string VertexTableProducer::svName_
private

Definition at line 74 of file VertexTableProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<edm::View<reco::VertexCompositePtrCandidate> > VertexTableProducer::svs_
private

Definition at line 69 of file VertexTableProducer.cc.

Referenced by produce().