TrackQuality Class Reference

This class analyses the reconstruction quality for a given track. More...

#include <TrackQuality.h>

Classes
struct	Layer

Public Types
typedef std::vector< TrackingParticleRef >	SimParticleTrail

Public Member Functions
void	evaluate (SimParticleTrail const &, reco::TrackBaseRef const &, const TrackerTopology *tTopo)
	Compute information about the track reconstruction quality. More...
const Layer &	layer (unsigned int index) const
	Return information about the given layer by index. More...
void	newEvent (const edm::Event &, const edm::EventSetup &)
	Pre-process event information (for accessing reconstruction information) More...
unsigned int	numberOfLayers () const
	Return the number of layers with simulated and/or reconstructed hits. More...
	TrackQuality (const edm::ParameterSet &, edm::ConsumesCollector &iC)
	Constructor by pset. More...

Private Attributes
std::unique_ptr< TrackerHitAssociator >	associator_
std::vector< Layer >	layers_
TrackerHitAssociator::Config	trackerHitAssociatorConfig_

Detailed Description

This class analyses the reconstruction quality for a given track.

Definition at line 28 of file TrackQuality.h.

Member Typedef Documentation

typedef std::vector<TrackingParticleRef> TrackQuality::SimParticleTrail

Definition at line 30 of file TrackQuality.h.

Constructor & Destructor Documentation

TrackQuality::TrackQuality	(	const edm::ParameterSet &	config,
		edm::ConsumesCollector &	iC
	)

Constructor by pset.

Definition at line 133 of file TrackQuality.cc.

    : trackerHitAssociatorConfig_(config.getParameter<edm::ParameterSet>("hitAssociator"), std::move(iC)) {}

Member Function Documentation

void TrackQuality::evaluate	(	SimParticleTrail const &	spt,
		reco::TrackBaseRef const &	tr,
		const TrackerTopology *	tTopo
	)

Compute information about the track reconstruction quality.

Definition at line 140 of file TrackQuality.cc.

References associator_, TrackQuality::Layer::Bad, TrackingRecHit::bad, gather_cfg::cout, TrackQuality::Layer::Dead, DetId::det(), getDetLayer(), TrackQuality::Layer::hits, mps_fire::i, TrackingRecHit::inactive, TrackQuality::Layer::layer, layer(), layers_, TrackQuality::Layer::Misassoc, TrackQuality::Layer::Missed, DetId::Muon, TrackQuality::Layer::Noise, TrackQuality::Layer::Hit::recHitId, reco::Track::recHitsBegin(), reco::Track::recHitsEnd(), TrackQuality::Layer::Hit::state, TrackQuality::Layer::subDet, and TrackQuality::Layer::Unknown.

