TrackDensityVertexFinder.hpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2020 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/EventData/TrackParameters.hpp"
#include "Acts/Utilities/Definitions.hpp"
#include "Acts/Utilities/Result.hpp"
#include "Acts/Vertexing/GaussianTrackDensity.hpp"
#include "Acts/Vertexing/Vertex.hpp"
#include "Acts/Vertexing/VertexFitterConcept.hpp"
#include "Acts/Vertexing/VertexingOptions.hpp"

namespace Acts {

template <typename vfitter_t,
          typename track_density_t =
              GaussianTrackDensity<typename vfitter_t::InputTrack_t>>
class TrackDensityVertexFinder {
  // Provided vertex fitter type should comply with the VertexFitterConcept
  // to ensure providing an input track type InputTrack_t

  // static_assert(VertexFitterConcept<vfitter_t>,
  //              "Vertex fitter does not fulfill vertex fitter concept.");

  using InputTrack_t = typename vfitter_t::InputTrack_t;

 public:
  struct Config {
    // The track density estimator
    track_density_t trackDensityEstimator;
  };

  struct State {};

  Result<std::vector<Vertex<InputTrack_t>>> find(
      const std::vector<const InputTrack_t*>& trackVector,
      const VertexingOptions<InputTrack_t>& vertexingOptions,
      State& state) const;

  template <
      typename T = InputTrack_t,
      std::enable_if_t<std::is_same<T, BoundTrackParameters>::value, int> = 0>
  TrackDensityVertexFinder(const Config& cfg)
      : m_cfg(cfg), m_extractParameters([](T params) { return params; }) {}

  template <
      typename T = InputTrack_t,
      std::enable_if_t<std::is_same<T, BoundTrackParameters>::value, int> = 0>
  TrackDensityVertexFinder()
      : m_extractParameters([](T params) { return params; }) {}

  TrackDensityVertexFinder(
      const Config& cfg,
      const std::function<BoundTrackParameters(InputTrack_t)>& func)
      : m_cfg(cfg), m_extractParameters(func) {}

  TrackDensityVertexFinder(
      const std::function<BoundTrackParameters(InputTrack_t)>& func)
      : m_extractParameters(func) {}

 private:
  Config m_cfg;

  std::function<BoundTrackParameters(InputTrack_t)> m_extractParameters;
};

}  // namespace Acts

#include "TrackDensityVertexFinder.ipp"