TrackAssociatorByPositionImpl Class Reference

#include <TrackAssociatorByPositionImpl.h>

Inheritance diagram for TrackAssociatorByPositionImpl:

Public Types
enum	Method { Method::chi2, Method::dist, Method::momdr, Method::posdr }
typedef std::vector< SimHitTPPair >	SimHitTPAssociationList
typedef std::pair< TrackingParticleRef, TrackPSimHitRef >	SimHitTPPair

Public Member Functions
reco::RecoToSimCollection	associateRecoToSim (const edm::RefToBaseVector< reco::Track > &, const edm::RefVector< TrackingParticleCollection > &) const override
	compare reco to sim the handle of reco::Track and TrackingParticle collections More...
reco::SimToRecoCollection	associateSimToReco (const edm::RefToBaseVector< reco::Track > &, const edm::RefVector< TrackingParticleCollection > &) const override
	compare reco to sim the handle of reco::Track and TrackingParticle collections More...
	TrackAssociatorByPositionImpl (const TrackingGeometry *geo, const Propagator *prop, const SimHitTPAssociationList *assocList, double qMinCut, double qCut, double positionMinimumDistance, Method method, bool minIfNoMatch, bool considerAllSimHits)
Public Member Functions inherited from reco::TrackToTrackingParticleAssociatorBaseImpl
virtual reco::RecoToSimCollection	associateRecoToSim (const edm::Handle< edm::View< reco::Track >> &tCH, const edm::Handle< TrackingParticleCollection > &tPCH) const
virtual reco::RecoToSimCollectionSeed	associateRecoToSim (const edm::Handle< edm::View< TrajectorySeed >> &, const edm::Handle< TrackingParticleCollection > &) const
virtual reco::RecoToSimCollectionTCandidate	associateRecoToSim (const edm::Handle< TrackCandidateCollection > &, const edm::Handle< TrackingParticleCollection > &) const
virtual reco::SimToRecoCollection	associateSimToReco (const edm::Handle< edm::View< reco::Track >> &tCH, const edm::Handle< TrackingParticleCollection > &tPCH) const
virtual reco::SimToRecoCollectionSeed	associateSimToReco (const edm::Handle< edm::View< TrajectorySeed >> &, const edm::Handle< TrackingParticleCollection > &) const
virtual reco::SimToRecoCollectionTCandidate	associateSimToReco (const edm::Handle< TrackCandidateCollection > &, const edm::Handle< TrackingParticleCollection > &) const
	TrackToTrackingParticleAssociatorBaseImpl ()
	Constructor. More...
virtual	~TrackToTrackingParticleAssociatorBaseImpl ()
	Destructor. More...

Private Member Functions
FreeTrajectoryState	getState (const reco::Track &) const
TrajectoryStateOnSurface	getState (const TrackingParticleRef &, const SimHitTPAssociationList &simHitsTPAssoc) const
double	quality (const TrajectoryStateOnSurface &, const TrajectoryStateOnSurface &) const

Private Attributes
bool	theConsiderAllSimHits
const TrackingGeometry *	theGeometry
Method	theMethod
bool	theMinIfNoMatch
double	thePositionMinimumDistance
const Propagator *	thePropagator
double	theQCut
double	theQminCut
const SimHitTPAssociationList *	theSimHitsTPAssoc

Detailed Description

Class that performs the association of reco::Tracks and TrackingParticles based on position in muon detector

Author

vlimant

Definition at line 30 of file TrackAssociatorByPositionImpl.h.

Member Typedef Documentation

SimHitTPAssociationList

typedef std::vector<SimHitTPPair> TrackAssociatorByPositionImpl::SimHitTPAssociationList

Definition at line 33 of file TrackAssociatorByPositionImpl.h.

SimHitTPPair

typedef std::pair<TrackingParticleRef, TrackPSimHitRef> TrackAssociatorByPositionImpl::SimHitTPPair

Definition at line 32 of file TrackAssociatorByPositionImpl.h.

Member Enumeration Documentation

Method

enum TrackAssociatorByPositionImpl::Method

strong

Enumerator
chi2
dist
momdr
posdr

Definition at line 34 of file TrackAssociatorByPositionImpl.h.

{ chi2, dist, momdr, posdr };

Constructor & Destructor Documentation

TrackAssociatorByPositionImpl()

TrackAssociatorByPositionImpl::TrackAssociatorByPositionImpl ( const TrackingGeometry *  geo, const Propagator *  prop, const SimHitTPAssociationList *  assocList, double  qMinCut, double  qCut, double  positionMinimumDistance, Method  method, bool  minIfNoMatch, bool  considerAllSimHits  )

inline

