TrackHistory.cc

Go to the documentation of this file.

 
#include "SimTracker/Records/interface/TrackAssociatorRecord.h"
#include "SimTracker/TrackHistory/interface/TrackHistory.h"
 
TrackHistory::TrackHistory(const edm::ParameterSet &config, edm::ConsumesCollector &&collector) : HistoryBase() {
  // Name of the track collection
  trackProducer_ = config.getUntrackedParameter<edm::InputTag>("trackProducer");
  collector.consumes<edm::View<reco::Track>>(trackProducer_);
 
  // Name of the traking pariticle collection
  trackingTruth_ = config.getUntrackedParameter<edm::InputTag>("trackingTruth");
  collector.consumes<TrackingParticleCollection>(trackingTruth_);
 
  // Track association record
  trackAssociator_ = config.getUntrackedParameter<edm::InputTag>("trackAssociator");
 
  // Association by max. value
  bestMatchByMaxValue_ = config.getUntrackedParameter<bool>("bestMatchByMaxValue");
 
  // Enable RecoToSim association
  enableRecoToSim_ = config.getUntrackedParameter<bool>("enableRecoToSim");
 
  // Enable SimToReco association
  enableSimToReco_ = config.getUntrackedParameter<bool>("enableSimToReco");
 
  if (enableRecoToSim_ or enableSimToReco_) {
    collector.consumes<reco::TrackToTrackingParticleAssociator>(trackAssociator_);
  }
 
  quality_ = 0.;
}
 
void TrackHistory::newEvent(const edm::Event &event, const edm::EventSetup &setup) {
  if (enableRecoToSim_ || enableSimToReco_) {
    // Track collection
    edm::Handle<edm::View<reco::Track>> trackCollection;
    event.getByLabel(trackProducer_, trackCollection);
 
    // Tracking particle information
    edm::Handle<TrackingParticleCollection> TPCollection;
    event.getByLabel(trackingTruth_, TPCollection);
 
    // Get the track associator
    edm::Handle<reco::TrackToTrackingParticleAssociator> associator;
    event.getByLabel(trackAssociator_, associator);
 
    // Calculate the map between recotracks -> tp
    if (enableRecoToSim_)
      recoToSim_ = associator->associateRecoToSim(trackCollection, TPCollection);
 
    // Calculate the map between recotracks <- tp
    if (enableSimToReco_)
      simToReco_ = associator->associateSimToReco(trackCollection, TPCollection);
  }
}
 
bool TrackHistory::evaluate(reco::TrackBaseRef tr) {
  if (!enableRecoToSim_)
    return false;
 
  std::pair<TrackingParticleRef, double> result = match(tr, recoToSim_, bestMatchByMaxValue_);
 
  TrackingParticleRef tpr(result.first);
  quality_ = result.second;
  trackingParticle_ = tpr;
 
  if (!tpr.isNull()) {
    HistoryBase::evaluate(tpr);
 
    recotrack_ = tr;
 
    return true;
  }
 
  return false;
}