teaching/mtl-aas
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add helper function for going to an MTL formula

Browse source
This commit is contained in:
Marcell Vazquez-Chanlatte 2016-12-23 18:32:04 -08:00
parent 014110cf90
commit 078eb11b5c
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
stl/utils.py
View file

@ -4,6 +4,7 @@ from lenses import lens, Lens
import funcy as fn import funcy as fn
import sympy import sympy
import stl.ast
from stl.ast import LinEq, And, Or, NaryOpSTL, F, G, Interval, Neg from stl.ast import LinEq, And, Or, NaryOpSTL, F, G, Interval, Neg
def walk(stl, bfs=False): def walk(stl, bfs=False):
@ -108,3 +109,12 @@ def f_neg_or_canonical_form(phi):
return Neg(F(phi.interval, children[0])) return Neg(F(phi.interval, children[0]))
else: else:
raise NotImplementedError raise NotImplementedError
def to_mtl(phi):
focus = lineq_lens(phi)
to_ap = lambda i: stl.ast.AtomicPred("AP{}".format(i))
ap_map = {to_ap(i): leq for i, leq in enumerate(focus.get_all())}
lineq_map = {v:k for k,v in ap_map.items()}
return focus.modify(lineq_map.get), ap_map