|
|
|
#include <TrackAssociatorByPosition.h>
Class that performs the association of reco::Tracks and TrackingParticles based on position in muon detector
Definition at line 30 of file TrackAssociatorByPosition.h.
| 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(), 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] |
| 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;
}
bool TrackAssociatorByPosition::theConsiderAllSimHits [private] |
Definition at line 84 of file TrackAssociatorByPosition.h.
Referenced by TrackAssociatorByPosition().
const TrackingGeometry* TrackAssociatorByPosition::theGeometry [private] |
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().
bool TrackAssociatorByPosition::theMinIfNoMatch [private] |
Definition at line 82 of file TrackAssociatorByPosition.h.
Referenced by TrackAssociatorByPosition().
double TrackAssociatorByPosition::thePositionMinimumDistance [private] |
Definition at line 83 of file TrackAssociatorByPosition.h.
Referenced by TrackAssociatorByPosition().
const Propagator* TrackAssociatorByPosition::thePropagator [private] |
Definition at line 78 of file TrackAssociatorByPosition.h.
Referenced by TrackAssociatorByPosition().
double TrackAssociatorByPosition::theQCut [private] |
Definition at line 81 of file TrackAssociatorByPosition.h.
Referenced by TrackAssociatorByPosition().
double TrackAssociatorByPosition::theQminCut [private] |
Definition at line 80 of file TrackAssociatorByPosition.h.
Referenced by TrackAssociatorByPosition().
1.7.3