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 (std::string name, std::vector< std::string > input_variables)

Enumeration Type Documentation

ConstituentsEDMType

enum 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 InDetGNNHardScatterSelection::ConstituentsSelection

strong

Enumerator
ALL

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

                                     {
        ALL
    };

ConstituentsSortOrder

enum InDetGNNHardScatterSelection::ConstituentsSortOrder

strong

Enumerator
PT_DESCENDING

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

                                     {
        PT_DESCENDING
    };

ConstituentsType

enum 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 InDetGNNHardScatterSelection::EDMType

strong

Enumerator
CHAR
UCHAR
INT
FLOAT
DOUBLE
CUSTOM_GETTER

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

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

Function Documentation

createConstituentsLoaderConfig()

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

Definition at line 49 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|vertexWeight|ntracks_ga|photon_deltaZ|photon_deltaZ_wBeamSpot)"_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;
    }