TrkrClusterFindingAlgorithmFinderFunction.cpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file TrkrClusterFindingAlgorithmFinderFunction.cpp

#include "ActsExamples/TrackFinding/TrkrClusterFindingAlgorithm.hpp"

namespace {
  template <typename Finder>
  struct TrkrFindingFunctionImpl
  {
    Finder finder;

    TrkrFindingFunctionImpl(Finder&& f) : finder(std::move(f)) {}

    ActsExamples::TrkrClusterFindingAlgorithm::FinderResult
    operator()(
       const std::vector<SourceLink>& sourceLinks,
       const ActsExamples::TrackParameters& initialParameters,
       const Acts::CombinatorialKalmanFilterOptions<Acts::CKFSourceLinkSelector>&       options) const
  {
      return finder.findTracks(sourceLinks, initialParameters, options);
    };
  };
}  // namespace

ActsExamples::TrkrClusterFindingAlgorithm::FinderFunction
ActsExamples::TrkrClusterFindingAlgorithm::makeFinderFunction(
    std::shared_ptr<const Acts::TrackingGeometry> trackingGeometry,
    Options::BFieldVariant magneticField)
{
  using Updater  = Acts::GainMatrixUpdater;
  using Smoother = Acts::GainMatrixSmoother;

  return std::visit(
      [trackingGeometry](auto&& inputField) -> FinderFunction {
        using InputMagneticField = typename std::decay_t<decltype(inputField)>::element_type;
        using MagneticField      = Acts::SharedBField<InputMagneticField>;
        using Stepper            = Acts::EigenStepper<MagneticField>;
        using Navigator          = Acts::Navigator;
        using Propagator         = Acts::Propagator<Stepper, Navigator>;
  using SourceLinkSelector = Acts::CKFSourceLinkSelector;
        using Finder             = Acts::CombinatorialKalmanFilter
    <Propagator, Updater, Smoother, SourceLinkSelector>;

        MagneticField field(std::move(inputField));
        Stepper       stepper(std::move(field));
        Navigator     navigator(trackingGeometry);
        navigator.resolvePassive   = false;
        navigator.resolveMaterial  = true;
        navigator.resolveSensitive = true;
        Propagator propagator(std::move(stepper), std::move(navigator));
        Finder     finder(std::move(propagator));

        return TrkrFindingFunctionImpl<Finder>(std::move(finder));
      },
      std::move(magneticField));
}