OniaVtxReProducer Class Reference

#include <OniaVtxReProducer.h>

Public Member Functions
const edm::InputTag &	inputBeamSpot () const
	Get the InputTag of the BeamSpot used in the VertexProducer. More...
const edm::ParameterSet &	inputConfig () const
	Get the configuration used in the VertexProducer. More...
const edm::InputTag &	inputTracks () const
	Get the InputTag of the TrackCollection used in the VertexProducer. More...
std::vector< TransientVertex >	makeVertices (const reco::TrackCollection &tracks, const reco::BeamSpot &bs, const edm::EventSetup &iSetup) 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< PrimaryVertexProducerAlgorithm >	algo_
edm::InputTag	beamSpotTag_
edm::ParameterSet	config_
edm::InputTag	tracksTag_

Detailed Description

Definition at line 15 of file OniaVtxReProducer.h.

Constructor & Destructor Documentation

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

This is the real constructor to be used.

Definition at line 11 of file OniaVtxReProducer.cc.

References beamSpotTag_, configure(), Exception, edm::Provenance::friendlyClassName(), edm::Event::getProvenance(), HLT_2018_cff::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::HandleBase::provenance(), and tracksTag_.

                                                                                                          {
  const edm::Provenance *prov = handle.provenance();
  if (prov == nullptr)
    throw cms::Exception("CorruptData") << "Vertex handle doesn't have provenance.";
  edm::ParameterSet psetFromProvenance = edm::parameterSet(*prov);

  bool is_primary_available = false;
  const edm::Provenance *parent_prov = prov;
  if (edm::moduleName(*prov) != "PrimaryVertexProducer") {
    std::vector<edm::BranchID> parents = prov->productProvenance()->parentage().parents();
    for (std::vector<edm::BranchID>::const_iterator it = parents.begin(), ed = parents.end(); it != ed; ++it) {
      edm::Provenance parprov = iEvent.getProvenance(*it);
      if (parprov.friendlyClassName() == "recoVertexs") {  // for AOD actually this the parent we should look for
        parent_prov = &parprov;
        psetFromProvenance = edm::parameterSet(parprov);
        is_primary_available = true;
        break;
      }
    }
  } else
    is_primary_available = true;
  if (is_primary_available)
    prov = parent_prov;
  else
    throw cms::Exception("Configuration") << "Vertices to re-produce don't come from a PrimaryVertexProducer \n";

  configure(psetFromProvenance);

  // Now we also dig out the ProcessName used for the reco::Tracks and reco::Vertices
  std::vector<edm::BranchID> parents = prov->productProvenance()->parentage().parents();
  bool foundTracks = false;
  bool foundBeamSpot = false;
  for (std::vector<edm::BranchID>::const_iterator it = parents.begin(), ed = parents.end(); it != ed; ++it) {
    edm::Provenance parprov = iEvent.getProvenance(*it);
    if (parprov.friendlyClassName() == "recoTracks") {
      tracksTag_ = edm::InputTag(parprov.moduleLabel(), parprov.productInstanceName(), parprov.processName());
      foundTracks = true;
      if (parprov.moduleLabel() != "generalTracks")
        foundTracks = false;  // this is necessary since we are asking for that in onia2mumu
    } else if (parprov.friendlyClassName() == "recoBeamSpot") {
      beamSpotTag_ = edm::InputTag(parprov.moduleLabel(), parprov.productInstanceName(), parprov.processName());
      foundBeamSpot = true;
      if (parprov.moduleLabel() != "offlineBeamSpot")
        foundBeamSpot = false;  // this is necessary since we are asking for that in onia2mumu
    }
  }
  if (!foundTracks || !foundBeamSpot) {
    //edm::LogWarning("OniaVtxReProducer_MissingParentage") <<
    throw cms::Exception("Configuration")
        << "Can't find correct parentage info for vertex collection inputs: " << (foundTracks ? "" : "generalTracks ")
        << (foundBeamSpot ? "" : "offlineBeamSpot") << "\n";
  }
}

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

inline

This is only for testing.

Definition at line 21 of file OniaVtxReProducer.h.

References configure(), makeVertices(), and PDWG_EXOHSCP_cff::tracks.

{ configure(configByHand); }

Member Function Documentation

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

private

Definition at line 65 of file OniaVtxReProducer.cc.

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

Referenced by inputBeamSpot(), and OniaVtxReProducer().

                                                                {
  config_ = iConfig;
  tracksTag_ = iConfig.getParameter<edm::InputTag>("TrackLabel");
  beamSpotTag_ = iConfig.getParameter<edm::InputTag>("beamSpotLabel");
  algo_.reset(new PrimaryVertexProducerAlgorithm(iConfig));
}

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_, and configure().

{ return beamSpotTag_; }

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

inline

Get the configuration used in the VertexProducer.

Definition at line 29 of file OniaVtxReProducer.h.

References config_.

{ return config_; }

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_.

{ return tracksTag_; }

std::vector< TransientVertex > OniaVtxReProducer::makeVertices	(	const reco::TrackCollection &	tracks,
		const reco::BeamSpot &	bs,
		const edm::EventSetup &	iSetup
	)		const

Make the vertices.

Definition at line 72 of file OniaVtxReProducer.cc.

References algo_, and edm::EventSetup::get().

Referenced by OniaVtxReProducer(), and Onia2MuMuPAT::produce().

                                                                                                {
  edm::ESHandle<TransientTrackBuilder> theB;
  iSetup.get<TransientTrackRecord>().get("TransientTrackBuilder", theB);

  std::vector<reco::TransientTrack> t_tks;
  t_tks.reserve(tracks.size());
  for (reco::TrackCollection::const_iterator it = tracks.begin(), ed = tracks.end(); it != ed; ++it) {
    t_tks.push_back((*theB).build(*it));
    t_tks.back().setBeamSpot(bs);
  }

  return algo_->vertices(t_tks, bs, "AdaptiveVertexFitter");
}

Member Data Documentation

std::unique_ptr<PrimaryVertexProducerAlgorithm> OniaVtxReProducer::algo_

private

Definition at line 43 of file OniaVtxReProducer.h.

Referenced by configure(), and makeVertices().

edm::InputTag OniaVtxReProducer::beamSpotTag_

private

Definition at line 42 of file OniaVtxReProducer.h.

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

edm::ParameterSet OniaVtxReProducer::config_

private

Definition at line 40 of file OniaVtxReProducer.h.

Referenced by configure(), and inputConfig().

edm::InputTag OniaVtxReProducer::tracksTag_

private

Definition at line 41 of file OniaVtxReProducer.h.

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