CMS 3D CMS Logo

Public Member Functions | Private Member Functions | Private Attributes

TrackAssociatorByPosition Class Reference

#include <TrackAssociatorByPosition.h>

Inheritance diagram for TrackAssociatorByPosition:
TrackAssociatorBase

List of all members.

Public Member Functions

reco::RecoToSimCollection associateRecoToSim (const edm::RefToBaseVector< reco::Track > &, const edm::RefVector< TrackingParticleCollection > &, const edm::Event *event=0, const edm::EventSetup *setup=0) const
 compare reco to sim the handle of reco::Track and TrackingParticle collections
reco::SimToRecoCollection associateSimToReco (const edm::RefToBaseVector< reco::Track > &, const edm::RefVector< TrackingParticleCollection > &, const edm::Event *event=0, const edm::EventSetup *setup=0) const
 compare reco to sim the handle of reco::Track and TrackingParticle collections
double quality (const TrajectoryStateOnSurface &, const TrajectoryStateOnSurface &) const
 TrackAssociatorByPosition (const edm::ParameterSet &iConfig, const TrackingGeometry *geo, const Propagator *prop)
 Constructor with propagator and PSet.
 ~TrackAssociatorByPosition ()
 Destructor.

Private Member Functions

FreeTrajectoryState getState (const reco::Track &) const
TrajectoryStateOnSurface getState (const TrackingParticle &) const

Private Attributes

bool theConsiderAllSimHits
const TrackingGeometrytheGeometry
unsigned int theMethod
bool theMinIfNoMatch
double thePositionMinimumDistance
const PropagatorthePropagator
double theQCut
double theQminCut

Detailed Description

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

Date:
2010/10/12 12:10:00
Revision:
1.11
Author:
vlimant

Definition at line 30 of file TrackAssociatorByPosition.h.


Constructor & Destructor Documentation

TrackAssociatorByPosition::TrackAssociatorByPosition ( const edm::ParameterSet iConfig,
const TrackingGeometry geo,
const Propagator prop 
) [inline]

Constructor with propagator and PSet.

Definition at line 35 of file TrackAssociatorByPosition.h.

References edm::ParameterSet::getParameter(), AlCaHLTBitMon_QueryRunRegistry::string, theConsiderAllSimHits, theGeometry, theMethod, theMinIfNoMatch, thePositionMinimumDistance, thePropagator, theQCut, and theQminCut.

                                                     {
     theGeometry = geo;
     thePropagator = prop;
     theMinIfNoMatch = iConfig.getParameter<bool>("MinIfNoMatch");
     theQminCut = iConfig.getParameter<double>("QminCut");
     theQCut = iConfig.getParameter<double>("QCut");
     thePositionMinimumDistance = iConfig.getParameter<double>("positionMinimumDistance");
     std::string  meth= iConfig.getParameter<std::string>("method");
     if (meth=="chi2"){ theMethod =0; }
     else if (meth=="dist"){theMethod =1;}
     else if (meth=="momdr"){theMethod = 2;}
     else if (meth=="posdr"){theMethod = 3;}
     else{
       edm::LogError("TrackAssociatorByPosition")<<meth<<" mothed not recognized. Use dr or chi2.";     }

     theConsiderAllSimHits = iConfig.getParameter<bool>("ConsiderAllSimHits");
   };
TrackAssociatorByPosition::~TrackAssociatorByPosition ( ) [inline]

Destructor.

Definition at line 57 of file TrackAssociatorByPosition.h.

                              {
  };

Member Function Documentation

RecoToSimCollection TrackAssociatorByPosition::associateRecoToSim ( const edm::RefToBaseVector< reco::Track > &  tCH,
const edm::RefVector< TrackingParticleCollection > &  tPCH,
const edm::Event event = 0,
const edm::EventSetup setup = 0 
) const [virtual]

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

Implements TrackAssociatorBase.

Definition at line 99 of file TrackAssociatorByPosition.cc.

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

                                                                                                    {
  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;
  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]);
      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(edm::Ref<TrackingParticleCollection>(tPCH,TPi),-dQ));//association map with quality, is order greater-first
        edm::LogVerbatim("TrackAssociatorByPosition")<<"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(edm::Ref<TrackingParticleCollection>(tPCH,minPair.second),-dQmin));}
  }//loop over tracks
  outputCollection.post_insert();
  return outputCollection;
}
SimToRecoCollection TrackAssociatorByPosition::associateSimToReco ( const edm::RefToBaseVector< reco::Track > &  tCH,
const edm::RefVector< TrackingParticleCollection > &  tPCH,
const edm::Event event = 0,
const edm::EventSetup setup = 0 
) const [virtual]

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

Implements TrackAssociatorBase.

