Examples/src/Input.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/EventData/Cluster.hpp"
0010 #include "ActsExamples/Framework/BufferedReader.hpp"
0011 #include "ActsExamples/Io/Csv/CsvGnnGraphReader.hpp"
0012 #include "ActsExamples/Io/Csv/CsvMeasurementReader.hpp"
0013 #include "ActsExamples/Io/Csv/CsvMuonSegmentReader.hpp"
0014 #include "ActsExamples/Io/Csv/CsvMuonSpacePointReader.hpp"
0015 #include "ActsExamples/Io/Csv/CsvParticleReader.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/TrackFinding/ITrackParamsLookupReader.hpp"
0020 #include "ActsPython/Utilities/Helpers.hpp"
0021 #include "ActsPython/Utilities/Macros.hpp"
0022
0023 #include <pybind11/pybind11.h>
0024 #include <pybind11/stl.h>
0025
0026 namespace py = pybind11;
0027 using namespace pybind11::literals;
0028
0029 using namespace ActsExamples;
0030
0031 namespace ActsPython {
0032
0033 void addInput(py::module& mex) {
0034   // Buffered reader
0035   ACTS_PYTHON_DECLARE_READER(BufferedReader, mex, "BufferedReader",
0036                              upstreamReader, selectionSeed, bufferSize);
0037
0038   ACTS_PYTHON_DECLARE_READER(CsvParticleReader, mex, "CsvParticleReader",
0039                              inputDir, inputStem, outputParticles);
0040
0041   ACTS_PYTHON_DECLARE_READER(CsvMeasurementReader, mex, "CsvMeasurementReader",
0042                              inputDir, outputMeasurements,
0043                              outputMeasurementSimHitsMap, outputClusters,
0044                              outputMeasurementParticlesMap, inputSimHits);
0045
0046   ACTS_PYTHON_DECLARE_READER(CsvSimHitReader, mex, "CsvSimHitReader", inputDir,
0047                              inputStem, outputSimHits);
0048   ACTS_PYTHON_DECLARE_READER(CsvMuonSegmentReader, mex, "CsvMuonSegmentReader",
0049                              inputDir, inputStem, outputSegments);
0050   ACTS_PYTHON_DECLARE_READER(CsvMuonSpacePointReader, mex,
0051                              "CsvMuonSpacePointReader", inputDir, inputStem,
0052                              outputSpacePoints);
0053
0054   ACTS_PYTHON_DECLARE_READER(CsvSpacePointReader, mex, "CsvSpacePointReader",
0055                              inputDir, inputStem, inputCollection,
0056                              outputSpacePoints, extendCollection);
0057
0058   ACTS_PYTHON_DECLARE_READER(CsvTrackParameterReader, mex,
0059                              "CsvTrackParameterReader", inputDir, inputStem,
0060                              outputTrackParameters, beamspot);
0061
0062   ACTS_PYTHON_DECLARE_READER(CsvGnnGraphReader, mex, "CsvGnnGraphReader",
0063                              inputDir, inputStem, outputGraph);
0064
0065   py::class_<ITrackParamsLookupReader,
0066              std::shared_ptr<ITrackParamsLookupReader>>(
0067       mex, "ITrackParamsLookupReader");
0068 }
0069
0070 }  // namespace ActsPython