InDet::GNNTrackFinderTritonTool Class Reference

InDet::GNNTrackFinderTritonTool is a tool that produces track candidates with graph neural networks-based pipeline using space points as inputs. More...

#include <GNNTrackFinderTritonTool.h>

Inheritance diagram for InDet::GNNTrackFinderTritonTool:

Collaboration diagram for InDet::GNNTrackFinderTritonTool:

Public Member Functions
virtual StatusCode	initialize () override
virtual StatusCode	getTracks (const std::vector< const Trk::SpacePoint * > &spacepoints, std::vector< std::vector< uint32_t > > &tracks) const override
	Get track candidates from a list of space points. More...
virtual MsgStream &	dump (MsgStream &out) const override
virtual std::ostream &	dump (std::ostream &out) const override

Private Member Functions
MsgStream &	dumpevent (MsgStream &out) const

Private Attributes
ToolHandle< AthInfer::IAthInferenceTool >	m_gnnTrackingTritonTool
ToolHandle< ISpacepointFeatureTool >	m_spacepointFeatureTool
StringProperty	m_featureNames
std::vector< std::string >	m_featureNamesVec

Detailed Description

InDet::GNNTrackFinderTritonTool is a tool that produces track candidates with graph neural networks-based pipeline using space points as inputs.

Author

xiang.nosp@m.yang.nosp@m..ju@c.nosp@m.ern..nosp@m.ch

Definition at line 28 of file GNNTrackFinderTritonTool.h.

Member Function Documentation

dump() [1/2]

MsgStream & InDet::GNNTrackFinderTritonTool::dump ( MsgStream &  out ) const

overridevirtual

Definition at line 75 of file GNNTrackFinderTritonTool.cxx.

{
  out<<std::endl;
  return dumpevent(out);
}

dump() [2/2]

std::ostream & InDet::GNNTrackFinderTritonTool::dump ( std::ostream &  out ) const

overridevirtual

Definition at line 81 of file GNNTrackFinderTritonTool.cxx.

{
  return out;
}

dumpevent()

MsgStream & InDet::GNNTrackFinderTritonTool::dumpevent ( MsgStream &  out ) const

private

Definition at line 86 of file GNNTrackFinderTritonTool.cxx.

{
  out<<"|---------------------------------------------------------------------|"
       <<std::endl;
  out<<"| Number output tracks    | "<<std::setw(12)
     <<"                              |"<<std::endl;
  out<<"|---------------------------------------------------------------------|"
     <<std::endl;
  return out;
}

getTracks()

StatusCode InDet::GNNTrackFinderTritonTool::getTracks ( const std::vector< const Trk::SpacePoint * > &  spacepoints, std::vector< std::vector< uint32_t > > &  tracks  ) const

overridevirtual

Get track candidates from a list of space points.

Parameters

spacepoints	a list of spacepoints as inputs to the GNN-based track finder.
tracks	a list of track candidates.

Returns

Definition at line 24 of file GNNTrackFinderTritonTool.cxx.

                                                   {
  int64_t numSpacepoints = (int64_t)spacepoints.size();
  std::vector<float> inputValues;
  std::vector<uint32_t> spacepointIDs;
 
  int64_t spacepointFeatures = m_featureNamesVec.size();
  int sp_idx = 0;
  for (const auto& sp : spacepoints) {
    // depending on the trained embedding and GNN models, the input features
    // may need to be updated.
    auto featureMap = m_spacepointFeatureTool->getFeatures(sp);
    for (int i = 0; i < spacepointFeatures; i++){
      inputValues.push_back(featureMap[m_featureNamesVec[i]]);
    }
 
    spacepointIDs.push_back(sp_idx++);
  }
 
  AthInfer::InputDataMap inputData;
  inputData["FEATURES"] = std::make_pair(
      std::vector<int64_t>{numSpacepoints, spacepointFeatures}, std::move(inputValues));
 
  AthInfer::OutputDataMap outputData;
  outputData["LABELS"] = std::make_pair(std::vector<int64_t>{numSpacepoints, 1}, std::vector<int64_t>{});
 
  ATH_CHECK(m_gnnTrackingTritonTool->inference(inputData, outputData));
 
  auto& trackLabels = std::get<std::vector<int64_t>>(outputData["LABELS"].second);
  if (trackLabels.size() == 0){
    ATH_MSG_DEBUG("No tracks found in the event.");
    return StatusCode::SUCCESS;
  }
 
  tracks.clear();
  std::vector<uint32_t> this_track;
  for (auto label : trackLabels) {
    if (label == -1) {
      if (this_track.size() > 0) {
        tracks.push_back(this_track);
        this_track.clear();
      }
    } else {
      this_track.push_back(label);
    }
  }
 
  return StatusCode::SUCCESS;
}

initialize()

StatusCode InDet::GNNTrackFinderTritonTool::initialize ( )

overridevirtual

Definition at line 15 of file GNNTrackFinderTritonTool.cxx.

                                                     {
  ATH_CHECK(m_gnnTrackingTritonTool.retrieve());
  ATH_CHECK(m_spacepointFeatureTool.retrieve());
 
  // tokenize the feature names by comma and push to the vector
  m_featureNamesVec = CxxUtils::tokenize(m_featureNames, ",");
  return StatusCode::SUCCESS;
}

Member Data Documentation

m_featureNames

StringProperty InDet::GNNTrackFinderTritonTool::m_featureNames

private

Initial value:

{this, "FeatureNames",
                                "r,phi,z,cluster_x_1,cluster_y_1,cluster_z_1,cluster_x_2,cluster_y_2,cluster_z_2,count_1,charge_count_1,loc_eta_1,loc_phi_1,localDir0_1,localDir1_1,localDir2_1,lengthDir0_1,lengthDir1_1,lengthDir2_1,glob_eta_1,glob_phi_1,eta_angle_1,phi_angle_1,count_2,charge_count_2,loc_eta_2,loc_phi_2,localDir0_2,localDir1_2,localDir2_2,lengthDir0_2,lengthDir1_2,lengthDir2_2,glob_eta_2,glob_phi_2,eta_angle_2,phi_angle_2,eta,cluster_r_1,cluster_phi_1,cluster_eta_1,cluster_r_2,cluster_phi_2,cluster_eta_2",
                                "Feature names for the GNN pipeline"}

Definition at line 59 of file GNNTrackFinderTritonTool.h.

m_featureNamesVec

std::vector<std::string> InDet::GNNTrackFinderTritonTool::m_featureNamesVec

private

Definition at line 63 of file GNNTrackFinderTritonTool.h.

m_gnnTrackingTritonTool

ToolHandle<AthInfer::IAthInferenceTool> InDet::GNNTrackFinderTritonTool::m_gnnTrackingTritonTool

private

Initial value:

{
      this, "TritonTool", "AthInfer::TritonTool"}

Definition at line 54 of file GNNTrackFinderTritonTool.h.

m_spacepointFeatureTool

ToolHandle<ISpacepointFeatureTool> InDet::GNNTrackFinderTritonTool::m_spacepointFeatureTool

private

Initial value:

{
      this, "SpacepointFeatureTool", "InDet::SpacepointFeatureTool"}

Definition at line 56 of file GNNTrackFinderTritonTool.h.

---

The documentation for this class was generated from the following files:

• GNNTrackFinderTritonTool.h
• GNNTrackFinderTritonTool.cxx