Definition at line 36 of file TrackAssociatorByPositionImpl.h.

      : theGeometry(geo),
        thePropagator(prop),
        theSimHitsTPAssoc(assocList),
        theQminCut(qMinCut),
        theQCut(qCut),
        thePositionMinimumDistance(positionMinimumDistance),
        theMethod(method),
        theMinIfNoMatch(minIfNoMatch),
        theConsiderAllSimHits(considerAllSimHits) {}

Member Function Documentation

associateRecoToSim()

RecoToSimCollection TrackAssociatorByPositionImpl::associateRecoToSim ( const edm::RefToBaseVector< reco::Track > &  tCH, const edm::RefVector< TrackingParticleCollection > &  tPCH  ) const

overridevirtual

compare reco to sim the handle of reco::Track and TrackingParticle collections

Implements reco::TrackToTrackingParticleAssociatorBaseImpl.

Definition at line 117 of file TrackAssociatorByPositionImpl.cc.

                                                                                                          {
  RecoToSimCollection outputCollection;
  //for each reco track find a matching tracking particle
  std::pair<unsigned int, unsigned int> minPair;
  const double dQmin_default = 1542543;
  double dQmin = dQmin_default;
 
  //cdj edm::Handle<SimHitTPAssociationProducer::SimHitTPAssociationList> simHitsTPAssoc;
 
  //warning: make sure the TP collection used in the map is the same used in the associator!
  //e->getByLabel(_simHitTpMapTag,simHitsTPAssoc);
 
  for (unsigned int Ti = 0; Ti != tCH.size(); ++Ti) {
    //initial state (initial OR inner OR outter)
    FreeTrajectoryState iState = getState(*(tCH)[Ti]);
 
    bool atLeastOne = false;
    //    for each tracking particle, find a state position and the plane to propagate the track to.
    for (unsigned int TPi = 0; TPi != tPCH.size(); ++TPi) {
      //get a state in the muon system
      TrajectoryStateOnSurface simReferenceState = getState((tPCH)[TPi], *theSimHitsTPAssoc);
      if (!simReferenceState.isValid())
        continue;
 
      //propagate the TRACK to the surface
      TrajectoryStateOnSurface trackReferenceState = thePropagator->propagate(iState, simReferenceState.surface());
      if (!trackReferenceState.isValid())
        continue;
 
      //comparison
      double dQ = quality(trackReferenceState, simReferenceState);
      if (dQ < theQCut) {
        atLeastOne = true;
        outputCollection.insert(tCH[Ti],
                                std::make_pair(tPCH[TPi], -dQ));  //association map with quality, is order greater-first
        edm::LogVerbatim("TrackAssociatorByPositionImpl")
            << "track number: " << Ti << " associated with dQ: " << dQ << " to TrackingParticle number: " << TPi;
      }
      if (dQ < dQmin) {
        dQmin = dQ;
        minPair = std::make_pair(Ti, TPi);
      }
    }  //loop over tracking particles
    if (theMinIfNoMatch && !atLeastOne && dQmin != dQmin_default) {
      outputCollection.insert(tCH[minPair.first], std::make_pair(tPCH[minPair.second], -dQmin));
    }
  }  //loop over tracks
  outputCollection.post_insert();
  return outputCollection;
}

References edm::AssociationMap< Tag >::insert(), TrajectoryStateOnSurface::isValid(), edm::AssociationMap< Tag >::post_insert(), qcdUeDQM_cfi::quality, edm::RefToBaseVector< T >::size(), edm::RefVector< C, T, F >::size(), and TrajectoryStateOnSurface::surface().

associateSimToReco()

SimToRecoCollection TrackAssociatorByPositionImpl::associateSimToReco ( const edm::RefToBaseVector< reco::Track > &  tCH, const edm::RefVector< TrackingParticleCollection > &  tPCH  ) const

overridevirtual

compare reco to sim the handle of reco::Track and TrackingParticle collections

Implements reco::TrackToTrackingParticleAssociatorBaseImpl.

