Support for Interaction class

Harvard-Neutrino · Dec 17, 2024 · b3ef5ca · b3ef5ca
1 parent 6210de4
commit b3ef5ca
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 2 deletions.
diff --git a/projects/interactions/private/pybindings/CrossSection.h b/projects/interactions/private/pybindings/CrossSection.h
@@ -6,6 +6,7 @@
 #include <pybind11/operators.h>
 #include <pybind11/stl.h>
 
+#include "../../public/SIREN/interactions/Interaction.h"
 #include "../../public/SIREN/interactions/CrossSection.h"
 #include "../../public/SIREN/interactions/pyCrossSection.h"
 #include "../../../dataclasses/public/SIREN/dataclasses/Particle.h"
@@ -18,7 +19,7 @@ void register_CrossSection(pybind11::module_ & m) {
     using namespace pybind11;
     using namespace siren::interactions;
 
-    class_<CrossSection, std::shared_ptr<CrossSection>, pyCrossSection>(m, "CrossSection")
+    class_<CrossSection, std::shared_ptr<CrossSection>, pyCrossSection, Interaction>(m, "CrossSection")
         .def(init<>())
         .def("__eq__", [](const CrossSection &self, const CrossSection &other){ return self == other; })
         .def("equal", &CrossSection::equal)

diff --git a/projects/interactions/private/pybindings/Decay.h b/projects/interactions/private/pybindings/Decay.h
@@ -6,6 +6,7 @@
 #include <pybind11/operators.h>
 #include <pybind11/stl.h>
 
+#include "../../public/SIREN/interactions/Interaction.h"
 #include "../../public/SIREN/interactions/CrossSection.h"
 #include "../../public/SIREN/interactions/Decay.h"
 #include "../../public/SIREN/interactions/pyDecay.h"
@@ -17,7 +18,7 @@ void register_Decay(pybind11::module_ & m) {
     using namespace pybind11;
     using namespace siren::interactions;
 
-    class_<Decay, std::shared_ptr<Decay>, pyDecay>(m, "Decay")
+    class_<Decay, std::shared_ptr<Decay>, pyDecay, Interaction>(m, "Decay")
         .def("TotalDecayLength",&Decay::TotalDecayLength)
         .def("TotalDecayLengthForFinalState",&Decay::TotalDecayLengthForFinalState);
 

diff --git a/projects/interactions/private/pybindings/InteractionCollection.h b/projects/interactions/private/pybindings/InteractionCollection.h
@@ -6,6 +6,7 @@
 #include <pybind11/operators.h>
 #include <pybind11/stl.h>
 
+#include "../../public/SIREN/interactions/Interaction.h"
 #include "../../public/SIREN/interactions/CrossSection.h"
 #include "../../public/SIREN/interactions/InteractionCollection.h"
 #include "../../public/SIREN/interactions/Decay.h"
@@ -22,6 +23,7 @@ void register_InteractionCollection(pybind11::module_ & m) {
         .def(init<siren::dataclasses::ParticleType, std::vector<std::shared_ptr<CrossSection>>>())
         .def(init<siren::dataclasses::ParticleType, std::vector<std::shared_ptr<Decay>>>())
         .def(init<siren::dataclasses::ParticleType, std::vector<std::shared_ptr<CrossSection>>, std::vector<std::shared_ptr<Decay>>>())
+        .def(init<siren::dataclasses::ParticleType, std::vector<std::shared_ptr<Interaction>>>())
         .def(self == self)
         .def("GetCrossSections",&InteractionCollection::GetCrossSections, return_value_policy::reference_internal)
         .def("GetDecays",&InteractionCollection::GetDecays, return_value_policy::reference_internal)

diff --git a/projects/interactions/private/pybindings/interactions.cxx b/projects/interactions/private/pybindings/interactions.cxx
@@ -1,6 +1,7 @@
 
 #include <vector>
 
+#include "../../public/SIREN/interactions/Interaction.h"
 #include "../../public/SIREN/interactions/CrossSection.h"
 #include "../../public/SIREN/interactions/NeutrissimoDecay.h"
 #include "../../public/SIREN/interactions/InteractionCollection.h"
@@ -10,6 +11,7 @@
 #include "../../public/SIREN/interactions/DarkNewsCrossSection.h"
 #include "../../public/SIREN/interactions/DarkNewsDecay.h"
 
+#include "./Interaction.h"
 #include "./CrossSection.h"
 #include "./DipoleFromTable.h"
 #include "./DarkNewsCrossSection.h"
@@ -32,6 +34,7 @@ using namespace pybind11;
 PYBIND11_MODULE(interactions,m) {
     using namespace siren::interactions;
 
+    register_Interaction(m);
     register_CrossSection(m);
     register_Decay(m);
     register_DipoleFromTable(m);