VertexDecoratorAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// Always protect against multiple includes!
#ifndef HSGNN_VERTEXDECORATORALG
#define HSGNN_VERTEXDECORATORALG
 
#include <utility>
#include <vector>
 
#include <AthenaBaseComps/AthHistogramAlgorithm.h>
 
#include <xAODEventInfo/EventInfo.h>
#include <xAODTracking/VertexContainer.h>
#include <xAODEgamma/ElectronContainer.h>
#include <xAODEgamma/PhotonContainer.h>
#include <xAODJet/JetContainer.h>
#include <xAODMuon/MuonContainer.h>
 
#include <InDetTruthVertexValidation/IInDetVertexTruthMatchTool.h>
#include <InDetGNNHardScatterSelection/GNNTool.h>
#include <TrackVertexAssociationTool/TrackVertexAssociationTool.h>
 
namespace InDetGNNHardScatterSelection
{
 
  class VertexDecoratorAlg final : public AthHistogramAlgorithm{
public:
    VertexDecoratorAlg(const std::string& name, ISvcLocator *pSvcLocator);
 
    StatusCode initialize() override;
    StatusCode execute() override;
private:
 
    // Members for configurable properties
    SG::ReadHandleKey<xAOD::VertexContainer> m_vertexInKey{
        this, "vertexIn", "PrimaryVertices", "containerName to read"};
    SG::ReadHandleKey<xAOD::EventInfo> m_eventInKey{
        this, "eventIn", "EventInfo", "containerName to read"};
    SG::ReadHandleKey<xAOD::ElectronContainer> m_electronsInKey{
        this, "electronsIn", "Electrons", "containerName to read"};
    SG::ReadHandleKey<xAOD::MuonContainer> m_muonsInKey{
        this, "muonsIn", "Muons", "containerName to read"};
    SG::ReadHandleKey<xAOD::JetContainer> m_jetsInKey{
        this, "jetsIn", "AntiKt4EMTopoJets", "containerName to read"};
    SG::ReadHandleKey<xAOD::PhotonContainer> m_photonsInKey{
        this, "photonsIn", "Photons", "containerName to read"};
 
    SG::WriteDecorHandleKey<xAOD::VertexContainer> m_photonLinksKey{
      this, "photonLinks", "", "" };
    SG::WriteDecorHandleKey<xAOD::VertexContainer> m_jetLinksKey{
      this, "jetLinks", "", "" };
    SG::WriteDecorHandleKey<xAOD::VertexContainer> m_electronLinksKey{
      this, "electronLinks", "", "" };
    SG::WriteDecorHandleKey<xAOD::VertexContainer> m_muonLinksKey{
      this, "muonLinks", "", "" };
 
    SG::ReadDecorHandleKey<xAOD::VertexContainer> m_deltaZKey{
      this, "deltaZKey", "", "" };
    SG::ReadDecorHandleKey<xAOD::VertexContainer> m_deltaPhiKey{
      this, "deltaPhiKey", "", "" };
 
    // additional ReadHandles for the Photons to declare dependencies
    // to the scheduler
    SG::ReadDecorHandleKey<xAOD::PhotonContainer> m_caloPointingZKey{
      this, "caloPointingZKey", "", "" };
    SG::ReadDecorHandleKey<xAOD::PhotonContainer> m_zCommonKey{
      this, "zCommonKey", "", "" };
    SG::ReadDecorHandleKey<xAOD::PhotonContainer> m_zCommonErrorKey{
      this, "zCommonErrorKey", "", "" };
 
    // Internal members
    ToolHandle<CP::TrackVertexAssociationTool> m_trkVtxAssociationTool{
        this, "TrackVertexAssociationTool", 
        "CP::TrackVertexAssociationTool/TrackVertexAssociationTool", 
        "Track vertex association tool to use"};
 
    ToolHandle<GNNTool> m_gnnTool{
        this, "gnnTool",
        "InDetGNNHardScatterSelection/GNNTool",
        "GNN tool to use"};
  };
} // namespace InDetGNNHardScatterSelection
 
#endif