PtAssignmentEngineDxy Class Reference

#include <PtAssignmentEngineDxy.h>

Public Member Functions
const PtAssignmentEngineAux2017 &	aux () const
virtual void	calculate_pt_dxy (const EMTFTrack &track, emtf::Feature &feature, emtf::Prediction &prediction) const
virtual void	call_tensorflow_dxy (const emtf::Feature &feature, emtf::Prediction &prediction) const
void	configure (int verbose, const std::string pbFileNameDxy)
virtual void	preprocessing_dxy (const EMTFTrack &track, emtf::Feature &feature) const
	PtAssignmentEngineDxy ()
virtual	~PtAssignmentEngineDxy ()

Protected Attributes
tensorflow::GraphDef *	graphDefDxy_
std::string	inputNameDxy_
std::vector< std::string >	outputNamesDxy_
std::string	pbFileNameDxy_
std::string	pbFilePathDxy_
tensorflow::Session *	sessionDxy_
int	verbose_

Detailed Description

Definition at line 16 of file PtAssignmentEngineDxy.h.

Constructor & Destructor Documentation

PtAssignmentEngineDxy()

PtAssignmentEngineDxy::PtAssignmentEngineDxy ( )

explicit

Definition at line 9 of file PtAssignmentEngineDxy.cc.

: graphDefDxy_(nullptr), sessionDxy_(nullptr) {}

~PtAssignmentEngineDxy()

PtAssignmentEngineDxy::~PtAssignmentEngineDxy ( )

virtual

Definition at line 11 of file PtAssignmentEngineDxy.cc.

References tensorflow::closeSession(), graphDefDxy_, and sessionDxy_.

                                              {
  if (sessionDxy_ != nullptr) {
    tensorflow::closeSession(sessionDxy_);
  }
  delete graphDefDxy_;
}

Member Function Documentation

aux()

const PtAssignmentEngineAux2017 & PtAssignmentEngineDxy::aux

(

)

const

Definition at line 39 of file PtAssignmentEngineDxy.cc.

References instance.

                                                                  {
  static const PtAssignmentEngineAux2017 instance;
  return instance;
}

calculate_pt_dxy()

void PtAssignmentEngineDxy::calculate_pt_dxy ( const EMTFTrack &  track, emtf::Feature &  feature, emtf::Prediction &  prediction  ) const

virtual

Definition at line 44 of file PtAssignmentEngineDxy.cc.

References call_tensorflow_dxy(), preprocessing_dxy(), and HLT_2024v13_cff::track.

Referenced by PtAssignment::process().

                                                                               {
  // This is called for each track instead of for entire track collection as was done in Phase-2 implementation
  preprocessing_dxy(track, feature);
  call_tensorflow_dxy(feature, prediction);
  return;
}

call_tensorflow_dxy()

void PtAssignmentEngineDxy::call_tensorflow_dxy ( const emtf::Feature &  feature, emtf::Prediction &  prediction  ) const

virtual

Definition at line 156 of file PtAssignmentEngineDxy.cc.

References filterCSVwithJSON::copy, ztail::d, data, emtf_assert, input, inputNameDxy_, emtf::NUM_FEATURES, emtf::NUM_PREDICTIONS, outputNamesDxy_, PatBasicFWLiteJetAnalyzer_Selector_cfg::outputs, tensorflow::run(), and sessionDxy_.

Referenced by calculate_pt_dxy().

                                                                                                            {
  tensorflow::Tensor input(tensorflow::DT_FLOAT, {1, emtf::NUM_FEATURES});
  std::vector<tensorflow::Tensor> outputs;
  emtf_assert(feature.size() == emtf::NUM_FEATURES);

  float* d = input.flat<float>().data();
  std::copy(feature.begin(), feature.end(), d);
  tensorflow::run(sessionDxy_, {{inputNameDxy_, input}}, outputNamesDxy_, &outputs);
  emtf_assert(outputs.size() == 1);
  emtf_assert(prediction.size() == emtf::NUM_PREDICTIONS);

  prediction.at(0) = outputs[0].matrix<float>()(0, 0);
  prediction.at(1) = outputs[0].matrix<float>()(0, 1);

  return;
}

