#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 More...

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

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.

◆ 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.

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

Public Member Functions