You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
3.1 KiB
46 lines
3.1 KiB
#include "labeling.h"
|
|
#include "src/helpers.h"
|
|
|
|
#include "storm/models/sparse/ItemLabeling.h"
|
|
#include "storm/models/sparse/StateLabeling.h"
|
|
#include "storm/models/sparse/ChoiceLabeling.h"
|
|
|
|
// Define python bindings
|
|
void define_labeling(py::module& m) {
|
|
|
|
// ItemLabeling
|
|
py::class_<storm::models::sparse::ItemLabeling, std::shared_ptr<storm::models::sparse::ItemLabeling>> labeling(m,"ItemLabeling", "Labeling");
|
|
labeling.def("add_label", [](storm::models::sparse::ItemLabeling& labeling, std::string label) {
|
|
labeling.addLabel(label);
|
|
}, py::arg("label"), "Add label")
|
|
.def("get_labels", &storm::models::sparse::ItemLabeling::getLabels, "Get all labels")
|
|
.def("contains_label", &storm::models::sparse::ItemLabeling::containsLabel, "Check if the given label is contained in the labeling", py::arg("label"))
|
|
.def("__str__", &streamToString<storm::models::sparse::ItemLabeling>)
|
|
;
|
|
|
|
|
|
// StateLabeling
|
|
py::class_<storm::models::sparse::StateLabeling, std::shared_ptr<storm::models::sparse::StateLabeling>>(m, "StateLabeling", "Labeling for states", labeling)
|
|
.def(py::init<uint_fast64_t>(), "state_count"_a)
|
|
.def("get_labels_of_state", &storm::models::sparse::StateLabeling::getLabelsOfState, "Get labels of given state", py::arg("state"))
|
|
.def("add_label_to_state", &storm::models::sparse::StateLabeling::addLabelToState, "Add label to state", py::arg("label"), py::arg("state"))
|
|
.def("has_state_label", &storm::models::sparse::StateLabeling::getStateHasLabel, "Check if the given state has the given label", py::arg("label"), py::arg("state"))
|
|
.def("get_states", &storm::models::sparse::StateLabeling::getStates, "Get all states which have the given label", py::arg("label"))
|
|
.def("set_states", [](storm::models::sparse::StateLabeling& labeling, std::string const& label, storm::storage::BitVector const& states) {
|
|
labeling.setStates(label, states);
|
|
}, "Add a label to the given states", py::arg("label"), py::arg("states"))
|
|
.def("__str__", &streamToString<storm::models::sparse::StateLabeling>)
|
|
;
|
|
|
|
|
|
py::class_<storm::models::sparse::ChoiceLabeling>(m, "ChoiceLabeling", "Labeling for choices", labeling)
|
|
.def(py::init<uint_fast64_t>(), "choice_count"_a)
|
|
.def("get_labels_of_choice", &storm::models::sparse::ChoiceLabeling::getLabelsOfChoice, py::arg("choice"), "Get labels of the given choice")
|
|
.def("add_label_to_choice", &storm::models::sparse::ChoiceLabeling::addLabelToChoice, "Adds a label to a given choice", py::arg("label"), py::arg("state"))
|
|
.def("get_choices", &storm::models::sparse::ChoiceLabeling::getChoices, py::arg("label"), "Get all choices which have the given label")
|
|
.def("set_choices", [](storm::models::sparse::ChoiceLabeling& labeling, std::string const& label, storm::storage::BitVector const& choices) {
|
|
labeling.setChoices(label, choices);
|
|
}, "Add a label to a the given choices", py::arg("label"), py::arg("choices"))
|
|
.def("__str__", &streamToString<storm::models::sparse::ChoiceLabeling>)
|
|
;
|
|
}
|