#include <TrackWithVertexSelector.h>

Public Member Functions
void	init (const edm::Event &event, const edm::EventSetup &)

void	init (const edm::Event &event)

bool	operator() (const reco::Track &t) const

bool	operator() (const reco::Track &t, const edm::Event &iEvent)

bool	operator() (const reco::TrackRef &t) const

bool	operator() (const reco::TrackRef &t, const edm::Event &iEvent)

bool	testTrack (const reco::Track &t) const

bool	testTrack (const reco::TrackRef &t) const

bool	testVertices (const reco::Track &t, const reco::VertexCollection &vtxs) const

bool	testVertices (const reco::TrackRef &t, const reco::VertexCollection &vtxs) const

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

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

	~TrackWithVertexSelector ()

Private Types
typedef math::XYZPoint	Point

Private Attributes
double	d0Max_

double	dzMax_

double	etaMax_

double	etaMin_

double	normalizedChi2_

double	nSigmaDtVertex_

uint32_t	numberOfLostHits_

uint32_t	numberOfValidHits_

uint32_t	numberOfValidPixelHits_

uint32_t	nVertices_

double	ptErrorCut_

double	ptMax_

double	ptMin_

std::string	quality_

double	rhoVtx_

edm::ValueMap< float > const *	timeresoscoll_ = 0

edm::EDGetTokenT< edm::ValueMap< float > >	timeResosToken_

edm::ValueMap< float > const *	timescoll_ = 0

edm::EDGetTokenT< edm::ValueMap< float > >	timesToken_

reco::VertexCollection const *	vcoll_ = 0

edm::EDGetTokenT< reco::VertexCollection >	vertexToken_

bool	vtxFallback_

double	zetaVtx_

Detailed Description

Definition at line 18 of file TrackWithVertexSelector.h.

Member Typedef Documentation

typedef math::XYZPoint TrackWithVertexSelector::Point

private

Definition at line 61 of file TrackWithVertexSelector.h.

Constructor & Destructor Documentation

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

inlineexplicit

Definition at line 20 of file TrackWithVertexSelector.h.

References ~TrackWithVertexSelector().

20 :

21 TrackWithVertexSelector(iConfig, iC) {}

TrackWithVertexSelector::TrackWithVertexSelector

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

Definition: TrackWithVertexSelector.h:20

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

explicit

Definition at line 11 of file TrackWithVertexSelector.cc.

                                                                                                           :
   numberOfValidHits_(iConfig.getParameter<uint32_t>("numberOfValidHits")),
   numberOfValidPixelHits_(iConfig.getParameter<uint32_t>("numberOfValidPixelHits")),
   numberOfLostHits_(iConfig.getParameter<uint32_t>("numberOfLostHits")),
   normalizedChi2_(iConfig.getParameter<double>("normalizedChi2")),
   ptMin_(iConfig.getParameter<double>("ptMin")),
   ptMax_(iConfig.getParameter<double>("ptMax")),
   etaMin_(iConfig.getParameter<double>("etaMin")),
   etaMax_(iConfig.getParameter<double>("etaMax")),
   dzMax_(iConfig.getParameter<double>("dzMax")),
   d0Max_(iConfig.getParameter<double>("d0Max")),
   ptErrorCut_(iConfig.getParameter<double>("ptErrorCut")),
   quality_(iConfig.getParameter<std::string>("quality")),
   nVertices_(iConfig.getParameter<bool>("useVtx") ? iConfig.getParameter<uint32_t>("nVertices") : 0),
   vertexToken_(iC.consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("vertexTag"))),
   timesToken_(iC.consumes<edm::ValueMap<float> >(iConfig.getParameter<edm::InputTag>("timesTag"))),
   timeResosToken_(iC.consumes<edm::ValueMap<float> >(iConfig.getParameter<edm::InputTag>("timeResosTag"))),
   vtxFallback_(iConfig.getParameter<bool>("vtxFallback")),
   zetaVtx_(iConfig.getParameter<double>("zetaVtx")),
   rhoVtx_(iConfig.getParameter<double>("rhoVtx")),
   nSigmaDtVertex_(iConfig.getParameter<double>("nSigmaDtVertex")) {
 }

