ATLAS Offline Software
Loading...
Searching...
No Matches
InDetGNNHardScatterSelection Namespace Reference

Namespaces

namespace  dataprep
namespace  getter_utils
 Utils to produce Constituent -> vector<double> functions.
namespace  internal

Classes

struct  ConstituentsInputConfig
class  ElectronsLoader
class  GNN
 Implementation of the GNN used by the InDetGNNHardScatterSelection::GNNTool. More...
class  GNNTool
struct  HSGNNInputConfig
class  IConstituentsLoader
struct  InputVariableConfig
class  IParticlesLoader
class  JetsLoader
class  MuonsLoader
class  PhotonsLoader
class  TracksLoader
class  VertexDecoratorAlg

Enumerations

enum class  ConstituentsEDMType {
  CHAR , UCHAR , INT , FLOAT ,
  DOUBLE , CUSTOM_GETTER
}
enum class  ConstituentsSortOrder { PT_DESCENDING }
enum class  ConstituentsSelection { ALL }
enum class  ConstituentsType {
  IPARTICLE , TRACK , ELECTRON , MUON ,
  JET , PHOTON
}
enum class  EDMType {
  CHAR , UCHAR , INT , FLOAT ,
  DOUBLE , CUSTOM_GETTER
}

Functions

ConstituentsInputConfig createConstituentsLoaderConfig (const std::string &name, const std::vector< std::string > &input_variables)

Enumeration Type Documentation

◆ ConstituentsEDMType

◆ ConstituentsSelection

Enumerator
ALL 

Definition at line 28 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/ConstituentsLoader.h.

28 {
29 ALL
30 };
int ALL
message levels --------------------------------------------------------—

◆ ConstituentsSortOrder

◆ ConstituentsType

◆ EDMType

Enumerator
CHAR 
UCHAR 
INT 
FLOAT 
DOUBLE 
CUSTOM_GETTER 

Definition at line 28 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/DataPrepUtilities.h.

Function Documentation

◆ createConstituentsLoaderConfig()

ConstituentsInputConfig InDetGNNHardScatterSelection::createConstituentsLoaderConfig ( const std::string & name,
const std::vector< std::string > & input_variables )

Definition at line 49 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/ConstituentsLoader.cxx.

52 {
54
55 TypeRegexes iparticle_type_regexes {
56 // iparticle variables
57 // ConstituentsEDMType picked correspond to the first matching regex
58 // {"(photon_deltaZ|photon_deltaZ_wBeamSpot)"_r, ConstituentsEDMType::FLOAT},
59 {"(pt|eta|phi|energy|deltaZ0|vertexWeight|ntracks_ga|photon_deltaZ|photon_deltaZ_wBeamSpot)"_r, ConstituentsEDMType::CUSTOM_GETTER}
60 };
61
62 if (name.find("tracks_all_sd0sort") != std::string::npos){
63 config = get_iparticle_input_config(
64 name, input_variables,
65 iparticle_type_regexes);
67 config.link_name = "trackParticleLinks";
68 config.output_name = "track_features";
69 }
70 else if (name.find("electrons") != std::string::npos){
71 config = get_iparticle_input_config(
72 name, input_variables,
73 iparticle_type_regexes);
75 config.link_name = "electronLinks";
76 config.output_name = "electron_features";
77 }
78 else if (name.find("muons") != std::string::npos){
79 config = get_iparticle_input_config(
80 name, input_variables,
81 iparticle_type_regexes);
83 config.link_name = "muonLinks";
84 config.output_name = "muon_features";
85 }
86 else if (name.find("jets") != std::string::npos){
87 config = get_iparticle_input_config(
88 name, input_variables,
89 iparticle_type_regexes);
91 config.link_name = "jetLinks";
92 config.output_name = "jet_features";
93 }
94 else if (name.find("photons") != std::string::npos){
95 config = get_iparticle_input_config(
96 name, input_variables,
97 iparticle_type_regexes);
99 config.link_name = "photonLinks";
100 config.output_name = "photon_features";
101 }
102 else{
103 throw std::runtime_error(
104 "Unknown constituent type: " + name
105 );
106 }
107 return config;
108 }