TruthParticleChildAssociationTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id$
#include "TruthParticleChildAssociationTool.h"
#include "xAODTruth/TruthVertex.h"
#include "AthenaKernel/errorcheck.h"
#include "TruthUtils/MagicNumbers.h"
 
namespace D3PD {
 
 
TruthParticleChildAssociationTool::TruthParticleChildAssociationTool
  (const std::string& type,
   const std::string& name,
   const IInterface* parent)
    : Base (type, name, parent),
      m_i (0)
{}
 
 
StatusCode TruthParticleChildAssociationTool::initialize()
{
  return Base::initialize();
}
 
 
StatusCode
TruthParticleChildAssociationTool::reset (const xAOD::TruthParticle& p)
{
  m_i = 0;
 
  const xAOD::TruthVertex* vx = p.decayVtx();
  if (!vx) {
    m_xaod_children.clear();
    return StatusCode::SUCCESS;
  }
 
  size_t sz = vx->nOutgoingParticles();
  m_xaod_children.reserve (sz);
  for (unsigned int i = 0; i < sz; i++) {
    const xAOD::TruthParticle* p2 = vx->outgoingParticle (i);
    if (p2)
      m_xaod_children[i] =  p2;
  }
  std::sort (m_xaod_children.begin(), m_xaod_children.end(), [](const auto & a, const auto & b) -> bool{ return HepMC::uniqueID(a) < HepMC::uniqueID(b); }); // FIXME barcode-based - NB ordering may change when this function switches to return id rather than barcode
  return StatusCode::SUCCESS;
}
 
 
const xAOD::TruthParticle* TruthParticleChildAssociationTool::next()
{
  if (m_i < m_xaod_children.size())
    return m_xaod_children[m_i++];
  return 0;
}
 
 
} // namespace D3PD