InDetGNNHardScatterSelection::getter_utils::CustomSequenceGetter< T > Class Template Reference

Template class to extract features from sequence of constituents. More...

#include <CustomGetterUtils.h>

Collaboration diagram for InDetGNNHardScatterSelection::getter_utils::CustomSequenceGetter< T >:

Public Types
using	Constituents = std::vector< const T * >
using	NamedSequenceFromConstituents = std::function< std::pair< std::string, std::vector< double > >(const xAOD::Vertex &, const Constituents &)>

Public Member Functions
	CustomSequenceGetter (const std::vector< InputVariableConfig > &inputs)
std::pair< std::vector< float >, std::vector< int64_t > >	getFeats (const xAOD::Vertex &jet, const Constituents &constituents) const

Private Member Functions
std::pair< NamedSequenceFromConstituents, std::set< std::string > >	customNamedSeqGetterWithDeps (const std::string &name)
std::pair< NamedSequenceFromConstituents, std::set< std::string > >	seqFromConsituents (const InputVariableConfig &cfg)

Private Attributes
std::vector< NamedSequenceFromConstituents >	m_sequencesFromConstituents

Detailed Description

template<typename T>
class InDetGNNHardScatterSelection::getter_utils::CustomSequenceGetter< T >

Template class to extract features from sequence of constituents.

Template Parameters

T	constituent type

It supports the following types of constituents:

• xAOD::IParticle
• xAOD::TrackParticle

Definition at line 81 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/CustomGetterUtils.h.

Member Typedef Documentation

Constituents

template<typename T >

using InDetGNNHardScatterSelection::getter_utils::CustomSequenceGetter< T >::Constituents = std::vector<const T*>

Definition at line 83 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/CustomGetterUtils.h.

NamedSequenceFromConstituents

template<typename T >

using InDetGNNHardScatterSelection::getter_utils::CustomSequenceGetter< T >::NamedSequenceFromConstituents = std::function<std::pair<std::string, std::vector<double> >( const xAOD::Vertex&, const Constituents&)>

Definition at line 84 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/CustomGetterUtils.h.

Constructor & Destructor Documentation

CustomSequenceGetter()

template<typename T >

InDetGNNHardScatterSelection::getter_utils::CustomSequenceGetter< T >::CustomSequenceGetter

(

const std::vector< InputVariableConfig > &

inputs

)

Definition at line 263 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/CustomGetterUtils.cxx.

    {
        for (const InputVariableConfig& input_cfg: inputs) {
          auto [seqGetter, seq_deps] = seqFromConsituents(input_cfg);
 
          m_sequencesFromConstituents.push_back(seqGetter);
        }
    }

Member Function Documentation

customNamedSeqGetterWithDeps()

template<typename T >

std::pair< typename CustomSequenceGetter< T >::NamedSequenceFromConstituents, std::set< std::string > > InDetGNNHardScatterSelection::getter_utils::CustomSequenceGetter< T >::customNamedSeqGetterWithDeps ( const std::string &  name )

private

Definition at line 224 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/CustomGetterUtils.cxx.

                                                                               {
      auto [getter, deps] = customSequenceGetterWithDeps<T>(name);
      return {
        [n=name, g=getter](const xAOD::Vertex& j,
                       const std::vector<const T*>& t) {
          return std::make_pair(n, g(j, t));
        },
        deps
      };
    }

getFeats()

template<typename T >

std::pair< std::vector< float >, std::vector< int64_t > > InDetGNNHardScatterSelection::getter_utils::CustomSequenceGetter< T >::getFeats	(	const xAOD::Vertex &	jet,
		const Constituents &	constituents
	)		const

Definition at line 274 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/CustomGetterUtils.cxx.

    {
      std::vector<float> cnsts_feats;
      int num_vars = m_sequencesFromConstituents.size();
      int num_cnsts = 0;
 
      int cnst_var_idx = 0;
      for (const auto& seq_builder: m_sequencesFromConstituents){
        auto double_vec = seq_builder(vertex, constituents).second;
 
        if (cnst_var_idx==0){
            num_cnsts = static_cast<int>(double_vec.size());
            cnsts_feats.resize(num_cnsts * num_vars);
        }
 
        // need to transpose + flatten
        for (unsigned int cnst_idx=0; cnst_idx<double_vec.size(); cnst_idx++){
          cnsts_feats.at(cnst_idx*num_vars + cnst_var_idx)
              = double_vec.at(cnst_idx);
        }
        cnst_var_idx++;
      }
      std::vector<int64_t> cnsts_feat_dim = {num_cnsts, num_vars};
      return {cnsts_feats, cnsts_feat_dim};
    }

seqFromConsituents()

template<typename T >

std::pair< typename CustomSequenceGetter< T >::NamedSequenceFromConstituents, std::set< std::string > > InDetGNNHardScatterSelection::getter_utils::CustomSequenceGetter< T >::seqFromConsituents ( const InputVariableConfig &  cfg )

private

Definition at line 237 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/CustomGetterUtils.cxx.

                                       {
      switch (cfg.type) {
        case ConstituentsEDMType::INT: return {
            SequenceGetter<int, T>(cfg.name), {cfg.name}
          };
        case ConstituentsEDMType::FLOAT: return {
            SequenceGetter<float, T>(cfg.name), {cfg.name}
          };
        case ConstituentsEDMType::CHAR: return {
            SequenceGetter<char, T>(cfg.name), {cfg.name}
          };
        case ConstituentsEDMType::UCHAR: return {
            SequenceGetter<unsigned char, T>(cfg.name), {cfg.name}
          };
        case ConstituentsEDMType::CUSTOM_GETTER: {
          return customNamedSeqGetterWithDeps(
            cfg.name);
        }
        default: {
          throw std::logic_error("Unknown EDM type for constituent.");
        }
      }
    }

Member Data Documentation

m_sequencesFromConstituents

template<typename T >

std::vector<NamedSequenceFromConstituents> InDetGNNHardScatterSelection::getter_utils::CustomSequenceGetter< T >::m_sequencesFromConstituents

private

Definition at line 96 of file InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/CustomGetterUtils.h.

---

The documentation for this class was generated from the following files:

• InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/InDetGNNHardScatterSelection/CustomGetterUtils.h
• InnerDetector/InDetRecTools/InDetGNNHardScatterSelection/Root/CustomGetterUtils.cxx