Back to home page

EIC code displayed by LXR

 
 

    


File indexing completed on 2025-01-18 09:12:01

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 "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 }  // namespace Acts::Python