ATLAS Offline Software
Public Member Functions | Private Attributes | List of all members
DerivationFramework::ClassifyAndCalculateHFAugmentation Class Reference

#include <ClassifyAndCalculateHFAugmentation.h>

Inheritance diagram for DerivationFramework::ClassifyAndCalculateHFAugmentation:
Collaboration diagram for DerivationFramework::ClassifyAndCalculateHFAugmentation:

Public Member Functions

 ClassifyAndCalculateHFAugmentation (const std::string &t, const std::string &n, const IInterface *p)
 
 ~ClassifyAndCalculateHFAugmentation ()
 
virtual StatusCode initialize () override
 
virtual StatusCode finalize () override
 
virtual StatusCode addBranches (const EventContext &ctx) const override
 

Private Attributes

SG::ReadHandleKey< xAOD::TruthParticleContainerm_truthParticlesKey {this, "TruthParticleContainerName", "TruthParticles", "Name of the truth particles collection that is used to compute the HF Classification"}
 
SG::ReadHandleKey< xAOD::JetContainerm_jetCollectionKey {this, "jetCollectionName", "AntiKt4TruthDressedWZJets", "Name of the jet collection that is used to compute the HF Classification"}
 
SG::ReadHandleKey< xAOD::EventInfom_eventInfoKey {this, "EventInfo", "EventInfo", ""}
 
SG::WriteDecorHandleKey< xAOD::EventInfom_hfDecorKey {this, "hfDecorationName", m_eventInfoKey , "HF_Classification", "Name that is used to store the HF Classification."}
 
SG::WriteDecorHandleKey< xAOD::EventInfom_SimplehfDecorKey {this, "SimplehfDecorationName", m_eventInfoKey ,"SimpleHFClassification", "Name that is used to store the simple HF Classification."}
 
SG::WriteDecorHandleKey< xAOD::JetContainerm_jetIDDecorationKey {this, "HadronOriginIDDecorationName", m_jetCollectionKey, "HFHadronOriginID", "jet origin ID decoration key"}
 
std::string m_hfDecorationName {""}
 
ToolHandle< DerivationFramework::JetMatchingToolm_JetMatchingTool_Tool
 
ToolHandle< DerivationFramework::ClassifyAndCalculateHFToolm_HFClassification_tool
 
ToolHandle< DerivationFramework::HadronOriginClassifierm_HadronOriginClassifier_Tool
 

Detailed Description

Definition at line 60 of file ClassifyAndCalculateHFAugmentation.h.

Constructor & Destructor Documentation

◆ ClassifyAndCalculateHFAugmentation()

DerivationFramework::ClassifyAndCalculateHFAugmentation::ClassifyAndCalculateHFAugmentation ( const std::string &  t,
const std::string &  n,
const IInterface *  p 
)

Definition at line 27 of file ClassifyAndCalculateHFAugmentation.cxx.

27  :
28  base_class(t,n,p), // Athena tool.
29  m_JetMatchingTool_Tool(""), // Hadron-jet matching tool.
30  m_HFClassification_tool(""), // HF classifier tool.
31  m_HadronOriginClassifier_Tool("") // HF hadron origin tool.
32  {
33 
34  // Declare a set of tool properties to set them exertanally:
35  // -m_HFClassification_tool: The tool to compute the HF classifier.
36  // -m_HadronOriginClassifier_Tool: The tool to determine the origin of the HF hadrons.
37  // -m_JetMatchingTool_Tool: The tool to match the hadrons with the jets.
38 
39  declareProperty("ClassifyAndComputeHFtool", m_HFClassification_tool);
40  declareProperty("HadronOriginClassifierTool", m_HadronOriginClassifier_Tool);
41  declareProperty("JetMatchingTool", m_JetMatchingTool_Tool);
42  }

◆ ~ClassifyAndCalculateHFAugmentation()

DerivationFramework::ClassifyAndCalculateHFAugmentation::~ClassifyAndCalculateHFAugmentation ( )

Definition at line 44 of file ClassifyAndCalculateHFAugmentation.cxx.

44 {}

Member Function Documentation

◆ addBranches()

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

Definition at line 99 of file ClassifyAndCalculateHFAugmentation.cxx.

