Python/src/Input.cpp

0001 // This file is part of the Acts project.
0002 //
0003 // Copyright (C) 2021 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 http://mozilla.org/MPL/2.0/.
0008
0009 #include "Acts/Plugins/Python/Utilities.hpp"
0010 #include "ActsExamples/EventData/Cluster.hpp"
0011 #include "ActsExamples/Io/Csv/CsvDriftCircleReader.hpp"
0012 #include "ActsExamples/Io/Csv/CsvMeasurementReader.hpp"
0013 #include "ActsExamples/Io/Csv/CsvMuonSimHitReader.hpp"
0014 #include "ActsExamples/Io/Csv/CsvParticleReader.hpp"
0015 #include "ActsExamples/Io/Csv/CsvPlanarClusterReader.hpp"
0016 #include "ActsExamples/Io/Csv/CsvSimHitReader.hpp"
0017 #include "ActsExamples/Io/Csv/CsvSpacePointReader.hpp"
0018 #include "ActsExamples/Io/Csv/CsvTrackParameterReader.hpp"
0019 #include "ActsExamples/Io/Root/RootAthenaNTupleReader.hpp"
0020 #include "ActsExamples/Io/Root/RootMaterialTrackReader.hpp"
0021 #include "ActsExamples/Io/Root/RootParticleReader.hpp"
0022 #include "ActsExamples/Io/Root/RootSimHitReader.hpp"
0023 #include "ActsExamples/Io/Root/RootTrackSummaryReader.hpp"
0024 #include "ActsExamples/Io/Root/RootVertexReader.hpp"
0025
0026 #include <memory>
0027
0028 #include <pybind11/pybind11.h>
0029 #include <pybind11/stl.h>
0030
0031 namespace py = pybind11;
0032 using namespace pybind11::literals;
0033
0034 using namespace ActsExamples;
0035
0036 namespace Acts::Python {
0037
0038 void addInput(Context& ctx) {
0039   auto mex = ctx.get("examples");
0040
0041   // ROOT READERS
0042   ACTS_PYTHON_DECLARE_READER(ActsExamples::RootParticleReader, mex,
0043                              "RootParticleReader", outputParticles, treeName,
0044                              filePath);
0045
0046   ACTS_PYTHON_DECLARE_READER(ActsExamples::RootVertexReader, mex,
0047                              "RootVertexReader", outputVertices, treeName,
0048                              filePath);
0049
0050   ACTS_PYTHON_DECLARE_READER(ActsExamples::RootMaterialTrackReader, mex,
0051                              "RootMaterialTrackReader", outputMaterialTracks,
0052                              treeName, fileList, readCachedSurfaceInformation);
0053
0054   ACTS_PYTHON_DECLARE_READER(ActsExamples::RootTrackSummaryReader, mex,
0055                              "RootTrackSummaryReader", outputTracks,
0056                              outputParticles, treeName, filePath);
0057
0058   // CSV READERS
0059   ACTS_PYTHON_DECLARE_READER(ActsExamples::CsvParticleReader, mex,
0060                              "CsvParticleReader", inputDir, inputStem,
0061                              outputParticles);
0062
0063   ACTS_PYTHON_DECLARE_READER(
0064       ActsExamples::CsvMeasurementReader, mex, "CsvMeasurementReader", inputDir,
0065       outputMeasurements, outputMeasurementSimHitsMap, outputSourceLinks,
0066       outputClusters, outputMeasurementParticlesMap, inputSimHits);
0067
0068   ACTS_PYTHON_DECLARE_READER(ActsExamples::CsvPlanarClusterReader, mex,
0069                              "CsvPlanarClusterReader", inputDir, outputClusters,
0070                              outputHitIds, outputMeasurementParticlesMap,
0071                              outputSimHits, trackingGeometry);
0072
0073   ACTS_PYTHON_DECLARE_READER(ActsExamples::CsvSimHitReader, mex,
0074                              "CsvSimHitReader", inputDir, inputStem,
0075                              outputSimHits);
0076   ACTS_PYTHON_DECLARE_READER(ActsExamples::CsvMuonSimHitReader, mex,
0077                              "CsvMuonSimHitReader", inputDir, inputStem,
0078                              outputSimHits);
0079   ACTS_PYTHON_DECLARE_READER(ActsExamples::CsvDriftCircleReader, mex,
0080                              "CsvDriftCircleReader", inputDir, inputStem,
0081                              outputDriftCircles);
0082
0083   ACTS_PYTHON_DECLARE_READER(
0084       ActsExamples::CsvSpacePointReader, mex, "CsvSpacePointReader", inputDir,
0085       inputStem, inputCollection, outputSpacePoints, extendCollection);
0086
0087   ACTS_PYTHON_DECLARE_READER(ActsExamples::CsvTrackParameterReader, mex,
0088                              "CsvTrackParameterReader", inputDir, inputStem,
0089                              outputTrackParameters, beamspot);
0090
0091   ACTS_PYTHON_DECLARE_READER(ActsExamples::RootAthenaNTupleReader, mex,
0092                              "RootAthenaNTupleReader", inputTreeName,
0093                              inputFilePath, outputTrackParameters,
0094                              outputTruthVtxParameters, outputRecoVtxParameters,
0095                              outputBeamspotConstraint);
0096
0097   ACTS_PYTHON_DECLARE_READER(ActsExamples::RootSimHitReader, mex,
0098                              "RootSimHitReader", treeName, filePath,
0099                              outputSimHits);
0100 }
0101
0102 }  // namespace Acts::Python