d0/d70/FullBilloirVertexFitter_8hpp_source.html

// This file is part of the Acts project.

//

// Copyright (C) 2019 CERN for the benefit of the Acts project

//

// This Source Code Form is subject to the terms of the Mozilla Public

// License, v. 2.0. If a copy of the MPL was not distributed with this

// file, You can obtain one at http://mozilla.org/MPL/2.0/.


#pragma once


#include "Acts/Propagator/EigenStepper.hpp"

#include "Acts/Propagator/Propagator.hpp"

#include "Acts/Utilities/Result.hpp"

#include "Acts/Vertexing/HelicalTrackLinearizer.hpp"

#include "Acts/Vertexing/LinearizerConcept.hpp"

#include "Acts/Vertexing/Vertex.hpp"

#include "Acts/Vertexing/VertexingOptions.hpp"

namespace Acts {


template <typename input_track_t, typename linearizer_t>

class FullBilloirVertexFitter {

  static_assert(LinearizerConcept<linearizer_t>,

                "Linearizer does not fulfill linearizer concept.");


 public:

  using InputTrack_t = input_track_t;

  using Propagator_t = typename linearizer_t::Propagator_t;

  using BField_t = typename linearizer_t::BField_t;

  using Linearizer_t = linearizer_t;


  struct State {

    State(const Acts::MagneticFieldContext& mctx) : linearizerState(mctx) {}

    typename Linearizer_t::State linearizerState;

  };


  struct Config {

    int maxIterations = 5;

  };


  template <

      typename T = input_track_t,

      std::enable_if_t<std::is_same<T, BoundTrackParameters>::value, int> = 0>

  FullBilloirVertexFitter(const Config& cfg)

      : m_cfg(cfg), extractParameters([](T params) { return params; }) {}


  FullBilloirVertexFitter(

      const Config& cfg,

      std::function<BoundTrackParameters(input_track_t)> func)

      : m_cfg(cfg), extractParameters(func) {}


  Result<Vertex<input_track_t>> fit(

      const std::vector<const input_track_t*>& paramVector,

      const linearizer_t& linearizer,

      const VertexingOptions<input_track_t>& vertexingOptions,

      State& state) const;


 private:

  Config m_cfg;


  std::function<BoundTrackParameters(input_track_t)> extractParameters;

};


}  // namespace Acts


#include "FullBilloirVertexFitter.ipp"