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

#include <TrackQuality.h>

Classes
struct	Layer

Public Types
enum	QualityAlgorithm { QualityAlgorithm::Cut, QualityAlgorithm::GBDT, QualityAlgorithm::NN, QualityAlgorithm::None }

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...

std::vector< float >	featureTransform (TTTrack< Ref_Phase2TrackerDigi_ > &aTrack, std::vector< std::string > const &featureNames)

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...

void	setCutParameters (std::string const &AlgorithmString, float maxZ0, float maxEta, float chi2dofMax, float bendchi2Max, float minPt, int nStubmin)

void	setONNXModel (std::string const &AlgorithmString, edm::FileInPath const &ONNXmodel, std::string const &ONNXInputName, std::vector< std::string > const &featureNames)

void	setTrackQuality (TTTrack< Ref_Phase2TrackerDigi_ > &aTrack)

	TrackQuality ()

	TrackQuality (const edm::ParameterSet &, edm::ConsumesCollector &iC)
	Constructor by pset. More...

	TrackQuality (const edm::ParameterSet &qualityParams)

	~TrackQuality ()=default

Private Attributes
std::unique_ptr< TrackerHitAssociator >	associator_

float	bendchi2Max_

float	chi2dofMax_

std::vector< std::string >	featureNames_

std::vector< Layer >	layers_

float	maxEta_

float	maxZ0_

float	minPt_

int	nStubsmin_

std::string	ONNXInputName_

float	ONNXInvRScaling_

edm::FileInPath	ONNXmodel_

QualityAlgorithm	qualityAlgorithm_ = QualityAlgorithm::None

TrackerHitAssociator::Config	trackerHitAssociatorConfig_

Detailed Description

This class analyses the reconstruction quality for a given track.

Definition at line 29 of file TrackQuality.h.

Member Typedef Documentation

◆ SimParticleTrail

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

Definition at line 30 of file TrackQuality.h.

Member Enumeration Documentation

◆ QualityAlgorithm

enum TrackQuality::QualityAlgorithm

strong

Enumerator
Cut
GBDT
NN
None

Definition at line 32 of file TrackQuality.h.

32 { Cut, GBDT, NN, None };

Constructor & Destructor Documentation

◆ TrackQuality() [1/3]

TrackQuality::TrackQuality ( )

Definition at line 12 of file TrackQuality.cc.

12 {}

◆ TrackQuality() [2/3]

TrackQuality::TrackQuality ( const edm::ParameterSet & qualityParams )

Definition at line 14 of file TrackQuality.cc.

                                                                {
   std::string AlgorithmString = qualityParams.getParameter<std::string>("qualityAlgorithm");
   // Unpacks EDM parameter set itself to save unecessary processing within TrackProducers
   if (AlgorithmString == "Cut") {
     setCutParameters(AlgorithmString,
                      (float)qualityParams.getParameter<double>("maxZ0"),
                      (float)qualityParams.getParameter<double>("maxEta"),
                      (float)qualityParams.getParameter<double>("chi2dofMax"),
                      (float)qualityParams.getParameter<double>("bendchi2Max"),
                      (float)qualityParams.getParameter<double>("minPt"),
                      qualityParams.getParameter<int>("nStubsmin"));
   }
  
   else {
     setONNXModel(AlgorithmString,
                  qualityParams.getParameter<edm::FileInPath>("ONNXmodel"),
                  qualityParams.getParameter<std::string>("ONNXInputName"),
                  qualityParams.getParameter<std::vector<std::string>>("featureNames"));
     ONNXInvRScaling_ = qualityParams.getParameter<double>("ONNXInvRScale");
   }
 }

References edm::ParameterSet::getParameter(), ONNXInvRScaling_, setCutParameters(), setONNXModel(), and AlCaHLTBitMon_QueryRunRegistry::string.

◆ ~TrackQuality()

TrackQuality::~TrackQuality ( )

default

◆ TrackQuality() [3/3]

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

Constructor by pset.

Definition at line 135 of file TrackQuality.cc.

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

Member Function Documentation

◆ evaluate()

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

Compute information about the track reconstruction quality.

Definition at line 142 of file TrackQuality.cc.

                                                                                                                  {
   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);
   }
 }

Referenced by TrackClassifier::qualityInformation().