Definition at line 146 of file TrackAssociatorByPosition.cc.

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

                                                                                                     {
  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;
  for (unsigned int TPi=0;TPi!=tPCH.size();++TPi){
    //get a state in the muon system
    TrajectoryStateOnSurface simReferenceState= getState(*(tPCH)[TPi]);
      
    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(edm::Ref<TrackingParticleCollection>(tPCH,TPi),
                                std::make_pair(tCH[Ti],-dQ));//association map with quality, is order greater-first
        edm::LogVerbatim("TrackAssociatorByPosition")<<"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(edm::Ref<TrackingParticleCollection>(tPCH,minPair.first),
                              std::make_pair(tCH[minPair.second],-dQmin));}
  }//loop over tracking particles
  
  outputCollection.post_insert();
  return outputCollection;
}
FreeTrajectoryState TrackAssociatorByPosition::getState ( const reco::Track track) const [private]

Definition at line 58 of file TrackAssociatorByPosition.cc.

References trajectoryStateTransform::initialFreeState().

                                                                                   {
  //may be you want to do more than that if track does not go to IP
  return trajectoryStateTransform::initialFreeState(track,thePropagator->magneticField());
}
TrajectoryStateOnSurface TrackAssociatorByPosition::getState ( const TrackingParticle st) const [private]

Definition at line 13 of file TrackAssociatorByPosition.cc.

References SurfaceSideDefinition::atCenterOfSurface, prof2calltree::count, createTree::dd, end, PSimHit::localPosition(), LogDebug, mag(), PSimHit::momentumAtEntry(), PSimHit::particleType(), dqm_diff::start, GeomDet::surface(), Surface::toGlobal(), DetId::Tracker, and TrackingParticle::trackPSimHit().

                                                                                            {
  //  TrackingParticle* simtrack = const_cast<TrackingParticle*>(&st);
  //loop over PSimHits
  const PSimHit * psimhit=0;
  const BoundPlane * plane=0;
  double dLim=thePositionMinimumDistance;

  //    look for the further most hit beyond a certain limit
  std::vector<PSimHit> pSimHit = st.trackPSimHit();
  if (!theConsiderAllSimHits) pSimHit=st.trackPSimHit(DetId::Tracker);
  std::vector<PSimHit> ::const_iterator start=pSimHit.begin();
  std::vector<PSimHit> ::const_iterator end=pSimHit.end();
  LogDebug("TrackAssociatorByPosition")<<pSimHit.size()<<" PSimHits.";

  unsigned int count=0;
  for (std::vector<PSimHit> ::const_iterator psit=start;psit!=end;++psit){    
    //get the detid
    DetId dd(psit->detUnitId());
    LogDebug("TrackAssociatorByPosition")<<count++<<"] PSimHit on: "<<dd.rawId();
    //get the surface from the global geometry
    const GeomDet * gd=theGeometry->idToDet(dd);
    if (!gd){edm::LogError("TrackAssociatorByPosition")<<"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("TrackAssociatorByPosition")<<"no corresponding PSimHit for a tracking particle. will fail.";
    return TrajectoryStateOnSurface();}
}
double TrackAssociatorByPosition::quality ( const TrajectoryStateOnSurface tr,
const TrajectoryStateOnSurface sim 
) const

Definition at line 63 of file TrackAssociatorByPosition.cc.

References PV3DBase< T, PVType, FrameType >::eta(), TrajectoryStateOnSurface::globalDirection(), TrajectoryStateOnSurface::globalPosition(), TrajectoryStateOnSurface::localError(), TrajectoryStateOnSurface::localParameters(), m, mag(), LocalTrajectoryError::matrix(), PV3DBase< T, PVType, FrameType >::phi(), v, and LocalTrajectoryParameters::vector().

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

Member Data Documentation

Definition at line 84 of file TrackAssociatorByPosition.h.

Referenced by TrackAssociatorByPosition().

Definition at line 77 of file TrackAssociatorByPosition.h.

Referenced by TrackAssociatorByPosition().

unsigned int TrackAssociatorByPosition::theMethod [private]

Definition at line 79 of file TrackAssociatorByPosition.h.

Referenced by TrackAssociatorByPosition().

Definition at line 82 of file TrackAssociatorByPosition.h.

Referenced by TrackAssociatorByPosition().

Definition at line 83 of file TrackAssociatorByPosition.h.

Referenced by TrackAssociatorByPosition().

Definition at line 78 of file TrackAssociatorByPosition.h.

Referenced by TrackAssociatorByPosition().

Definition at line 81 of file TrackAssociatorByPosition.h.

Referenced by TrackAssociatorByPosition().

Definition at line 80 of file TrackAssociatorByPosition.h.

Referenced by TrackAssociatorByPosition().