DerivationFramework::ClassifyAndCalculateHFAugmentation Class Reference

#include <ClassifyAndCalculateHFAugmentation.h>

Inheritance diagram for DerivationFramework::ClassifyAndCalculateHFAugmentation:

Collaboration diagram for DerivationFramework::ClassifyAndCalculateHFAugmentation:

Public Member Functions
virtual StatusCode	initialize () override final
virtual StatusCode	addBranches (const EventContext &ctx) const override final

Private Attributes
SG::ReadHandleKey< xAOD::TruthParticleContainer >	m_truthParticlesKey {this, "TruthParticleContainerName", "TruthParticles", "Name of the truth particles collection that is used to compute the HF Classification"}
SG::ReadHandleKey< xAOD::JetContainer >	m_jetCollectionKey {this, "jetCollectionName", "AntiKt4TruthDressedWZJets", "Name of the jet collection that is used to compute the HF Classification"}
SG::ReadHandleKey< xAOD::EventInfo >	m_eventInfoKey {this, "EventInfo", "EventInfo", ""}
SG::WriteDecorHandleKey< xAOD::EventInfo >	m_hfDecorKey {this, "hfDecorationName", m_eventInfoKey , "HF_Classification", "Name that is used to store the HF Classification."}
SG::WriteDecorHandleKey< xAOD::EventInfo >	m_SimplehfDecorKey {this, "SimplehfDecorationName", m_eventInfoKey ,"SimpleHFClassification", "Name that is used to store the simple HF Classification."}
SG::WriteDecorHandleKey< xAOD::JetContainer >	m_jetIDDecorationKey {this, "HadronOriginIDDecorationName", m_jetCollectionKey, "HFHadronOriginID", "jet origin ID decoration key"}
std::string	m_hfDecorationName {""}
PublicToolHandle< DerivationFramework::JetMatchingTool >	m_JetMatchingTool_Tool {this, "JetMatchingTool", ""}
PublicToolHandle< DerivationFramework::ClassifyAndCalculateHFTool >	m_HFClassification_tool {this, "ClassifyAndComputeHFtool", ""}
PublicToolHandle< DerivationFramework::HadronOriginClassifier >	m_HadronOriginClassifier_Tool {this, "HadronOriginClassifierTool", ""}

Detailed Description

Definition at line 56 of file ClassifyAndCalculateHFAugmentation.h.

Member Function Documentation

addBranches()

StatusCode DerivationFramework::ClassifyAndCalculateHFAugmentation::addBranches ( const EventContext &  ctx ) const

finaloverridevirtual

