GNNVertexFitterAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "GNNVertexFitterAlg.h"
#include "xAODJet/JetContainer.h"
#include "xAODTracking/VertexAuxContainer.h"
#include "xAODTracking/VertexContainer.h"
 
namespace Rec {
 
GNNVertexFitterAlg::GNNVertexFitterAlg(const std::string &name, ISvcLocator *pSvcLocator)
    : AthReentrantAlgorithm(name, pSvcLocator), m_VtxTool("Rec::GNNVertexFitterTool/VtxTool", this) {
  declareProperty("VtxTool", m_VtxTool, "The GNN Vtxing Tool");
}
 
StatusCode GNNVertexFitterAlg::initialize() {
 
  // Retrieving the tool
  ATH_CHECK(m_VtxTool.retrieve());
 
  // Initializing Keys
  ATH_CHECK(m_inJetsKey.initialize());
  ATH_CHECK(m_outVertexKey.initialize());
  ATH_CHECK(m_pvContainerKey.initialize());
 
  return StatusCode::SUCCESS;
}
 
StatusCode GNNVertexFitterAlg::execute(const EventContext &ctx) const {
 
  ATH_MSG_DEBUG("In GNNVertexFitterAlg::execute()");
 
  // Extract Jets
  SG::ReadHandle<xAOD::JetContainer> inJetContainer(m_inJetsKey, ctx);
  if (!inJetContainer.isValid()) {
    ATH_MSG_WARNING("No xAOD::JetContainer named " << m_inJetsKey.key() << " found in StoreGate");
    return StatusCode::FAILURE;
  }
 
  // Write new GNN Vertice Container
  SG::WriteHandle<xAOD::VertexContainer> outVertexContainer(m_outVertexKey, ctx);
  if (outVertexContainer.record(std::make_unique<xAOD::VertexContainer>(), std::make_unique<xAOD::VertexAuxContainer>())
          .isFailure()) {
    ATH_MSG_ERROR("Storegate record of VertexContainer failed.");
    return StatusCode::FAILURE;
  }
 
  const xAOD::Vertex *pv = nullptr;
  //-- Extract Primary Vertices
  SG::ReadHandle<xAOD::VertexContainer> pv_cont(m_pvContainerKey, ctx);
  if (!pv_cont.isValid()) {
    ATH_MSG_ERROR("No Primary Vertices container found in TDS");
    return StatusCode::FAILURE;
  } else {
    //-- Extract PV itself
    for (auto v : *pv_cont) {
      if (v->vertexType() == xAOD::VxType::PriVtx) {
        pv = v;
        break;
      }
    }
  }
  if (!pv) pv = pv_cont->front();
 
  // Perform a Vertex fit
  ATH_CHECK(m_VtxTool->fitAllVertices(inJetContainer.ptr(), outVertexContainer.ptr(), *pv, ctx));
 
  return StatusCode::SUCCESS;
}
 
StatusCode GNNVertexFitterAlg::finalize() {
  ATH_MSG_DEBUG("GNNVertexFitter::finalize()");
  return StatusCode::SUCCESS;
}
 
} // namespace Rec