Sebastian Junges 7 years ago
parent
commit
b012d0143c
  1. 10
      src/core/environment.cpp
  2. 8
      src/core/environment.h
  3. 2
      src/mod_core.cpp
  4. 19
      src/pars/pla.cpp
  5. 6
      tests/core/test_environment.py
  6. 9
      tests/pars/test_pla.py

10
src/core/environment.cpp

@ -0,0 +1,10 @@
#include "environment.h"
#include "src/helpers.h"
#include "storm/environment/Environment.h"
void define_environment(py::module& m) {
py::class_<storm::Environment>(m, "Environment", "Environment")
.def(py::init<>(), "Construct default environment")
;
}

8
src/core/environment.h

@ -0,0 +1,8 @@
#ifndef PYTHON_CORE_ENVIRONMENT_H_
#define PYTHON_CORE_ENVIRONMENT_H_
#include "common.h"
void define_environment(py::module& m);
#endif /* PYTHON_CORE_ENVIRONMENT_H_ */

2
src/mod_core.cpp

@ -6,6 +6,7 @@
#include "core/bisimulation.h" #include "core/bisimulation.h"
#include "core/input.h" #include "core/input.h"
#include "core/analysis.h" #include "core/analysis.h"
#include "core/environment.h"
PYBIND11_MODULE(core, m) { PYBIND11_MODULE(core, m) {
m.doc() = "core"; m.doc() = "core";
@ -25,4 +26,5 @@ PYBIND11_MODULE(core, m) {
define_bisimulation(m); define_bisimulation(m);
define_input(m); define_input(m);
define_graph_constraints(m); define_graph_constraints(m);
define_environment(m);
} }

19
src/pars/pla.cpp

@ -2,21 +2,22 @@
#include "src/helpers.h" #include "src/helpers.h"
#include "storm/api/storm.h" #include "storm/api/storm.h"
typedef storm::modelchecker::SparseDtmcParameterLiftingModelChecker<storm::models::sparse::Dtmc<storm::RationalFunction>, double> SparseDtmcRegionChecker; typedef storm::modelchecker::SparseDtmcParameterLiftingModelChecker<storm::models::sparse::Dtmc<storm::RationalFunction>, double> SparseDtmcRegionChecker;
typedef storm::modelchecker::RegionModelChecker<storm::RationalFunction> RegionModelChecker; typedef storm::modelchecker::RegionModelChecker<storm::RationalFunction> RegionModelChecker;
typedef storm::storage::ParameterRegion<storm::RationalFunction> Region; typedef storm::storage::ParameterRegion<storm::RationalFunction> Region;
// Thin wrappers // Thin wrappers
std::shared_ptr<RegionModelChecker> createRegionChecker(std::shared_ptr<storm::models::sparse::Model<storm::RationalFunction>> const& model, std::shared_ptr<storm::logic::Formula> const& formula) {
return storm::api::initializeParameterLiftingRegionModelChecker<storm::RationalFunction, double>(model, storm::api::createTask<storm::RationalFunction>(formula, true));
std::shared_ptr<RegionModelChecker> createRegionChecker(storm::Environment const& env, std::shared_ptr<storm::models::sparse::Model<storm::RationalFunction>> const& model, std::shared_ptr<storm::logic::Formula> const& formula) {
return storm::api::initializeParameterLiftingRegionModelChecker<storm::RationalFunction, double>(env, model, storm::api::createTask<storm::RationalFunction>(formula, true));
} }
storm::modelchecker::RegionResult checkRegion(std::shared_ptr<RegionModelChecker>& checker, Region const& region, storm::modelchecker::RegionResultHypothesis const& hypothesis, storm::modelchecker::RegionResult const& initialResult, bool sampleVertices) {
return checker->analyzeRegion(region, hypothesis, initialResult, sampleVertices);
storm::modelchecker::RegionResult checkRegion(std::shared_ptr<RegionModelChecker>& checker, storm::Environment const& env, Region const& region, storm::modelchecker::RegionResultHypothesis const& hypothesis, storm::modelchecker::RegionResult const& initialResult, bool sampleVertices) {
return checker->analyzeRegion(env, region, hypothesis, initialResult, sampleVertices);
} }
storm::RationalFunction getBound(std::shared_ptr<RegionModelChecker>& checker, Region const& region, bool maximise) {
return checker->getBoundAtInitState(region, maximise ? storm::solver::OptimizationDirection::Maximize : storm::solver::OptimizationDirection::Minimize);
storm::RationalFunction getBound(std::shared_ptr<RegionModelChecker>& checker, storm::Environment const& env, Region const& region, bool maximise) {
return checker->getBoundAtInitState(env, region, maximise ? storm::solver::OptimizationDirection::Maximize : storm::solver::OptimizationDirection::Minimize);
} }
@ -74,11 +75,11 @@ void define_pla(py::module& m) {
auto tmp = storm::api::initializeParameterLiftingRegionModelChecker<storm::RationalFunction, double>(model, task); auto tmp = storm::api::initializeParameterLiftingRegionModelChecker<storm::RationalFunction, double>(model, task);
new (&instance) std::unique_ptr<SparseDtmcRegionChecker>(tmp); new (&instance) std::unique_ptr<SparseDtmcRegionChecker>(tmp);
}, py::arg("model"), py::arg("task")*/ }, py::arg("model"), py::arg("task")*/
.def("check_region", &checkRegion, "Check region", py::arg("region"), py::arg("hypothesis") = storm::modelchecker::RegionResultHypothesis::Unknown, py::arg("initialResult") = storm::modelchecker::RegionResult::Unknown, py::arg("sampleVertices") = false)
.def("get_bound", &getBound, "Get bound", py::arg("region"), py::arg("maximise")= true);
.def("check_region", &checkRegion, "Check region", py::arg("environment"), py::arg("region"), py::arg("hypothesis") = storm::modelchecker::RegionResultHypothesis::Unknown, py::arg("initialResult") = storm::modelchecker::RegionResult::Unknown, py::arg("sampleVertices") = false)
.def("get_bound", &getBound, "Get bound", py::arg("environment"), py::arg("region"), py::arg("maximise")= true);
; ;
m.def("create_region_checker", &createRegionChecker, "Create region checker", py::arg("model"), py::arg("formula"));
m.def("create_region_checker", &createRegionChecker, "Create region checker", py::arg("environment"), py::arg("model"), py::arg("formula"));
//m.def("is_parameter_lifting_sound", &storm::utility::parameterlifting::validateParameterLiftingSound, "Check if parameter lifting is sound", py::arg("model"), py::arg("formula")); //m.def("is_parameter_lifting_sound", &storm::utility::parameterlifting::validateParameterLiftingSound, "Check if parameter lifting is sound", py::arg("model"), py::arg("formula"));
m.def("gather_derivatives", &gatherDerivatives, "Gather all derivatives of transition probabilities", py::arg("model"), py::arg("var")); m.def("gather_derivatives", &gatherDerivatives, "Gather all derivatives of transition probabilities", py::arg("model"), py::arg("var"));
} }

