feat: Python bindings for geantino hypothesis (acts-project#2508)

AJPfleger · Oct 6, 2023 · 577c162 · 577c162
1 parent 3201e75
commit 577c162
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/Core/include/Acts/EventData/ParticleHypothesis.hpp b/Core/include/Acts/EventData/ParticleHypothesis.hpp
@@ -110,6 +110,9 @@ class NonNeutralChargedParticleHypothesis
                                                pion().mass(), absQ);
   }
 
+  static NonNeutralChargedParticleHypothesis chargedGeantino() {
+    return chargedGeantino(Acts::UnitConstants::e);
+  }
   static NonNeutralChargedParticleHypothesis chargedGeantino(float absQ) {
     return NonNeutralChargedParticleHypothesis(PdgParticle::eInvalid, 0, absQ);
   }
@@ -154,6 +157,9 @@ class ParticleHypothesis : public GenericParticleHypothesis<AnyCharge> {
   static ParticleHypothesis geantino() {
     return NeutralParticleHypothesis::geantino();
   }
+  static ParticleHypothesis chargedGeantino() {
+    return chargedGeantino(Acts::UnitConstants::e);
+  }
   static ParticleHypothesis chargedGeantino(float absQ) {
     return ParticleHypothesis(PdgParticle::eInvalid, 0, absQ);
   }

diff --git a/Examples/Python/src/EventData.cpp b/Examples/Python/src/EventData.cpp
@@ -36,9 +36,20 @@ void addEventData(Context& ctx) {
                                     [](py::object /* self */) {
                                       return Acts::ParticleHypothesis::pion();
                                     })
-      .def_property_readonly_static("electron", [](py::object /* self */) {
-        return Acts::ParticleHypothesis::electron();
-      });
+      .def_property_readonly_static(
+          "electron",
+          [](py::object /* self */) {
+            return Acts::ParticleHypothesis::electron();
+          })
+      .def_property_readonly_static(
+          "geantino",
+          [](py::object /* self */) {
+            return Acts::ParticleHypothesis::geantino();
+          })
+      .def_property_readonly_static(
+          "chargedGeantino", [](py::object /* self */) {
+            return Acts::ParticleHypothesis::chargedGeantino();
+          });
 }
 
 }  // namespace Acts::Python