configure()

void PtAssignmentEngineDxy::configure	(	int	verbose,
		const std::string	pbFileNameDxy
	)

Definition at line 18 of file PtAssignmentEngineDxy.cc.

References tensorflow::createSession(), emtf_assert, contentValuesFiles::fullPath, graphDefDxy_, inputNameDxy_, tensorflow::loadGraphDef(), outputNamesDxy_, pbFileNameDxy_, pbFilePathDxy_, sessionDxy_, AlCaHLTBitMon_QueryRunRegistry::string, verbose, and verbose_.

Referenced by PtAssignment::configure().

                                                                                {
  verbose_ = verbose;

  pbFileNameDxy_ = pbFileNameDxy;
  std::string pbFilePathDxy_ = "L1Trigger/L1TMuon/data/emtf_luts/" + pbFileNameDxy_;

  inputNameDxy_ = "input1";
  outputNamesDxy_ = {"Identity"};

  if (graphDefDxy_ == nullptr) {
    graphDefDxy_ = tensorflow::loadGraphDef(edm::FileInPath(pbFilePathDxy_).fullPath());
  }
  emtf_assert(graphDefDxy_ != nullptr);

  if (sessionDxy_ == nullptr) {
    sessionDxy_ = tensorflow::createSession(graphDefDxy_);
  }

  emtf_assert(sessionDxy_ != nullptr);
}

preprocessing_dxy()

void PtAssignmentEngineDxy::preprocessing_dxy ( const EMTFTrack &  track, emtf::Feature &  feature  ) const

virtual

Definition at line 53 of file PtAssignmentEngineDxy.cc.

References data, mps_fire::i, and HLT_2024v13_cff::track.

