TrackExtrapolator Class Reference

#include <RecoTracker/TrackExtrapolator/src/TrackExtrapolator.cc>

Inheritance diagram for TrackExtrapolator:

Public Member Functions
	TrackExtrapolator (const edm::ParameterSet &)
	~TrackExtrapolator () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndRuns () const final

Private Member Functions
void	produce (edm::Event &, const edm::EventSetup &) override
bool	propagateTrackToVolume (const reco::Track &fTrack, const MagneticField &fField, const Propagator &fPropagator, const FiducialVolume &volume, reco::TrackBase::Point &resultPos, reco::TrackBase::Vector &resultMom)
	track quality of the tracks we care about More...

Private Attributes
reco::TrackBase::TrackQuality	trackQuality_
	Input tracks. More...
edm::EDGetTokenT< reco::TrackCollection >	tracksSrc_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

Description: Extrapolates tracks to Calo Face. Migrating this functionality from RecoJets/JetAssociationAlgorithms/JetTracksAssociatorDRCalo.h, which will now essentially be defunct.

Implementation:

Definition at line 57 of file TrackExtrapolator.h.

Constructor & Destructor Documentation

TrackExtrapolator::TrackExtrapolator ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 10 of file TrackExtrapolator.cc.

References Exception, edm::ParameterSet::getParameter(), reco::TrackBase::qualityByName(), AlCaHLTBitMon_QueryRunRegistry::string, trackQuality_, tracksSrc_, and reco::TrackBase::undefQuality.

                                                                   {
  tracksSrc_ = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("trackSrc"));
  trackQuality_ = reco::TrackBase::qualityByName(iConfig.getParameter<std::string>("trackQuality"));
  if (trackQuality_ == reco::TrackBase::undefQuality) {  // we have a problem
    throw cms::Exception("InvalidInput") << "Unknown trackQuality value '"
                                         << iConfig.getParameter<std::string>("trackQuality")
                                         << "'. See possible values in 'reco::TrackBase::qualityByName'";
  }

  produces<std::vector<reco::TrackExtrapolation>>();
}

TrackExtrapolator::~TrackExtrapolator ( )

override

Definition at line 22 of file TrackExtrapolator.cc.

{}

Member Function Documentation

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

overrideprivate

Definition at line 29 of file TrackExtrapolator.cc.

References j2tParametersCALO_cfi::extrapolations, edm::EventSetup::get(), edm::Event::getByToken(), goodTracks_cfi::goodTracks, eostools::move(), propagateTrackToVolume(), edm::Event::put(), StandaloneTrackMonitor_cfi::trackQuality, trackQuality_, tracksSrc_, and DetIdAssociator::volume().

                                                                             {
  // get stuff from Event Setup
  edm::ESHandle<MagneticField> field_h;
  iSetup.get<IdealMagneticFieldRecord>().get(field_h);
  edm::ESHandle<Propagator> propagator_h;
  iSetup.get<TrackingComponentsRecord>().get("SteppingHelixPropagatorAlong", propagator_h);
  edm::ESHandle<DetIdAssociator> ecalDetIdAssociator_h;
  iSetup.get<DetIdAssociatorRecord>().get("EcalDetIdAssociator", ecalDetIdAssociator_h);
  FiducialVolume const& ecalvolume = ecalDetIdAssociator_h->volume();

  // get stuff from Event
  edm::Handle<reco::TrackCollection> tracks_h;
  iEvent.getByToken(tracksSrc_, tracks_h);

  auto extrapolations = std::make_unique<std::vector<reco::TrackExtrapolation>>();

  // Get list of tracks we want to extrapolate
  std::vector<reco::TrackRef> goodTracks;
  for (reco::TrackCollection::const_iterator trkBegin = tracks_h->begin(), trkEnd = tracks_h->end(), itrk = trkBegin;
       itrk != trkEnd;
       ++itrk) {
    reco::TrackBase::TrackQuality trackQuality = reco::TrackBase::TrackQuality(trackQuality_);

    // Cut on track quality
    if (itrk->quality(trackQuality)) {
      goodTracks.push_back(reco::TrackRef(tracks_h, itrk - trkBegin));
    }
  }
  std::vector<reco::TrackBase::Point> vresultPos(1);
  std::vector<reco::TrackBase::Vector> vresultMom(1);

  // Now loop through the list of tracks and extrapolate them
  for (std::vector<reco::TrackRef>::const_iterator trkBegin = goodTracks.begin(),
                                                   trkEnd = goodTracks.end(),
                                                   itrk = trkBegin;
       itrk != trkEnd;
       ++itrk) {
    if (propagateTrackToVolume(**itrk, *field_h, *propagator_h, ecalvolume, vresultPos[0], vresultMom[0])) {
      extrapolations->push_back(reco::TrackExtrapolation(*itrk, vresultPos, vresultMom));
    }
  }
  iEvent.put(std::move(extrapolations));
}

