Produces and/or checks pat::VertexAssociation's. More...

#include "PhysicsTools/PatAlgos/interface/VertexingHelper.h"

Public Member Functions
bool	enabled () const
	returns true if this was given a non dummy configuration More...

void	newEvent (const edm::Event &event)
	To be called for each new event, reads in the vertex collection. More...

void	newEvent (const edm::Event &event, const edm::EventSetup &setup)

template<typename AnyCandRef >
pat::VertexAssociation	operator() (const AnyCandRef &) const

	VertexingHelper ()

	VertexingHelper (const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC)

Private Member Functions
pat::VertexAssociation	associate (const reco::Candidate &) const

reco::TrackBaseRef	getTrack_ (const reco::Candidate &c) const
	Get out the track from the Candidate / RecoCandidate / PFCandidate. More...

Private Attributes
pat::VertexAssociationSelector	assoSelector_
	selector of associations More...

bool	enabled_
	true if it has non null configuration More...

bool	playback_
	true if it's just reading the associations from the event More...

edm::ESHandle< TransientTrackBuilder >	ttBuilder_

bool	useTracks_
	use tracks inside candidates More...

edm::EDGetTokenT< edm::ValueMap< pat::VertexAssociation > >	vertexAssociationsToken_

edm::Handle< edm::ValueMap< pat::VertexAssociation > >	vertexAssoMap_

edm::Handle< reco::VertexCollection >	vertexHandle_

edm::EDGetTokenT< reco::VertexCollection >	verticesToken_

Detailed Description

Produces and/or checks pat::VertexAssociation's.

The VertexingHelper produces pat::VertexAssociation, or reads them from the event, and can use them to select if a candidate is good or not.

Author: Giovanni Petrucciani

Version

Id: VertexingHelper.h,v 1.3 2008/06/06 14:13:40 gpetrucc Exp

Definition at line 46 of file VertexingHelper.h.

Constructor & Destructor Documentation

pat::helper::VertexingHelper::VertexingHelper ( )

inline

Definition at line 48 of file VertexingHelper.h.

48 : enabled_(false) {}

pat::helper::VertexingHelper::enabled_

bool enabled_

true if it has non null configuration

Definition: VertexingHelper.h:68

pat::helper::VertexingHelper::VertexingHelper	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector &&	iC
	)

Definition at line 11 of file VertexingHelper.cc.

References assoSelector_, edm::ParameterSet::empty(), enabled_, Exception, edm::ParameterSet::existsAs(), edm::ParameterSet::getParameter(), playback_, useTracks_, vertexAssociationsToken_, and verticesToken_.

 {
     if (!iConfig.empty()) {
         enabled_ = true;
         if ( iConfig.existsAs<edm::InputTag>("vertexAssociations") == iConfig.existsAs<edm::InputTag>("vertices")) {
             throw cms::Exception("Configuration") <<
                 "VertexingHelper: you must configure either 'vertices' (to produce associations) or 'vertexAssociations' (to read them from disk), " <<
                 "you can't specify both, nor you can specify none!\n";
         }
 
         if (iConfig.existsAs<edm::InputTag>("vertexAssociations")) {
             playback_ = true;
             vertexAssociationsToken_ = iC.consumes<edm::ValueMap<pat::VertexAssociation> >(iConfig.getParameter<edm::InputTag>("vertexAssociations"));
         }
         if (iConfig.existsAs<edm::InputTag>("vertices")) { // vertex have been specified, so run on the fly
             playback_ = false;
             verticesToken_ =  iC.consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("vertices"));
             // ------ MODE ------------------
             useTracks_ = iConfig.getParameter<bool>("useTracks");
             // ------ CUTS (fully optional) ------------------
         }
         assoSelector_ = reco::modules::make<pat::VertexAssociationSelector>(iConfig);
     } else {
         enabled_ = false;
     }
 }

Member Function Documentation

pat::VertexAssociation pat::helper::VertexingHelper::associate ( const reco::Candidate & c ) const

private

Try to associated this candidate to a vertex. If no association is found passing all cuts, return a null association

Definition at line 55 of file VertexingHelper.cc.

