TruthDecoratorAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETTRACKPERFMON_TruthDECORATORALG_H
#define INDETTRACKPERFMON_TruthDECORATORALG_H


#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "MCTruthClassifier/IMCTruthClassifier.h"

#include "xAODTruth/TruthParticleContainer.h"
 

#include <string>
#include <vector>

#include "SafeDecorator.h"
 
 
namespace IDTPM {
 
  class TruthDecoratorAlg :
      public AthReentrantAlgorithm {
 
  public:
 
    TruthDecoratorAlg( const std::string& name, ISvcLocator* pSvcLocator );
 
    virtual ~TruthDecoratorAlg() = default;
 
    virtual StatusCode initialize() override;
 
    virtual StatusCode execute( const EventContext& ctx ) const override;
 
  private:
 
    SG::ReadHandleKey<xAOD::TruthParticleContainer> m_truthParticlesName {
        this, "TruthParticleContainerName", "TruthParticles", "Name of container of truth particles" };
 
    StringProperty m_prefix { this, "Prefix", "Truth_", "Decoration prefix to avoid clashes" };
 
    StatusCode decorateTruthParticle(
        const xAOD::TruthParticle& truth,
        std::vector< IDTPM::OptionalDecoration< xAOD::TruthParticleContainer, int >>& truth_decor ) const;
 
 
    enum TruthDecorations : size_t {
      Type,
      Origin,
      NDecorations
    };
 
    const std::vector< std::string > m_decor_truth_names {
      "truthType",
      "truthOrigin"
    };
 
    std::vector< IDTPM::WriteKeyAccessorPair< xAOD::TruthParticleContainer,
                                              int > > m_decor_truth{};
 
 
    PublicToolHandle< IMCTruthClassifier > m_truthClassifier {
        this, "MCTruthClassifier", "MCTruthClassifier/MCTruthClassifier", "Truth classification tool" };
 
  };
 
} // namespace IDTPM
 
#endif // > ! INDETTRACKPERFMON_TruthDECORATORALG_H