Definition at line 70 of file ClassifyAndCalculateHFAugmentation.cxx.

  {
 
 
    // Retrieve the truth particle container
    SG::ReadHandle<xAOD::TruthParticleContainer> truthParticlesHandle(m_truthParticlesKey, ctx);
    if (!truthParticlesHandle.isValid()) {
      ATH_MSG_ERROR ("Could not retrieve TruthParticleContainer " << truthParticlesHandle.key());
      return StatusCode::FAILURE;
    }
    const xAOD::TruthParticleContainer* xTruthParticleContainer = truthParticlesHandle.cptr();
 
    // Retrieve the jets container
    SG::ReadHandle<xAOD::JetContainer> jetInputHandle(m_jetCollectionKey, ctx);
    if (!jetInputHandle.isValid()) {
      ATH_MSG_ERROR ("Could not retrieve JetContainer " << jetInputHandle.key());
      return StatusCode::FAILURE;
    }
    const xAOD::JetContainer* JetCollection = jetInputHandle.cptr();
 
    // Compute a map that associates the HF hadrons with their origin using the tool m_HadronOriginClassifier_Tool.
    std::map<const xAOD::TruthParticle*, DerivationFramework::HadronOriginClassifier::HF_id> hadronMap = m_HadronOriginClassifier_Tool->GetOriginMap();
 
    // Create a map with a list of matched hadrons for each jet.
    std::map<const xAOD::Jet*, std::vector<xAOD::TruthParticleContainer::const_iterator>> particleMatch = m_JetMatchingTool_Tool->matchHadronsToJets(xTruthParticleContainer, JetCollection);
 
    // Calculate the necessary information from the jets to compute the HF classifier.
    m_HFClassification_tool->flagJets(JetCollection, particleMatch, hadronMap, m_hfDecorationName);
 
    // Compute the HF classifier and the simple HF classifier.
    int hfclassif     = m_HFClassification_tool->computeHFClassification(JetCollection, m_hfDecorationName);
    int simpleclassif = m_HFClassification_tool->getSimpleClassification(hfclassif);
 
    // Retrieve EventInfo
    SG::ReadHandle<xAOD::EventInfo> eventInfoHandle(m_eventInfoKey, ctx);
    if (!eventInfoHandle.isValid()) {
      ATH_MSG_ERROR ("Could not retrieve EventInfo " << eventInfoHandle.key());
      return StatusCode::FAILURE;
    }
    const xAOD::EventInfo* EventInfo = eventInfoHandle.cptr();
 
    // Decorate EventInfo with the HF Classification and the simple version
    SG::WriteDecorHandle<xAOD::EventInfo, int> decorator_HFClassification(m_hfDecorKey, ctx);
    decorator_HFClassification(*EventInfo) = hfclassif;
 
    SG::WriteDecorHandle<xAOD::EventInfo, int> decorator_SimpleHFClassification(m_SimplehfDecorKey, ctx);
    decorator_SimpleHFClassification(*EventInfo) = simpleclassif;
 
    // Decorate truth jets with origin ID
    SG::WriteDecorHandle<xAOD::JetContainer, int> jetIdDecorator(m_jetIDDecorationKey, ctx);
    for (const auto jet : *JetCollection) {
      int id = -999;
      SG::ConstAccessor<int> hfidAcc(m_hfDecorationName + "_id");
      if(hfidAcc.isAvailable(*jet)){
        id = hfidAcc(*jet);
      }
      jetIdDecorator(*jet) = id;
    }
 
    return StatusCode::SUCCESS;
  }

initialize()

StatusCode DerivationFramework::ClassifyAndCalculateHFAugmentation::initialize ( )

finaloverridevirtual

Definition at line 27 of file ClassifyAndCalculateHFAugmentation.cxx.

                                                           {
 
    ATH_MSG_INFO("Initialize HF computation");
 
    ATH_MSG_INFO("Jets Container Name "            << m_jetCollectionKey.key());
    ATH_MSG_INFO("Truth Particles Container Name " << m_truthParticlesKey.key());
    ATH_MSG_INFO("HF Classifier Name "             << m_hfDecorKey.key());
    ATH_MSG_INFO("Simple HF Classifier Name "      << m_SimplehfDecorKey.key());
    ATH_MSG_INFO("Jet Origin ID Decoration Name "  << m_jetIDDecorationKey.key());
    std::size_t pos = m_hfDecorKey.key().find(".");
    if (pos != std::string::npos) { m_hfDecorationName = m_hfDecorKey.key().substr (pos+1); }
    ATH_CHECK( m_truthParticlesKey.initialize() );
    ATH_CHECK( m_jetCollectionKey.initialize() );
    ATH_CHECK( m_eventInfoKey.initialize() );
    ATH_CHECK( m_hfDecorKey.initialize() );
    ATH_CHECK( m_SimplehfDecorKey.initialize() );
    ATH_CHECK( m_jetIDDecorationKey.initialize() );
 
    // Retrieve the necessary tools
    if(m_HFClassification_tool.retrieve().isFailure()){
      ATH_MSG_ERROR("Unable to retrieve the tool " << m_HFClassification_tool);
      return StatusCode::FAILURE;
    }
 
    if(m_HadronOriginClassifier_Tool.retrieve().isFailure()){
      ATH_MSG_ERROR("Unable to retrieve the tool " << m_HadronOriginClassifier_Tool);
      return StatusCode::FAILURE;
    }
 
    if(m_JetMatchingTool_Tool.retrieve().isFailure()){
      ATH_MSG_ERROR("Unable to retrieve the tool " << m_JetMatchingTool_Tool);
      return StatusCode::FAILURE;
    }
 
    return StatusCode::SUCCESS;
  }

