Examples/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(py::module& mex) {
0029   ACTS_PYTHON_DECLARE_ALGORITHM(TrajectoriesToPrototracks, mex,
0030                                 "TrajectoriesToPrototracks", inputTrajectories,
0031                                 outputProtoTracks);
0032
0033   ACTS_PYTHON_DECLARE_ALGORITHM(TracksToTrajectories, mex,
0034                                 "TracksToTrajectories", inputTracks,
0035                                 outputTrajectories);
0036
0037   ACTS_PYTHON_DECLARE_ALGORITHM(TracksToParameters, mex, "TracksToParameters",
0038                                 inputTracks, outputTrackParameters);
0039
0040   ACTS_PYTHON_DECLARE_ALGORITHM(SeedsToPrototracks, mex, "SeedsToPrototracks",
0041                                 inputSeeds, outputProtoTracks);
0042
0043   ACTS_PYTHON_DECLARE_ALGORITHM(PrototracksToSeeds, mex, "PrototracksToSeeds",
0044                                 inputProtoTracks, inputSpacePoints, outputSeeds,
0045                                 outputProtoTracks);
0046
0047   ACTS_PYTHON_DECLARE_ALGORITHM(
0048       MeasurementMapSelector, mex, "MeasurementMapSelector", inputMeasurements,
0049       inputMeasurementParticleMap, outputMeasurementParticleMap,
0050       geometrySelection);
0051
0052   ACTS_PYTHON_DECLARE_ALGORITHM(PrototracksToTracks, mex, "PrototracksToTracks",
0053                                 inputMeasurements, inputProtoTracks,
0054                                 inputTrackParameters, outputTracks);
0055 }
0056
0057 }  // namespace ActsPython