VertexFromTrackProducer Class Reference

#include <RecoVertex/PrimaryVertexProducer/src/VertexFromTrackProducer.cc>

Inheritance diagram for VertexFromTrackProducer:

Public Member Functions
edm::ParameterSet	config () const
virtual void	produce (edm::Event &, const edm::EventSetup &) override
	VertexFromTrackProducer (const edm::ParameterSet &)
	~VertexFromTrackProducer ()
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
ModuleDescription const &	moduleDescription () const
virtual	~EDProducer ()
Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...
virtual	~ProducerBase ()
Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()
ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const
bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)
virtual	~EDConsumerBase ()

Public Attributes
edm::EDGetTokenT< reco::BeamSpot >	beamSpotLabel
edm::EDGetTokenT< edm::View < reco::RecoCandidate > >	candidateToken
edm::InputTag	trackLabel
edm::EDGetTokenT< edm::View < reco::Track > >	trackToken
edm::EDGetTokenT < trigger::TriggerFilterObjectWithRefs >	triggerFilterElectronsSrc
edm::EDGetTokenT < trigger::TriggerFilterObjectWithRefs >	triggerFilterMuonsSrc
edm::EDGetTokenT< edm::View < reco::Vertex > >	vertexLabel

Private Attributes
bool	fIsRecoCandidate
bool	fUseBeamSpot
bool	fUseTriggerFilterElectrons
bool	fUseTriggerFilterMuons
bool	fUseVertex
bool	fVerbose
edm::ParameterSet	theConfig

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)
Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)
EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)
ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...
template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()
void	consumesMany (const TypeToGet &id)
template<BranchType B>
void	consumesMany (const TypeToGet &id)
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Description: produces a primary vertex extrapolating the track of a candidate on the beam axis

Implementation: <Notes on="" implementation>="">

Definition at line 44 of file VertexFromTrackProducer.h.

Constructor & Destructor Documentation

VertexFromTrackProducer::VertexFromTrackProducer ( const edm::ParameterSet &  conf )

explicit

Definition at line 27 of file VertexFromTrackProducer.cc.

References beamSpotLabel, candidateToken, fIsRecoCandidate, fUseBeamSpot, fUseTriggerFilterElectrons, fUseTriggerFilterMuons, fUseVertex, fVerbose, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), trackLabel, trackToken, triggerFilterElectronsSrc, triggerFilterMuonsSrc, and vertexLabel.

  : theConfig(conf)
{
  edm::LogInfo("PVDebugInfo") 
    << "Initializing  VertexFromTrackProducer" << "\n";
  fVerbose = conf.getUntrackedParameter<bool>("verbose", false);
  trackLabel = conf.getParameter<edm::InputTag>("trackLabel");
  trackToken = consumes<edm::View<reco::Track> >(trackLabel);
  candidateToken = consumes<edm::View<reco::RecoCandidate> >(trackLabel);
  fIsRecoCandidate = conf.getParameter<bool>("isRecoCandidate");
  fUseBeamSpot = conf.getParameter<bool>("useBeamSpot");
  fUseVertex = conf.getParameter<bool>("useVertex");
  fUseTriggerFilterElectrons = conf.getParameter<bool>("useTriggerFilterElectrons");
  fUseTriggerFilterMuons = conf.getParameter<bool>("useTriggerFilterMuons");
  triggerFilterElectronsSrc = consumes<trigger::TriggerFilterObjectWithRefs>(conf.getParameter<edm::InputTag>("triggerFilterElectronsSrc"));
  triggerFilterMuonsSrc = consumes<trigger::TriggerFilterObjectWithRefs>(conf.getParameter<edm::InputTag>("triggerFilterMuonsSrc"));
  vertexLabel = consumes<edm::View<reco::Vertex> >(conf.getParameter<edm::InputTag>("vertexLabel"));
  beamSpotLabel = consumes<reco::BeamSpot>(conf.getParameter<edm::InputTag>("beamSpotLabel"));
 
  produces<reco::VertexCollection>();

}

VertexFromTrackProducer::~VertexFromTrackProducer

(

)

Definition at line 51 of file VertexFromTrackProducer.cc.

{}

Member Function Documentation

edm::ParameterSet VertexFromTrackProducer::config ( void  ) const

inline

Definition at line 52 of file VertexFromTrackProducer.h.

References theConfig.

{ return theConfig; }

void VertexFromTrackProducer::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overridevirtual

Implements edm::EDProducer.

Definition at line 59 of file VertexFromTrackProducer.cc.

