5 changed files with 7 additions and 67 deletions
-
5examples/shields/01_pre_shield_extraction.py
-
6examples/shields/07_pre_shield_simulator.py
-
53examples/shields/09_optimal_shield_simulator.py
-
9examples/shields/10_optimal_controller.py
-
1lib/stormpy/examples/files.py
@ -1,53 +0,0 @@ |
|||
import stormpy |
|||
import stormpy.core |
|||
import stormpy.simulator |
|||
|
|||
|
|||
import stormpy.shields |
|||
|
|||
import stormpy.examples |
|||
import stormpy.examples.files |
|||
|
|||
import random |
|||
|
|||
|
|||
def optimal_shield_simulator(): |
|||
path = stormpy.examples.files.prism_smg_lights |
|||
formula_str = "<optimal, Optimal> <<shield>> R{\"differenceWithInterferenceCost\"}min=? [ LRA ]" |
|||
|
|||
program = stormpy.parse_prism_program(path) |
|||
formulas = stormpy.parse_properties_for_prism_program(formula_str, program) |
|||
|
|||
options = stormpy.BuilderOptions([p.raw_formula for p in formulas]) |
|||
options.set_build_state_valuations(True) |
|||
options.set_build_choice_labels(True) |
|||
options.set_build_all_labels() |
|||
model = stormpy.build_sparse_model_with_options(program, options) |
|||
|
|||
result = stormpy.model_checking(model, formulas[0], extract_scheduler=True) |
|||
|
|||
assert result.has_scheduler |
|||
assert result.has_shield |
|||
|
|||
shield = result.shield |
|||
|
|||
scheduler = shield.construct() |
|||
simulator = stormpy.simulator.create_simulator(model)#, seed=42) |
|||
|
|||
print(simulator) |
|||
while not simulator.is_done(): |
|||
current_state = simulator.get_current_state() |
|||
state_string = model.state_valuations.get_string(current_state) |
|||
# print(F"Simulator is in state {state_string}.") |
|||
temp = scheduler.get_choice(current_state) |
|||
# print(F"Correction map is {temp.choice_map}") |
|||
# print([model.get_label_of_choice(current_state, x) for x in simulator.available_actions()]) |
|||
print(F"Available actions {simulator.available_actions()}") |
|||
for action in simulator.available_actions(): |
|||
print(F"Action: {action} ActionLabel: {model.get_label_of_choice(current_state, action)}") |
|||
observation, reward = simulator.step() |
|||
|
|||
|
|||
|
|||
if __name__ == '__main__': |
|||
optimal_shield_simulator() |
@ -1,9 +0,0 @@ |
|||
import stormpy |
|||
import stormpy.core |
|||
import stormpy.simulator |
|||
|
|||
|
|||
import stormpy.shields |
|||
|
|||
import stormpy.examples |
|||
import stormpy.examples.files |
Reference in new issue
xxxxxxxxxx