diff --git a/src/storage/expressions.cpp b/src/storage/expressions.cpp index 88a5176..4cc2941 100644 --- a/src/storage/expressions.cpp +++ b/src/storage/expressions.cpp @@ -32,6 +32,7 @@ void define_expressions(py::module& m) { // Variable py::class_>(m, "Variable", "Represents a variable") .def_property_readonly("name", &storm::expressions::Variable::getName, "Variable name") + .def_property_readonly("manager", &storm::expressions::Variable::getManager, "Variable manager") .def("has_boolean_type", &storm::expressions::Variable::hasBooleanType, "Check if the variable is of boolean type") .def("has_integer_type", &storm::expressions::Variable::hasIntegerType, "Check if the variable is of integer type") .def("has_rational_type", &storm::expressions::Variable::hasRationalType, "Check if the variable is of rational type") diff --git a/src/storage/prism.cpp b/src/storage/prism.cpp index aab935c..109ee8f 100644 --- a/src/storage/prism.cpp +++ b/src/storage/prism.cpp @@ -87,10 +87,12 @@ void define_prism(py::module& m) { ; py::class_ update(m, "PrismUpdate", "An update in a Prism command"); - update.def_property_readonly("assignments", [](Update const& update) { + update.def(py::init const&>()) + .def_property_readonly("assignments", [](Update const& update) { return update.getAssignments(); }, "Assignments in the update") .def_property_readonly("probability_expression", &Update::getLikelihoodExpression, "The probability expression for this update") + .def_property_readonly("global_index", &Update::getGlobalIndex, "Retrieves the global index of the update, that is, a unique index over all modules") .def("substitute", &Update::substitute, "Substitutes all identifiers in the update according to the given map") .def("simplify", &Update::simplify, "Simplifies the update in various ways (also removes identity assignments)") .def("get_assignment", &Update::getAssignment, py::arg("variable_name"), "Retrieves a reference to the assignment for the variable with the given name") @@ -99,7 +101,8 @@ void define_prism(py::module& m) { ; py::class_ assignment(m, "PrismAssignment", "An assignment in prism"); - assignment.def_property_readonly("variable", &Assignment::getVariable, "Variable that is updated") + assignment.def(py::init()) + .def_property_readonly("variable", &Assignment::getVariable, "Variable that is updated") .def_property_readonly("expression", &Assignment::getExpression, "Expression for the update") .def("__str__", &streamToString) ;