GaussianTrackDensity.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file GaussianTrackDensity.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/Vertexing/TrackDensity.hpp"

namespace Acts {

class GaussianTrackDensity {
 public:
  struct State {
    State(unsigned int nTracks) : trackDensityState(nTracks) {}
    // The track density
    // Defaulted to Gaussian shaped density function
    TrackDensity trackDensity;

    // The track density state
    TrackDensity::State trackDensityState;
  };

  struct Config {
    // Maximum d0 impact parameter significance to use a track
    double d0MaxSignificance = 3.5;

    // Maximum z0 impact parameter significance to use a track
    double z0MaxSignificance = 12.;
  };

  GaussianTrackDensity() = default;

  GaussianTrackDensity(const Config& cfg) : m_cfg(cfg) {}

  double globalMaximum(const std::vector<Acts::BoundParameters>& trackList,
                       State& state) const;

  std::pair<double, double> globalMaximumWithWidth(
      const std::vector<Acts::BoundParameters>& trackList, State& state) const;

 private:
  Config m_cfg;

  void addTracks(const std::vector<Acts::BoundParameters>& trackList,
                 State& state) const;
};
}  // namespace Acts