ActsToTrkConverterTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSEVENTCNV_ActsToTrkConverterTool_H
#define ACTSEVENTCNV_ActsToTrkConverterTool_H
 
#include <tuple>
 
// ATHENA
#include "AthenaBaseComps/AthAlgTool.h"
 
 
#include "TrkToolInterfaces/IExtendedTrackSummaryTool.h"
#include "TrkToolInterfaces/IRIO_OnTrackCreator.h"
 
#include "TrkParameters/TrackParameters.h" //typedef, cannot fwd declare
#include "xAODTracking/TrackJacobianContainer.h"
#include "xAODTracking/TrackParametersContainer.h"
#include "xAODTracking/TrackStateContainer.h"
#include "xAODTracking/TrackMeasurementContainer.h"
 
#include "TrkPrepRawData/PrepRawData.h"
 
 
// PACKAGE
#include "ActsToolInterfaces/IActsToTrkConverterTool.h"
#include "ActsGeometryInterfaces/ITrackingGeometryTool.h"
 
#include "ActsCalibBase/SourceLinkType.h"
#include "ActsCalibrators/TrkMeasurementCalibrator.h"
#include "ActsCalibrators/TrkPrepRawDataCalibrator.h"
#include "ActsCalibrators/xAODUncalibMeasCalibrator.h"
 
#include "Acts/EventData/BoundTrackParameters.hpp"
 
 
#include "MuonRecToolInterfaces/IMuonCompetingClustersOnTrackCreator.h"
#include "MuonReadoutGeometry/MuonDetectorManager.h"
#include "MuonPrepRawData/MuonPrepDataContainer.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
 
namespace ActsTrk {
class ActsToTrkConverterTool : public extends<AthAlgTool, IActsToTrkConverterTool>
{
 
public:
  virtual StatusCode initialize() override;
 
  using base_class::base_class;
 
 

  virtual SurfacePtr_t actsSurfaceToTrkSurface(const EventContext& ctx,
                                               const Acts::Surface &actsSurface) const override;

  virtual std::shared_ptr<const Acts::Surface> trkSurfaceToActsSurface(const Trk::Surface &atlasSurface) const override;

  virtual std::vector<Acts::SourceLink> trkTrackToSourceLinks(const Trk::Track& track) const override;
 
  virtual void toSourceLinks(const std::vector<const Trk::MeasurementBase*>& measSet,
                              std::vector<Acts::SourceLink>& links) const override final;
 
  virtual void toSourceLinks(const std::vector<const Trk::PrepRawData*>& prdSet,
                             std::vector<Acts::SourceLink>& links) const override final;
 
  virtual std::unique_ptr<Trk::Track> convertFitResult(const EventContext& ctx,
                                                       TrackFitResult_t& fitResult,
                                                       const Trk::TrackInfo::TrackFitter fitAuthor,
                                                       const detail::SourceLinkType slType) const override final;
  virtual
  const Acts::BoundTrackParameters
  trkTrackParametersToActsParameters(const Trk::TrackParameters &atlasParameter, const Acts::GeometryContext& gctx, Trk::ParticleHypothesis = Trk::pion) const override;

  virtual
  std::unique_ptr<Trk::TrackParameters>
  actsTrackParametersToTrkParameters(const EventContext& ctx, const Acts::BoundTrackParameters &actsParameter, const Acts::GeometryContext& gctx) const override;

  virtual 
  void trkTrackCollectionToActsTrackContainer(ActsTrk::MutableTrackContainer &tc, const TrackCollection& trackColl, const Acts::GeometryContext& gctx) const override;
 
 virtual std::unique_ptr<TrackCollection> 
      convertActsToTrkContainer(const EventContext& ctx,
                                const ActsTrk::TrackContainer& trackCont) const override final;
private:
  template <typename Proxy_t>
  std::unique_ptr<Trk::Track> convertActsTrack(const EventContext& ctx,
                                               const Proxy_t& track,
                                               const Trk::TrackInfo::TrackFitter fitAuthor,
                                               const detail::SourceLinkType slType) const;
 