TrackWithVertexSelector::~TrackWithVertexSelector ( )

Definition at line 34 of file TrackWithVertexSelector.cc.

Referenced by TrackWithVertexSelector().

34 { }

Member Function Documentation

void TrackWithVertexSelector::init	(	const edm::Event &	event,
		const edm::EventSetup &
	)

inline

Definition at line 25 of file TrackWithVertexSelector.h.

References init(), operator()(), and protons_cff::t.

Referenced by init(), and operator()().

25 {init(event);}

TrackWithVertexSelector::init

void init(const edm::Event &event, const edm::EventSetup &)

Definition: TrackWithVertexSelector.h:25

void TrackWithVertexSelector::init ( const edm::Event & event )

Definition at line 37 of file TrackWithVertexSelector.cc.

References edm::HandleBase::isValid(), edm::Handle< T >::product(), timeresoscoll_, timeResosToken_, timescoll_, timesToken_, vcoll_, and vertexToken_.

                                                          {
     edm::Handle<reco::VertexCollection> hVtx;
     event.getByToken(vertexToken_, hVtx);
     vcoll_ = hVtx.product();
 
     edm::Handle<edm::ValueMap<float> > hTimes;
     event.getByToken(timesToken_, hTimes);
     timescoll_ = hTimes.isValid() ? hTimes.product() : nullptr;
 
     edm::Handle<edm::ValueMap<float> > hTimeResos;
     event.getByToken(timeResosToken_, hTimeResos);
     timeresoscoll_ = hTimeResos.isValid() ? hTimeResos.product() : nullptr;
 }

bool TrackWithVertexSelector::operator() ( const reco::Track & t ) const

Definition at line 126 of file TrackWithVertexSelector.cc.

References nVertices_, testTrack(), testVertices(), and vcoll_.

Referenced by init(), and operator()().

                                                                  {
   if (!testTrack(t)) return false;
   if (nVertices_ == 0) return true;
   return testVertices(t, *vcoll_);
 }

bool TrackWithVertexSelector::operator()	(	const reco::Track &	t,
		const edm::Event &	iEvent
	)

inline

Definition at line 29 of file TrackWithVertexSelector.h.

References init(), operator()(), and protons_cff::t.

                                                                   {
          init(iEvent); return (*this)(t);   
       }

bool TrackWithVertexSelector::operator() ( const reco::TrackRef & t ) const

Definition at line 132 of file TrackWithVertexSelector.cc.

References nVertices_, testTrack(), testVertices(), and vcoll_.

                                                                        {
   if (!testTrack(tref)) return false;
   if (nVertices_ == 0) return true;
   return testVertices(tref, *vcoll_);
 }

bool TrackWithVertexSelector::operator()	(	const reco::TrackRef &	t,
		const edm::Event &	iEvent
	)

inline

Definition at line 34 of file TrackWithVertexSelector.h.

References init(), protons_cff::t, testTrack(), and testVertices().

                                                                      {
          init(iEvent); return (*this)(t);   
       }

bool TrackWithVertexSelector::testTrack ( const reco::Track & t ) const

Definition at line 51 of file TrackWithVertexSelector.cc.

References funct::abs(), reco::TrackBase::d0(), d0Max_, reco::TrackBase::dz(), dzMax_, reco::TrackBase::eta(), etaMax_, etaMin_, reco::TrackBase::hitPattern(), SiStripPI::max, reco::TrackBase::normalizedChi2(), normalizedChi2_, reco::TrackBase::numberOfLostHits(), numberOfLostHits_, reco::TrackBase::numberOfValidHits(), numberOfValidHits_, reco::HitPattern::numberOfValidPixelHits(), numberOfValidPixelHits_, reco::TrackBase::pt(), reco::TrackBase::ptError(), ptErrorCut_, ptMax_, ptMin_, reco::TrackBase::quality(), quality_, and reco::TrackBase::qualityByName().