◆ featureTransform()

std::vector< float > TrackQuality::featureTransform	(	TTTrack< Ref_Phase2TrackerDigi_ > &	aTrack,
		std::vector< std::string > const &	featureNames
	)

Definition at line 36 of file TrackQuality.cc.

                                                                                             {
   // List input features for MVA in proper order below, the features options are
   // {"log_chi2","log_chi2rphi","log_chi2rz","log_bendchi2","nstubs","lay1_hits","lay2_hits",
   // "lay3_hits","lay4_hits","lay5_hits","lay6_hits","disk1_hits","disk2_hits","disk3_hits",
   // "disk4_hits","disk5_hits","rinv","tanl","z0","dtot","ltot","chi2","chi2rz","chi2rphi",
   // "bendchi2","pt","eta","nlaymiss_interior","phi","bendchi2_bin","chi2rz_bin","chi2rphi_bin"}
  
   std::vector<float> transformedFeatures;
  
   // Define feature map, filled as features are generated
   std::map<std::string, float> feature_map;
  
   // The following converts the 7 bit hitmask in the TTTrackword to an expected
   // 11 bit hitmask based on the eta of the track
   std::vector<int> hitpattern_binary = {0, 0, 0, 0, 0, 0, 0};
   std::vector<int> hitpattern_expanded_binary = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
   std::vector<float> eta_bins = {0.0, 0.2, 0.41, 0.62, 0.9, 1.26, 1.68, 2.08, 2.4};
  
   // Expected hitmap table, each row corresponds to an eta bin, each value corresponds to
   // the expected layer in the expanded hit pattern. The expanded hit pattern should be
   // 11 bits but contains a 12th element so this hitmap table is symmetric
   int hitmap[8][7] = {{0, 1, 2, 3, 4, 5, 11},
                       {0, 1, 2, 3, 4, 5, 11},
                       {0, 1, 2, 3, 4, 5, 11},
                       {0, 1, 2, 3, 4, 5, 11},
                       {0, 1, 2, 3, 4, 5, 11},
                       {0, 1, 2, 6, 7, 8, 9},
                       {0, 1, 7, 8, 9, 10, 11},
                       {0, 6, 7, 8, 9, 10, 11}};
  
   // iterate through bits of the hitpattern and compare to 1 filling the hitpattern binary vector
   int tmp_trk_hitpattern = aTrack.hitPattern();
   for (int i = 6; i >= 0; i--) {
     int k = tmp_trk_hitpattern >> i;
     if (k & 1)
       hitpattern_binary[i] = 1;
   }
  
   // calculate number of missed interior layers from hitpattern
   int nbits = floor(log2(tmp_trk_hitpattern)) + 1;
   int lay_i = 0;
   int tmp_trk_nlaymiss_interior = 0;
   bool seq = false;
   for (int i = 0; i < nbits; i++) {
     lay_i = ((1 << i) & tmp_trk_hitpattern) >> i;  //0 or 1 in ith bit (right to left)
  
     if (lay_i && !seq)
       seq = true;  //sequence starts when first 1 found
     if (!lay_i && seq)
       tmp_trk_nlaymiss_interior++;
   }
  
   float eta = abs(aTrack.eta());
   int eta_size = static_cast<int>(eta_bins.size());
   // First iterate through eta bins
  
   for (int j = 1; j < eta_size; j++) {
     if (eta < eta_bins[j] && eta >= eta_bins[j - 1])  // if track in eta bin
     {
       // Iterate through hitpattern binary
       for (int k = 0; k <= 6; k++)
         // Fill expanded binary entries using the expected hitmap table positions
         hitpattern_expanded_binary[hitmap[j - 1][k]] = hitpattern_binary[k];
       break;
     }
   }
  
   int tmp_trk_ltot = 0;
   //calculate number of layer hits
   for (int i = 0; i < 6; ++i) {
     tmp_trk_ltot += hitpattern_expanded_binary[i];
   }
  
   int tmp_trk_dtot = 0;
   //calculate number of disk hits
   for (int i = 6; i < 11; ++i) {
     tmp_trk_dtot += hitpattern_expanded_binary[i];
   }
  
   // bin bendchi2 variable (bins from https://twiki.cern.ch/twiki/bin/viewauth/CMS/HybridDataFormat#Fitted_Tracks_written_by_KalmanF)
   float tmp_trk_bendchi2 = aTrack.stubPtConsistency();
   std::array<float, 8> bendchi2_bins{{0, 0.5, 1.25, 2, 3, 5, 10, 50}};
   int n_bendchi2 = static_cast<int>(bendchi2_bins.size());
   float tmp_trk_bendchi2_bin = -1;
   for (int i = 0; i < n_bendchi2; i++) {
     if (tmp_trk_bendchi2 >= bendchi2_bins[i] && tmp_trk_bendchi2 < bendchi2_bins[i + 1]) {
       tmp_trk_bendchi2_bin = i;
       break;
     }
   }
   if (tmp_trk_bendchi2_bin < 0)
     tmp_trk_bendchi2_bin = n_bendchi2;
  
   // bin chi2rphi variable (bins from https://twiki.cern.ch/twiki/bin/viewauth/CMS/HybridDataFormat#Fitted_Tracks_written_by_KalmanF)
   float tmp_trk_chi2rphi = aTrack.chi2XY();
   std::array<float, 16> chi2rphi_bins{{0, 0.25, 0.5, 1, 2, 3, 5, 7, 10, 20, 40, 100, 200, 500, 1000, 3000}};
   int n_chi2rphi = static_cast<int>(chi2rphi_bins.size());
   float tmp_trk_chi2rphi_bin = -1;
   for (int i = 0; i < n_chi2rphi; i++) {
     if (tmp_trk_chi2rphi >= chi2rphi_bins[i] && tmp_trk_chi2rphi < chi2rphi_bins[i + 1]) {
       tmp_trk_chi2rphi_bin = i;
       break;
     }
   }
   if (tmp_trk_chi2rphi_bin < 0)
     tmp_trk_chi2rphi_bin = n_chi2rphi;
  
   // bin chi2rz variable (bins from https://twiki.cern.ch/twiki/bin/viewauth/CMS/HybridDataFormat#Fitted_Tracks_written_by_KalmanF)
   float tmp_trk_chi2rz = aTrack.chi2Z();
   std::array<float, 16> chi2rz_bins{{0, 0.25, 0.5, 1, 2, 3, 5, 7, 10, 20, 40, 100, 200, 500, 1000, 3000}};
   int n_chi2rz = static_cast<int>(chi2rz_bins.size());
   float tmp_trk_chi2rz_bin = -1;
   for (int i = 0; i < n_chi2rz; i++) {
     if (tmp_trk_chi2rz >= chi2rz_bins[i] && tmp_trk_chi2rz < chi2rz_bins[i + 1]) {
       tmp_trk_chi2rz_bin = i;
       break;
     }
   }
   if (tmp_trk_chi2rz_bin < 0)
     tmp_trk_chi2rz_bin = n_chi2rz;
  
   // get the nstub
   std::vector<edm::Ref<edmNew::DetSetVector<TTStub<Ref_Phase2TrackerDigi_>>, TTStub<Ref_Phase2TrackerDigi_>>> stubRefs =
       aTrack.getStubRefs();
  
   // fill the feature map
   feature_map["nstub"] = stubRefs.size();
   feature_map["rinv"] = ONNXInvRScaling_ * abs(aTrack.rInv());
   feature_map["tanl"] = abs(aTrack.tanL());
   feature_map["z0"] = aTrack.z0();
   feature_map["phi"] = aTrack.phi();
   feature_map["pt"] = aTrack.momentum().perp();
   feature_map["eta"] = aTrack.eta();
  
   float tmp_trk_chi2 = aTrack.chi2();
   feature_map["chi2"] = tmp_trk_chi2;
   feature_map["log_chi2"] = log(tmp_trk_chi2);
  
   feature_map["chi2rphi"] = tmp_trk_chi2rphi;
   feature_map["log_chi2rphi"] = log(tmp_trk_chi2rphi);
  
   feature_map["chi2rz"] = tmp_trk_chi2rz;
   feature_map["log_chi2rz"] = log(tmp_trk_chi2rz);
  
   feature_map["chi2rz"] = tmp_trk_chi2rz;
   feature_map["log_chi2rz"] = log(tmp_trk_chi2rz);
  
   feature_map["bendchi2"] = tmp_trk_bendchi2;
   feature_map["log_bendchi2"] = log(tmp_trk_bendchi2);
  
   feature_map["lay1_hits"] = float(hitpattern_expanded_binary[0]);
   feature_map["lay2_hits"] = float(hitpattern_expanded_binary[1]);
   feature_map["lay3_hits"] = float(hitpattern_expanded_binary[2]);
   feature_map["lay4_hits"] = float(hitpattern_expanded_binary[3]);
   feature_map["lay5_hits"] = float(hitpattern_expanded_binary[4]);
   feature_map["lay6_hits"] = float(hitpattern_expanded_binary[5]);
   feature_map["disk1_hits"] = float(hitpattern_expanded_binary[6]);
   feature_map["disk2_hits"] = float(hitpattern_expanded_binary[7]);
   feature_map["disk3_hits"] = float(hitpattern_expanded_binary[8]);
   feature_map["disk4_hits"] = float(hitpattern_expanded_binary[9]);
   feature_map["disk5_hits"] = float(hitpattern_expanded_binary[10]);
  
   feature_map["dtot"] = float(tmp_trk_dtot);
   feature_map["ltot"] = float(tmp_trk_ltot);
  
   feature_map["nlaymiss_interior"] = float(tmp_trk_nlaymiss_interior);
   feature_map["bendchi2_bin"] = tmp_trk_bendchi2_bin;
   feature_map["chi2rphi_bin"] = tmp_trk_chi2rphi_bin;
   feature_map["chi2rz_bin"] = tmp_trk_chi2rz_bin;
  
   // fill tensor with track params
   transformedFeatures.reserve(featureNames.size());
   for (const std::string& feature : featureNames)
     transformedFeatures.push_back(feature_map[feature]);
  
   return transformedFeatures;
 }

