L1TrackQuality Class Reference

#include <L1TrackQuality.h>

Public Member Functions
std::vector< float >	featureTransform (TTTrack< Ref_Phase2TrackerDigi_ > &aTrack, std::vector< std::string > const &featureNames)
	L1TrackQuality ()
	L1TrackQuality (const edm::ParameterSet &qualityParams)
float	runEmulatedTQ (std::vector< ap_fixed< 10, 5 >> inputFeatures)
void	setBonusFeatures (std::vector< float > bonusFeatures)
void	setL1TrackQuality (TTTrack< Ref_Phase2TrackerDigi_ > &aTrack)
void	setModel (edm::FileInPath const &model, std::vector< std::string > const &featureNames)
	~L1TrackQuality ()=default

Static Public Member Functions
static constexpr std::array< double, 1<< TTTrack_TrackWord::TrackBitWidths::kMVAQualitySize >	getTqMVAPreSigBins ()
static constexpr double	invSigmoid (double value)

Private Attributes
std::vector< float >	bonusFeatures_
std::vector< std::string >	featureNames_
edm::FileInPath	model_
bool	useHPH_

Detailed Description

Definition at line 28 of file L1TrackQuality.h.

Constructor & Destructor Documentation

L1TrackQuality() [1/2]

L1TrackQuality::L1TrackQuality

(

)

Definition at line 10 of file L1TrackQuality.cc.

{}

L1TrackQuality() [2/2]

L1TrackQuality::L1TrackQuality

(

const edm::ParameterSet &

qualityParams

)

Definition at line 12 of file L1TrackQuality.cc.

References edm::ParameterSet::getParameter(), and setModel().

                                                                   : useHPH_(false), bonusFeatures_() {
  // Unpacks EDM parameter set itself to save unecessary processing within TrackProducers
  setModel(qualityParams.getParameter<edm::FileInPath>("model"),
           qualityParams.getParameter<std::vector<std::string>>("featureNames"));
}

~L1TrackQuality()

L1TrackQuality::~L1TrackQuality ( )

default

Member Function Documentation

featureTransform()

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

Definition at line 18 of file L1TrackQuality.cc.

References funct::abs(), TTTrack< T >::eta(), TrackQualityParams_cfi::featureNames, nano_mu_digi_cff::float, TTTrack_TrackWord::getBendChi2Bits(), TTTrack_TrackWord::getChi2RPhiBits(), TTTrack_TrackWord::getChi2RZBits(), TTTrack< T >::getStubRefs(), TTTrack< T >::hitPattern(), mps_fire::i, TTTrack_TrackWord::minZ0, TTTrack< T >::phi(), cmsswSequenceInfo::seq, AlCaHLTBitMon_QueryRunRegistry::string, TTTrack< T >::tanL(), and TTTrack< T >::z0().

Referenced by setL1TrackQuality().

                                                                                              {
  // List input features for MVA in proper order below, the current features options are
  // {"phi", "eta", "z0", "bendchi2_bin", "nstub", "nlaymiss_interior", "chi2rphi_bin",
  // "chi2rz_bin"}
  //
  // To use more features, they must be created here and added to feature_map below

  std::vector<float> transformedFeatures;

  // Define feature map, filled as features are generated
  std::map<std::string, float> feature_map;

  // -------- calculate feature variables --------

  // calculate number of missed interior layers from hitpattern
  int tmp_trk_hitpattern = aTrack.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++;
  }

  // binned chi2 variables
  int tmp_trk_bendchi2_bin = aTrack.getBendChi2Bits();
  int tmp_trk_chi2rphi_bin = aTrack.getChi2RPhiBits();
  int tmp_trk_chi2rz_bin = aTrack.getChi2RZBits();

  // get the nstub
  std::vector<edm::Ref<edmNew::DetSetVector<TTStub<Ref_Phase2TrackerDigi_>>, TTStub<Ref_Phase2TrackerDigi_>>> stubRefs =
      aTrack.getStubRefs();
  int tmp_trk_nstub = stubRefs.size();

  // get other variables directly from TTTrack
  float tmp_trk_z0 = aTrack.z0();
  float tmp_trk_z0_scaled = tmp_trk_z0 / abs(aTrack.minZ0);
  float tmp_trk_phi = aTrack.phi();
  float tmp_trk_eta = aTrack.eta();
  float tmp_trk_tanl = aTrack.tanL();

  // -------- fill the feature map ---------

  feature_map["nstub"] = float(tmp_trk_nstub);
  feature_map["z0"] = tmp_trk_z0;
  feature_map["z0_scaled"] = tmp_trk_z0_scaled;
  feature_map["phi"] = tmp_trk_phi;
  feature_map["eta"] = tmp_trk_eta;
  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;
  feature_map["tanl"] = tmp_trk_tanl;

  // fill tensor with track params
  transformedFeatures.reserve(featureNames.size());
  for (const std::string& feature : featureNames)
    transformedFeatures.push_back(feature_map[feature]);

  return transformedFeatures;
}