Definition at line 169 of file TrackAssociatorByPositionImpl.cc.

                                                                                                          {
  SimToRecoCollection outputCollection;
  //for each tracking particle, find matching tracks.
 
  std::pair<unsigned int, unsigned int> minPair;
  const double dQmin_default = 1542543;
  double dQmin = dQmin_default;
 
  //edm::Handle<SimHitTPAssociationProducer::SimHitTPAssociationList> simHitsTPAssoc;
 
  //warning: make sure the TP collection used in the map is the same used in the associator!
  //e->getByLabel(_simHitTpMapTag,simHitsTPAssoc);
 
  for (unsigned int TPi = 0; TPi != tPCH.size(); ++TPi) {
    //get a state in the muon system
    TrajectoryStateOnSurface simReferenceState = getState((tPCH)[TPi], *theSimHitsTPAssoc);
 
    if (!simReferenceState.isValid())
      continue;
    bool atLeastOne = false;
    //  propagate every track from any state (initial, inner, outter) to the surface
    //  and make the position test
    for (unsigned int Ti = 0; Ti != tCH.size(); ++Ti) {
      //initial state
      FreeTrajectoryState iState = getState(*(tCH)[Ti]);
 
      //propagation to surface
      TrajectoryStateOnSurface trackReferenceState = thePropagator->propagate(iState, simReferenceState.surface());
      if (!trackReferenceState.isValid())
        continue;
 
      //comparison
      double dQ = quality(trackReferenceState, simReferenceState);
      if (dQ < theQCut) {
        atLeastOne = true;
        outputCollection.insert(tPCH[TPi],
                                std::make_pair(tCH[Ti], -dQ));  //association map with quality, is order greater-first
        edm::LogVerbatim("TrackAssociatorByPositionImpl")
            << "TrackingParticle number: " << TPi << " associated with dQ: " << dQ << " to track number: " << Ti;
      }
      if (dQ < dQmin) {
        dQmin = dQ;
        minPair = std::make_pair(TPi, Ti);
      }
    }  //loop over tracks
    if (theMinIfNoMatch && !atLeastOne && dQmin != dQmin_default) {
      outputCollection.insert(tPCH[minPair.first], std::make_pair(tCH[minPair.second], -dQmin));
    }
  }  //loop over tracking particles
 
  outputCollection.post_insert();
  return outputCollection;
}

References edm::AssociationMap< Tag >::insert(), TrajectoryStateOnSurface::isValid(), edm::AssociationMap< Tag >::post_insert(), qcdUeDQM_cfi::quality, edm::RefToBaseVector< T >::size(), edm::RefVector< C, T, F >::size(), and TrajectoryStateOnSurface::surface().

getState() [1/2]

FreeTrajectoryState TrackAssociatorByPositionImpl::getState ( const reco::Track &  track ) const

private

Definition at line 76 of file TrackAssociatorByPositionImpl.cc.

                                                                                        {
  //may be you want to do more than that if track does not go to IP
  return trajectoryStateTransform::initialFreeState(track, thePropagator->magneticField());
}

References trajectoryStateTransform::initialFreeState(), and HLT_2018_cff::track.

getState() [2/2]

TrajectoryStateOnSurface TrackAssociatorByPositionImpl::getState ( const TrackingParticleRef &  st, const SimHitTPAssociationList &  simHitsTPAssoc  ) const

private

Definition at line 13 of file TrackAssociatorByPositionImpl.cc.

                                                                                                                      {
  using SimHitTPPair = std::pair<TrackingParticleRef, TrackPSimHitRef>;
  SimHitTPPair clusterTPpairWithDummyTP(st, TrackPSimHitRef());  //SimHit is dummy:
  // sorting only the cluster is needed
  auto range = std::equal_range(
      simHitsTPAssoc.begin(),
      simHitsTPAssoc.end(),
      clusterTPpairWithDummyTP,
      [](const SimHitTPPair& iLHS, const SimHitTPPair& iRHS) -> bool { return iLHS.first.key() > iRHS.first.key(); });
 
  //  TrackingParticle* simtrack = const_cast<TrackingParticle*>(&st);
  //loop over PSimHits
  const PSimHit* psimhit = nullptr;
  const BoundPlane* plane = nullptr;
  double dLim = thePositionMinimumDistance;
 
  //    look for the further most hit beyond a certain limit
  auto start = range.first;
  auto end = range.second;
  LogDebug("TrackAssociatorByPositionImpl") << range.second - range.first << " PSimHits.";
 
  unsigned int count = 0;
  for (auto ip = start; ip != end; ++ip) {
    TrackPSimHitRef psit = ip->second;
 
    //get the detid
    DetId dd(psit->detUnitId());
 
    if (!theConsiderAllSimHits && dd.det() != DetId::Tracker)
      continue;
 
    LogDebug("TrackAssociatorByPositionImpl") << count++ << "] PSimHit on: " << dd.rawId();
    //get the surface from the global geometry
    const GeomDet* gd = theGeometry->idToDet(dd);
    if (!gd) {
      edm::LogError("TrackAssociatorByPositionImpl") << "no geomdet for: " << dd.rawId() << ". will skip.";
      continue;
    }
    double d = gd->surface().toGlobal(psit->localPosition()).mag();
    if (d > dLim) {
      dLim = d;
      psimhit = &(*psit);
      plane = &gd->surface();
    }
  }
 
  if (psimhit && plane) {
    //build a trajectorystate on this surface
    SurfaceSideDefinition::SurfaceSide surfaceside = SurfaceSideDefinition::atCenterOfSurface;
    GlobalPoint initialPoint = plane->toGlobal(psimhit->localPosition());
    GlobalVector initialMomentum = plane->toGlobal(psimhit->momentumAtEntry());
    int initialCharge = (psimhit->particleType() > 0) ? -1 : 1;
    CartesianTrajectoryError initialCartesianErrors;  //no error at initial state
    const GlobalTrajectoryParameters initialParameters(
        initialPoint, initialMomentum, initialCharge, thePropagator->magneticField());
    return TrajectoryStateOnSurface(initialParameters, initialCartesianErrors, *plane, surfaceside);
  } else {
    //    edm::LogError("TrackAssociatorByPositionImpl")<<"no corresponding PSimHit for a tracking particle. will fail.";
    return TrajectoryStateOnSurface();
  }
}

