Examples/src/AmbiguityResolution.cpp

0001 // This file is part of the ACTS project.
0002 //
0003 // Copyright (C) 2016 CERN for the benefit of the ACTS project
0004 //
0005 // This Source Code Form is subject to the terms of the Mozilla Public
0006 // License, v. 2.0. If a copy of the MPL was not distributed with this
0007 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
0008
0009 #include "ActsExamples/AmbiguityResolution/GreedyAmbiguityResolutionAlgorithm.hpp"
0010 #include "ActsExamples/AmbiguityResolution/ScoreBasedAmbiguityResolutionAlgorithm.hpp"
0011 #include "ActsPython/Utilities/Helpers.hpp"
0012 #include "ActsPython/Utilities/Macros.hpp"
0013
0014 #include <memory>
0015
0016 #include <pybind11/pybind11.h>
0017 #include <pybind11/stl.h>
0018
0019 namespace py = pybind11;
0020
0021 using namespace Acts;
0022 using namespace ActsExamples;
0023
0024 namespace ActsPython {
0025
0026 void addAmbiguityResolution(py::module& mex) {
0027   ACTS_PYTHON_DECLARE_ALGORITHM(
0028       ActsExamples::GreedyAmbiguityResolutionAlgorithm, mex,
0029       "GreedyAmbiguityResolutionAlgorithm", inputTracks, outputTracks,
0030       maximumSharedHits, maximumIterations, nMeasurementsMin);
0031
0032   ACTS_PYTHON_DECLARE_ALGORITHM(
0033       ActsExamples::ScoreBasedAmbiguityResolutionAlgorithm, mex,
0034       "ScoreBasedAmbiguityResolutionAlgorithm", inputTracks, configFile,
0035       outputTracks, minScore, minScoreSharedTracks, maxShared, minUnshared,
0036       maxSharedTracksPerMeasurement, useAmbiguityScoring);
0037 }
0038
0039 }  // namespace ActsPython