HGTDTruthTrackDecorationAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration.
*/
 
#ifndef HGTDTRACKEXTENSIONALG_TRUTHDECORATION_H
#define HGTDTRACKEXTENSIONALG_TRUTHDECORATION_H
 
// Base Class
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "StoreGate/WriteDecorHandleKey.h"
 
namespace ActsTrk {
  
  class HGTDTruthTrackDecorationAlg
    : public AthReentrantAlgorithm {
 
  public:
    using AthReentrantAlgorithm::AthReentrantAlgorithm;
    virtual ~HGTDTruthTrackDecorationAlg() = default;
    
    virtual StatusCode initialize() override;
    virtual StatusCode execute(const EventContext&) const override;
    
  private:
    SG::ReadHandleKey<xAOD::TrackParticleContainer> m_trackParticleContainerName {this, "TrackParticleContainerName", "", "Name of the TrackParticle container"};
    
    // Truth decorations
    SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_layerClusterTruthClassKey { this, "HGTD_cluster_truth_class", m_trackParticleContainerName, "HGTD_cluster_truth_class", "Decoration for cluster truth classification" };
    SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_layerClusterShadowedKey { this, "HGTD_cluster_shadowed", m_trackParticleContainerName, "HGTD_cluster_shadowed", "Decoration for shadowed cluster" };
    SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_layerClusterMergedKey { this, "HGTD_cluster_merged", m_trackParticleContainerName, "HGTD_cluster_merged", "Decoration for merged cluster" };
    SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_layerPrimaryExpectedKey { this, "HGTD_primary_expected", m_trackParticleContainerName, "HGTD_primary_expected", "Decoration for primary expected cluster" };  
    
    struct TruthTrackExtensionData {
      std::vector<int> truthClassVec = {-1, -1, -1, -1};             
      std::vector<bool> isShadowedVec = {false, false, false, false}; 
      std::vector<bool> isMergedVec = {false, false, false, false};   
      std::vector<bool> primaryExistsVec = {false, false, false, false}; 
    };
    
  };
  
} // namespace ActsTrk
 
#endif