InDetGNNHardScatterSelection::dataprep Namespace Reference

Typedefs
typedef std::vector< std::pair< std::regex, std::string > >	StringRegexes

Functions
std::tuple< std::vector< HSGNNInputConfig >, std::vector< ConstituentsInputConfig > >	createGetterConfig (lwt::GraphConfig &graph_config)
std::vector< internal::VarFromVertex >	createVertexVarGetters (const std::vector< HSGNNInputConfig > &inputs)

Typedef Documentation

StringRegexes

typedef std::vector<std::pair<std::regex, std::string> > InDetGNNHardScatterSelection::dataprep::StringRegexes

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

Function Documentation

createGetterConfig()

std::tuple< std::vector< HSGNNInputConfig >, std::vector< ConstituentsInputConfig > > InDetGNNHardScatterSelection::dataprep::createGetterConfig

(

lwt::GraphConfig &

graph_config

)

Definition at line 118 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/DataPrepUtilities.cxx.

    {
 
      // build the standard inputs
 
      // type and default value-finding regexes are hardcoded for now
      TypeRegexes type_regexes = {
        {".*_isDefaults"_r, EDMType::CHAR},
        {"sumPt2|sumPt|chi2Over_ndf|z_asymmetry|weighted_z_asymmetry|z_kurtosis|z_skewness|photon_deltaz|photon_deltaPhi|actualIntPerXing"_r, EDMType::FLOAT},
        {"ntrk"_r, EDMType::INT},
        {"(log_)?pt|abs_eta|eta|phi|energy|mass"_r, EDMType::CUSTOM_GETTER},
      };
 
      StringRegexes default_flag_regexes{
        {"ntrk|sumPt2|sumPt|chi2Over_ndf|z_asymmetry|weighted_z_asymmetry|z_kurtosis|z_skewness|photon_deltaz|photon_deltaPhi|actualIntPerXing"_r, ""},
        {"((log_)?pt|abs_eta|eta|phi|energy|mass)"_r, ""}}; // no default for custom cases
 
      std::vector<HSGNNInputConfig> input_config;
      for (auto& node: config.inputs){
        std::vector<std::string> input_names;
        for (const auto& var: node.variables) {
          input_names.push_back(var.name);
        }
        input_config = get_input_config(input_names, type_regexes, default_flag_regexes);
      }
 
      // build the constituents inputs
      std::vector<std::pair<std::string, std::vector<std::string>>> constituent_names;
      for (auto& node: config.input_sequences) {
 
        std::vector<std::string> names;
        for (const auto& var: node.variables) {
          names.push_back(var.name);
        }
        constituent_names.emplace_back(node.name, names);
      }
 
      std::vector<ConstituentsInputConfig> constituent_configs;
      for (auto el: constituent_names){
        constituent_configs.push_back(
          createConstituentsLoaderConfig(el.first, el.second));
      }
 
      return std::make_tuple(input_config, constituent_configs);
    }

createVertexVarGetters()

std::vector< internal::VarFromVertex > InDetGNNHardScatterSelection::dataprep::createVertexVarGetters

(

const std::vector< HSGNNInputConfig > &

inputs

)

Definition at line 168 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/DataPrepUtilities.cxx.

    {
      std::vector<internal::VarFromVertex> varsFromVertex;
 
      for (const auto& input: inputs) {
        if (input.type != EDMType::CUSTOM_GETTER) {
          auto filler = internal::get::varFromVertex(input.name, input.type,
                                         input.default_flag);
          varsFromVertex.push_back(filler);
        } else {
          varsFromVertex.push_back(getter_utils::customGetterAndName(input.name));
        }
      }
 
      return varsFromVertex;
    }