File indexing completed on 2025-01-18 09:12:01
0001
0002
0003
0004
0005
0006
0007
0008
0009 #include "Acts/Plugins/Python/Utilities.hpp"
0010 #include "ActsExamples/AmbiguityResolution/GreedyAmbiguityResolutionAlgorithm.hpp"
0011 #include "ActsExamples/AmbiguityResolution/ScoreBasedAmbiguityResolutionAlgorithm.hpp"
0012
0013 #include <memory>
0014
0015 #include <pybind11/pybind11.h>
0016 #include <pybind11/stl.h>
0017
0018 namespace py = pybind11;
0019
0020 using namespace Acts;
0021 using namespace ActsExamples;
0022
0023 namespace Acts::Python {
0024
0025 void addAmbiguityResolution(Context& ctx) {
0026 auto [m, mex] = ctx.get("main", "examples");
0027
0028 ACTS_PYTHON_DECLARE_ALGORITHM(
0029 ActsExamples::GreedyAmbiguityResolutionAlgorithm, mex,
0030 "GreedyAmbiguityResolutionAlgorithm", inputTracks, outputTracks,
0031 maximumSharedHits, maximumIterations, nMeasurementsMin);
0032
0033 ACTS_PYTHON_DECLARE_ALGORITHM(
0034 ActsExamples::ScoreBasedAmbiguityResolutionAlgorithm, mex,
0035 "ScoreBasedAmbiguityResolutionAlgorithm", inputTracks, configFile,
0036 outputTracks, minScore, minScoreSharedTracks, maxShared,
0037 maxSharedTracksPerMeasurement, pTMin, pTMax, phiMin, phiMax, etaMin,
0038 etaMax, useAmbiguityFunction);
0039 }
0040
0041 }