Python/src/EDM4hepComponent.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 #include "ActsExamples/DD4hepDetector/DD4hepDetector.hpp"
0011 #include "ActsExamples/Io/EDM4hep/EDM4hepMeasurementReader.hpp"
0012 #include "ActsExamples/Io/EDM4hep/EDM4hepMeasurementWriter.hpp"
0013 #include "ActsExamples/Io/EDM4hep/EDM4hepMultiTrajectoryWriter.hpp"
0014 #include "ActsExamples/Io/EDM4hep/EDM4hepParticleWriter.hpp"
0015 #include "ActsExamples/Io/EDM4hep/EDM4hepReader.hpp"
0016 #include "ActsExamples/Io/EDM4hep/EDM4hepSimHitWriter.hpp"
0017 #include "ActsExamples/Io/EDM4hep/EDM4hepTrackReader.hpp"
0018 #include "ActsExamples/Io/EDM4hep/EDM4hepTrackWriter.hpp"
0019
0020 #include <pybind11/pybind11.h>
0021 #include <pybind11/stl.h>
0022
0023 namespace py = pybind11;
0024 using namespace pybind11::literals;
0025
0026 using namespace Acts;
0027 using namespace Acts::Python;
0028
0029 PYBIND11_MODULE(ActsPythonBindingsEDM4hep, m) {
0030   ACTS_PYTHON_DECLARE_READER(
0031       ActsExamples::EDM4hepReader, m, "EDM4hepReader", inputPath,
0032       inputParticles, inputSimHits, outputParticlesGenerator,
0033       outputParticlesSimulation, outputSimHits, graphvizOutput, dd4hepDetector,
0034       trackingGeometry, sortSimHitsInTime);
0035
0036   ACTS_PYTHON_DECLARE_WRITER(
0037       ActsExamples::EDM4hepSimHitWriter, m, "EDM4hepSimHitWriter", inputSimHits,
0038       inputParticles, outputPath, outputParticles, outputSimTrackerHits);
0039
0040   ACTS_PYTHON_DECLARE_READER(ActsExamples::EDM4hepMeasurementReader, m,
0041                              "EDM4hepMeasurementReader", inputPath,
0042                              outputMeasurements, outputMeasurementSimHitsMap,
0043                              outputClusters);
0044
0045   ACTS_PYTHON_DECLARE_WRITER(ActsExamples::EDM4hepMeasurementWriter, m,
0046                              "EDM4hepMeasurementWriter", inputMeasurements,
0047                              inputClusters, outputPath);
0048
0049   ACTS_PYTHON_DECLARE_WRITER(ActsExamples::EDM4hepParticleWriter, m,
0050                              "EDM4hepParticleWriter", inputParticles,
0051                              outputPath, outputParticles);
0052
0053   ACTS_PYTHON_DECLARE_WRITER(ActsExamples::EDM4hepMultiTrajectoryWriter, m,
0054                              "EDM4hepMultiTrajectoryWriter", inputTrajectories,
0055                              inputMeasurementParticlesMap, outputPath, Bz);
0056
0057   ACTS_PYTHON_DECLARE_WRITER(ActsExamples::EDM4hepTrackWriter, m,
0058                              "EDM4hepTrackWriter", inputTracks, outputPath, Bz);
0059
0060   ACTS_PYTHON_DECLARE_READER(ActsExamples::EDM4hepTrackReader, m,
0061                              "EDM4hepTrackReader", inputTracks, outputTracks,
0062                              inputPath, Bz);
0063 }