bool TrackExtrapolator::propagateTrackToVolume ( const reco::Track &  fTrack, const MagneticField &  fField, const Propagator &  fPropagator, const FiducialVolume &  volume, reco::TrackBase::Point &  resultPos, reco::TrackBase::Vector &  resultMom  )

private

track quality of the tracks we care about

Propagate a track to a given radius, given the magnetic field and the propagator. Store the resulting position, momentum, and direction.

Definition at line 75 of file TrackExtrapolator.cc.

References Plane::build(), Cylinder::build(), reco::TrackBase::charge(), DEFINE_FWK_MODULE, reco::Track::extra(), TrajectoryStateOnSurface::globalMomentum(), TrajectoryStateOnSurface::globalPosition(), edm::Ref< C, T, F >::isAvailable(), TrajectoryStateOnSurface::isValid(), FiducialVolume::minR(), FiducialVolume::minZ(), reco::Track::outerPx(), reco::Track::outerPy(), reco::Track::outerPz(), reco::Track::outerX(), reco::Track::outerY(), reco::Track::outerZ(), Propagator::propagate(), reco::TrackBase::px(), reco::TrackBase::py(), reco::TrackBase::pz(), reco::btau::trackMomentum, reco::TrackBase::vx(), reco::TrackBase::vy(), reco::TrackBase::vz(), and PV3DBase< T, PVType, FrameType >::z().

Referenced by produce().

                                                                               {
  GlobalPoint trackPosition(fTrack.vx(), fTrack.vy(), fTrack.vz());   // reference point
  GlobalVector trackMomentum(fTrack.px(), fTrack.py(), fTrack.pz());  // reference momentum
  if (fTrack.extra().isAvailable()) {                                 // use outer point information, if available
    trackPosition = GlobalPoint(fTrack.outerX(), fTrack.outerY(), fTrack.outerZ());
    trackMomentum = GlobalVector(fTrack.outerPx(), fTrack.outerPy(), fTrack.outerPz());
  }

  GlobalTrajectoryParameters trackParams(trackPosition, trackMomentum, fTrack.charge(), &fField);
  FreeTrajectoryState trackState(trackParams);

  TrajectoryStateOnSurface propagatedInfo = fPropagator.propagate(
      trackState, *Cylinder::build(volume.minR(), Surface::PositionType(0, 0, 0), Surface::RotationType()));

  // if the track went through either side of the endcaps, repropagate the track
  double minz = volume.minZ();
  if (propagatedInfo.isValid() && propagatedInfo.globalPosition().z() > minz) {
    propagatedInfo =
        fPropagator.propagate(trackState, *Plane::build(Surface::PositionType(0, 0, minz), Surface::RotationType()));

  } else if (propagatedInfo.isValid() && propagatedInfo.globalPosition().z() < -minz) {
    propagatedInfo =
        fPropagator.propagate(trackState, *Plane::build(Surface::PositionType(0, 0, -minz), Surface::RotationType()));
  }

  if (propagatedInfo.isValid()) {
    resultPos = propagatedInfo.globalPosition();
    resultMom = propagatedInfo.globalMomentum();
    return true;
  } else {
    return false;
  }
}

Member Data Documentation

reco::TrackBase::TrackQuality TrackExtrapolator::trackQuality_

private

Input tracks.

Definition at line 68 of file TrackExtrapolator.h.

Referenced by produce(), and TrackExtrapolator().

edm::EDGetTokenT<reco::TrackCollection> TrackExtrapolator::tracksSrc_

private

Definition at line 67 of file TrackExtrapolator.h.

Referenced by produce(), and TrackExtrapolator().