FlavorTagDiscriminants::HbbTag Class Reference

#include <HbbTag.h>

Collaboration diagram for FlavorTagDiscriminants::HbbTag:

Public Member Functions
	HbbTag (const HbbTagConfig &config)
	HbbTag (HbbTag &&)
	~HbbTag ()
void	decorate (const xAOD::Jet &jet) const

Private Types
using	JetLink = ElementLink< xAOD::JetContainer >
using	PartLink = std::vector< ElementLink< xAOD::IParticleContainer > >
using	Pg = std::function< std::pair< std::string, double >(const xAOD::Jet &)>
template<typename T >
using	Decorator = SG::AuxElement::Decorator< T >
using	NodeWriter = std::vector< std::pair< std::string, Decorator< float > >>

Private Attributes
SG::AuxElement::ConstAccessor< JetLink >	m_parent_link
SG::AuxElement::ConstAccessor< PartLink >	m_subjet_link_getter
std::vector< Pg >	m_fat_jet_getters
std::vector< Pg >	m_subjet_getters
size_t	m_n_subjets
double	m_min_subjet_pt
std::unique_ptr< lwt::LightweightGraph >	m_graph
std::map< std::string, std::map< std::string, double > >	m_defaults
std::vector< std::pair< std::string, NodeWriter > >	m_outputs

Detailed Description

Definition at line 22 of file HbbTag.h.

Member Typedef Documentation

Decorator

template<typename T >

using FlavorTagDiscriminants::HbbTag::Decorator = SG::AuxElement::Decorator<T>

private

Definition at line 48 of file HbbTag.h.

JetLink

using FlavorTagDiscriminants::HbbTag::JetLink = ElementLink<xAOD::JetContainer>

private

Definition at line 32 of file HbbTag.h.

NodeWriter

using FlavorTagDiscriminants::HbbTag::NodeWriter = std::vector<std::pair<std::string, Decorator<float> >>

private

Definition at line 49 of file HbbTag.h.

PartLink

using FlavorTagDiscriminants::HbbTag::PartLink = std::vector<ElementLink<xAOD::IParticleContainer> >

private

Definition at line 34 of file HbbTag.h.

Pg

using FlavorTagDiscriminants::HbbTag::Pg = std::function<std::pair<std::string, double>(const xAOD::Jet&)>

private

Definition at line 36 of file HbbTag.h.

Constructor & Destructor Documentation

HbbTag() [1/2]

FlavorTagDiscriminants::HbbTag::HbbTag

(

const HbbTagConfig &

config

)

Definition at line 49 of file HbbTag.cxx.

                                          :
    m_parent_link("Parent"),
    m_subjet_link_getter(config.subjet_link_name),
    m_n_subjets(0),
    m_min_subjet_pt(config.min_subjet_pt)
  {
    namespace fs = std::filesystem;
    // setup NN
    fs::path nn_path = config.input_file_path;
    if (!fs::exists(nn_path)) {
      nn_path = PathResolverFindCalibFile(nn_path.string());
      if (nn_path.empty()) {
        throw std::runtime_error(
          "no file found at '" + config.input_file_path.string() + "'");
      }
    }
    std::ifstream input_stream(nn_path.string());
    lwt::GraphConfig graph_cfg = lwt::parse_json_graph(input_stream);
    m_graph.reset(new lwt::LightweightGraph(graph_cfg));
 
    // setup large-R jet getters and defaults
    HbbGraphConfig keys = getHbbGraphConfig(graph_cfg);
    // add the getters
    for (const std::string& key: keys.fatjet) {
      m_fat_jet_getters.push_back(makePairGetter(key));
    }
    for (const std::string& key: keys.subjet) {
      m_subjet_getters.push_back(makePairGetter(key));
    }
    m_defaults = keys.defaults;
    m_n_subjets = keys.n_subjets;
 
    // setup outputs
    for (const auto& output: graph_cfg.outputs) {
      const std::string& node_name = output.first;
      const lwt::OutputNodeConfig& node = output.second;
      NodeWriter node_writer;
      for (const std::string& varname: node.labels) {
        std::string write_name = node_name + "_" + varname;
        node_writer.emplace_back(varname, write_name);
      }
      m_outputs.emplace_back(node_name, node_writer);
    }
  }

