DerivationFramework::TruthLinkRepointTool Class Reference

#include <TruthLinkRepointTool.h>

Inheritance diagram for DerivationFramework::TruthLinkRepointTool:

Collaboration diagram for DerivationFramework::TruthLinkRepointTool:

Public Member Functions
	TruthLinkRepointTool (const std::string &t, const std::string &n, const IInterface *p)
	~TruthLinkRepointTool ()
virtual StatusCode	addBranches () const override final
virtual StatusCode	initialize () 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...
Gaudi::Property< std::string >	m_decOutput
	Parameter: output decoration. More...
SG::ReadHandleKeyArray< xAOD::TruthParticleContainer >	m_targetKeys {this, "TargetCollections", {"TruthMuons","TruthPhotons","TruthElectrons"}, "Name of target truth collections"}
	Parameter: target collection. More...
SG::WriteDecorHandleKey< xAOD::IParticleContainer >	m_decorKey {this, "TargetDecorKeys", "", "Will be overwritten during initialize"}

Detailed Description

Definition at line 23 of file TruthLinkRepointTool.h.

Constructor & Destructor Documentation

TruthLinkRepointTool()

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

Definition at line 24 of file TruthLinkRepointTool.cxx.

                              :
    base_class(t,n,p) {
}

~TruthLinkRepointTool()

DerivationFramework::TruthLinkRepointTool::~TruthLinkRepointTool ( )

default

Member Function Documentation

addBranches()

StatusCode DerivationFramework::TruthLinkRepointTool::addBranches ( ) const

finaloverridevirtual

Definition at line 45 of file TruthLinkRepointTool.cxx.

                                                                      {
  const EventContext& ctx = Gaudi::Hive::currentContext();
  // 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 retrive "<<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 84 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 29 of file TruthLinkRepointTool.cxx.

                                                              {
  ATH_CHECK(m_recoKey.initialize());
  if (m_decOutput.value().empty()) {
     ATH_MSG_FATAL("Please enter a a valid output decorator");
     return StatusCode::FAILURE;
  }
  ATH_CHECK(m_targetKeys.initialize());
  m_decorKey = m_recoKey.key() + "." + m_decOutput;
  ATH_CHECK(m_decorKey.initialize());
  return StatusCode::SUCCESS;
}

Member Data Documentation

m_decorKey

SG::WriteDecorHandleKey<xAOD::IParticleContainer> DerivationFramework::TruthLinkRepointTool::m_decorKey {this, "TargetDecorKeys", "", "Will be overwritten during initialize"}

private

Definition at line 40 of file TruthLinkRepointTool.h.

m_decOutput

Gaudi::Property<std::string> DerivationFramework::TruthLinkRepointTool::m_decOutput

private

Initial value:

{this, "OutputDecoration",
                                  "TruthLink", "Name of the output decoration on the reco object"}

Parameter: output decoration.

Definition at line 35 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 32 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