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

added var_names property

Browse source
This commit is contained in:
Marcell Vazquez-Chanlatte 2017-11-26 20:01:54 -08:00
parent 034bc22955
commit 9098cff929
2 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
stl/ast.py
View file

@ -5,7 +5,7 @@ from functools import lru_cache
import funcy as fn
import lenses
from lenses import lens
from lenses import lens, bind
def flatten_binary(phi, op, dropT, shortT):
@ -77,6 +77,12 @@ class AST(object):
def atomic_predicates(self):
return set(AP_lens(self).Each().collect())
@property
def var_names(self):
symbols = set(bind(self.lineqs).Each().terms.Each().collect())
symbols |= self.atomic_predicates
return set(bind(symbols).Each().id.collect())
def inline_context(self, context):
phi, phi2 = self, None