TrkrClusterFittingAlgorithm.hpp

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

#pragma once

#include <iostream>
#include <map>
#include <random>
#include <stdexcept>
#include <boost/program_options.hpp>

#include "Acts/TrackFitting/GainMatrixSmoother.hpp"
#include "Acts/TrackFitting/GainMatrixUpdater.hpp"
#include "Acts/Geometry/GeometryIdentifier.hpp"
#include "Acts/MagneticField/ConstantBField.hpp"
#include "Acts/MagneticField/InterpolatedBFieldMap.hpp"
#include "Acts/MagneticField/SharedBField.hpp"
#include "Acts/Propagator/EigenStepper.hpp"
#include "Acts/Propagator/Navigator.hpp"
#include "Acts/Propagator/Propagator.hpp"
#include "Acts/Surfaces/Surface.hpp"
#include "Acts/Utilities/Helpers.hpp"
#include "Acts/Utilities/ParameterDefinitions.hpp"
#include "Acts/TrackFitting/KalmanFitter.hpp"
#include "Acts/Geometry/TrackingGeometry.hpp"

#include "ActsExamples/Plugins/BField/ScalableBField.hpp"
#include "ActsExamples/EventData/Track.hpp"
#include "ActsExamples/Framework/BareAlgorithm.hpp"
#include "ActsExamples/Plugins/BField/BFieldOptions.hpp"
#include "ActsExamples/EventData/TrkrClusterSourceLink.hpp"

namespace ActsExamples {

class TrkrClusterFittingAlgorithm : public BareAlgorithm
{
 public:
  using FitterResult
    = Acts::Result<Acts::KalmanFitterResult<ActsExamples::TrkrClusterSourceLink>>;
  using FitterFunction
    = std::function<FitterResult(
        const std::vector<ActsExamples::TrkrClusterSourceLink>&,
        const TrackParameters&,
        const Acts::KalmanFitterOptions<Acts::VoidOutlierFinder>&)>;

  using DirectedFitterFunction
  = std::function<FitterResult(
        const std::vector<ActsExamples::TrkrClusterSourceLink>&,
        const TrackParameters&,
        const Acts::KalmanFitterOptions<Acts::VoidOutlierFinder>&,
        const std::vector<const Acts::Surface*>&)>;
      

  static FitterFunction makeFitterFunction(
      std::shared_ptr<const Acts::TrackingGeometry> trackingGeometry,
      Options::BFieldVariant magneticField);

  static DirectedFitterFunction makeFitterFunction(
      Options::BFieldVariant magneticField);

  struct Config
  {
    FitterFunction fit;
    DirectedFitterFunction dFit;
  };

  TrkrClusterFittingAlgorithm(Config cfg, Acts::Logging::Level lvl);


private:
  Config m_cfg;
};

}