diff --git a/src/storage/prism.cpp b/src/storage/prism.cpp index b982bd0..307ce94 100644 --- a/src/storage/prism.cpp +++ b/src/storage/prism.cpp @@ -9,6 +9,7 @@ void define_prism(py::module& m) { py::class_> program(m, "PrismProgram", "A Prism Program"); program.def_property_readonly("constants", &Program::getConstants, "Get Program Constants") .def_property_readonly("nr_modules", &storm::prism::Program::getNumberOfModules, "Number of modules") + .def_property_readonly("modules", &storm::prism::Program::getModules, "Modules in the program") .def_property_readonly("model_type", &storm::prism::Program::getModelType, "Model type") .def_property_readonly("has_undefined_constants", &storm::prism::Program::hasUndefinedConstants, "Flag if program has undefined constants") .def_property_readonly("undefined_constants_are_graph_preserving", &storm::prism::Program::undefinedConstantsAreGraphPreserving, "Flag if the undefined constants do not change the graph structure") @@ -17,6 +18,23 @@ void define_prism(py::module& m) { .def_property_readonly("expression_manager", &Program::getManager, "Get the expression manager for expressions in this program") .def("__str__", &streamToString); + py::class_ module(m, "PrismModule", "A module in a Prism program"); + module.def_property_readonly("commands", [](Module const& module) {return module.getCommands();}, "Commands in the module") + .def_property_readonly("name", &Module::getName, "Name of the module") + ; + + py::class_ command(m, "PrismCommand", "A command in a Prism program"); + command.def_property_readonly("global_index", &Command::getGlobalIndex, "Get global index") + .def_property_readonly("guard_expression", &Command::getGuardExpression, "Get guard expression") + .def_property_readonly("updates", &Command::getUpdates, "Updates in the command"); + + py::class_ update(m, "PrismUpdate", "An update in a Prism command"); + update.def_property_readonly("assignments", &Update::getAssignments, "Assignments in the update"); + + py::class_ assignment(m, "PrismAssignment", "An assignment in prism"); + assignment.def_property_readonly("variable", &Assignment::getVariable, "Variable that is updated") + .def_property_readonly("expression", &Assignment::getExpression, "Expression for the update"); + // PrismType py::enum_(m, "PrismModelType", "Type of the prism model")