ParticleLevelPtEtaPhiDecoratorAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
 
#include "TruthParticleLevelAnalysisAlgorithms/ParticleLevelPtEtaPhiDecoratorAlg.h"
 
namespace CP {
 
StatusCode ParticleLevelPtEtaPhiDecoratorAlg::initialize() {
 
  ANA_CHECK(m_particlesKey.initialize());
 
  return StatusCode::SUCCESS;
}
 
StatusCode ParticleLevelPtEtaPhiDecoratorAlg::execute() {
 
  SG::ReadHandle<xAOD::TruthParticleContainer> particles(m_particlesKey);
 
  // decorators
  static const SG::AuxElement::Decorator<float> dec_pt("pt");
  static const SG::AuxElement::Decorator<float> dec_eta("eta");
  static const SG::AuxElement::Decorator<float> dec_phi("phi");
 
  for (const auto* particle : *particles) {
 
    // decorate pT/eta/phi of the truth particle
    // (Energy is already decorated at DAOD-level!)
    dec_pt(*particle) = particle->pt();
    dec_eta(*particle) = particle->eta();
    dec_phi(*particle) = particle->phi();
  }
  return StatusCode::SUCCESS;
}
 
}  // namespace CP