References TrackValidation_cff::association, assoSelector_, TransientTrackBuilder::build(), end, Exception, getTrack_(), edm::RefToBase< T >::isNull(), edm::ESHandleBase::isValid(), playback_, pat::VertexAssociation::setDistances(), reco::TransientTrack::trajectoryStateClosestToPoint(), groupFilesInBlocks::tt, ttBuilder_, useTracks_, reco::Candidate::vertex(), vertexHandle_, and extraflags_cff::vtx.

                                                                 {
     if (playback_) throw cms::Exception("Configuration") << "VertexingHelper: if this module was configured to read associations from the event," <<
                                                             " you must use 'operator()' passing a candidate ref, and not 'associate()' directly!\n";
 
     reco::VertexCollection::const_iterator vtx, end;
     size_t ivtx;
     reco::TrackBaseRef tk;
     reco::TransientTrack tt;
     if (useTracks_) {
         if (!ttBuilder_.isValid()) throw cms::Exception("Configuration") << "VertexingHelper: If you use 'useTracks', you must call newEvent(iEvent,iSetup)!\n";
         tk = getTrack_(c);
         if (tk.isNull()) return pat::VertexAssociation();
         tt = ttBuilder_->build(*tk);
     }
     for (vtx = vertexHandle_->begin(), end = vertexHandle_->end(), ivtx = 0; vtx != end; ++vtx, ++ivtx) {
         pat::VertexAssociation association(reco::VertexRef(vertexHandle_, ivtx), tk);
         if (useTracks_ == false) {
             association.setDistances(c.vertex(), vtx->position(), vtx->error());
         } else {
             GlobalPoint vtxGP(vtx->x(), vtx->y(), vtx->z()); // need to convert XYZPoint to GlobalPoint
             TrajectoryStateClosestToPoint tscp = tt.trajectoryStateClosestToPoint(vtxGP);
             GlobalPoint          trackPos = tscp.theState().position();
             AlgebraicSymMatrix33 trackErr = tscp.theState().cartesianError().matrix().Sub<AlgebraicSymMatrix33>(0,0);
             association.setDistances(trackPos, vtx->position(), trackErr + vtx->error());
         }
         if (assoSelector_(association)) return association;
     }
     return pat::VertexAssociation();
 }

bool pat::helper::VertexingHelper::enabled ( ) const

inline

returns true if this was given a non dummy configuration

Definition at line 52 of file VertexingHelper.h.

References GeneralSetup::setup().

Referenced by pat::PATGenericParticleProducer::produce().

52 { return enabled_; }

pat::helper::VertexingHelper::enabled_

bool enabled_

true if it has non null configuration

Definition: VertexingHelper.h:68

reco::TrackBaseRef pat::helper::VertexingHelper::getTrack_ ( const reco::Candidate & c ) const

private

Get out the track from the Candidate / RecoCandidate / PFCandidate.

Definition at line 85 of file VertexingHelper.cc.

References reco::RecoCandidate::bestTrackRef(), EnergyCorrector::c, and reco::PFCandidate::trackRef().

Referenced by associate().

                                                                                   {
     const reco::RecoCandidate   *rc  = dynamic_cast<const reco::RecoCandidate *>(&c);
     if (rc  != nullptr)  { return rc->bestTrackRef(); }
     const reco::PFCandidate *pfc = dynamic_cast<const reco::PFCandidate *>(&c);
     if (pfc != nullptr) { return reco::TrackBaseRef(pfc->trackRef()); }
 
     return reco::TrackBaseRef();
 }

void pat::helper::VertexingHelper::newEvent ( const edm::Event & event )

To be called for each new event, reads in the vertex collection.

Definition at line 39 of file VertexingHelper.cc.

References edm::Event::getByToken(), playback_, vertexAssociationsToken_, vertexAssoMap_, vertexHandle_, and verticesToken_.

Referenced by newEvent(), pat::PATVertexAssociationProducer::produce(), and pat::PATGenericParticleProducer::produce().

                                                          {
     if (playback_) {
         iEvent.getByToken(vertexAssociationsToken_, vertexAssoMap_);
     } else {
         iEvent.getByToken(verticesToken_, vertexHandle_);
     }
 }

void pat::helper::VertexingHelper::newEvent	(	const edm::Event &	event,
		const edm::EventSetup &	setup
	)

To be called for each new event, reads in the vertex collection and the tracking info You need this if 'useTrack' is true

Definition at line 48 of file VertexingHelper.cc.

References edm::EventSetup::get(), newEvent(), playback_, and ttBuilder_.

                                                                                        {
     newEvent(iEvent);
     if (!playback_) iSetup.get<TransientTrackRecord>().get("TransientTrackBuilder", ttBuilder_);
 }

template<typename AnyCandRef >

pat::VertexAssociation pat::helper::VertexingHelper::operator() ( const AnyCandRef & cand ) const

Return true if this candidate is associated to a valid vertex AnyCandRef should be a Ref<>, RefToBase<> or Ptr to a Candidate object

Definition at line 98 of file VertexingHelper.h.

References trackingPlots::assoc.

     {
         if (playback_) {
             const pat::VertexAssociation &assoc = (*vertexAssoMap_)[cand];
             return assoSelector_(assoc) ? assoc : pat::VertexAssociation();
         } else {
             return associate( *cand );
         }
 
     }