ActsExtrapolationAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#include "ActsGeometry/ActsExtrapolationAlg.h"
 
// ATHENA
#include "AthenaKernel/IAthRNGSvc.h"
#include "AthenaKernel/RNGWrapper.h"
#include "GaudiKernel/EventContext.h"
#include "GaudiKernel/ISvcLocator.h"
 
// ACTS
#include "Acts/Propagator/MaterialInteractor.hpp"
#include "Acts/Propagator/detail/SteppingLogger.hpp"
#include "Acts/Surfaces/PerigeeSurface.hpp"
#include "Acts/Utilities/Helpers.hpp"
#include "Acts/Definitions/Units.hpp"
#include "Acts/Utilities/Logger.hpp"
 
// PACKAGE
#include "ActsGeometryInterfaces/GeometryContext.h"
#include "ActsGeometry/IActsPropStepRootWriterSvc.h"
#include "ActsGeometryInterfaces/ITrackingGeometryTool.h"
#include "ActsInterop/Logger.h"
 
// OTHER
#include "CLHEP/Random/RandomEngine.h"
 
// STL
#include <fstream>
#include <string>
 
using namespace Acts::UnitLiterals;
 
StatusCode ActsExtrapolationAlg::initialize() {
 
  ATH_MSG_DEBUG(name() << "::" << __FUNCTION__);
 
  ATH_CHECK(m_rndmGenSvc.retrieve());
  ATH_CHECK(m_extrapolationTool.retrieve());
  ATH_CHECK(m_propStepWriterSvc.retrieve());
  ATH_CHECK(m_trackingGeometryTool.retrieve());
  ATH_CHECK( m_materialTrackCollectionKey.initialize() );
 
  return StatusCode::SUCCESS;
}
 
StatusCode ActsExtrapolationAlg::execute(const EventContext &ctx) const {
 
  ATH_MSG_VERBOSE(name() << "::" << __FUNCTION__);
 
  ATHRNG::RNGWrapper *rngWrapper = m_rndmGenSvc->getEngine(this);
  rngWrapper->setSeed(name(), ctx);
  CLHEP::HepRandomEngine *rngEngine = rngWrapper->getEngine(ctx);
 
  ATH_MSG_VERBOSE("Extrapolating " << m_nParticlePerEvent << " particles");
 
  // Write to the collection to the EventStore
  SG::WriteHandle<ActsTrk::RecordedMaterialTrackCollection> materialTracks(m_materialTrackCollectionKey, ctx);
 
  // Record the collection once per event if not already there
  if (!materialTracks.isPresent()) {
      auto coll = std::make_unique<ActsTrk::RecordedMaterialTrackCollection>();
      ATH_CHECK(materialTracks.record(std::move(coll)));
  }
 
  // Add the track to the recorded collection
  auto* coll = materialTracks.ptr();
  if (!coll) {
      ATH_MSG_ERROR("RecordedMaterialTrackCollection ptr() is null for key "
                    << m_materialTrackCollectionKey.key());
      return StatusCode::FAILURE;
  }
 
  for (size_t i = 0; i < m_nParticlePerEvent; i++) {
    double d0 = 0;
    double z0 = 0;
    double phi = rngEngine->flat() * 2 * M_PI - M_PI;
    std::vector<double> etaRange = m_etaRange;
    double etaMin = etaRange.at(0);
    double etaMax = etaRange.at(1);
    double eta = rngEngine->flat() * std::abs(etaMax - etaMin) + etaMin;
 
    std::vector<double> ptRange = m_ptRange;
    double ptMin = ptRange.at(0) * 1_GeV;
    double ptMax = ptRange.at(1) * 1_GeV;
 
    double pt = rngEngine->flat() * std::abs(ptMax - ptMin) + ptMin;
 
    Acts::Vector3 momentum(pt * std::cos(phi), pt * std::sin(phi),
                            pt * std::sinh(eta));
 
    double theta = Acts::VectorHelpers::theta(momentum);
 
    double charge = rngEngine->flat() > 0.5 ? -1 : 1;
 
    double qop = charge / momentum.norm();
 
    std::shared_ptr<Acts::PerigeeSurface> surface =
        Acts::Surface::makeShared<Acts::PerigeeSurface>(
            Acts::Vector3(0, 0, 0));
 
    double t = 0;
    ATH_MSG_VERBOSE("Pseudo-particle: eta: " << eta << " phi: " << phi);
 
    Acts::BoundVector pars;
    // cppcheck-suppress constStatement; will be able to initialize this directly with eigen 3.4
    pars << d0, z0, phi, theta, qop, t;
    std::optional<Acts::BoundMatrix> cov = std::nullopt;
 
    if (charge != 0.) {
      // Perigee, no alignment -> default geo context
      Acts::GenericBoundTrackParameters startParameters(std::move(surface), std::move(pars), std::move(cov), Acts::ParticleHypothesis::pion());
      auto result = m_extrapolationTool->propagationSteps(ctx, startParameters);
      if (!result.ok()) {
        ATH_MSG_WARNING("Extrapolation tool failed to extrapolate the track: "
                        << result.error().message());
        continue;
      }
      auto &output = result.value();
      if(output.first.size() == 0) {
        ATH_MSG_WARNING("Got ZERO steps from the extrapolation tool");
      }
      if (m_writePropStep) {
        m_propStepWriterSvc->write(output.first);
      }
 
      if(m_writeMaterialTracks){
        Acts::RecordedMaterialTrack track;
        track.first.first = Acts::Vector3::Zero();
        track.first.second = momentum;
        track.second = std::move(output.second);
        coll->push_back(track);
      }
    }
 
    ATH_MSG_VERBOSE(name() << " execute done");
  }
 
  return StatusCode::SUCCESS;
}
 
 
void ActsExtrapolationAlg::writeStepsObj(
    const std::vector<Acts::detail::Step>& steps) const {
 
  std::lock_guard<std::mutex> lock(m_writeMutex);
 
  static std::ofstream out ATLAS_THREAD_SAFE ("steps.obj");
  std::stringstream lstr;
  lstr << "l";
  for (const auto &step : steps) {
    const auto &pos = step.position;
    out << "v " << pos.x() << " " << pos.y() << " " << pos.z() << std::endl;
    lstr << " " << m_objVtxCount;
    m_objVtxCount++;
  }
 
  lstr << std::endl;
 
  out << lstr.str() << std::endl;
}