CMS 3D CMS Logo

List of all members | Public Member Functions | Private Member Functions | Private Attributes
OniaVtxReProducer Class Reference

#include <OniaVtxReProducer.h>

Public Member Functions

const edm::InputTaginputBeamSpot () const
 Get the InputTag of the BeamSpot used in the VertexProducer. More...
 
const edm::ParameterSetinputConfig () const
 Get the configuration used in the VertexProducer. More...
 
const edm::InputTaginputTracks () const
 Get the InputTag of the TrackCollection used in the VertexProducer. More...
 
std::vector< TransientVertexmakeVertices (const reco::TrackCollection &tracks, const reco::BeamSpot &bs, const TransientTrackBuilder &theB) const
 Make the vertices. More...
 
 OniaVtxReProducer (const edm::Handle< reco::VertexCollection > &configFromOriginalVertexCollection, const edm::Event &iEvent)
 This is the real constructor to be used. More...
 
 OniaVtxReProducer (const edm::ParameterSet &configByHand)
 This is only for testing. More...
 

Private Member Functions

void configure (const edm::ParameterSet &iConfig)
 

Private Attributes

std::unique_ptr< PrimaryVertexProducerAlgorithmalgo_
 
edm::InputTag beamSpotTag_
 
edm::ParameterSet config_
 
edm::InputTag tracksTag_
 

Detailed Description

Definition at line 15 of file OniaVtxReProducer.h.

Constructor & Destructor Documentation

◆ OniaVtxReProducer() [1/2]

OniaVtxReProducer::OniaVtxReProducer ( const edm::Handle< reco::VertexCollection > &  configFromOriginalVertexCollection,
const edm::Event iEvent 
)

This is the real constructor to be used.

Definition at line 10 of file OniaVtxReProducer.cc.

References beamSpotTag_, configure(), Exception, edm::Provenance::friendlyClassName(), patZpeak::handle, iEvent, ProducerED_cfi::InputTag, edm::Provenance::moduleLabel(), edm::moduleName(), edm::parameterSet(), edm::ProductProvenance::parentage(), parents, edm::Parentage::parents(), edm::Provenance::processName(), edm::Provenance::productInstanceName(), edm::Provenance::productProvenance(), edm::Provenance::stable(), and tracksTag_.

10  {
11  const edm::Provenance *prov = handle.provenance();
12  if (prov == nullptr)
13  throw cms::Exception("CorruptData") << "Vertex handle doesn't have provenance.";
14  edm::ParameterSet psetFromProvenance = edm::parameterSet(prov->stable(), iEvent.processHistory());
15 
16  bool is_primary_available = false;
17  const edm::Provenance *parent_prov = prov;
18  if (edm::moduleName(prov->stable(), iEvent.processHistory()) != "PrimaryVertexProducer") {
19  std::vector<edm::BranchID> parents = prov->productProvenance()->parentage().parents();
20  for (std::vector<edm::BranchID>::const_iterator it = parents.begin(), ed = parents.end(); it != ed; ++it) {
21  edm::Provenance parprov = iEvent.getProvenance(*it);
22  if (parprov.friendlyClassName() == "recoVertexs") { // for AOD actually this the parent we should look for
23  parent_prov = &parprov;
24  psetFromProvenance = edm::parameterSet(parprov.stable(), iEvent.processHistory());
25  is_primary_available = true;
26  break;
27  }
28  }
29  } else
30  is_primary_available = true;
31  if (is_primary_available)
32  prov = parent_prov;
33  else
34  throw cms::Exception("Configuration") << "Vertices to re-produce don't come from a PrimaryVertexProducer \n";
35 
36  configure(psetFromProvenance);
37 
38  // Now we also dig out the ProcessName used for the reco::Tracks and reco::Vertices
39  std::vector<edm::BranchID> parents = prov->productProvenance()->parentage().parents();
40  bool foundTracks = false;
41  bool foundBeamSpot = false;
42  for (std::vector<edm::BranchID>::const_iterator it = parents.begin(), ed = parents.end(); it != ed; ++it) {
43  const edm::Provenance &parprov = iEvent.getProvenance(*it);
44  if (parprov.friendlyClassName() == "recoTracks") {
45  tracksTag_ = edm::InputTag(parprov.moduleLabel(), parprov.productInstanceName(), parprov.processName());
46  foundTracks = true;
47  if (parprov.moduleLabel() != "generalTracks")
48  foundTracks = false; // this is necessary since we are asking for that in onia2mumu
49  } else if (parprov.friendlyClassName() == "recoBeamSpot") {
50  beamSpotTag_ = edm::InputTag(parprov.moduleLabel(), parprov.productInstanceName(), parprov.processName());
51  foundBeamSpot = true;
52  if (parprov.moduleLabel() != "offlineBeamSpot")
53  foundBeamSpot = false; // this is necessary since we are asking for that in onia2mumu
54  }
55  }
56  if (!foundTracks || !foundBeamSpot) {
57  //edm::LogWarning("OniaVtxReProducer_MissingParentage") <<
58  throw cms::Exception("Configuration")
59  << "Can't find correct parentage info for vertex collection inputs: " << (foundTracks ? "" : "generalTracks ")
60  << (foundBeamSpot ? "" : "offlineBeamSpot") << "\n";
61  }
62 }
TPRegexp parents
Definition: eve_filter.cc:21
ProductProvenance const * productProvenance() const
Definition: Provenance.cc:24
std::string const & processName() const
Definition: Provenance.h:57
std::vector< BranchID > const & parents() const
Definition: Parentage.h:44
ParameterSet const & parameterSet(StableProvenance const &provenance, ProcessHistory const &history)
Definition: Provenance.cc:11
int iEvent
Definition: GenABIO.cc:224
std::string const & productInstanceName() const
Definition: Provenance.h:58
void configure(const edm::ParameterSet &iConfig)
edm::InputTag tracksTag_
std::string moduleName(StableProvenance const &provenance, ProcessHistory const &history)
Definition: Provenance.cc:27
std::string const & moduleLabel() const
Definition: Provenance.h:55
edm::InputTag beamSpotTag_
StableProvenance const & stable() const
Definition: Provenance.h:42
std::string const & friendlyClassName() const
Definition: Provenance.h:59
Parentage const & parentage() const