100  {
101 
102 
103  // Retrieve the truth particle container
105  if (!truthParticlesHandle.isValid()) {
106  ATH_MSG_ERROR ("Could not retrieve TruthParticleContainer " << truthParticlesHandle.key());
107  return StatusCode::FAILURE;
108  }
109  const xAOD::TruthParticleContainer* xTruthParticleContainer = truthParticlesHandle.cptr();
110 
111  // Retrieve the jets container
113  if (!jetInputHandle.isValid()) {
114  ATH_MSG_ERROR ("Could not retrieve JetContainer " << jetInputHandle.key());
115  return StatusCode::FAILURE;
116  }
117  const xAOD::JetContainer* JetCollection = jetInputHandle.cptr();
118 
119  // Compute a map that associates the HF hadrons with their origin using the tool m_HadronOriginClassifier_Tool.
120  std::map<const xAOD::TruthParticle*, DerivationFramework::HadronOriginClassifier::HF_id> hadronMap = m_HadronOriginClassifier_Tool->GetOriginMap();
121 
122  // Create a map with a list of matched hadrons for each jet.
123  std::map<const xAOD::Jet*, std::vector<xAOD::TruthParticleContainer::const_iterator>> particleMatch = m_JetMatchingTool_Tool->matchHadronsToJets(xTruthParticleContainer, JetCollection);
124 
125  // Calculate the necessary information from the jets to compute the HF classifier.
126  m_HFClassification_tool->flagJets(JetCollection, particleMatch, hadronMap, m_hfDecorationName);
127 
128  // Compute the HF classifier and the simple HF classifier.
129  int hfclassif = m_HFClassification_tool->computeHFClassification(JetCollection, m_hfDecorationName);
130  int simpleclassif = m_HFClassification_tool->getSimpleClassification(hfclassif);
131 
132  // Retrieve EventInfo
133  SG::ReadHandle<xAOD::EventInfo> eventInfoHandle(m_eventInfoKey, ctx);
134  if (!eventInfoHandle.isValid()) {
135  ATH_MSG_ERROR ("Could not retrieve EventInfo " << eventInfoHandle.key());
136  return StatusCode::FAILURE;
137  }
138  const xAOD::EventInfo* EventInfo = eventInfoHandle.cptr();
139 
140  // Decorate EventInfo with the HF Classification and the simple version
141  SG::WriteDecorHandle<xAOD::EventInfo, int> decorator_HFClassification(m_hfDecorKey, ctx);
142  decorator_HFClassification(*EventInfo) = hfclassif;
143 
144  SG::WriteDecorHandle<xAOD::EventInfo, int> decorator_SimpleHFClassification(m_SimplehfDecorKey, ctx);
145  decorator_SimpleHFClassification(*EventInfo) = simpleclassif;
146 
147  // Decorate truth jets with origin ID
149  for (const auto jet : *JetCollection) {
150  int id = -999;
152  if(hfidAcc.isAvailable(*jet)){
153  id = hfidAcc(*jet);
154  }
155  jetIdDecorator(*jet) = id;
156  }
157 
158  return StatusCode::SUCCESS;
159  }

◆ finalize()

StatusCode DerivationFramework::ClassifyAndCalculateHFAugmentation::finalize ( )
overridevirtual

Definition at line 89 of file ClassifyAndCalculateHFAugmentation.cxx.

89  {
90  return StatusCode::SUCCESS;
91  }

◆ initialize()

StatusCode DerivationFramework::ClassifyAndCalculateHFAugmentation::initialize ( )
overridevirtual

Definition at line 52 of file ClassifyAndCalculateHFAugmentation.cxx.

52  {
53 
54  ATH_MSG_INFO("Initialize HF computation");
55 
56  ATH_MSG_INFO("Jets Container Name " << m_jetCollectionKey.key());
57  ATH_MSG_INFO("Truth Particles Container Name " << m_truthParticlesKey.key());
58  ATH_MSG_INFO("HF Classifier Name " << m_hfDecorKey.key());
59  ATH_MSG_INFO("Simple HF Classifier Name " << m_SimplehfDecorKey.key());
60  ATH_MSG_INFO("Jet Origin ID Decoration Name " << m_jetIDDecorationKey.key());
61  std::size_t pos = m_hfDecorKey.key().find(".");
62  if (pos != std::string::npos) { m_hfDecorationName = m_hfDecorKey.key().substr (pos+1); }
69 
70  // Retrieve the necessary tools
71  if(m_HFClassification_tool.retrieve().isFailure()){
72  ATH_MSG_ERROR("Unable to retrieve the tool " << m_HFClassification_tool);
73  return StatusCode::FAILURE;
74  }
75 
76  if(m_HadronOriginClassifier_Tool.retrieve().isFailure()){
77  ATH_MSG_ERROR("Unable to retrieve the tool " << m_HadronOriginClassifier_Tool);
78  return StatusCode::FAILURE;
79  }
80 
81  if(m_JetMatchingTool_Tool.retrieve().isFailure()){
82  ATH_MSG_ERROR("Unable to retrieve the tool " << m_JetMatchingTool_Tool);
83  return StatusCode::FAILURE;
84  }
85 
86  return StatusCode::SUCCESS;
87  }

Member Data Documentation

◆ m_eventInfoKey

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

Definition at line 94 of file ClassifyAndCalculateHFAugmentation.h.

◆ m_HadronOriginClassifier_Tool

ToolHandle<DerivationFramework::HadronOriginClassifier> DerivationFramework::ClassifyAndCalculateHFAugmentation::m_HadronOriginClassifier_Tool
private

Definition at line 111 of file ClassifyAndCalculateHFAugmentation.h.

◆ m_HFClassification_tool

ToolHandle<DerivationFramework::ClassifyAndCalculateHFTool> DerivationFramework::ClassifyAndCalculateHFAugmentation::m_HFClassification_tool
private

Definition at line 110 of file ClassifyAndCalculateHFAugmentation.h.

◆ m_hfDecorationName

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

