TruthParticleFillerTool.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "TruthParticleFillerTool.h"
#include "AthenaKernel/errorcheck.h"
#include "GaudiKernel/IPartPropSvc.h"
#include "HepPDT/ParticleData.hh"
#include "HepPDT/ParticleDataTable.hh"
#include "TruthUtils/MagicNumbers.h"
 
namespace D3PD {
 
 
TruthParticleFillerTool::TruthParticleFillerTool (const std::string& type,
                                                  const std::string& name,
                                                  const IInterface* parent)
  : Base (type, name, parent),
    m_ppsvc ("PartPropSvc", name)
{
  declareProperty ("PDGIDVariable",  m_PDGIDVariable = "pdgId");
  declareProperty ("PartPropSvc", m_ppsvc,
                   "Particle property service.");
 
  book().ignore(); // Avoid coverity warnings.
}
 
 
StatusCode TruthParticleFillerTool::initialize()
{
  CHECK( m_ppsvc.retrieve() );
  return StatusCode::SUCCESS;
}
 
 
StatusCode TruthParticleFillerTool::book()
{
  CHECK( addVariable("status",    m_status) );
  CHECK( addVariable("barcode",   m_barcode) );
  CHECK( addVariable(m_PDGIDVariable,     m_pdgId) );
  CHECK( addVariable("charge",    m_charge) );
  return StatusCode::SUCCESS;
}
 
 
StatusCode TruthParticleFillerTool::fill (const xAOD::TruthParticle& p)
{
  *m_status = p.status();
  *m_barcode = HepMC::barcode(p); // FIXME barcode-based
  *m_pdgId = p.pdgId();
 
  const HepPDT::ParticleDataTable* pdt = m_ppsvc->PDT();
  const HepPDT::ParticleData* pd = pdt->particle (std::abs(p.pdgId()));
  *m_charge = pd ? pd->charge() : 0;
  if (p.pdgId() < 0)
    *m_charge = - *m_charge;
 
  return StatusCode::SUCCESS;
}
 
 
} // namespace D3PD