#include <TruthQGDecorationTool.h>

Inheritance diagram for DerivationFramework::TruthQGDecorationTool:

Collaboration diagram for DerivationFramework::TruthQGDecorationTool:

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

Private Attributes
SG::ReadHandleKey< xAOD::JetContainer >	m_jetsKey {this, "JetCollection", "AntiKt4TruthWZJets", "Name of jet collection for decoration"}
	input collection key
SG::WriteDecorHandleKey< xAOD::JetContainer >	m_decOutput {this, "TrueFlavor", m_jetsKey, "TrueFlavor", "Name of the output decoration on the jet"}
	output decoration

Detailed Description

Definition at line 24 of file TruthQGDecorationTool.h.

Member Function Documentation

◆ addBranches()

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

virtual

Definition at line 26 of file TruthQGDecorationTool.cxx.

{
  // Event context
 
  // Retrieve the jet container
  SG::ReadHandle<xAOD::JetContainer> inputJets(m_jetsKey, ctx);
  if (!inputJets.isValid()) {
    ATH_MSG_ERROR("Couldn't retrieve container with name " << m_jetsKey);
    return StatusCode::FAILURE;
  }
 
  SG::WriteDecorHandle<xAOD::JetContainer,int> output_decorator(m_decOutput, ctx);
  static const SG::ConstAccessor<int> PartonTruthLabelIDAcc("PartonTruthLabelID");
  static const SG::ConstAccessor<int> HadronConeExclTruthLabelIDAcc("HadronConeExclTruthLabelID");
 
  for (const auto *ajet : *inputJets){
    if (!PartonTruthLabelIDAcc.isAvailable(*ajet) ){
      ATH_MSG_ERROR("Did not have input PartonTruthLabelID decorations available");
      return StatusCode::FAILURE;
    }
    else if (!HadronConeExclTruthLabelIDAcc.isAvailable(*ajet) ){
      ATH_MSG_ERROR("Did not have input HadronConeExclTruthLabelID decorations available");
      return StatusCode::FAILURE;
    } // Now we have the input decorations
    /* Agreement from the HF-tagging and Jet/MET group:
        - If it is non-zero, use the label from the HF-tagging group (b, c, tau)
        - If it is zero, use the label from the Jet/MET group (q/g)
        - In the case that the two disagree (e.g. Jet/MET says b and HF says light),
           multiply the Jet/MET label by 100 to ensure this case is kept separate
    */
    if (HadronConeExclTruthLabelIDAcc(*ajet)!=0){
      output_decorator(*ajet) = HadronConeExclTruthLabelIDAcc(*ajet);
    } else {
      int labelid = PartonTruthLabelIDAcc(*ajet);
      if (std::abs(labelid)!=5 &&
          std::abs(labelid)!=4 &&
          std::abs(labelid)!=15){
        output_decorator(*ajet) = labelid;
      } else {
        output_decorator(*ajet) = labelid*100;
      }
    }
  } // Loop over jets
 
  return StatusCode::SUCCESS;
}

◆ initialize()

StatusCode DerivationFramework::TruthQGDecorationTool::initialize ( )

Definition at line 17 of file TruthQGDecorationTool.cxx.

                                                                {
 
  ATH_CHECK(m_jetsKey.initialize());
  ATH_CHECK(m_decOutput.initialize());
  return StatusCode::SUCCESS;
 
}

Member Data Documentation

◆ m_decOutput

SG::WriteDecorHandleKey<xAOD::JetContainer> DerivationFramework::TruthQGDecorationTool::m_decOutput {this, "TrueFlavor", m_jetsKey, "TrueFlavor", "Name of the output decoration on the jet"}

private

output decoration

Definition at line 37 of file TruthQGDecorationTool.h.

38{this, "TrueFlavor", m_jetsKey, "TrueFlavor", "Name of the output decoration on the jet"};

◆ m_jetsKey

SG::ReadHandleKey<xAOD::JetContainer> DerivationFramework::TruthQGDecorationTool::m_jetsKey {this, "JetCollection", "AntiKt4TruthWZJets", "Name of jet collection for decoration"}

private

input collection key

Definition at line 34 of file TruthQGDecorationTool.h.

35{this, "JetCollection", "AntiKt4TruthWZJets", "Name of jet collection for decoration"};

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

Public Member Functions