ParticleLevelChargeDecoratorAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
 
#include "TruthParticleLevelAnalysisAlgorithms/ParticleLevelChargeDecoratorAlg.h"
 
namespace CP {
 
StatusCode ParticleLevelChargeDecoratorAlg::initialize() {
 
  ANA_CHECK(m_particlesKey.initialize());
 
  return StatusCode::SUCCESS;
}
 
StatusCode ParticleLevelChargeDecoratorAlg::execute() {
 
  SG::ReadHandle<xAOD::TruthParticleContainer> particles(m_particlesKey);
 
  // decorators
  static const SG::AuxElement::Decorator<float> dec_charge("charge");
 
  for (const auto* particle : *particles) {
 
    // decorate the charge so we can save it later
    dec_charge(*particle) = particle->charge();
  }
  return StatusCode::SUCCESS;
}
 
}  // namespace CP