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

enum class InDetGNNHardScatterSelection::ConstituentsEDMType

strong

Enumerator
CHAR
UCHAR
INT
FLOAT
DOUBLE
CUSTOM_GETTER

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

{CHAR, UCHAR, INT, FLOAT, DOUBLE, CUSTOM_GETTER};

ConstituentsSelection

enum class InDetGNNHardScatterSelection::ConstituentsSelection

strong

Enumerator
ALL

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

                                     {
        ALL
    };

ConstituentsSortOrder

enum class InDetGNNHardScatterSelection::ConstituentsSortOrder

strong

Enumerator
PT_DESCENDING

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

                                     {
        PT_DESCENDING
    };

ConstituentsType

enum class InDetGNNHardScatterSelection::ConstituentsType

strong

Enumerator
IPARTICLE
TRACK
ELECTRON
MUON
JET
PHOTON

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

                                {
        IPARTICLE,
        TRACK,
        ELECTRON,
        MUON,
        JET,
        PHOTON
    };

EDMType

enum class InDetGNNHardScatterSelection::EDMType

strong

Enumerator
CHAR
UCHAR
INT
FLOAT
DOUBLE
CUSTOM_GETTER

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

{CHAR, UCHAR, INT, FLOAT, DOUBLE, CUSTOM_GETTER};

Function Documentation

createConstituentsLoaderConfig()

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

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

     {
      ConstituentsInputConfig config;
 
      TypeRegexes iparticle_type_regexes {
          // iparticle variables
          // ConstituentsEDMType picked correspond to the first matching regex
  //        {"(photon_deltaZ|photon_deltaZ_wBeamSpot)"_r, ConstituentsEDMType::FLOAT},
          {"(pt|eta|phi|energy|deltaZ0|trkpt|trketa|trkphi|gsftrkpt|gsftrketa|gsftrkphi|cluster_eta|cluster_phi|isAmbiguous|vertexWeight|ntracks_ga|photon_deltaZ|photon_deltaZ_wBeamSpot|d0|z0)"_r, ConstituentsEDMType::CUSTOM_GETTER}
      };
      
      if (name.find("tracks_all_sd0sort") != std::string::npos){
        config = get_iparticle_input_config(
          name, input_variables,
          iparticle_type_regexes);
        config.type = ConstituentsType::TRACK;
        config.link_name = "trackParticleLinks";
        config.output_name = "track_features";
      }
      else if (name.find("electrons") != std::string::npos){
        config = get_iparticle_input_config(
          name, input_variables,
          iparticle_type_regexes);
        config.type = ConstituentsType::ELECTRON;
        config.link_name = "electronLinks";
        config.output_name = "electron_features";
      }
      else if (name.find("muons") != std::string::npos){
        config = get_iparticle_input_config(
          name, input_variables,
          iparticle_type_regexes);
        config.type = ConstituentsType::MUON;
        config.link_name = "muonLinks";
        config.output_name = "muon_features";
      }
      else if (name.find("jets") != std::string::npos){
        config = get_iparticle_input_config(
          name, input_variables,
          iparticle_type_regexes);
        config.type = ConstituentsType::JET;
        config.link_name = "jetLinks";
        config.output_name = "jet_features";
      }
      else if (name.find("photons") != std::string::npos){
        config = get_iparticle_input_config(
          name, input_variables,
          iparticle_type_regexes);
        config.type = ConstituentsType::PHOTON;
        config.link_name = "photonLinks";
        config.output_name = "photon_features";
      }
      else{
        throw std::runtime_error(
          "Unknown constituent type: " + name
          );
      }
      return config;
    }