GNNVertexFitterAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef VKalVrt_GNNVertexFitterAlg_H
#define VKalVrt_GNNVertexFitterAlg_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GNNVertexFitter/GNNVertexFitterTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "xAODJet/JetContainer.h"
 
namespace Rec {
 
class GNNVertexFitterAlg : public AthReentrantAlgorithm {
public:
  GNNVertexFitterAlg(const std::string &name, ISvcLocator *pSvcLocator);
  StatusCode initialize() override;
  StatusCode execute(const EventContext &ctx) const override;
  StatusCode finalize() override;
 
private:
  // Tools
  ToolHandle<Rec::GNNVertexFitterTool> m_VtxTool;
  // Input jets
  SG::ReadHandleKey<xAOD::JetContainer> m_inJetsKey{this, "inputJetContainer", "", "Input jet container"};
  // Output vertices
  SG::WriteHandleKey<xAOD::VertexContainer> m_outVertexKey{this, "outputVertexContainer", "GNNVertices",
                                                           "Output vertex container"};
 
  // Input Primary Vertices
  SG::ReadHandleKey<xAOD::VertexContainer> m_pvContainerKey{this, "PrimaryVertexContainer", "PrimaryVertices",
                                                            "Read PrimaryVertices container"};
 
};
 
} // namespace Rec
 
#endif