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

implement to_mtl, from_mtl and fix ast lens to allow AP

Browse source
This commit is contained in:
Marcell Vazquez-Chanlatte 2016-12-23 21:40:24 -08:00
parent 078eb11b5c
commit c31874d138
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
stl/utils.py
View file

@ -48,7 +48,7 @@ def _ast_lens(phi, *, pred, focus=lens(), focus_lens):
psi = focus.get(state=phi) psi = focus.get(state=phi)
ret_lens = [focus.add_lens(l) for l in focus_lens(psi)] if pred(psi) else [] ret_lens = [focus.add_lens(l) for l in focus_lens(psi)] if pred(psi) else []
if isinstance(psi, LinEq): if isinstance(psi, (LinEq, stl.ast.AtomicPred)):
return ret_lens return ret_lens
child_lenses = list(_child_lens(psi, focus=focus)) child_lenses = list(_child_lens(psi, focus=focus))
@ -58,6 +58,7 @@ def _ast_lens(phi, *, pred, focus=lens(), focus_lens):
lineq_lens = fn.partial(ast_lens, pred=type_pred(LinEq)) lineq_lens = fn.partial(ast_lens, pred=type_pred(LinEq))
AP_lens = fn.partial(ast_lens, pred=type_pred(stl.ast.AtomicPred))
and_or_lens = fn.partial(ast_lens, pred=type_pred(And, Or)) and_or_lens = fn.partial(ast_lens, pred=type_pred(And, Or))
def terms_lens(phi:"STL", bind=True) -> lens: def terms_lens(phi:"STL", bind=True) -> lens:
@ -118,3 +119,8 @@ def to_mtl(phi):
lineq_map = {v:k for k,v in ap_map.items()} lineq_map = {v:k for k,v in ap_map.items()}
return focus.modify(lineq_map.get), ap_map return focus.modify(lineq_map.get), ap_map
def from_mtl(phi, ap_map):
focus = lineq_lens(phi)
return focus.modify(ap_map.get)