ActsToTrkFitterWrapTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSEVENTCNV_ActsToTrkWrappingTool_H
#define ACTSEVENTCNV_ActsToTrkWrappingTool_H
 
// ATHENA
#include "AthenaBaseComps/AthAlgTool.h"
#include "ActsToolInterfaces/IFitterTool.h"
 
#include "ActsEvent/ContextUtility.h"
#include "TrkFitterInterfaces/ITrackFitter.h"
#include "ActsToolInterfaces/ITrackConverterTool.h"
#include "ActsGeometryInterfaces/IGeometryRealmConvTool.h"
 
namespace ActsTrk {
 
class ActsToTrkFitterWrapTool : public extends<AthAlgTool, Trk::ITrackFitter> {
 
  public:
    using base_class::base_class;
    ~ActsToTrkFitterWrapTool() = default;
 
    StatusCode initialize() override;

    virtual std::unique_ptr<Trk::Track> 
        fit (const EventContext& ctx,
             const Trk::Track& track,
             const Trk::RunOutlierRemoval runOutlier = false,
             const Trk::ParticleHypothesis hypothesis = Trk::nonInteracting) const override;

    virtual std::unique_ptr<Trk::Track> 
        fit(const EventContext& ctx,
            const Trk::Track& track,
            const Trk::MeasurementSet& measSet,
            const Trk::RunOutlierRemoval runOutlier = false,
            const Trk::ParticleHypothesis hypothesis = Trk::nonInteracting) const override;
    
    virtual std::unique_ptr<Trk::Track> 
        fit(const EventContext& ctx,
            const Trk::Track& track,
            const Trk::PrepRawDataSet& prepRawSet,
            const Trk::RunOutlierRemoval runOutlier = false,
            const Trk::ParticleHypothesis hypothesis = Trk::nonInteracting) const override;    
    
    virtual std::unique_ptr<Trk::Track> 
        fit(const EventContext& ctx,
            const Trk::PrepRawDataSet& prepRawSet,
            const Trk::TrackParameters& params,
            const Trk::RunOutlierRemoval runOutlier = false,
            const Trk::ParticleHypothesis hypothesis = Trk::nonInteracting) const override;

    virtual std::unique_ptr<Trk::Track> 
        fit(const EventContext& ctx,
            const Trk::MeasurementSet& measSet,
            const Trk::TrackParameters& params,
            const Trk::RunOutlierRemoval runOutlier = false,
            const Trk::ParticleHypothesis hypothesis = Trk::nonInteracting) const override;
    
    virtual std::unique_ptr<Trk::Track> 
        fit(const EventContext& ctx,
            const Trk::Track& track1,
            const Trk::Track& track2,
            const Trk::RunOutlierRemoval runOutlier = false,
            const Trk::ParticleHypothesis hypothesis = Trk::nonInteracting) const override;
    
  private:
    std::unique_ptr<Trk::Track> 
        reFitImpl(const EventContext& ctx,
                  const Trk::Track& track,
                  const std::vector<Acts::SourceLink>& measColl,
                  const Trk::ParticleHypothesis hypothesis,
                  const bool useScaledCov = false) const;
    
    std::unique_ptr<Trk::Track> 
        fitImpl(const EventContext& ctx,
                const std::vector<Acts::SourceLink>& measColl,
                const Acts::BoundTrackParameters& params) const;
                                          

    ToolHandle<IFitterTool> m_actsFitterTool{this, "ActsFitterTool", ""};
    
    ToolHandle<ITrackConverterTool> m_ATLASConverterTool{this, "ATLASConverterTool", ""};
    PublicToolHandle<IGeometryRealmConvTool> m_geometryConvTool{this, "GeometryRealmConvTool", ""};
    ContextUtility m_ctxProvider{this};
    Gaudi::Property< double > m_option_seedCovarianceScale {this, "SeedCovarianceScale", 1.,
      "Scale factor for the input seed covariance when doing refitting"};
 
};
 
}
#endif