References funct::abs(), TTTrack< T >::chi2(), TTTrack< T >::chi2XY(), TTTrack< T >::chi2Z(), PVValHelper::eta, TTTrack< T >::eta(), TrackQualityParams_cfi::featureNames, dqmMemoryStats::float, TTTrack< T >::getStubRefs(), TTTrack< T >::hitPattern(), mps_fire::i, dqmiolumiharvest::j, dqmdumpme::k, dqm-mbProfile::log, TTTrack< T >::momentum(), ONNXInvRScaling_, PV3DBase< T, PVType, FrameType >::perp(), TTTrack< T >::phi(), TTTrack< T >::rInv(), cmsswSequenceInfo::seq, AlCaHLTBitMon_QueryRunRegistry::string, TTTrack< T >::stubPtConsistency(), TTTrack< T >::tanL(), and TTTrack< T >::z0().

Referenced by setTrackQuality().

◆ layer()

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.

77 { return layers_[index]; }

References layers_.

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

◆ newEvent()

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

Pre-process event information (for accessing reconstruction information)

Definition at line 138 of file TrackQuality.cc.

                                                                        {
   associator_ = std::make_unique<TrackerHitAssociator>(ev, trackerHitAssociatorConfig_);
 }

References associator_, ev, and trackerHitAssociatorConfig_.