References SiPixelRawToDigiRegional_cfi::beamSpot, beamSpotLabel, candidateToken, gather_cfg::cout, fIsRecoCandidate, fUseBeamSpot, fUseTriggerFilterElectrons, fUseTriggerFilterMuons, fUseVertex, fVerbose, edm::Event::getByToken(), i, edm::HandleBase::isValid(), reco::BeamSpot::position(), reco::Vertex::position(), RecoTauCleanerPlugins::pt, edm::Event::put(), query::result, trackToken, trigger::TriggerElectron, triggerFilterElectronsSrc, triggerFilterMuonsSrc, trigger::TriggerMuon, findQualityFiles::v, vertexLabel, reco::Vertex::xError(), reco::Vertex::yError(), and reco::Vertex::zError().

{
  using namespace edm;

  std::auto_ptr<reco::VertexCollection> result(new reco::VertexCollection);
  reco::VertexCollection vColl;

  math::XYZPoint vertexPoint;
  bool vertexAvailable = false;

  // get the BeamSpot
  edm::Handle<reco::BeamSpot> recoBeamSpotHandle;
  iEvent.getByToken(beamSpotLabel,recoBeamSpotHandle);
  if (recoBeamSpotHandle.isValid()){
    reco::BeamSpot beamSpot = *recoBeamSpotHandle;
    vertexPoint = beamSpot.position();
  }else{
    edm::LogError("UnusableBeamSpot") << "No beam spot found in Event";
  }

  if(fUseVertex)
  {
    // get the Vertex
    edm::Handle<edm::View<reco::Vertex> > recoVertexHandle;
    iEvent.getByToken(vertexLabel,recoVertexHandle);
    if ((recoVertexHandle.isValid()) && (recoVertexHandle->size()>0)){
      reco::Vertex vertex = recoVertexHandle->at(0);
      vertexPoint = vertex.position();
      vertexAvailable = true;
    }else {
      edm::LogInfo("UnusableVertex")
    << "No vertex found in Event, beam spot used instaed" << "\n";
    }
  }

  const reco::Track* track = 0;
  if(fIsRecoCandidate)
  {
    edm::Handle<edm::View<reco::RecoCandidate> > candidateHandle;
    iEvent.getByToken(candidateToken, candidateHandle);
    if ((candidateHandle.isValid())&&(candidateHandle->size()>0)){
      double maxpt=0.;
      unsigned i_maxpt=0;
      for (unsigned i = 0; i < candidateHandle->size(); ++i) {
        double pt=candidateHandle->ptrAt(i)->pt();
        if(pt>maxpt)
    {
      i_maxpt=i;
      maxpt=pt;
    }
      }
      track = dynamic_cast<const reco::Track*>(candidateHandle->ptrAt(i_maxpt)->bestTrack());
    }
  } 
  else if(fUseTriggerFilterElectrons) {
    edm::Handle<trigger::TriggerFilterObjectWithRefs> triggerfilter;
    iEvent.getByToken(triggerFilterElectronsSrc, triggerfilter);
    std::vector<reco::ElectronRef> recocandidates;
    triggerfilter->getObjects(trigger::TriggerElectron,recocandidates);
    if ((recocandidates.size()>0)){
      double maxpt=0.;
      unsigned i_maxpt=0;
      for (unsigned i = 0; i < recocandidates.size(); ++i) {
    double pt=recocandidates.at(i)->pt();
    if(pt>maxpt) 
      {
        i_maxpt=i;
        maxpt=pt;
      }
    track = dynamic_cast<const reco::Track*>(recocandidates.at(i_maxpt)->bestTrack());
      }
    }
  }  
  else if(fUseTriggerFilterMuons) {
    edm::Handle<trigger::TriggerFilterObjectWithRefs> triggerfilter;
    iEvent.getByToken(triggerFilterMuonsSrc, triggerfilter);
    std::vector<reco::RecoChargedCandidateRef> recocandidates;
    triggerfilter->getObjects(trigger::TriggerMuon,recocandidates);
    if ((recocandidates.size()>0)){
      double maxpt=0.;
      unsigned i_maxpt=0;
      for (unsigned i = 0; i < recocandidates.size(); ++i) {
    double pt=recocandidates.at(i)->pt();
    if(pt>maxpt) 
      {
        i_maxpt=i;
        maxpt=pt;
      }
    track = dynamic_cast<const reco::Track*>(recocandidates.at(i_maxpt)->bestTrack());
      }
    }
  }
  else {
    edm::Handle<edm::View<reco::Track> > trackHandle;
    iEvent.getByToken(trackToken, trackHandle);
    if ((trackHandle.isValid())&&(trackHandle->size()>0)){
      double maxpt=0.;
      unsigned i_maxpt=0;
      for (unsigned i = 0; i < trackHandle->size(); ++i) {
        double pt=trackHandle->ptrAt(i)->pt();
        if(pt>maxpt)
    {
      i_maxpt=i;
      maxpt=pt;
    }
      }
      track = dynamic_cast<const reco::Track*>(&*trackHandle->ptrAt(i_maxpt));
    }
  }

  if(track) {
    if(fUseBeamSpot || (fUseVertex && vertexAvailable) ) {
      vertexPoint.SetZ(vertexPoint.z()+track->dz(vertexPoint));
    }
    else {
      vertexPoint.SetZ(track->vz());
    }
  }
  math::Error<3>::type noErrors;
  reco::Vertex v(vertexPoint, noErrors);
  vColl.push_back(v);

  // provide beamspot or primary vertex if no candidate found
  //if(vColl.size()==0)
  //{
  //    math::Error<3>::type noErrors;
  //    reco::Vertex v(vertexPoint, noErrors);
  //    vColl.push_back(v);
  //}

  if(fVerbose){
    int ivtx=0;
    for(reco::VertexCollection::const_iterator v=vColl.begin(); 
    v!=vColl.end(); ++v){
      std::cout << "recvtx "<< ivtx++ 
        << " x "  << std::setw(6) << v->position().x() 
        << " dx " << std::setw(6) << v->xError()
        << " y "  << std::setw(6) << v->position().y() 
        << " dy " << std::setw(6) << v->yError()
        << " z "  << std::setw(6) << v->position().z() 
        << " dz " << std::setw(6) << v->zError()
        << std::endl;
    }
  }

  
  *result = vColl;
  iEvent.put(result);
  
}