Referenced by TrackClassifier::qualityInformation().

                                                                                                                 {
  std::vector<MatchedHit> matchedHits;

  // iterate over reconstructed hits
  for (trackingRecHit_iterator hit = tr->recHitsBegin(); hit != tr->recHitsEnd(); ++hit) {
    // on which module the hit lies
    DetId detId = (*hit)->geographicalId();

    // FIXME: check for double-sided modules?

    // didn't find a hit on that module
    if (!(*hit)->isValid()) {
      MatchedHit matchedHit;
      matchedHit.detId = detId;
      matchedHit.simTrackId = NonMatchedTrackId;
      // check why hit wasn't valid and propagate information
      switch ((*hit)->getType()) {
        case TrackingRecHit::inactive:
          matchedHit.state = Layer::Dead;
          break;

        case TrackingRecHit::bad:
          matchedHit.state = Layer::Bad;
          break;

        default:
          matchedHit.state = Layer::Missed;
      }
      matchedHit.recHitId = hit - tr->recHitsBegin();
      matchedHits.push_back(matchedHit);
      continue;
    }

    // find simulated tracks causing hit that was reconstructed
    std::vector<SimHitIdpr> simIds = associator_->associateHitId(**hit);

    // must be noise or so
    if (simIds.empty()) {
      MatchedHit matchedHit;
      matchedHit.detId = detId;
      matchedHit.simTrackId = NonMatchedTrackId;
      matchedHit.state = Layer::Noise;
      matchedHit.recHitId = hit - tr->recHitsBegin();
      matchedHits.push_back(matchedHit);
      continue;
    }

    // register all simulated tracks contributing
    for (std::vector<SimHitIdpr>::const_iterator i = simIds.begin(); i != simIds.end(); ++i) {
      MatchedHit matchedHit;
      matchedHit.detId = detId;
      matchedHit.simTrackId = i->first;
      matchedHit.collision = i->second;
      // RecHit <-> SimHit matcher currently doesn't support muon system
      if (detId.det() == DetId::Muon)
        matchedHit.state = Layer::Unknown;
      else
        // assume hit was mismatched (until possible confirmation)
        matchedHit.state = Layer::Misassoc;
      matchedHit.recHitId = hit - tr->recHitsBegin();
      matchedHits.push_back(matchedHit);
    }
  }

  // sort hits found so far by module id
  std::stable_sort(matchedHits.begin(), matchedHits.end());

  // in case we added missed modules, re-sort
  std::stable_sort(matchedHits.begin(), matchedHits.end());

  // prepare for ordering results by layer enum and layer/disk number
  typedef std::multimap<DetLayer, const MatchedHit *> LayerHitMap;
  LayerHitMap layerHitMap;

  // iterate over all simulated/reconstructed hits again
  for (std::vector<MatchedHit>::const_iterator hit = matchedHits.begin(); hit != matchedHits.end();) {
    // we can have multiple reco-to-sim matches per module, find best one
    const MatchedHit *best = nullptr;

    // this loop iterates over all subsequent hits in the same module
    do {
      // update our best hit pointer
      if (!best || statePriorities[hit->state] > statePriorities[best->state] ||
          best->simTrackId == NonMatchedTrackId) {
        best = &*hit;
      }
      ++hit;
    } while (hit != matchedHits.end() && hit->detId == best->detId);

    // ignore hit in case track reco was looking at the wrong module
    if (best->simTrackId != NonMatchedTrackId || best->state != Layer::Missed) {
      layerHitMap.insert(std::make_pair(getDetLayer(best->detId, tTopo), best));
    }
  }

  layers_.clear();

#ifdef DEBUG_TRACK_QUALITY
  std::cout << "---------------------" << std::endl;
#endif
  // now prepare final collection
  for (LayerHitMap::const_iterator hit = layerHitMap.begin(); hit != layerHitMap.end(); ++hit) {
#ifdef DEBUG_TRACK_QUALITY
    std::cout << "detLayer (" << hit->first.first << ", " << hit->first.second << ")"
              << " [" << (uint32_t)hit->second->detId << "] sim(" << (int)hit->second->simTrackId << ")"
              << " hit(" << hit->second->recHitId << ") -> " << hit->second->state << std::endl;
#endif

    // find out if we need to start a new layer
    Layer *layer = layers_.empty() ? nullptr : &layers_.back();
    if (!layer || hit->first.first != layer->subDet || hit->first.second != layer->layer) {
      Layer newLayer;
      newLayer.subDet = hit->first.first;
      newLayer.layer = hit->first.second;
      layers_.push_back(newLayer);
      layer = &layers_.back();
    }

    // create hit and add it to layer
    Layer::Hit newHit;
    newHit.recHitId = hit->second->recHitId;
    newHit.state = hit->second->state;

    layer->hits.push_back(newHit);
  }
}

const Layer& TrackQuality::layer ( unsigned int  index ) const

inline

Return information about the given layer by index.

Definition at line 77 of file TrackQuality.h.

References layers_.

Referenced by evaluate(), geometryXMLparser.DTAlignable::index(), geometryXMLparser.CSCAlignable::index(), and TrackClassifier::qualityInformation().

{ return layers_[index]; }

void TrackQuality::newEvent	(	const edm::Event &	ev,
		const edm::EventSetup &	es
	)

Pre-process event information (for accessing reconstruction information)

Definition at line 136 of file TrackQuality.cc.

References associator_, and trackerHitAssociatorConfig_.

Referenced by TrackClassifier::newEvent().

                                                                       {
  associator_.reset(new TrackerHitAssociator(ev, trackerHitAssociatorConfig_));
}

unsigned int TrackQuality::numberOfLayers ( ) const

inline

Return the number of layers with simulated and/or reconstructed hits.

Definition at line 74 of file TrackQuality.h.

References layers_.

Referenced by TrackClassifier::qualityInformation().

{ return layers_.size(); }

Member Data Documentation

std::unique_ptr<TrackerHitAssociator> TrackQuality::associator_

private

Definition at line 81 of file TrackQuality.h.

Referenced by evaluate(), and newEvent().

std::vector<Layer> TrackQuality::layers_

private

Definition at line 83 of file TrackQuality.h.

Referenced by evaluate(), layer(), and numberOfLayers().

TrackerHitAssociator::Config TrackQuality::trackerHitAssociatorConfig_

private

Definition at line 80 of file TrackQuality.h.

Referenced by newEvent().