FullBilloirVertexFitter.hpp

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

// 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/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 Config {
    int maxIterations = 5;
  };

  template <typename T = input_track_t,
            std::enable_if_t<std::is_same<T, BoundParameters>::value, int> = 0>
  FullBilloirVertexFitter(const Config& cfg)
      : m_cfg(cfg), extractParameters([](T params) { return params; }) {}

  FullBilloirVertexFitter(const Config& cfg,
                          std::function<BoundParameters(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) const;

 private:
  Config m_cfg;

  std::function<BoundParameters(input_track_t)> extractParameters;
};

}  // namespace Acts

#include "FullBilloirVertexFitter.ipp"