TrackExtensionAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSTRACKRECONSTRUCTION_TRACKEXTENSIONALG_H
#define ACTSTRACKRECONSTRUCTION_TRACKEXTENSIONALG_H
// Base Class
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
 
// Gaudi includes
#include "GaudiKernel/ToolHandle.h"
 
// Tools
#include "ActsGeometryInterfaces/IExtrapolationTool.h"
#include "ActsGeometryInterfaces/ITrackingGeometryTool.h"
 
// ACTS
#include "Acts/EventData/TrackContainer.hpp"
 
// ActsTrk
#include "ActsEvent/ProtoTrackCollection.h"
#include "ActsEvent/TrackContainer.h"
#include "ActsGeometry/DetectorElementToActsGeometryIdMap.h"
#include "ActsToolInterfaces/IOnTrackCalibratorTool.h"
#include "ActsToolInterfaces/IFitterTool.h"
 
// Athena
#include "GaudiKernel/EventContext.h"
#include "xAODInDetMeasurement/PixelClusterContainer.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "BeamSpotConditionsData/BeamSpotData.h"
 
// STL
#include <memory>
#include <string>
 
// Handle Keys
#include "ActsEvent/TrackContainerHandlesHelper.h"
#include "StoreGate/WriteHandleKey.h"
#include "src/detail/Definitions.h"
#include "src/detail/DuplicateSeedDetector.h"
#include "src/detail/TrackFindingMeasurements.h"
#include "src/TrackStatePrinterTool.h"

namespace ActsTrk {
class TrackExtensionAlg : public AthReentrantAlgorithm {
 public:
 
  using AthReentrantAlgorithm::AthReentrantAlgorithm;
  virtual StatusCode initialize() override;
  virtual StatusCode execute(const EventContext& context) const override;
  using CKFOptions = Acts::CombinatorialKalmanFilterOptions<detail::RecoTrackContainer>;
 
 private:
  SG::ReadHandleKey<xAOD::PixelClusterContainer> m_pixelClusters{
      this, "PixelClusterContainer", "", "the pix clusters"};
  SG::ReadHandleKey<ActsTrk::ProtoTrackCollection> m_protoTrackCollectionKey{
      this, "ProtoTracksLocation", "", "Input proto tracks"};
  SG::ReadHandleKey<xAOD::TruthParticleContainer> m_truthParticlesKey{
      this, "TruthLocation", "",
      "Truth container (to be enabled only for debugging)"};
  SG::WriteHandleKey<ActsTrk::TrackContainer> m_trackContainerKey{
      this, "ACTSTracksLocation", "",
      "Output track collection (ActsTrk variant)"};
  ActsTrk::MutableTrackContainerHandlesHelper m_tracksBackendHandlesHelper{this};
 
  SG::ReadCondHandleKey< InDet::BeamSpotData > m_beamSpotKey{
    this, "BeamSpotKey", "BeamSpotData",
    "SG key for beam spot"};
  
  PublicToolHandle<ActsTrk::ITrackingGeometryTool> m_trackingGeometryTool{
      this, "TrackingGeometryTool", ""};
  ToolHandle<ActsTrk::IExtrapolationTool> m_extrapolationTool{
      this, "ExtrapolationTool", ""};
  ToolHandle<ActsTrk::IOnTrackCalibratorTool<detail::RecoTrackStateContainer>>
      m_pixelCalibTool{this, "PixelCalibrator", "",
                       "Opt. pixel measurement calibrator"};
  ToolHandle<ActsTrk::IOnTrackCalibratorTool<detail::RecoTrackStateContainer>>
      m_stripCalibTool{this, "StripCalibrator", "",
                       "Opt. strip measurement calibrator"};
  ToolHandle<ActsTrk::IOnTrackCalibratorTool<detail::RecoTrackStateContainer>>
      m_hgtdCalibTool{this, "HGTDCalibrator", "",
                       "Opt. HGTD measurement calibrator"};                       
  ToolHandle<ActsTrk::TrackStatePrinterTool> m_trackStatePrinter{
      this, "TrackStatePrinter", "", "optional track state printer"};
  ToolHandle<ActsTrk::IFitterTool> m_actsFitter{
      this, "ActsFitter", "", "Choice of Acts Fitter (Kalman by default)"};
 
  Gaudi::Property<bool> m_propagateForward{this, "PropagateForward", false,
                                           "If true propagate forward"};
 
  std::unique_ptr<detail::CKF_config> m_ckfConfig;
  std::unique_ptr<const Acts::Logger> m_logger;
 
};
}  // namespace ActsTrk
#endif  // ACTSTRACKRECONSTRUCTION_TRACKEXTENSIONALG_H