From 77dff27866a769b91611d06dbacf9b2ad0a67906 Mon Sep 17 00:00:00 2001
From: Tom Janson <tom.janson@rwth-aachen.de>
Date: Thu, 15 Dec 2016 00:09:40 +0100
Subject: [PATCH] Path (from storm::utility::ksp) pybind proof of concept

this stuff is not fun.
---
 src/mod_utility.cpp           |  9 ++++++
 src/utility/shortestPaths.cpp | 61 +++++++++++++++++++++++++++++++++++
 src/utility/shortestPaths.h   |  8 +++++
 3 files changed, 78 insertions(+)
 create mode 100644 src/mod_utility.cpp
 create mode 100644 src/utility/shortestPaths.cpp
 create mode 100644 src/utility/shortestPaths.h

diff --git a/src/mod_utility.cpp b/src/mod_utility.cpp
new file mode 100644
index 0000000..f222c8f
--- /dev/null
+++ b/src/mod_utility.cpp
@@ -0,0 +1,9 @@
+#include "common.h"
+
+#include "utility/shortestPaths.h"
+
+PYBIND11_PLUGIN(utility) {
+    py::module m("utility", "Dumping ground of stuff that really should be somewhere more reasonable");
+    define_ksp(m);
+    return m.ptr();
+}
diff --git a/src/utility/shortestPaths.cpp b/src/utility/shortestPaths.cpp
new file mode 100644
index 0000000..13b8c34
--- /dev/null
+++ b/src/utility/shortestPaths.cpp
@@ -0,0 +1,61 @@
+#include "shortestPaths.h"
+#include "storm/utility/shortestPaths.h"
+
+#include <string>
+
+using namespace pybind11::literals;
+
+void define_ksp(py::module& m) {
+
+    using Path = storm::utility::ksp::Path<double>;
+    using state_t = storm::utility::ksp::state_t;
+
+    // (k-shortest) Path
+    py::class_<Path>(m, "Path")
+        // Fuck all this. FIXME
+
+        //.def(py::init<boost::optional<state_t>, unsigned long, double>()) // does not work (because it's an aggregate initialized struct?)
+
+        // this may or may not be working (i.e., initializing with the values as expected)
+        // https://pybind11.readthedocs.io/en/latest/advanced/classes.html#custom-constructors
+        .def("__init__", [](Path &instance, state_t preNode, unsigned long preK, double distance) {
+                new (&instance) Path { boost::optional<state_t>(preNode), preK, distance };
+            }, "predecessorNode"_a, "predecessorK"_a, "distance"_a)
+
+        .def("__init__", [](Path &instance, unsigned long preK, double distance) {
+                new (&instance) Path { boost::none, preK, distance };
+            }, "predecessorK"_a, "distance"_a)
+
+        // this actually seemed to work, once!, but not anymore
+        .def(py::self == py::self)
+
+        .def("__repr__",
+            [](Path const& a) {
+                // shit this is ugly. but it's a proof of concept -- at least it works
+                std::string str = "<Path with";
+                if (a.predecessorNode) {
+                    str += " predecessorNode: " + std::to_string(a.predecessorNode.get());
+                }
+                str += " predecessorK " + std::to_string(a.predecessorK);
+                str += " distance " + std::to_string(a.distance);
+                str += ">";
+                return str;
+            })
+
+        .def("debug_distance",
+            [](Path const& a) {
+                return a.distance; // yeah, that's it, that's all i want you to do you lazy fuck. it's just a double. jesus
+            })
+
+        .def_readwrite("predecessorNode", &Path::predecessorNode) // does not work (again due to struct??) FIXME
+        .def_readwrite("predecessorK", &Path::predecessorK)
+        .def_readwrite("distance", &Path::distance)
+        ;
+
+    // TODO continue
+    /*
+    // ShortestPathsGenerator
+    py::class_<storm::utility::ksp::ShortestPathsGenerator<double>>(m, "ShortestPathsGenerator")
+            ;
+    */
+}
\ No newline at end of file
diff --git a/src/utility/shortestPaths.h b/src/utility/shortestPaths.h
new file mode 100644
index 0000000..5cc0a97
--- /dev/null
+++ b/src/utility/shortestPaths.h
@@ -0,0 +1,8 @@
+#ifndef PYTHON_UTILITY_SHORTESTPATHS_H_
+#define PYTHON_UTILITY_SHORTESTPATHS_H_
+
+#include "src/common.h"
+
+void define_ksp(py::module& m);
+
+#endif /* PYTHON_UTILITY_SHORTESTPATHS_H_ */