Python/src/ModuleEntry.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/ActsVersion.hpp"
0010 #include "Acts/Plugins/Python/Utilities.hpp"
0011
0012 #include <tuple>
0013 #include <unordered_map>
0014
0015 #include <pybind11/detail/common.h>
0016 #include <pybind11/functional.h>
0017 #include <pybind11/operators.h>
0018 #include <pybind11/pybind11.h>
0019 #include <pybind11/pytypes.h>
0020 #include <pybind11/stl.h>
0021 #include <pyerrors.h>
0022
0023 namespace py = pybind11;
0024 using namespace Acts::Python;
0025
0026 namespace Acts::Python {
0027 void addContext(Context& ctx);
0028 void addAny(Context& ctx);
0029 void addUnits(Context& ctx);
0030 void addFramework(Context& ctx);
0031 void addLogging(Context& ctx);
0032 void addPdgParticle(Context& ctx);
0033 void addAlgebra(Context& ctx);
0034 void addBinning(Context& ctx);
0035 void addEventData(Context& ctx);
0036
0037 void addPropagation(Context& ctx);
0038 void addNavigation(Context& ctx);
0039
0040 void addGeometry(Context& ctx);
0041 void addGeometryBuildingGen1(Context& ctx);
0042 void addExperimentalGeometry(Context& ctx);
0043
0044 void addMagneticField(Context& ctx);
0045
0046 void addMaterial(Context& ctx);
0047 void addOutput(Context& ctx);
0048 void addDetector(Context& ctx);
0049 void addExampleAlgorithms(Context& ctx);
0050 void addInput(Context& ctx);
0051 void addGenerators(Context& ctx);
0052 void addTruthTracking(Context& ctx);
0053 void addTrackFitting(Context& ctx);
0054 void addTrackFinding(Context& ctx);
0055 void addVertexing(Context& ctx);
0056 void addAmbiguityResolution(Context& ctx);
0057 void addUtilities(Context& ctx);
0058
0059 // Plugins
0060 void addDigitization(Context& ctx);
0061 void addPythia8(Context& ctx);
0062 void addGeoModel(Context& ctx);
0063 void addTGeo(Context& ctx);
0064 void addJson(Context& ctx);
0065 void addDetray(Context& ctx);
0066 void addHepMC3(Context& ctx);
0067 void addExaTrkXTrackFinding(Context& ctx);
0068 void addSvg(Context& ctx);
0069 void addObj(Context& ctx);
0070 void addOnnx(Context& ctx);
0071 void addOnnxNeuralCalibrator(Context& ctx);
0072 void addCovfie(Context& ctx);
0073 void addTraccc(Context& ctx);
0074 void addHashing(Context& ctx);
0075
0076 }  // namespace Acts::Python
0077
0078 PYBIND11_MODULE(ActsPythonBindings, m) {
0079   Acts::Python::Context ctx;
0080   ctx.modules["main"] = m;
0081   auto mex = m.def_submodule("_examples");
0082   ctx.modules["examples"] = mex;
0083   auto prop = m.def_submodule("_propagator");
0084   ctx.modules["propagation"] = prop;
0085   m.doc() = "Acts";
0086
0087   m.attr("__version__") =
0088       std::tuple{Acts::VersionMajor, Acts::VersionMinor, Acts::VersionPatch};
0089
0090   {
0091     auto mv = m.def_submodule("version");
0092
0093     mv.attr("major") = Acts::VersionMajor;
0094     mv.attr("minor") = Acts::VersionMinor;
0095     mv.attr("patch") = Acts::VersionPatch;
0096
0097     mv.attr("commit_hash") = Acts::CommitHash;
0098     mv.attr("commit_hash_short") = Acts::CommitHashShort;
0099   }
0100
0101   addContext(ctx);
0102   addAny(ctx);
0103   addUnits(ctx);
0104   addFramework(ctx);
0105   addLogging(ctx);
0106   addPdgParticle(ctx);
0107   addAlgebra(ctx);
0108   addBinning(ctx);
0109   addEventData(ctx);
0110   addOutput(ctx);
0111
0112   addPropagation(ctx);
0113   addNavigation(ctx);
0114   addGeometryBuildingGen1(ctx);
0115   addGeometry(ctx);
0116   addExperimentalGeometry(ctx);
0117
0118   addMagneticField(ctx);
0119   addMaterial(ctx);
0120   addDetector(ctx);
0121   addExampleAlgorithms(ctx);
0122   addInput(ctx);
0123   addGenerators(ctx);
0124   addTruthTracking(ctx);
0125   addTrackFitting(ctx);
0126   addTrackFinding(ctx);
0127   addVertexing(ctx);
0128   addAmbiguityResolution(ctx);
0129   addUtilities(ctx);
0130
0131   addDigitization(ctx);
0132   addPythia8(ctx);
0133   addJson(ctx);
0134   addGeoModel(ctx);
0135   addTGeo(ctx);
0136   addDetray(ctx);
0137   addHepMC3(ctx);
0138   addExaTrkXTrackFinding(ctx);
0139   addObj(ctx);
0140   addSvg(ctx);
0141   addOnnx(ctx);
0142   addOnnxNeuralCalibrator(ctx);
0143   addCovfie(ctx);
0144   addTraccc(ctx);
0145   addHashing(ctx);
0146 }