6
tests/core/test_environment.py

@ -0,0 +1,6 @@
import stormpy
from helpers.helper import get_example_path
class TestEnvironment:
def test_environment(self):
env = stormpy.Environment()

9
tests/pars/test_pla.py

@ -16,16 +16,17 @@ class TestPLA:
assert model.nr_transitions == 803 assert model.nr_transitions == 803
assert model.model_type == stormpy.ModelType.DTMC assert model.model_type == stormpy.ModelType.DTMC
assert model.has_parameters assert model.has_parameters
checker = stormpy.pars.create_region_checker(model, formulas[0].raw_formula)
env = stormpy.Environment()
checker = stormpy.pars.create_region_checker(env, model, formulas[0].raw_formula)
parameters = model.collect_probability_parameters() parameters = model.collect_probability_parameters()
assert len(parameters) == 2 assert len(parameters) == 2
region = stormpy.pars.ParameterRegion("0.7<=pL<=0.9,0.75<=pK<=0.95", parameters) region = stormpy.pars.ParameterRegion("0.7<=pL<=0.9,0.75<=pK<=0.95", parameters)
result = checker.check_region(region)
result = checker.check_region(env, region)
assert result == stormpy.pars.RegionResult.ALLSAT assert result == stormpy.pars.RegionResult.ALLSAT
region = stormpy.pars.ParameterRegion("0.4<=pL<=0.65,0.75<=pK<=0.95", parameters) region = stormpy.pars.ParameterRegion("0.4<=pL<=0.65,0.75<=pK<=0.95", parameters)
result = checker.check_region(region, stormpy.pars.RegionResultHypothesis.UNKNOWN,
result = checker.check_region(env, region, stormpy.pars.RegionResultHypothesis.UNKNOWN,
stormpy.pars.RegionResult.UNKNOWN, True) stormpy.pars.RegionResult.UNKNOWN, True)
assert result == stormpy.pars.RegionResult.EXISTSBOTH assert result == stormpy.pars.RegionResult.EXISTSBOTH
region = stormpy.pars.ParameterRegion("0.1<=pL<=0.73,0.2<=pK<=0.715", parameters) region = stormpy.pars.ParameterRegion("0.1<=pL<=0.73,0.2<=pK<=0.715", parameters)
result = checker.check_region(region)
result = checker.check_region(env, region)
assert result == stormpy.pars.RegionResult.ALLVIOLATED assert result == stormpy.pars.RegionResult.ALLVIOLATED
Loading…
Cancel
Save