Referenced by TrackClassifier::newEvent().

◆ numberOfLayers()

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.

74 { return layers_.size(); }

References layers_.

Referenced by TrackClassifier::qualityInformation().

◆ setCutParameters()

void TrackQuality::setCutParameters	(	std::string const &	AlgorithmString,
		float	maxZ0,
		float	maxEta,
		float	chi2dofMax,
		float	bendchi2Max,
		float	minPt,
		int	nStubmin
	)

Definition at line 275 of file TrackQuality.cc.

                                                   {
   qualityAlgorithm_ = QualityAlgorithm::Cut;
   maxZ0_ = maxZ0;
   maxEta_ = maxEta;
   chi2dofMax_ = chi2dofMax;
   bendchi2Max_ = bendchi2Max;
   minPt_ = minPt;
   nStubsmin_ = nStubmin;
 }

References TrackQualityParams_cfi::bendchi2Max, bendchi2Max_, L1TrackerEtMissProducer_cfi::chi2dofMax, chi2dofMax_, Cut, maxEta, maxEta_, L1TrackerEtMissProducer_cfi::maxZ0, maxZ0_, beam_dqm_sourceclient-live_cfg::minPt, minPt_, nStubsmin_, and qualityAlgorithm_.

Referenced by TrackQuality().

◆ setONNXModel()