Definition at line 103 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 100 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 93 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 102 of file ClassifyAndCalculateHFAugmentation.h.

◆ m_JetMatchingTool_Tool

ToolHandle<DerivationFramework::JetMatchingTool> DerivationFramework::ClassifyAndCalculateHFAugmentation::m_JetMatchingTool_Tool
private

Definition at line 109 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 101 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 92 of file ClassifyAndCalculateHFAugmentation.h.


The documentation for this class was generated from the following files:
DerivationFramework::ClassifyAndCalculateHFAugmentation::m_SimplehfDecorKey
SG::WriteDecorHandleKey< xAOD::EventInfo > m_SimplehfDecorKey
Definition: ClassifyAndCalculateHFAugmentation.h:101
ATH_MSG_INFO
#define ATH_MSG_INFO(x)
Definition: AthMsgStreamMacros.h:31
DerivationFramework::ClassifyAndCalculateHFAugmentation::m_JetMatchingTool_Tool
ToolHandle< DerivationFramework::JetMatchingTool > m_JetMatchingTool_Tool
Definition: ClassifyAndCalculateHFAugmentation.h:109
DerivationFramework::ClassifyAndCalculateHFAugmentation::m_eventInfoKey
SG::ReadHandleKey< xAOD::EventInfo > m_eventInfoKey
Definition: ClassifyAndCalculateHFAugmentation.h:94
SG::ReadHandle< xAOD::TruthParticleContainer >
DerivationFramework::ClassifyAndCalculateHFAugmentation::m_HFClassification_tool
ToolHandle< DerivationFramework::ClassifyAndCalculateHFTool > m_HFClassification_tool
Definition: ClassifyAndCalculateHFAugmentation.h:110
DerivationFramework::ClassifyAndCalculateHFAugmentation::m_HadronOriginClassifier_Tool
ToolHandle< DerivationFramework::HadronOriginClassifier > m_HadronOriginClassifier_Tool
Definition: ClassifyAndCalculateHFAugmentation.h:111
SG::ConstAccessor< int >
read_hist_ntuple.t
t
Definition: read_hist_ntuple.py:5
SG::VarHandleKey::key
const std::string & key() const
Return the StoreGate ID for the referenced object.
Definition: AthToolSupport/AsgDataHandles/Root/VarHandleKey.cxx:141
python.utils.AtlRunQueryDQUtils.p
p
Definition: AtlRunQueryDQUtils.py:209
jet
Definition: JetCalibTools_PlotJESFactors.cxx:23
ATH_MSG_ERROR
#define ATH_MSG_ERROR(x)
Definition: AthMsgStreamMacros.h:33
beamspotman.n
n
Definition: beamspotman.py:727
DerivationFramework::ClassifyAndCalculateHFAugmentation::m_jetCollectionKey
SG::ReadHandleKey< xAOD::JetContainer > m_jetCollectionKey
Definition: ClassifyAndCalculateHFAugmentation.h:93
SG::WriteDecorHandle
Handle class for adding a decoration to an object.
Definition: StoreGate/StoreGate/WriteDecorHandle.h:100
ATH_CHECK
#define ATH_CHECK
Definition: AthCheckMacros.h:40
SG::VarHandleKey::initialize
StatusCode initialize(bool used=true)
If this object is used as a property, then this should be called during the initialize phase.
Definition: AthToolSupport/AsgDataHandles/Root/VarHandleKey.cxx:103
DataVector
Derived DataVector<T>.
Definition: DataVector.h:795
DerivationFramework::ClassifyAndCalculateHFAugmentation::m_hfDecorationName
std::string m_hfDecorationName
Definition: ClassifyAndCalculateHFAugmentation.h:103
id
SG::auxid_t id
Definition: Control/AthContainers/Root/debug.cxx:239
EventInfo
This class provides general information about an event. Event information is provided by the accessor...
Definition: EventInfo/EventInfo/EventInfo.h:43
python.LumiBlobConversion.pos
pos
Definition: LumiBlobConversion.py:16
xAOD::EventInfo_v1
Class describing the basic event information.
Definition: EventInfo_v1.h:43
SG::WriteDecorHandleKey::initialize
StatusCode initialize(bool used=true)
If this object is used as a property, then this should be called during the initialize phase.
JetCollection
Container for Jets
Definition: JetCollection.h:30
DerivationFramework::ClassifyAndCalculateHFAugmentation::m_jetIDDecorationKey
SG::WriteDecorHandleKey< xAOD::JetContainer > m_jetIDDecorationKey
Definition: ClassifyAndCalculateHFAugmentation.h:102
DerivationFramework::ClassifyAndCalculateHFAugmentation::m_truthParticlesKey
SG::ReadHandleKey< xAOD::TruthParticleContainer > m_truthParticlesKey
Definition: ClassifyAndCalculateHFAugmentation.h:92
DerivationFramework::ClassifyAndCalculateHFAugmentation::m_hfDecorKey
SG::WriteDecorHandleKey< xAOD::EventInfo > m_hfDecorKey
Definition: ClassifyAndCalculateHFAugmentation.h:100