TrackingGeometryCondAlg.cxx

Go to the documentation of this file.

/*
 *   Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
 */
 
 
#include "TrkGeometry/TrackingGeometry.h"
#include "GaudiKernel/EventContext.h"
#include "AthenaKernel/IOVInfiniteRange.h"
#include "TrackingGeometryCondAlg/TrackingGeometryCondAlg.h"
 
Trk::TrackingGeometryCondAlg::TrackingGeometryCondAlg(const std::string& name,
                                                      ISvcLocator* pSvcLocator)
  : AthCondAlgorithm(name, pSvcLocator)
{
}
 
StatusCode Trk::TrackingGeometryCondAlg::initialize()
{
  ATH_MSG_DEBUG("initialize " << name());
 
  // Write Handle
  ATH_CHECK(m_trackingGeometryWriteKey.initialize());
  // Retrieve tools
  ATH_CHECK(m_trackingGeometryBuilder.retrieve());
  if (m_geometryProcessors.retrieve().isFailure()){
    ATH_MSG_FATAL( "Could not retrieve " << m_geometryProcessors );
    return StatusCode::FAILURE;
  }
  return StatusCode::SUCCESS;
}
 
StatusCode Trk::TrackingGeometryCondAlg::execute(const EventContext& ctx) const{
  //Set up write handle
  SG::WriteCondHandle<Trk::TrackingGeometry> writeHandle{m_trackingGeometryWriteKey, ctx};
  if (writeHandle.isValid()) {
    ATH_MSG_DEBUG("Found valid write handle");
    return StatusCode::SUCCESS;
  }
 
  // Start with infinite range and let the TG builder narrow it down.
  writeHandle.addDependency (IOVInfiniteRange::infiniteMixed());
 
  std::unique_ptr<Trk::TrackingGeometry> trackingGeometry =
    m_trackingGeometryBuilder->trackingGeometry(ctx, nullptr, writeHandle);
 
  // loop over the recursive geometry processors
  auto gpIter = m_geometryProcessors.begin();
  auto gpIterE = m_geometryProcessors.end();
  for (; gpIter != gpIterE; ++gpIter) {
    if ((*gpIter)->process(*trackingGeometry).isFailure()) {
      ATH_MSG_FATAL("Processing of TrackingGeometry did not succeed. Abort.");
      return StatusCode::FAILURE;
    } else {
      ATH_MSG_VERBOSE("Successfully processed the TrackingGeometry with "
                      << (*gpIter));
    }
  }
  if (m_dumpGeo) trackingGeometry->dump(msgStream(), "TrackingGeometryCondAlg");
  ATH_CHECK(writeHandle.record(std::move(trackingGeometry)));
  
  return StatusCode::SUCCESS;
}