ATLAS Offline Software
Namespaces | Classes | Enumerations | Functions
InDetGNNHardScatterSelection Namespace Reference

Namespaces

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

Classes

struct  ConstituentsInputConfig
 
class  ElectronsLoader
 
class  GNN
 
class  GNNTool
 
struct  HSGNNInputConfig
 
class  IConstituentsLoader
 
struct  InputVariableConfig
 
class  IParticlesLoader
 
class  JetsLoader
 
class  MuonsLoader
 
class  PhotonsLoader
 
class  TracksLoader
 
class  VertexDecoratorAlg
 

Enumerations

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

Functions

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

Enumeration Type Documentation

◆ ConstituentsEDMType

Enumerator
CHAR 
UCHAR 
INT 
FLOAT 
DOUBLE 
CUSTOM_GETTER 

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

◆ ConstituentsSelection

◆ ConstituentsSortOrder

◆ ConstituentsType

Enumerator
IPARTICLE 
TRACK 
ELECTRON 
MUON 
JET 
PHOTON 

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

31  {
32  IPARTICLE,
33  TRACK,
34  ELECTRON,
35  MUON,
36  JET,
37  PHOTON
38  };

◆ EDMType

Enumerator
CHAR 
UCHAR 
INT 
FLOAT 
DOUBLE 
CUSTOM_GETTER 

Definition at line 29 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);
74  config.type = ConstituentsType::ELECTRON;
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);
98  config.type = ConstituentsType::PHOTON;
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  }
python.base_data.config
config
Definition: base_data.py:21
InDetGNNHardScatterSelection::ConstituentsInputConfig
Definition: InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/ConstituentsLoader.h:45
ALL
@ ALL
Definition: sTGCenumeration.h:14
InDetGNNHardScatterSelection::ConstituentsEDMType::CHAR
@ CHAR
EMAmbiguityType::ELECTRON
@ ELECTRON
Definition: EMAmbiguityToolDefs.h:11
InDetGNNHardScatterSelection::ConstituentsType::PHOTON
@ PHOTON
config
Definition: PhysicsAnalysis/AnalysisCommon/AssociationUtils/python/config.py:1
InDetGNNHardScatterSelection::ConstituentsEDMType::UCHAR
@ UCHAR
TCS::JET
@ JET
Definition: Trigger/TrigT1/L1Topo/L1TopoCommon/L1TopoCommon/Types.h:12
InDetGNNHardScatterSelection::ConstituentsEDMType::INT
@ INT
MUON
xAOD::Muon MUON
D3PD INCLUDES.
Definition: TileCellFillerTool.h:37
InDetGNNHardScatterSelection::ConstituentsEDMType::FLOAT
@ FLOAT
InDetGNNHardScatterSelection::ConstituentsEDMType::CUSTOM_GETTER
@ CUSTOM_GETTER
InDetGNNHardScatterSelection::ConstituentsEDMType::DOUBLE
@ DOUBLE
name
std::string name
Definition: Control/AthContainers/Root/debug.cxx:228
InDetGNNHardScatterSelection::ConstituentsType::IPARTICLE
@ IPARTICLE
InDetGNNHardScatterSelection::ConstituentsSortOrder::PT_DESCENDING
@ PT_DESCENDING
TRACK
xAOD::TrackParticle TRACK
Definition: D3PDMaker/TileD3PDMaker/src/ITrackTools.h:86