getTqMVAPreSigBins()

static constexpr std::array<double, 1 << TTTrack_TrackWord::TrackBitWidths::kMVAQualitySize> L1TrackQuality::getTqMVAPreSigBins ( )

inlinestatic

Definition at line 55 of file L1TrackQuality.h.

References invSigmoid(), and TTTrack_TrackWord::tqMVABins.

Referenced by trklet::ProducerKFout::produce().

                                                                                                              {
    return {{-16.,
             invSigmoid(TTTrack_TrackWord::tqMVABins[1]),
             invSigmoid(TTTrack_TrackWord::tqMVABins[2]),
             invSigmoid(TTTrack_TrackWord::tqMVABins[3]),
             invSigmoid(TTTrack_TrackWord::tqMVABins[4]),
             invSigmoid(TTTrack_TrackWord::tqMVABins[5]),
             invSigmoid(TTTrack_TrackWord::tqMVABins[6]),
             invSigmoid(TTTrack_TrackWord::tqMVABins[7])}};
  }

invSigmoid()

static constexpr double L1TrackQuality::invSigmoid ( double  value )

inlinestatic

Definition at line 54 of file L1TrackQuality.h.

References dqm-mbProfile::log.

Referenced by getTqMVAPreSigBins().

{ return -log(1. / value - 1.); }

runEmulatedTQ()

float L1TrackQuality::runEmulatedTQ

(

std::vector< ap_fixed< 10, 5 >>

inputFeatures

)

Definition at line 96 of file L1TrackQuality.cc.

References edm::FileInPath::fullPath(), and model_.

                                                                            {
  // load in bdt

  conifer::BDT<ap_fixed<10, 5>, ap_fixed<10, 5>> bdt(this->model_.fullPath());

  // collect features and classify using bdt
  std::vector<ap_fixed<10, 5>> output = bdt.decision_function(inputFeatures);
  return output.at(0).to_float();  // need logistic sigmoid fcn applied to xgb output
}

setBonusFeatures()

void L1TrackQuality::setBonusFeatures

(

std::vector< float >

bonusFeatures

)

Definition at line 112 of file L1TrackQuality.cc.

References bonusFeatures_, and useHPH_.

                                                                    {
  bonusFeatures_ = bonusFeatures;
  useHPH_ = true;
}

setL1TrackQuality()

void L1TrackQuality::setL1TrackQuality

(

TTTrack< Ref_Phase2TrackerDigi_ > &

aTrack

)

Definition at line 86 of file L1TrackQuality.cc.

References JetChargeProducer_cfi::exp, featureNames_, featureTransform(), edm::FileInPath::fullPath(), PixelMapPlotter::inputs, model_, and TTTrack< T >::settrkMVA1().

                                                                              {
  // load in bdt
  conifer::BDT<float, float> bdt(this->model_.fullPath());

  // collect features and classify using bdt
  std::vector<float> inputs = featureTransform(aTrack, this->featureNames_);
  std::vector<float> output = bdt.decision_function(inputs);
  aTrack.settrkMVA1(1. / (1. + exp(-output.at(0))));
}

setModel()

void L1TrackQuality::setModel	(	edm::FileInPath const &	model,
		std::vector< std::string > const &	featureNames
	)

Definition at line 106 of file L1TrackQuality.cc.

References TrackQualityParams_cfi::featureNames, featureNames_, ReggeGribovPartonMC_EposLHC_2760GeV_PbPb_cfi::model, and model_.

Referenced by L1TrackQuality().

                                                                                                  {
  //Convert algorithm string to Enum class for track by track comparison
  model_ = model;
  featureNames_ = featureNames;
}

Member Data Documentation

bonusFeatures_

std::vector<float> L1TrackQuality::bonusFeatures_

private

Definition at line 71 of file L1TrackQuality.h.

Referenced by setBonusFeatures().

featureNames_

std::vector<std::string> L1TrackQuality::featureNames_

private

Definition at line 69 of file L1TrackQuality.h.

Referenced by setL1TrackQuality(), and setModel().

model_

edm::FileInPath L1TrackQuality::model_

private

Definition at line 68 of file L1TrackQuality.h.

Referenced by runEmulatedTQ(), setL1TrackQuality(), and setModel().

useHPH_

bool L1TrackQuality::useHPH_

private

Definition at line 70 of file L1TrackQuality.h.

Referenced by setBonusFeatures().