  /*** @brief Translate the Acts surface bounds to its equivalent in the Trk realm.
   *          @note Not all bounds are implemented
   *  @param bounds: Refrence to the bounds to be translated */
  std::shared_ptr<Trk::SurfaceBounds> translateBounds(const Acts::SurfaceBounds& bounds) const;
  SurfacePtr_t translateFreeSurface(const Acts::Surface& surface) const;

  using  TrkTSOSMask = std::bitset<Trk::TrackStateOnSurface::NumberOfTrackStateOnSurfaceTypes>;
  void appendMeasTSOS(const EventContext& ctx,
                      const xAOD::UncalibratedMeasurement* meas,
                      const TrkTSOSMask typePattern,
                      Trk::FitQualityOnSurface&& quality,
                      std::unique_ptr<Trk::TrackParameters> trkPars,
                      Trk::TrackStates& states) const;
  template <typename PrdType_t>
  const Trk::PrepRawData* fetchPrd(const EventContext& ctx,
                                   const SG::ReadHandleKey<PrdType_t>& key,
                                   const Identifier& prdId,
                                   const IdentifierHash& hash) const;
 
 
  bool actsTrackParameterPositionCheck(
     const Acts::BoundTrackParameters& actsParameter,
     const Trk::TrackParameters& tsos, const Acts::GeometryContext& gctx) const;
 
  PublicToolHandle<ActsTrk::ITrackingGeometryTool> m_trackingGeometryTool{this, "TrackingGeometryTool", "ActsTrackingGeometryTool"};
  
  ToolHandle<Trk::IExtendedTrackSummaryTool> m_trkSummaryTool {this, "SummaryTool", "", "ToolHandle for track summary tool"};
  ToolHandle<Trk::IRIO_OnTrackCreator> m_ROTcreator {this, "RotCreatorTool", ""};
 
  std::shared_ptr<const Acts::TrackingGeometry> m_trackingGeometry{};
  std::unordered_map<Identifier, std::shared_ptr<const Acts::Surface>> m_actsSurfaceMap{};
 
  Gaudi::Property<bool> m_visualDebugOutput{
     this, "VisualDebugOutput", false,
     "Print additional output for debug plots"};
 
 
  Gaudi::Property<bool> m_extractMuonSurfaces{this, "ExtractMuonSurfaces", false,
     "If True, use the MuonDetectorManager to extract the Muon surfaces"};

  Gaudi::Property<int> m_sourceLinkType{this, "SourceLinkType", 
                                        Acts::toUnderlying(detail::SourceLinkType::nTypes) };

  Gaudi::Property<bool> m_convertHoles{this, "convertHoles", true };
  Gaudi::Property<bool> m_convertOutliers{this, "convertOutliers", true};
  Gaudi::Property<bool> m_convertMaterial{this, "convertMaterial", false};
 
  ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
  
  SG::ReadHandleKey<Muon::MdtPrepDataContainer> m_keyMdt{this, "MdtKey", "MDT_DriftCircles"};
  SG::ReadHandleKey<Muon::RpcPrepDataContainer> m_keyRpc{this, "RpcKey", "RPC_Measurements"};
  SG::ReadHandleKey<Muon::TgcPrepDataContainer> m_keyTgc{this, "TgcKey", "TGC_MeasurementsAllBCs"};
  SG::ReadHandleKey<Muon::MMPrepDataContainer> m_keyMm{this, "MmKey", "MM_Measurements"};
  SG::ReadHandleKey<Muon::sTgcPrepDataContainer> m_keyStgc{this, "sTgcKey", "STGC_Measurements"};
 
 
  ToolHandle<Muon::IMuonCompetingClustersOnTrackCreator> m_compRotCreator{this, "CompetingRotCreator", ""};  //<! competing clusters rio ontrack creator

  SG::ReadCondHandleKey<MuonGM::MuonDetectorManager> m_muonMgrKey{this, "MuonManagerKey", "MuonDetectorManager"};
 
  detail::SourceLinkType m_slType{detail::SourceLinkType::nTypes};
 
  detail::TrkMeasurementCalibrator m_measCalib{};
  detail::TrkPrepRawDataCalibrator m_prdCalib{};
 
  Trk::TrackInfo::TrackFitter m_fitAuthor{Trk::TrackInfo::TrackFitter::GlobalChi2Fitter};
 
};
 
}; // namespace ActsTrk
 
#endif