HbbTag() [2/2]

FlavorTagDiscriminants::HbbTag::HbbTag ( HbbTag &&  )

default

~HbbTag()

FlavorTagDiscriminants::HbbTag::~HbbTag

(

)

Definition at line 94 of file HbbTag.cxx.

{}

Member Function Documentation

decorate()

void FlavorTagDiscriminants::HbbTag::decorate

(

const xAOD::Jet &

jet

)

const

Definition at line 96 of file HbbTag.cxx.

                                                {
    namespace hk = hbb_key;
    std::map<std::string, std::map<std::string, double>> inputs = m_defaults;
    for (const auto& getter: m_fat_jet_getters) {
      requireOverwrite(inputs.at(hk::fatjet),getter(jet));
    }
 
    std::vector<const xAOD::IParticle*> subjets;
    const xAOD::Jet* parent = *m_parent_link(jet);
    if (!parent) throw std::runtime_error("can't resolve parent jet");
    for (const auto& link: m_subjet_link_getter(*parent)) {
      const xAOD::IParticle* subjet = *link;
      if (!subjet) throw std::runtime_error("can't resolve subjet link");
      if (subjet->pt() >= m_min_subjet_pt) {
        subjets.push_back(subjet);
      }
    }
    std::sort(subjets.begin(), subjets.end(),
              [](auto* a, auto* b) { return a->pt() > b->pt(); });
 
    size_t n_jets = std::min(subjets.size(), m_n_subjets);
    for (size_t jet_n = 0; jet_n < n_jets; jet_n++) {
      const auto* subjet = dynamic_cast<const xAOD::Jet*>(subjets.at(jet_n));
      if (!subjet) throw std::runtime_error("IParticle is not a Jet");
      std::string subjet_name = hk::subjet + std::to_string(jet_n);
      for (const auto& getter: m_subjet_getters) {
        requireOverwrite(inputs.at(subjet_name),getter(*subjet));
      }
    }
 
    // calculate and write
    for (const auto& node: m_outputs) {
      const auto& result = m_graph->compute(inputs, {}, node.first);
      for (const auto& var_writer: node.second) {
        var_writer.second(jet) = result.at(var_writer.first);
      }
    }
 
  }

Member Data Documentation

m_defaults

std::map<std::string, std::map<std::string,double> > FlavorTagDiscriminants::HbbTag::m_defaults

private

Definition at line 44 of file HbbTag.h.

m_fat_jet_getters

std::vector<Pg> FlavorTagDiscriminants::HbbTag::m_fat_jet_getters

private

Definition at line 37 of file HbbTag.h.

m_graph

std::unique_ptr<lwt::LightweightGraph> FlavorTagDiscriminants::HbbTag::m_graph

private

Definition at line 43 of file HbbTag.h.

m_min_subjet_pt

double FlavorTagDiscriminants::HbbTag::m_min_subjet_pt

private

Definition at line 40 of file HbbTag.h.

m_n_subjets

size_t FlavorTagDiscriminants::HbbTag::m_n_subjets

private

Definition at line 39 of file HbbTag.h.

m_outputs

std::vector<std::pair<std::string, NodeWriter> > FlavorTagDiscriminants::HbbTag::m_outputs

private

Definition at line 50 of file HbbTag.h.

m_parent_link

SG::AuxElement::ConstAccessor<JetLink> FlavorTagDiscriminants::HbbTag::m_parent_link

private

Definition at line 33 of file HbbTag.h.

m_subjet_getters

std::vector<Pg> FlavorTagDiscriminants::HbbTag::m_subjet_getters

private

Definition at line 38 of file HbbTag.h.

m_subjet_link_getter

SG::AuxElement::ConstAccessor<PartLink> FlavorTagDiscriminants::HbbTag::m_subjet_link_getter

private

Definition at line 35 of file HbbTag.h.

---

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

• HbbTag.h
• HbbTag.cxx