Referenced by operator()(), and testTrack().

                                                                 {
   using std::abs;
   if ((t.numberOfValidHits() >= numberOfValidHits_) &&
       (static_cast<unsigned int>(t.hitPattern().numberOfValidPixelHits()) >= numberOfValidPixelHits_) &&
       (t.numberOfLostHits() <= numberOfLostHits_) &&
       (t.normalizedChi2()    <= normalizedChi2_) &&
       (t.ptError()/t.pt()*std::max(1.,t.normalizedChi2()) <= ptErrorCut_) &&
       (t.quality(t.qualityByName(quality_))) &&
       (t.pt()              >= ptMin_)      &&
       (t.pt()              <= ptMax_)      &&
       (abs(t.eta())   <= etaMax_)     &&
       (abs(t.eta())   >= etaMin_)     &&
       (abs(t.dz())    <= dzMax_)      &&
       (abs(t.d0())    <= d0Max_)  ) {
     return true;
   }
   return false;
 }

bool TrackWithVertexSelector::testTrack ( const reco::TrackRef & t ) const

Definition at line 70 of file TrackWithVertexSelector.cc.

References testTrack().

                                                                       {
   return testTrack(*tref);
 }

bool TrackWithVertexSelector::testVertices	(	const reco::Track &	t,
		const reco::VertexCollection &	vtxs
	)		const

Definition at line 74 of file TrackWithVertexSelector.cc.

References funct::abs(), reco::TrackBase::dxy(), reco::TrackBase::dz(), nVertices_, convertSQLiteXML::ok, rhoVtx_, reco::TrackBase::vertex(), vtxFallback_, and zetaVtx_.

Referenced by operator()().

                                                                                                      {
   bool ok = false;
   if (!vtxs.empty()) {
     unsigned int tested = 1;
     for (reco::VertexCollection::const_iterator it = vtxs.begin(), ed = vtxs.end();
      it != ed; ++it) {
       if ( (std::abs(t.dxy(it->position())) < rhoVtx_) &&
            (std::abs(t.dz(it->position())) < zetaVtx_)  ) {
         ok = true; break;
       }
       if (tested++ >= nVertices_) break;
     }
   } else if (vtxFallback_) {
     return ( (std::abs(t.vertex().z()) < 15.9) && (t.vertex().Rho() < 0.2) );
   }
   return ok;
 }

bool TrackWithVertexSelector::testVertices	(	const reco::TrackRef &	t,
		const reco::VertexCollection &	vtxs
	)		const

Definition at line 92 of file TrackWithVertexSelector.cc.

References funct::abs(), MillePedeFileConverter_cfg::e, edm::isNotFinite(), nSigmaDtVertex_, nVertices_, convertSQLiteXML::ok, rhoVtx_, mathSSE::sqrt(), protons_cff::t, protons_cff::time, timeresoscoll_, timescoll_, vtxFallback_, and zetaVtx_.

                                                                                                            {
   const auto& t = *tref;
   const bool timeAvailable = timescoll_ != nullptr && timeresoscoll_ != nullptr;
   bool ok = false;
   if (!vtxs.empty()) {
     unsigned int tested = 1;
     for (reco::VertexCollection::const_iterator it = vtxs.begin(), ed = vtxs.end();
      it != ed; ++it) {
       const bool useTime = timeAvailable && it->t() != 0.;
       float time = useTime ? (*timescoll_)[tref] : -1.f;
       float timeReso = useTime ? (*timeresoscoll_)[tref] : -1.f;
       timeReso = ( timeReso > 1e-6 ? timeReso : fakeBeamSpotTimeWidth );
 
       if( edm::isNotFinite(time) ) {
     time = 0.0;
     timeReso = 2.0*fakeBeamSpotTimeWidth;
       }
 
       const double vtxSigmaT2 = it->tError() * it->tError();
       const double vtxTrackErr = std::sqrt( vtxSigmaT2 + timeReso*timeReso );
 
       if ( (std::abs(t.dxy(it->position())) < rhoVtx_) &&
            (std::abs(t.dz(it->position())) < zetaVtx_) &&
        ( !useTime || (std::abs(time - it->t())/vtxTrackErr < nSigmaDtVertex_) ) ) {
         ok = true; break;
       }
       if (tested++ >= nVertices_) break;
     }
   } else if (vtxFallback_) {
     return ( (std::abs(t.vertex().z()) < 15.9) && (t.vertex().Rho() < 0.2) );
   }
   return ok;
 }

