make canonic for pomdps

lib/stormpy/pomdp/__init__.py

@@ -1,3 +1,5 @@
 from . import pomdp
 from .pomdp import *
 
+def make_canonic(model):
+    return pomdp._make_canonic_Double(model)

src/mod_pomdp.cpp

@@ -1 +1,13 @@
 
+#include "common.h"
+#include "pomdp/transformations.h"
+
+PYBIND11_MODULE(pomdp, m) {
+    m.doc() = "Functionality for POMDP analysis";
+
+#ifdef STORMPY_DISABLE_SIGNATURE_DOC
+    py::options options;
+    options.disable_function_signatures();
+#endif
+    define_transformations<double>(m, "Double");
+}

src/pomdp/common.h

@@ -0,0 +1 @@
+#include "src/common.h"

src/pomdp/transformations.cpp

@@ -0,0 +1,15 @@
+#include "transformations.h"
+#include <storm-pomdp/transformer/MakePOMDPCanonic.h>
+
+template<typename ValueType>
+std::shared_ptr<storm::models::sparse::Pomdp<ValueType>> make_canonic(storm::models::sparse::Pomdp<ValueType> const& pomdp) {
+    storm::transformer::MakePOMDPCanonic<ValueType> makeCanonic(pomdp);
+    return makeCanonic.transform();
+}
+
+template<typename ValueType>
+void define_transformations(py::module& m, std::string const& vtSuffix) {
+    m.def(("_make_canonic_" + vtSuffix).c_str(), &make_canonic<ValueType>, "Return a canonicly-ordered POMDP", py::arg("pomdp"));
+}
+
+template void define_transformations<double>(py::module& m, std::string const& vtSuffix);

src/pomdp/transformations.h

@@ -0,0 +1,6 @@
+#pragma once
+
+#include "common.h"
+
+template<typename VT>
+void define_transformations(py::module& m, std::string const& vtSuffix);