◆ OniaVtxReProducer() [2/2]

OniaVtxReProducer::OniaVtxReProducer ( const edm::ParameterSet configByHand)
inline

This is only for testing.

Definition at line 21 of file OniaVtxReProducer.h.

References configure().

21 { configure(configByHand); }
void configure(const edm::ParameterSet &iConfig)

Member Function Documentation

◆ configure()

void OniaVtxReProducer::configure ( const edm::ParameterSet iConfig)
private

Definition at line 64 of file OniaVtxReProducer.cc.

References algo_, beamSpotTag_, config_, edm::ParameterSet::getParameter(), and tracksTag_.

Referenced by OniaVtxReProducer().

64  {
65  config_ = iConfig;
66  tracksTag_ = iConfig.getParameter<edm::InputTag>("TrackLabel");
67  beamSpotTag_ = iConfig.getParameter<edm::InputTag>("beamSpotLabel");
68  algo_ = std::make_unique<PrimaryVertexProducerAlgorithm>(iConfig);
69 }
T getParameter(std::string const &) const
Definition: ParameterSet.h:307
std::unique_ptr< PrimaryVertexProducerAlgorithm > algo_
edm::ParameterSet config_
edm::InputTag tracksTag_
edm::InputTag beamSpotTag_

◆ inputBeamSpot()

const edm::InputTag& OniaVtxReProducer::inputBeamSpot ( ) const
inline

Get the InputTag of the BeamSpot used in the VertexProducer.

Definition at line 35 of file OniaVtxReProducer.h.

References beamSpotTag_.

35 { return beamSpotTag_; }
edm::InputTag beamSpotTag_

◆ inputConfig()

const edm::ParameterSet& OniaVtxReProducer::inputConfig ( ) const
inline

Get the configuration used in the VertexProducer.

Definition at line 29 of file OniaVtxReProducer.h.

References config_.

29 { return config_; }
edm::ParameterSet config_

◆ inputTracks()

const edm::InputTag& OniaVtxReProducer::inputTracks ( ) const
inline

Get the InputTag of the TrackCollection used in the VertexProducer.

Definition at line 32 of file OniaVtxReProducer.h.

References tracksTag_.

32 { return tracksTag_; }
edm::InputTag tracksTag_

◆ makeVertices()

std::vector< TransientVertex > OniaVtxReProducer::makeVertices ( const reco::TrackCollection tracks,
const reco::BeamSpot bs,
const TransientTrackBuilder theB 
) const

Make the vertices.

Definition at line 71 of file OniaVtxReProducer.cc.

References algo_, cms::cuda::bs, TransientTrackBuilder::build(), and pwdgSkimBPark_cfi::tracks.

Referenced by Onia2MuMuPAT::produce().

73  {
74  std::vector<reco::TransientTrack> t_tks;
75  t_tks.reserve(tracks.size());
76  for (reco::TrackCollection::const_iterator it = tracks.begin(), ed = tracks.end(); it != ed; ++it) {
77  t_tks.push_back(theB.build(*it));
78  t_tks.back().setBeamSpot(bs);
79  }
80 
81  return algo_->vertices(t_tks, bs, "AdaptiveVertexFitter");
82 }
std::unique_ptr< PrimaryVertexProducerAlgorithm > algo_
reco::TransientTrack build(const reco::Track *p) const

Member Data Documentation

◆ algo_

std::unique_ptr<PrimaryVertexProducerAlgorithm> OniaVtxReProducer::algo_
private

Definition at line 43 of file OniaVtxReProducer.h.

Referenced by configure(), and makeVertices().

◆ beamSpotTag_

edm::InputTag OniaVtxReProducer::beamSpotTag_
private

Definition at line 42 of file OniaVtxReProducer.h.

Referenced by configure(), inputBeamSpot(), and OniaVtxReProducer().

◆ config_

edm::ParameterSet OniaVtxReProducer::config_
private

Definition at line 40 of file OniaVtxReProducer.h.

Referenced by configure(), and inputConfig().

◆ tracksTag_

edm::InputTag OniaVtxReProducer::tracksTag_
private

Definition at line 41 of file OniaVtxReProducer.h.

Referenced by configure(), inputTracks(), and OniaVtxReProducer().