Referenced by calculate_pt_dxy().

                                                                                                {
  // Mimic Phase-1 EMTF input calculations
  // 6 delta Phis: S1-S2, S1-S3, S1-S4, S2-S3, S2-S4, S3-S4
  // 6 delta Thetas: S1-S2, S1-S3, S1-S4, S2-S3, S2-S4, S3-S4
  // 6 delta Phi signs: S1-S2, S1-S3, S1-S4, S2-S3, S2-S4, S3-S4
  // 6 delta Theta signs: S1-S2, S1-S3, S1-S4, S2-S3, S2-S4, S3-S4
  // 1 track Theta taken from stub coordinate in ME2, ME3, ME4 (in this priority)
  // 4 CSC pattern values (Run 2 convention): S1, S2, S3, S4
  // Total: 29 variables
  std::array<float, 6> x_dphi;
  std::array<float, 6> x_dphi_sign;
  std::array<float, 6> x_dtheta;
  std::array<float, 6> x_dtheta_sign;
  std::array<float, 1> x_trk_theta;
  std::array<float, 4> x_csc_pattern;

  // Initialize x_csc_pattern to zeros
  x_csc_pattern.fill(0);

  EMTFPtLUT data = track.PtLUT();

  const int invalid_dtheta = 127;
  const int invalid_dphi = 8191;

  // // Which stations have hits
  bool st1 = (track.Mode() >= 8);
  bool st2 = ((track.Mode() % 8) >= 4);
  bool st3 = ((track.Mode() % 4) >= 2);
  bool st4 = ((track.Mode() % 2) == 1);

  // Get valid pattern values
  if (st1)
    x_csc_pattern[0] = data.cpattern[0];
  if (st2)
    x_csc_pattern[1] = data.cpattern[1];
  if (st3)
    x_csc_pattern[2] = data.cpattern[2];
  if (st4)
    x_csc_pattern[3] = data.cpattern[3];

  for (int i = 0; i < 6; ++i) {  // There are 6 deltas between 4 stations.
    // Calculate delta phi
    x_dphi[i] = (data.delta_ph[i] != invalid_dphi) ? data.delta_ph[i] : 0;

    // Calculate delta theta
    x_dtheta[i] = (data.delta_th[i] != invalid_dtheta) ? data.delta_th[i] : 0;

    // Get delta phi and theta signs
    x_dphi_sign[i] = data.sign_ph[i];
    x_dtheta_sign[i] = data.sign_th[i];
  }

  // Set dPhi and dTheta values to 0 if there was no hit in the station
  if (!st1) {
    x_dphi[0] = 0;
    x_dphi[1] = 0;
    x_dphi[2] = 0;

    x_dtheta[0] = 0;
    x_dtheta[1] = 0;
    x_dtheta[2] = 0;
  }
  if (!st2) {
    x_dphi[0] = 0;
    x_dphi[3] = 0;
    x_dphi[4] = 0;

    x_dtheta[0] = 0;
    x_dtheta[3] = 0;
    x_dtheta[4] = 0;
  }
  if (!st3) {
    x_dphi[1] = 0;
    x_dphi[3] = 0;
    x_dphi[5] = 0;

    x_dtheta[1] = 0;
    x_dtheta[3] = 0;
    x_dtheta[5] = 0;
  }
  if (!st4) {
    x_dphi[2] = 0;
    x_dphi[4] = 0;
    x_dphi[5] = 0;

    x_dtheta[2] = 0;
    x_dtheta[4] = 0;
    x_dtheta[5] = 0;
  }

  x_trk_theta[0] = track.Theta_fp();

  // Set NN inputs
  feature = {{x_dphi[0],        x_dphi[1],        x_dphi[2],        x_dphi[3],        x_dphi[4],
              x_dphi[5],        x_dphi_sign[0],   x_dphi_sign[1],   x_dphi_sign[2],   x_dphi_sign[3],
              x_dphi_sign[4],   x_dphi_sign[5],   x_dtheta[0],      x_dtheta[1],      x_dtheta[2],
              x_dtheta[3],      x_dtheta[4],      x_dtheta[5],      x_dtheta_sign[0], x_dtheta_sign[1],
              x_dtheta_sign[2], x_dtheta_sign[3], x_dtheta_sign[4], x_dtheta_sign[5], x_csc_pattern[0],
              x_csc_pattern[1], x_csc_pattern[2], x_csc_pattern[3], x_trk_theta[0]}};

  return;
}

Member Data Documentation

graphDefDxy_

tensorflow::GraphDef* PtAssignmentEngineDxy::graphDefDxy_

protected

Definition at line 34 of file PtAssignmentEngineDxy.h.

Referenced by configure(), and ~PtAssignmentEngineDxy().

inputNameDxy_

std::string PtAssignmentEngineDxy::inputNameDxy_

protected

Definition at line 38 of file PtAssignmentEngineDxy.h.

Referenced by call_tensorflow_dxy(), and configure().

outputNamesDxy_

std::vector<std::string> PtAssignmentEngineDxy::outputNamesDxy_

protected

Definition at line 39 of file PtAssignmentEngineDxy.h.

Referenced by call_tensorflow_dxy(), and configure().

pbFileNameDxy_

std::string PtAssignmentEngineDxy::pbFileNameDxy_

protected

Definition at line 36 of file PtAssignmentEngineDxy.h.

Referenced by configure().

pbFilePathDxy_

std::string PtAssignmentEngineDxy::pbFilePathDxy_

protected

Definition at line 37 of file PtAssignmentEngineDxy.h.

Referenced by configure().

sessionDxy_

tensorflow::Session* PtAssignmentEngineDxy::sessionDxy_

protected

Definition at line 35 of file PtAssignmentEngineDxy.h.

Referenced by call_tensorflow_dxy(), configure(), and ~PtAssignmentEngineDxy().

verbose_

int PtAssignmentEngineDxy::verbose_

protected

Definition at line 32 of file PtAssignmentEngineDxy.h.

Referenced by configure().