Examples/src/Vertexing.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/Index.hpp"
0010 #include "ActsExamples/Vertexing/AdaptiveMultiVertexFinderAlgorithm.hpp"
0011 #include "ActsExamples/Vertexing/HoughVertexFinderAlgorithm.hpp"
0012 #include "ActsExamples/Vertexing/IterativeVertexFinderAlgorithm.hpp"
0013 #include "ActsExamples/Vertexing/VertexFitterAlgorithm.hpp"
0014 #include "ActsPython/Utilities/Helpers.hpp"
0015 #include "ActsPython/Utilities/Macros.hpp"
0016
0017 #include <memory>
0018
0019 #include <pybind11/pybind11.h>
0020 #include <pybind11/stl.h>
0021
0022 namespace py = pybind11;
0023
0024 using namespace ActsExamples;
0025 using namespace Acts;
0026
0027 namespace ActsPython {
0028
0029 void addVertexing(py::module& mex) {
0030   using Seeder = AdaptiveMultiVertexFinderAlgorithm::SeedFinder;
0031
0032   py::enum_<Seeder>(mex, "VertexSeedFinder")
0033       .value("TruthSeeder", Seeder::TruthSeeder)
0034       .value("GaussianSeeder", Seeder::GaussianSeeder)
0035       .value("AdaptiveGridSeeder", Seeder::AdaptiveGridSeeder);
0036
0037   ACTS_PYTHON_DECLARE_ALGORITHM(
0038       AdaptiveMultiVertexFinderAlgorithm, mex,
0039       "AdaptiveMultiVertexFinderAlgorithm", inputTrackParameters,
0040       inputTruthParticles, inputTruthVertices, outputProtoVertices,
0041       outputVertices, seedFinder, bField, minWeight, doSmoothing, maxIterations,
0042       useTime, tracksMaxZinterval, initialVariances, doFullSplitting,
0043       tracksMaxSignificance, maxMergeVertexSignificance, spatialBinExtent,
0044       temporalBinExtent, simultaneousSeeds);
0045
0046   ACTS_PYTHON_DECLARE_ALGORITHM(IterativeVertexFinderAlgorithm, mex,
0047                                 "IterativeVertexFinderAlgorithm",
0048                                 inputTrackParameters, outputProtoVertices,
0049                                 outputVertices, bField, maxIterations);
0050
0051   ACTS_PYTHON_DECLARE_ALGORITHM(VertexFitterAlgorithm, mex,
0052                                 "VertexFitterAlgorithm", inputTrackParameters,
0053                                 inputProtoVertices, outputVertices, bField,
0054                                 doConstrainedFit, constraintPos, constraintCov);
0055
0056   ACTS_PYTHON_DECLARE_ALGORITHM(HoughVertexFinderAlgorithm, mex,
0057                                 "HoughVertexFinderAlgorithm", inputSpacepoints,
0058                                 outputVertices, targetSPs, minAbsEta, maxAbsEta,
0059                                 minHits, defVtxPosition);
0060 }
0061
0062 }  // namespace ActsPython