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