CMS 3D CMS Logo

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
 
bool hasAbilityToProduceInLumis () const final
 
bool hasAbilityToProduceInRuns () const final
 

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::VertexgoodPvCut_
 
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::CandidatesvCut_
 
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<>
typedef CacheContexts< T... > CacheTypes
 
typedef CacheTypes::GlobalCache GlobalCache
 
typedef AbilityChecker< T... > HasAbility
 
typedef CacheTypes::LuminosityBlockCache LuminosityBlockCache
 
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCacheLuminosityBlockContext
 
typedef CacheTypes::LuminosityBlockSummaryCache LuminosityBlockSummaryCache
 
typedef CacheTypes::RunCache RunCache
 
typedef RunContextT< RunCache, GlobalCacheRunContext
 
typedef CacheTypes::RunSummaryCache RunSummaryCache
 

Detailed Description

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 49 of file VertexTableProducer.cc.

Constructor & Destructor Documentation

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

Definition at line 86 of file VertexTableProducer.cc.

86  :
87  pvs_(consumes<std::vector<reco::Vertex>>( params.getParameter<edm::InputTag>("pvSrc") )),
88  pvsScore_(consumes<edm::ValueMap<float>>( params.getParameter<edm::InputTag>("pvSrc") )),
90  svCut_(params.getParameter<std::string>("svCut") , true),
91  goodPvCut_(params.getParameter<std::string>("goodPvCut") , true),
92  goodPvCutString_(params.getParameter<std::string>("goodPvCut") ),
93  pvName_(params.getParameter<std::string>("pvName") ),
94  svName_(params.getParameter<std::string>("svName") ),
95  svDoc_(params.getParameter<std::string>("svDoc") ),
96  dlenMin_(params.getParameter<double>("dlenMin") ),
97  dlenSigMin_(params.getParameter<double>("dlenSigMin") )
98 
99 {
100  produces<nanoaod::FlatTable>("pv");
101  produces<nanoaod::FlatTable>("otherPVs");
102  produces<nanoaod::FlatTable>("svs");
103  produces<edm::PtrVector<reco::Candidate> >();
104 
105 }
T getParameter(std::string const &) const
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_
const std::string svName_
const StringCutObjectSelector< reco::Vertex > goodPvCut_
const StringCutObjectSelector< reco::Candidate > svCut_
VertexTableProducer::~VertexTableProducer ( )
override

Definition at line 108 of file VertexTableProducer.cc.

109 {
110 
111  // do anything here that needs to be done at destruction time
112  // (e.g. close files, deallocate resources etc.)
113 
114 }

Member Function Documentation

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

Definition at line 196 of file VertexTableProducer.cc.

197 {
198 }
void VertexTableProducer::endStream ( )
overrideprivate

Definition at line 202 of file VertexTableProducer.cc.

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

Definition at line 207 of file VertexTableProducer.cc.

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

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

Definition at line 125 of file VertexTableProducer.cc.

References EnergyCorrector::c, RecoVertex::convertError(), RecoVertex::convertPos(), VertexDistance3D::distance(), VertexDistanceXY::distance(), dlenMin_, dlenSigMin_, PVValHelper::dx, PVValHelper::dxy, PVValHelper::dy, PVValHelper::dz, nanoaod::FlatTable::FloatColumn, edm::Event::getByToken(), goodPvCut_, goodPvCutString_, AlcaRecoSelection_cff::goodPVs, mps_fire::i, nanoaod::FlatTable::IntColumn, eostools::move(), edm::Event::put(), MetAnalyzer::pv(), pvName_, pvs_, pvsScore_, Measurement1D::significance(), mathSSE::sqrt(), pfDeepBoostedJetPreprocessParams_cfi::sv, svCut_, svName_, svs_, and Measurement1D::value().

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

Member Data Documentation

const double VertexTableProducer::dlenMin_
private

Definition at line 77 of file VertexTableProducer.cc.

Referenced by produce().

const double VertexTableProducer::dlenSigMin_
private

Definition at line 77 of file VertexTableProducer.cc.

Referenced by produce().

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

Definition at line 72 of file VertexTableProducer.cc.

Referenced by produce().

const std::string VertexTableProducer::goodPvCutString_
private

Definition at line 73 of file VertexTableProducer.cc.

Referenced by produce().

const std::string VertexTableProducer::pvName_
private

Definition at line 74 of file VertexTableProducer.cc.

Referenced by produce().

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

Definition at line 68 of file VertexTableProducer.cc.

Referenced by produce().

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

Definition at line 69 of file VertexTableProducer.cc.

Referenced by produce().

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

Definition at line 71 of file VertexTableProducer.cc.

Referenced by produce().

const std::string VertexTableProducer::svDoc_
private

Definition at line 76 of file VertexTableProducer.cc.

const std::string VertexTableProducer::svName_
private

Definition at line 75 of file VertexTableProducer.cc.

Referenced by produce().

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

Definition at line 70 of file VertexTableProducer.cc.

Referenced by produce().