CustomGetterUtils.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
 
 
// EDM includes
#include "xAODJet/JetFwd.h"
#include "xAODTracking/TrackParticleFwd.h"
#include "xAODBase/IParticle.h"
#include "AthContainers/AuxElement.h"
#include "FlavorTagDiscriminants/DataPrepUtilities.h"
 
#include <functional>
#include <string>
#include <set>
 
#ifndef CUSTOM_GETTER_UTILS_H
#define CUSTOM_GETTER_UTILS_H
 
namespace FlavorTagDiscriminants {
 
  namespace getter_utils {
    
    // -------------------------------------------------------
    // type aliases
    template <typename T>
    using Constituents = std::vector<const T*>;
 
    template <typename T>
    using SequenceGetterFunc = std::function<std::vector<double>(const xAOD::Jet&, const Constituents<T>&)>;
    // -------------------------------------------------------
 
    std::function<std::pair<std::string, double>(const xAOD::Jet&)>
    namedCustomJetGetter(const std::string&);
 
    template <typename T>
    std::pair<SequenceGetterFunc<T>, std::set<std::string>>
    buildCustomSeqGetter(const std::string& name, const std::string& prefix);
 
    template <typename T>
    class SeqGetter {
        public:
          using Const = Constituents<T>;
          using InputSequence = std::function<std::pair<std::string, std::vector<double>>(
              const xAOD::Jet&,
              const Const&)>;
 
          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:
          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);
          std::vector<InputSequence> m_sequence_getters;
          std::set<std::string> m_deps;
          std::set<std::string> m_used_remap;        
        };
    }
}
 
#endif