Member Data Documentation

double TrackWithVertexSelector::d0Max_

private

Definition at line 48 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::dzMax_

private

Definition at line 48 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::etaMax_

private

Definition at line 47 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::etaMin_

private

Definition at line 47 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::normalizedChi2_

private

Definition at line 46 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::nSigmaDtVertex_

private

Definition at line 56 of file TrackWithVertexSelector.h.

Referenced by testVertices().

uint32_t TrackWithVertexSelector::numberOfLostHits_

private

Definition at line 45 of file TrackWithVertexSelector.h.

Referenced by testTrack().

uint32_t TrackWithVertexSelector::numberOfValidHits_

private

Definition at line 43 of file TrackWithVertexSelector.h.

Referenced by testTrack().

uint32_t TrackWithVertexSelector::numberOfValidPixelHits_

private

Definition at line 44 of file TrackWithVertexSelector.h.

Referenced by testTrack().

uint32_t TrackWithVertexSelector::nVertices_

private

Definition at line 52 of file TrackWithVertexSelector.h.

Referenced by operator()(), and testVertices().

double TrackWithVertexSelector::ptErrorCut_

private

Definition at line 49 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::ptMax_

private

Definition at line 47 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::ptMin_

private

Definition at line 47 of file TrackWithVertexSelector.h.

Referenced by testTrack().

std::string TrackWithVertexSelector::quality_

private

Definition at line 50 of file TrackWithVertexSelector.h.

Referenced by testTrack().

double TrackWithVertexSelector::rhoVtx_

private

Definition at line 56 of file TrackWithVertexSelector.h.

Referenced by testVertices().

edm::ValueMap<float> const* TrackWithVertexSelector::timeresoscoll_ = 0

private

Definition at line 60 of file TrackWithVertexSelector.h.

Referenced by init(), and testVertices().

edm::EDGetTokenT<edm::ValueMap<float> > TrackWithVertexSelector::timeResosToken_

private

Definition at line 54 of file TrackWithVertexSelector.h.

Referenced by init().

edm::ValueMap<float> const* TrackWithVertexSelector::timescoll_ = 0

private

Definition at line 59 of file TrackWithVertexSelector.h.

Referenced by init(), and testVertices().

edm::EDGetTokenT<edm::ValueMap<float> > TrackWithVertexSelector::timesToken_

private

Definition at line 54 of file TrackWithVertexSelector.h.

Referenced by init().

reco::VertexCollection const* TrackWithVertexSelector::vcoll_ = 0

private

Definition at line 58 of file TrackWithVertexSelector.h.

Referenced by init(), and operator()().

edm::EDGetTokenT<reco::VertexCollection> TrackWithVertexSelector::vertexToken_

private

Definition at line 53 of file TrackWithVertexSelector.h.

Referenced by init().

bool TrackWithVertexSelector::vtxFallback_

private

Definition at line 55 of file TrackWithVertexSelector.h.

Referenced by testVertices().

double TrackWithVertexSelector::zetaVtx_

private

Definition at line 56 of file TrackWithVertexSelector.h.

Referenced by testVertices().

Public Member Functions

Private Types

Private Attributes

Detailed Description

Member Typedef Documentation

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation