Python/src/Utilities.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 "Acts/Plugins/Python/Utilities.hpp"
0010
0011 #include "ActsExamples/Utilities/MeasurementMapSelector.hpp"
0012 #include "ActsExamples/Utilities/ProtoTracksToTracks.hpp"
0013 #include "ActsExamples/Utilities/PrototracksToSeeds.hpp"
0014 #include "ActsExamples/Utilities/SeedsToPrototracks.hpp"
0015 #include "ActsExamples/Utilities/TracksToParameters.hpp"
0016 #include "ActsExamples/Utilities/TracksToTrajectories.hpp"
0017 #include "ActsExamples/Utilities/TrajectoriesToPrototracks.hpp"
0018
0019 #include <pybind11/pybind11.h>
0020 #include <pybind11/stl.h>
0021
0022 namespace py = pybind11;
0023
0024 using namespace ActsExamples;
0025 using namespace Acts;
0026
0027 namespace Acts::Python {
0028
0029 void addUtilities(Context& ctx) {
0030   auto [m, mex] = ctx.get("main", "examples");
0031
0032   ACTS_PYTHON_DECLARE_ALGORITHM(ActsExamples::TrajectoriesToPrototracks, mex,
0033                                 "TrajectoriesToPrototracks", inputTrajectories,
0034                                 outputProtoTracks);
0035
0036   ACTS_PYTHON_DECLARE_ALGORITHM(ActsExamples::TracksToTrajectories, mex,
0037                                 "TracksToTrajectories", inputTracks,
0038                                 outputTrajectories);
0039
0040   ACTS_PYTHON_DECLARE_ALGORITHM(ActsExamples::TracksToParameters, mex,
0041                                 "TracksToParameters", inputTracks,
0042                                 outputTrackParameters);
0043
0044   ACTS_PYTHON_DECLARE_ALGORITHM(ActsExamples::SeedsToPrototracks, mex,
0045                                 "SeedsToPrototracks", inputSeeds,
0046                                 outputProtoTracks);
0047
0048   ACTS_PYTHON_DECLARE_ALGORITHM(
0049       ActsExamples::PrototracksToSeeds, mex, "PrototracksToSeeds",
0050       inputProtoTracks, inputSpacePoints, outputSeeds, outputProtoTracks);
0051
0052   ACTS_PYTHON_DECLARE_ALGORITHM(
0053       ActsExamples::MeasurementMapSelector, mex, "MeasurementMapSelector",
0054       inputMeasurements, inputMeasurementParticleMap,
0055       outputMeasurementParticleMap, geometrySelection);
0056
0057   ACTS_PYTHON_DECLARE_ALGORITHM(ActsExamples::PrototracksToTracks, mex,
0058                                 "PrototracksToTracks", inputMeasurements,
0059                                 inputProtoTracks, outputTracks);
0060 }
0061
0062 }  // namespace Acts::Python