void TrackQuality::setONNXModel	(	std::string const &	AlgorithmString,
		edm::FileInPath const &	ONNXmodel,
		std::string const &	ONNXInputName,
		std::vector< std::string > const &	featureNames
	)

Definition at line 291 of file TrackQuality.cc.

                                                                           {
   //Convert algorithm string to Enum class for track by track comparison
   if (AlgorithmString == "NN") {
     qualityAlgorithm_ = QualityAlgorithm::NN;
   } else if (AlgorithmString == "GBDT") {
     qualityAlgorithm_ = QualityAlgorithm::GBDT;
   } else {
     qualityAlgorithm_ = QualityAlgorithm::None;
   }
   ONNXmodel_ = ONNXmodel;
   ONNXInputName_ = ONNXInputName;
   featureNames_ = featureNames;
 }

References TrackQualityParams_cfi::featureNames, featureNames_, GBDT, NN, None, TrackQualityParams_cfi::ONNXInputName, ONNXInputName_, TrackQualityParams_cfi::ONNXmodel, ONNXmodel_, and qualityAlgorithm_.

Referenced by TrackQuality().

◆ setTrackQuality()

void TrackQuality::setTrackQuality ( TTTrack< Ref_Phase2TrackerDigi_ > & aTrack )

Definition at line 215 of file TrackQuality.cc.

                                                                           {
   if (this->qualityAlgorithm_ == QualityAlgorithm::Cut) {
     // Get Track parameters
     float trk_pt = aTrack.momentum().perp();
     float trk_bend_chi2 = aTrack.stubPtConsistency();
     float trk_z0 = aTrack.z0();
     float trk_eta = aTrack.momentum().eta();
     float trk_chi2 = aTrack.chi2();
     const auto& stubRefs = aTrack.getStubRefs();
     int nStubs = stubRefs.size();
  
     float classification = 0.0;  // Default classification is 0
  
     if (trk_pt >= this->minPt_ && abs(trk_z0) < this->maxZ0_ && abs(trk_eta) < this->maxEta_ &&
         trk_chi2 < this->chi2dofMax_ && trk_bend_chi2 < this->bendchi2Max_ && nStubs >= this->nStubsmin_)
       classification = 1.0;
     // Classification updated to 1 if conditions are met
  
     aTrack.settrkMVA1(classification);
   }
  
   if ((this->qualityAlgorithm_ == QualityAlgorithm::NN) || (this->qualityAlgorithm_ == QualityAlgorithm::GBDT)) {
     // Setup ONNX input and output names and arrays
     std::vector<std::string> ortinput_names;
     std::vector<std::string> ortoutput_names;
  
     cms::Ort::FloatArrays ortinput;
     cms::Ort::FloatArrays ortoutputs;
  
     std::vector<float> Transformed_features = featureTransform(aTrack, this->featureNames_);
     cms::Ort::ONNXRuntime Runtime(this->ONNXmodel_.fullPath());  //Setup ONNX runtime
  
     ortinput_names.push_back(this->ONNXInputName_);
     ortoutput_names = Runtime.getOutputNames();
  
     //ONNX runtime recieves a vector of vectors of floats so push back the input
     // vector of float to create a 1,1,21 ortinput
     ortinput.push_back(Transformed_features);
  
     // batch_size 1 as only one set of transformed features is being processed
     int batch_size = 1;
     // Run classification
     ortoutputs = Runtime.run(ortinput_names, ortinput, {}, ortoutput_names, batch_size);
  
     if (this->qualityAlgorithm_ == QualityAlgorithm::NN) {
       aTrack.settrkMVA1(ortoutputs[0][0]);
     }
  
     else if (this->qualityAlgorithm_ == QualityAlgorithm::GBDT) {
       aTrack.settrkMVA1(ortoutputs[1][1]);
     }
     // Slight differences in the ONNX models of the GBDTs and NNs mean different
     // indices of the ortoutput need to be accessed
   }
  
   else {
     aTrack.settrkMVA1(-999);
   }
 }