References SurfaceSideDefinition::atCenterOfSurface, KineDebug3::count(), ztail::d, createTree::dd, end, LogDebug, mag(), FastTimerService_cff::range, GeomDet::surface(), Surface::toGlobal(), and DetId::Tracker.

quality()

double TrackAssociatorByPositionImpl::quality ( const TrajectoryStateOnSurface &  tr, const TrajectoryStateOnSurface &  sim  ) const

private

Definition at line 81 of file TrackAssociatorByPositionImpl.cc.

                                                                                         {
  switch (theMethod) {
    case Method::chi2: {
      AlgebraicVector5 v(tr.localParameters().vector() - sim.localParameters().vector());
      AlgebraicSymMatrix55 m(tr.localError().matrix());
      int ierr = !m.Invert();
      if (ierr != 0)
        edm::LogInfo("TrackAssociatorByPositionImpl") << "error inverting the error matrix:\n" << m;
      double est = ROOT::Math::Similarity(v, m);
      return est;
      break;
    }
    case Method::dist: {
      return (tr.globalPosition() - sim.globalPosition()).mag();
      break;
    }
    case Method::momdr: {
      return (deltaR<double>(tr.globalDirection().eta(),
                             tr.globalDirection().phi(),
                             sim.globalDirection().eta(),
                             sim.globalDirection().phi()));
      break;
    }
    case Method::posdr: {
      return (deltaR<double>(
          tr.globalPosition().eta(), tr.globalPosition().phi(), sim.globalPosition().eta(), sim.globalPosition().phi()));
      break;
    }
  }
  //should never be reached
  edm::LogError("TrackAssociatorByPositionImpl")
      << "option: " << static_cast<int>(theMethod) << " has not been recognized. association has no meaning.";
  return -1;
}

References hltPixelTracks_cff::chi2, PV3DBase< T, PVType, FrameType >::eta(), TrajectoryStateOnSurface::globalDirection(), TrajectoryStateOnSurface::globalPosition(), TrajectoryStateOnSurface::localError(), TrajectoryStateOnSurface::localParameters(), visualization-live-secondInstance_cfg::m, mag(), LocalTrajectoryError::matrix(), PV3DBase< T, PVType, FrameType >::phi(), findQualityFiles::v, and LocalTrajectoryParameters::vector().

Member Data Documentation

theConsiderAllSimHits

bool TrackAssociatorByPositionImpl::theConsiderAllSimHits

private

Definition at line 76 of file TrackAssociatorByPositionImpl.h.

theGeometry

const TrackingGeometry* TrackAssociatorByPositionImpl::theGeometry

private

Definition at line 68 of file TrackAssociatorByPositionImpl.h.

theMethod

Method TrackAssociatorByPositionImpl::theMethod

private

Definition at line 74 of file TrackAssociatorByPositionImpl.h.

theMinIfNoMatch

bool TrackAssociatorByPositionImpl::theMinIfNoMatch

private

Definition at line 75 of file TrackAssociatorByPositionImpl.h.

thePositionMinimumDistance

double TrackAssociatorByPositionImpl::thePositionMinimumDistance

private

Definition at line 73 of file TrackAssociatorByPositionImpl.h.

thePropagator

const Propagator* TrackAssociatorByPositionImpl::thePropagator

private

Definition at line 69 of file TrackAssociatorByPositionImpl.h.

theQCut

double TrackAssociatorByPositionImpl::theQCut

private

Definition at line 72 of file TrackAssociatorByPositionImpl.h.

theQminCut

double TrackAssociatorByPositionImpl::theQminCut

private

Definition at line 71 of file TrackAssociatorByPositionImpl.h.

theSimHitsTPAssoc

const SimHitTPAssociationList* TrackAssociatorByPositionImpl::theSimHitsTPAssoc

private

Definition at line 70 of file TrackAssociatorByPositionImpl.h.