Member Data Documentation

edm::EDGetTokenT<reco::BeamSpot> VertexFromTrackProducer::beamSpotLabel

Definition at line 59 of file VertexFromTrackProducer.h.

Referenced by produce(), and VertexFromTrackProducer().

edm::EDGetTokenT<edm::View<reco::RecoCandidate> > VertexFromTrackProducer::candidateToken

Definition at line 55 of file VertexFromTrackProducer.h.

Referenced by produce(), and VertexFromTrackProducer().

bool VertexFromTrackProducer::fIsRecoCandidate

private

Definition at line 63 of file VertexFromTrackProducer.h.

Referenced by produce(), and VertexFromTrackProducer().

bool VertexFromTrackProducer::fUseBeamSpot

private

Definition at line 64 of file VertexFromTrackProducer.h.

Referenced by produce(), and VertexFromTrackProducer().

bool VertexFromTrackProducer::fUseTriggerFilterElectrons

private

Definition at line 66 of file VertexFromTrackProducer.h.

Referenced by produce(), and VertexFromTrackProducer().

bool VertexFromTrackProducer::fUseTriggerFilterMuons

private

Definition at line 66 of file VertexFromTrackProducer.h.

Referenced by produce(), and VertexFromTrackProducer().

bool VertexFromTrackProducer::fUseVertex

private

Definition at line 65 of file VertexFromTrackProducer.h.

Referenced by produce(), and VertexFromTrackProducer().

bool VertexFromTrackProducer::fVerbose

private

Definition at line 68 of file VertexFromTrackProducer.h.

Referenced by produce(), and VertexFromTrackProducer().

edm::ParameterSet VertexFromTrackProducer::theConfig

private

Definition at line 67 of file VertexFromTrackProducer.h.

Referenced by config().

edm::InputTag VertexFromTrackProducer::trackLabel

Definition at line 53 of file VertexFromTrackProducer.h.

Referenced by VertexFromTrackProducer().

edm::EDGetTokenT<edm::View<reco::Track> > VertexFromTrackProducer::trackToken

Definition at line 54 of file VertexFromTrackProducer.h.

Referenced by produce(), and VertexFromTrackProducer().

edm::EDGetTokenT<trigger::TriggerFilterObjectWithRefs> VertexFromTrackProducer::triggerFilterElectronsSrc

Definition at line 56 of file VertexFromTrackProducer.h.

Referenced by produce(), and VertexFromTrackProducer().

edm::EDGetTokenT<trigger::TriggerFilterObjectWithRefs> VertexFromTrackProducer::triggerFilterMuonsSrc

Definition at line 57 of file VertexFromTrackProducer.h.

Referenced by produce(), and VertexFromTrackProducer().

edm::EDGetTokenT<edm::View<reco::Vertex> > VertexFromTrackProducer::vertexLabel

Definition at line 58 of file VertexFromTrackProducer.h.

Referenced by produce(), and VertexFromTrackProducer().