DerivationFramework::TruthLinkRepointTool Class Reference

#include <TruthLinkRepointTool.h>

Inheritance diagram for DerivationFramework::TruthLinkRepointTool:

Collaboration diagram for DerivationFramework::TruthLinkRepointTool:

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

Static Private Member Functions
static int	find_match (const xAOD::TruthParticle *p, const xAOD::TruthParticleContainer *c)

Private Attributes
SG::ReadHandleKey< xAOD::IParticleContainer >	m_recoKey
	Parameter: input collection key. More...
SG::WriteDecorHandleKey< xAOD::IParticleContainer >	m_decorKey {this, "OutputDecoration", m_recoKey, "TruthLink", "Name of the output decoration on the reco object"}
	Parameter: output decoration. More...
SG::ReadHandleKeyArray< xAOD::TruthParticleContainer >	m_targetKeys {this, "TargetCollections", {"TruthMuons","TruthPhotons","TruthElectrons"}, "Name of target truth collections"}
	Parameter: target collection. More...

Detailed Description

Definition at line 23 of file TruthLinkRepointTool.h.

Member Function Documentation

addBranches()

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

finaloverridevirtual

Definition at line 31 of file TruthLinkRepointTool.cxx.

                                                                                             {
  // Retrieve the truth collections
  std::vector<const xAOD::TruthParticleContainer*> targets{};
  targets.reserve(m_targetKeys.size());
 
  SG::WriteDecorHandle<xAOD::IParticleContainer, ElementLink<xAOD::TruthParticleContainer> >
    output_decorator (m_decorKey, ctx);
 
  for (const SG::ReadHandleKey<xAOD::TruthParticleContainer>& key : m_targetKeys) {
    SG::ReadHandle<xAOD::TruthParticleContainer> readHandle{key, ctx};
    if (!readHandle.isValid()) {
      ATH_MSG_FATAL("Failed to retrieve "<<key.fullKey());
      return StatusCode::FAILURE;
    }
    targets.emplace_back(readHandle.cptr());
  }
 
 
  SG::ReadHandle<xAOD::IParticleContainer> inputCont{m_recoKey, ctx};
  if (!inputCont.isValid()) {
    ATH_MSG_FATAL("Failed to retrieve "<<m_recoKey.fullKey());
    return StatusCode::FAILURE;
  }
  for ( const xAOD::IParticle* input : *inputCont) {
    const xAOD::TruthParticle* truthPart = xAOD::TruthHelpers::getTruthParticle(*input);
    output_decorator(*input) = ElementLink<xAOD::TruthParticleContainer>{};
    for (const xAOD::TruthParticleContainer* target : targets) {
        int index = find_match(truthPart, target);
 
        if (index >=0) output_decorator(*input) = ElementLink<xAOD::TruthParticleContainer>(*target, index);
 
    }
  }
 
  return StatusCode::SUCCESS;
}

find_match()

int DerivationFramework::TruthLinkRepointTool::find_match ( const xAOD::TruthParticle *  p, const xAOD::TruthParticleContainer *  c  )

staticprivate

Definition at line 69 of file TruthLinkRepointTool.cxx.

{
  // See if it's already gone
  if (!p) return -1;
  // Look through the mini-collection
  for (int i=0;i<int(c->size());++i){
    if (c->at(i) && HepMC::is_same_particle(p,c->at(i))) return i;
  }
  // Note: just fine if it wasn't in the mini-collection
  return -1;
}

initialize()

StatusCode DerivationFramework::TruthLinkRepointTool::initialize ( )

finaloverridevirtual

Definition at line 23 of file TruthLinkRepointTool.cxx.

                                                              {
  ATH_CHECK(m_recoKey.initialize());
  ATH_CHECK(m_targetKeys.initialize());
  ATH_CHECK(m_decorKey.initialize());
  return StatusCode::SUCCESS;
}

Member Data Documentation

m_decorKey

SG::WriteDecorHandleKey<xAOD::IParticleContainer> DerivationFramework::TruthLinkRepointTool::m_decorKey {this, "OutputDecoration", m_recoKey, "TruthLink", "Name of the output decoration on the reco object"}

private

Parameter: output decoration.

Definition at line 36 of file TruthLinkRepointTool.h.

m_recoKey

SG::ReadHandleKey<xAOD::IParticleContainer> DerivationFramework::TruthLinkRepointTool::m_recoKey

private

Initial value:

{this,"RecoCollection", "Muons",
        "Name of reco collection for decoration"}

Parameter: input collection key.

Definition at line 33 of file TruthLinkRepointTool.h.

m_targetKeys

SG::ReadHandleKeyArray<xAOD::TruthParticleContainer> DerivationFramework::TruthLinkRepointTool::m_targetKeys {this, "TargetCollections", {"TruthMuons","TruthPhotons","TruthElectrons"}, "Name of target truth collections"}

private

Parameter: target collection.

Definition at line 38 of file TruthLinkRepointTool.h.

---

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

• TruthLinkRepointTool.h
• TruthLinkRepointTool.cxx