FlavorTagDiscriminants::getter_utils::SeqGetter< T > Class Template Reference

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

#include <CustomGetterUtils.h>

Collaboration diagram for FlavorTagDiscriminants::getter_utils::SeqGetter< T >:

Public Types
using	Const = Constituents< T >
using	InputSequence = std::function< std::pair< std::string, std::vector< double > >(const xAOD::Jet &, const Const &)>

Public Member Functions
	SeqGetter (std::vector< InputVariableConfig > inputs, const FTagOptions &options)
std::pair< std::vector< float >, std::vector< int64_t > >	getFeats (const xAOD::Jet &jet, const Const &constituents) const
std::map< std::string, std::vector< double > >	getDL2Feats (const xAOD::Jet &jet, const Const &constituents) const
std::set< std::string >	getDependencies () const
std::set< std::string >	getUsedRemap () const

Private Member Functions
std::pair< InputSequence, std::set< std::string > >	getNamedCustomSeqGetter (const std::string &name, const std::string &prefix)
std::pair< InputSequence, std::set< std::string > >	seqFromConsituents (const InputVariableConfig &cfg, const FTagOptions &options)

Private Attributes
std::vector< InputSequence >	m_sequence_getters
std::set< std::string >	m_deps
std::set< std::string >	m_used_remap

Detailed Description

template<typename T>
class FlavorTagDiscriminants::getter_utils::SeqGetter< 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 70 of file CustomGetterUtils.h.

Member Typedef Documentation

Const

template<typename T >

using FlavorTagDiscriminants::getter_utils::SeqGetter< T >::Const = Constituents<T>

Definition at line 72 of file CustomGetterUtils.h.

InputSequence

template<typename T >

using FlavorTagDiscriminants::getter_utils::SeqGetter< T >::InputSequence = std::function<std::pair<std::string, std::vector<double> >( const xAOD::Jet&, const Const&)>

Definition at line 73 of file CustomGetterUtils.h.

Constructor & Destructor Documentation

SeqGetter()

template<typename T >

FlavorTagDiscriminants::getter_utils::SeqGetter< T >::SeqGetter	(	std::vector< InputVariableConfig >	inputs,
		const FTagOptions &	options
	)

Definition at line 389 of file CustomGetterUtils.cxx.

    {
      std::map<std::string, std::string> remap = options.remap_scalar;
      for (const InputVariableConfig& input_cfg: inputs) {
        auto [seqGetter, seq_deps] = seqFromConsituents(input_cfg, options);
 
        if(input_cfg.flip_sign){
          auto seqGetter_flip=[g=seqGetter](const xAOD::Jet&jet, const Const& constituents){
            auto [n,v] = g(jet,constituents);
            std::for_each(v.begin(), v.end(), [](double &n){ n=-1.0*n; });
            return std::make_pair(n,v);
          };
          m_sequence_getters.push_back(seqGetter_flip);
        }
        else{
          m_sequence_getters.push_back(seqGetter);
        }
        m_deps.merge(seq_deps);
        if (auto h = remap.extract(input_cfg.name)){
          m_used_remap.insert(h.key());
        }
      }
    }

Member Function Documentation

getDependencies()

template<typename T >

std::set< std::string > FlavorTagDiscriminants::getter_utils::SeqGetter< T >::getDependencies

Definition at line 452 of file CustomGetterUtils.cxx.

                                                          {
      return m_deps;
    }

getDL2Feats()

template<typename T >

std::map< std::string, std::vector< double > > FlavorTagDiscriminants::getter_utils::SeqGetter< T >::getDL2Feats	(	const xAOD::Jet &	jet,
		const Const &	constituents
	)		const

Definition at line 441 of file CustomGetterUtils.cxx.

    {
      std::map<std::string, std::vector<double>> feats;
      for (const auto& seq_getter: m_sequence_getters){
        feats.insert(seq_getter(jet, constituents));
      }
      return feats;
    }

getFeats()

template<typename T >

std::pair< std::vector< float >, std::vector< int64_t > > FlavorTagDiscriminants::getter_utils::SeqGetter< T >::getFeats	(	const xAOD::Jet &	jet,
		const Const &	constituents
	)		const

Definition at line 414 of file CustomGetterUtils.cxx.

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

getNamedCustomSeqGetter()

template<typename T >

std::pair< typename SeqGetter< T >::InputSequence, std::set< std::string > > FlavorTagDiscriminants::getter_utils::SeqGetter< T >::getNamedCustomSeqGetter ( const std::string &  name, const std::string &  prefix  )

private

Definition at line 351 of file CustomGetterUtils.cxx.

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

getUsedRemap()

template<typename T >

std::set< std::string > FlavorTagDiscriminants::getter_utils::SeqGetter< T >::getUsedRemap

Definition at line 456 of file CustomGetterUtils.cxx.

                                                       {
      return m_used_remap;
    }

seqFromConsituents()

template<typename T >

std::pair< typename SeqGetter< T >::InputSequence, std::set< std::string > > FlavorTagDiscriminants::getter_utils::SeqGetter< T >::seqFromConsituents ( const InputVariableConfig &  cfg, const FTagOptions &  options  )

private

Definition at line 363 of file CustomGetterUtils.cxx.

                                                                                              {
      const std::string prefix = options.track_prefix;
      switch (cfg.type) {
        case ConstituentsEDMType::INT: return {
            NamedSeqGetter<int, T>(cfg.name), {cfg.name}
          };
        case ConstituentsEDMType::FLOAT: return {
            NamedSeqGetter<float, T>(cfg.name), {cfg.name}
          };
        case ConstituentsEDMType::CHAR: return {
            NamedSeqGetter<char, T>(cfg.name), {cfg.name}
          };
        case ConstituentsEDMType::UCHAR: return {
            NamedSeqGetter<unsigned char, T>(cfg.name), {cfg.name}
          };
        case ConstituentsEDMType::CUSTOM_GETTER: {
          return getNamedCustomSeqGetter(
            cfg.name, options.track_prefix);
        }
        default: {
          throw std::logic_error("Unknown EDM type for constituent.");
        }
      }
    }

Member Data Documentation

m_deps

template<typename T >

std::set<std::string> FlavorTagDiscriminants::getter_utils::SeqGetter< T >::m_deps

private

Definition at line 93 of file CustomGetterUtils.h.

m_sequence_getters

template<typename T >

std::vector<InputSequence> FlavorTagDiscriminants::getter_utils::SeqGetter< T >::m_sequence_getters

private

Definition at line 92 of file CustomGetterUtils.h.

m_used_remap

template<typename T >

std::set<std::string> FlavorTagDiscriminants::getter_utils::SeqGetter< T >::m_used_remap

private

Definition at line 94 of file CustomGetterUtils.h.

---

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

• CustomGetterUtils.h
• CustomGetterUtils.cxx