References funct::abs(), bendchi2Max_, TTTrack< T >::chi2(), chi2dofMax_, Cut, PV3DBase< T, PVType, FrameType >::eta(), featureNames_, featureTransform(), edm::FileInPath::fullPath(), GBDT, cms::Ort::ONNXRuntime::getOutputNames(), TTTrack< T >::getStubRefs(), maxEta_, maxZ0_, minPt_, TTTrack< T >::momentum(), NN, nStubsmin_, ONNXInputName_, ONNXmodel_, PV3DBase< T, PVType, FrameType >::perp(), qualityAlgorithm_, TTTrack< T >::settrkMVA1(), TTTrack< T >::stubPtConsistency(), and TTTrack< T >::z0().

Member Data Documentation

◆ associator_

std::unique_ptr<TrackerHitAssociator> TrackQuality::associator_

private

Definition at line 81 of file TrackQuality.h.

Referenced by evaluate(), and newEvent().

◆ bendchi2Max_

float TrackQuality::bendchi2Max_

private

Definition at line 72 of file TrackQuality.h.

Referenced by setCutParameters(), and setTrackQuality().

◆ chi2dofMax_

float TrackQuality::chi2dofMax_

private

Definition at line 71 of file TrackQuality.h.

Referenced by setCutParameters(), and setTrackQuality().

◆ featureNames_

std::vector<std::string> TrackQuality::featureNames_

private

Definition at line 68 of file TrackQuality.h.

Referenced by setONNXModel(), and setTrackQuality().

◆ layers_

std::vector<Layer> TrackQuality::layers_

private

Definition at line 83 of file TrackQuality.h.

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

◆ maxEta_

float TrackQuality::maxEta_

private

Definition at line 70 of file TrackQuality.h.

Referenced by setCutParameters(), and setTrackQuality().

◆ maxZ0_

float TrackQuality::maxZ0_

private

Definition at line 69 of file TrackQuality.h.

Referenced by setCutParameters(), and setTrackQuality().

◆ minPt_

float TrackQuality::minPt_

private

Definition at line 73 of file TrackQuality.h.

Referenced by setCutParameters(), and setTrackQuality().

◆ nStubsmin_

int TrackQuality::nStubsmin_

private

Definition at line 74 of file TrackQuality.h.

Referenced by setCutParameters(), and setTrackQuality().

◆ ONNXInputName_

std::string TrackQuality::ONNXInputName_

private

Definition at line 67 of file TrackQuality.h.

Referenced by setONNXModel(), and setTrackQuality().

◆ ONNXInvRScaling_

float TrackQuality::ONNXInvRScaling_

private

Definition at line 75 of file TrackQuality.h.

Referenced by featureTransform(), and TrackQuality().

◆ ONNXmodel_

edm::FileInPath TrackQuality::ONNXmodel_

private

Definition at line 66 of file TrackQuality.h.

Referenced by setONNXModel(), and setTrackQuality().

◆ qualityAlgorithm_

QualityAlgorithm TrackQuality::qualityAlgorithm_ = QualityAlgorithm::None

private

Definition at line 65 of file TrackQuality.h.

Referenced by setCutParameters(), setONNXModel(), and setTrackQuality().

◆ trackerHitAssociatorConfig_

TrackerHitAssociator::Config TrackQuality::trackerHitAssociatorConfig_

private

Definition at line 80 of file TrackQuality.h.

Referenced by newEvent().

Classes

Public Types

Public Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ SimParticleTrail

Member Enumeration Documentation

◆ QualityAlgorithm

Constructor & Destructor Documentation

◆ TrackQuality() [1/3]

◆ TrackQuality() [2/3]

◆ ~TrackQuality()

◆ TrackQuality() [3/3]

Member Function Documentation

◆ evaluate()

◆ featureTransform()

◆ layer()

◆ newEvent()

◆ numberOfLayers()

◆ setCutParameters()

◆ setONNXModel()

◆ setTrackQuality()

Member Data Documentation

◆ associator_

◆ bendchi2Max_

◆ chi2dofMax_

◆ featureNames_

◆ layers_

◆ maxEta_

◆ maxZ0_

◆ minPt_

◆ nStubsmin_

◆ ONNXInputName_

◆ ONNXInvRScaling_

◆ ONNXmodel_

◆ qualityAlgorithm_

◆ trackerHitAssociatorConfig_