feat: Bind more EDM types, simple python only demo script

.github/workflows/builds.yml

@@ -635,4 +635,4 @@ jobs:
           path: "*.whl"
 
       - name: Test
-        run: CI/dependencies/run.sh .env python3 Python/Examples/scripts/hello_world_kalman.py
+        run: CI/dependencies/run.sh .env python3 Examples/Scripts/Python/track_finding_python_only.py

Examples/Scripts/Python/track_finding_python_only.py

@@ -0,0 +1,166 @@
+#!/usr/bin/env python3
+# This file is part of the ACTS project.
+#
+# Copyright (C) 2016 CERN for the benefit of the ACTS project
+#
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+import os
+from pathlib import Path
+
+os.environ["ACTS_SEQUENCER_DISABLE_FPEMON"] = "1"
+
+import acts
+import acts.examples
+from acts import UnitConstants as u
+
+
+def runTrackFindingPythonOnly(
+    trackingGeometry,
+    field,
+    digiConfigFile,
+    geoSelectionConfigFile,
+    outputDir,
+    decorators=[],
+    s=None,
+):
+    from acts.examples.simulation import (
+        addParticleGun,
+        MomentumConfig,
+        EtaConfig,
+        PhiConfig,
+        ParticleConfig,
+        addFatras,
+        addDigitization,
+    )
+
+    s = s or acts.examples.Sequencer(events=1, numThreads=1, logLevel=acts.logging.INFO)
+    outputDir = Path(outputDir)
+    rnd = acts.examples.RandomNumbers(seed=42)
+
+    for d in decorators:
+        s.addContextDecorator(d)
+
+    addParticleGun(
+        s,
+        MomentumConfig(1.0 * u.GeV, 10.0 * u.GeV, transverse=True),
+        EtaConfig(-2.0, 2.0, uniform=True),
+        PhiConfig(0.0, 360.0 * u.degree),
+        ParticleConfig(1, acts.PdgParticle.eMuon, randomizeCharge=True),
+        rnd=rnd,
+    )
+
+    addFatras(
+        s,
+        trackingGeometry,
+        field,
+        rnd=rnd,
+    )
+
+    addDigitization(
+        s,
+        trackingGeometry,
+        field,
+        digiConfigFile=digiConfigFile,
+        rnd=rnd,
+    )
+
+    s.addAlgorithm(
+        acts.examples.SpacePointMaker(
+            level=acts.logging.INFO,
+            trackingGeometry=trackingGeometry,
+            inputMeasurements="measurements",
+            outputSpacePoints="spacepoints",
+            geometrySelection=acts.examples.json.readJsonGeometryList(
+                str(geoSelectionConfigFile)
+            ),
+        )
+    )
+
+    class PythonTrackFinder(acts.examples.IAlgorithm):
+        def __init__(self, name, level):
+            acts.examples.IAlgorithm.__init__(self, name, level)
+
+            self.spacepoints = acts.examples.ReadDataHandle(
+                self, acts.SpacePointContainer2, "Spacepoints"
+            )
+            self.spacepoints.initialize("spacepoints")
+
+            self.prototracks = acts.examples.WriteDataHandle(
+                self, acts.examples.ProtoTrackContainer, "Prototracks"
+            )
+            self.prototracks.initialize("prototracks")
+
+        def execute(self, context):
+            spacepoints = self.spacepoints(context.eventStore)
+
+            track = acts.examples.ProtoTrack()
+            for sp in sorted(spacepoints, key=lambda sp: sp.r):
+                for sl in sp.sourceLinks:
+                    isl = acts.examples.IndexSourceLink.FromSourceLink(sl)
+                    track.append(isl.index())
+
+            prototracks = acts.examples.ProtoTrackContainer()
+            prototracks.append(track)
+
+            self.prototracks(context, prototracks)
+            return acts.examples.ProcessCode.SUCCESS
+
+    s.addAlgorithm(PythonTrackFinder("PythonTrackFinder", acts.logging.INFO))
+
+    class PythonTrackFitter(acts.examples.IAlgorithm):
+        def __init__(self, name, level):
+            acts.examples.IAlgorithm.__init__(self, name, level)
+
+            self.prototracks = acts.examples.ReadDataHandle(
+                self, acts.examples.ProtoTrackContainer, "Prototracks"
+            )
+            self.prototracks.initialize("prototracks")
+
+            self.tracks = acts.examples.WriteDataHandle(
+                self, acts.examples.ConstTrackContainer, "Tracks"
+            )
+            self.tracks.initialize("fitted_tracks")
+
+        def execute(self, context):
+            prototracks = self.prototracks(context.eventStore)
+
+            container = acts.examples.TrackContainer()
+            for prototrack in prototracks:
+                track = container.makeTrack()
+                track.parameters = acts.BoundVector(1.0, 1.0, 1.0, 1.0, 1.0, 1.0)
+                track.nMeasurements = len(prototrack)
+
+            self.tracks(context, container.makeConst())
+            return acts.examples.ProcessCode.SUCCESS
+
+    s.addAlgorithm(PythonTrackFitter("PythonTrackFitter", acts.logging.INFO))
+
+    return s
+
+
+if __name__ == "__main__":
+    srcdir = Path(__file__).resolve().parent.parent.parent.parent
+
+    detector = acts.examples.GenericDetector(acts.examples.GenericDetector.Config())
+    trackingGeometry = detector.trackingGeometry()
+    decorators = detector.contextDecorators()
+
+    field = acts.ConstantBField(acts.Vector3(0.0, 0.0, 2.0 * u.T))
+
+    digiConfigFile = srcdir / "Examples/Configs/generic-digi-smearing-config.json"
+    geoSelectionConfigFile = srcdir / "Examples/Configs/generic-seeding-config.json"
+
+    outputDir = Path.cwd() / "output_track_finding_python_only"
+    outputDir.mkdir(exist_ok=True)
+
+    runTrackFindingPythonOnly(
+        trackingGeometry=trackingGeometry,
+        field=field,
+        digiConfigFile=digiConfigFile,
+        geoSelectionConfigFile=geoSelectionConfigFile,
+        outputDir=outputDir,
+        decorators=decorators,
+    ).run()

Python/Core/src/EventData.cpp

@@ -222,7 +222,14 @@ void addEventData(py::module_& m) {
           static_cast<FloatGetter>(&ConstSpacePointProxy2::varianceZ))
       .def_property_readonly(
           "varianceR",
-          static_cast<FloatGetter>(&ConstSpacePointProxy2::varianceR));
+          static_cast<FloatGetter>(&ConstSpacePointProxy2::varianceR))
+      .def_property_readonly(
+          "sourceLinks", py::cpp_function(
+                             [](const ConstSpacePointProxy2& self) {
+                               auto sls = self.sourceLinks();
+                               return py::make_iterator(sls.begin(), sls.end());
+                             },
+                             py::keep_alive<0, 1>()));
 
   // SeedContainer2
   auto seedContainer2 =