Member Data Documentation

m_eventInfoKey

SG::ReadHandleKey<xAOD::EventInfo> DerivationFramework::ClassifyAndCalculateHFAugmentation::m_eventInfoKey {this, "EventInfo", "EventInfo", ""}

private

Definition at line 86 of file ClassifyAndCalculateHFAugmentation.h.

m_HadronOriginClassifier_Tool

PublicToolHandle<DerivationFramework::HadronOriginClassifier> DerivationFramework::ClassifyAndCalculateHFAugmentation::m_HadronOriginClassifier_Tool {this, "HadronOriginClassifierTool", ""}

private

Definition at line 103 of file ClassifyAndCalculateHFAugmentation.h.

m_HFClassification_tool

PublicToolHandle<DerivationFramework::ClassifyAndCalculateHFTool> DerivationFramework::ClassifyAndCalculateHFAugmentation::m_HFClassification_tool {this, "ClassifyAndComputeHFtool", ""}

private

Definition at line 102 of file ClassifyAndCalculateHFAugmentation.h.

m_hfDecorationName

std::string DerivationFramework::ClassifyAndCalculateHFAugmentation::m_hfDecorationName {""}

private

Definition at line 95 of file ClassifyAndCalculateHFAugmentation.h.

m_hfDecorKey

SG::WriteDecorHandleKey<xAOD::EventInfo> DerivationFramework::ClassifyAndCalculateHFAugmentation::m_hfDecorKey {this, "hfDecorationName", m_eventInfoKey , "HF_Classification", "Name that is used to store the HF Classification."}

private

Definition at line 92 of file ClassifyAndCalculateHFAugmentation.h.

m_jetCollectionKey

SG::ReadHandleKey<xAOD::JetContainer> DerivationFramework::ClassifyAndCalculateHFAugmentation::m_jetCollectionKey {this, "jetCollectionName", "AntiKt4TruthDressedWZJets", "Name of the jet collection that is used to compute the HF Classification"}

private

Definition at line 85 of file ClassifyAndCalculateHFAugmentation.h.

m_jetIDDecorationKey

SG::WriteDecorHandleKey<xAOD::JetContainer> DerivationFramework::ClassifyAndCalculateHFAugmentation::m_jetIDDecorationKey {this, "HadronOriginIDDecorationName", m_jetCollectionKey, "HFHadronOriginID", "jet origin ID decoration key"}

private

Definition at line 94 of file ClassifyAndCalculateHFAugmentation.h.

m_JetMatchingTool_Tool

PublicToolHandle<DerivationFramework::JetMatchingTool> DerivationFramework::ClassifyAndCalculateHFAugmentation::m_JetMatchingTool_Tool {this, "JetMatchingTool", ""}

private

Definition at line 101 of file ClassifyAndCalculateHFAugmentation.h.

m_SimplehfDecorKey

SG::WriteDecorHandleKey<xAOD::EventInfo> DerivationFramework::ClassifyAndCalculateHFAugmentation::m_SimplehfDecorKey {this, "SimplehfDecorationName", m_eventInfoKey ,"SimpleHFClassification", "Name that is used to store the simple HF Classification."}

private

Definition at line 93 of file ClassifyAndCalculateHFAugmentation.h.

m_truthParticlesKey

SG::ReadHandleKey<xAOD::TruthParticleContainer> DerivationFramework::ClassifyAndCalculateHFAugmentation::m_truthParticlesKey {this, "TruthParticleContainerName", "TruthParticles", "Name of the truth particles collection that is used to compute the HF Classification"}

private

Definition at line 84 of file ClassifyAndCalculateHFAugmentation.h.

---

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

• ClassifyAndCalculateHFAugmentation.h
• ClassifyAndCalculateHFAugmentation.cxx