rename ast

stl.py

@@ -1,144 +0,0 @@
-# -*- coding: utf-8 -*-
-# TODO: create iso lens between sugar and non-sugar
-# TODO: supress + given a + (-b). i.e. want a - b
-
-from collections import namedtuple, deque
-from itertools import repeat
-from typing import Union
-from enum import Enum
-from sympy import Symbol
-
-from lenses import lens
-
-import funcy as fn
-
-VarKind = Enum("VarKind", ["x", "u", "w"])
-str_to_varkind = {"x": VarKind.x, "u": VarKind.u, "w": VarKind.w}
-dt_sym = Symbol('dt', positive=True)
-t_sym = Symbol('t', positive=True)
-
-class LinEq(namedtuple("LinEquality", ["terms", "op", "const"])):
-    def __repr__(self):
-        n = len(self.terms)
-        rep = "{}"
-        if n > 1:
-            rep += " + {}"*(n - 1)
-        rep += " {op} {c}"
-        return rep.format(*self.terms, op=self.op, c=self.const)
-
-    def children(self):
-        return []
-
-
-class Var(namedtuple("Var", ["coeff", "id", "time"])):
-    def __repr__(self):
-        time_str = "[{}]".format(self.time)
-        return "{c}*{i}{t}".format(c=self.coeff, i=self.id, t=time_str)
-
-
-class Interval(namedtuple('I', ['lower', 'upper'])):
-    def __repr__(self):
-        return "[{},{}]".format(self.lower, self.upper)
-
-    def children(self):
-        return [self.lower, self.upper]
-
-
-class NaryOpSTL(namedtuple('NaryOp', ['args'])):
-    OP = "?"
-    def __repr__(self):
-        n = len(self.args)
-        if n == 1:
-            return "{}".format(self.args[0])
-        elif self.args:
-            rep = "({})" + " {op} ({})"*(len(self.args) - 1)
-            return rep.format(*self.args, op=self.OP)
-        else:
-            return ""
-
-    def children(self):
-        return self.args
-
-
-class Or(NaryOpSTL):
-    OP = "∨"
-
-class And(NaryOpSTL):
-    OP = "∧"
-
-
-class ModalOp(namedtuple('ModalOp', ['interval', 'arg'])):
-    def children(self):
-        return [self.arg]
-
-
-class F(ModalOp):
-    def __repr__(self):
-        return "◇{}({})".format(self.interval, self.arg)
-
-
-class G(ModalOp):
-    def __repr__(self):
-        return "□{}({})".format(self.interval, self.arg)
-
-
-class Neg(namedtuple('Neg', ['arg'])):
-    def __repr__(self):
-        return "¬({})".format(self.arg)
-
-    def children(self):
-        return [self.arg]
-
-
-def walk(stl, bfs=False):
-    """Walks Ast. Defaults to DFS unless BFS flag is set."""
-    pop = deque.popleft if bfs else deque.pop
-    children = deque([stl])
-    while len(children) != 0:
-        node = pop(children)
-        yield node
-        children.extend(node.children())
-
-
-def tree(stl):
-    return {x:set(x.children()) for x in walk(stl) if x.children()}
-
-
-def lineq_lens(phi:"STL", bind=True) -> lens:
-    return ast_lens(phi, bind=bind, types={LinEq})
-
-
-def terms_lens(phi:"STL", bind=True) -> lens:
-    return lineq_lens(phi, bind).terms.each_()
-
-
-def and_or_lens(phi:"STL", bind=True) -> lens:
-    return ast_lens(phi, bind=bind, types={And, Or})
-
-
-def _child_lens(psi, focus):
-    if psi is None:
-        return
-    if isinstance(psi, NaryOpSTL):
-        for j, _ in enumerate(psi.args):
-            yield focus.args[j]
-    else:
-        yield focus.arg
-
-
-def ast_lens(phi:"STL", bind=True, *, types) -> lens:
-    tls = list(fn.flatten(_ast_lens(phi, types=types)))
-    tl = lens().tuple_(*tls).each_()
-    return tl.bind(phi) if bind else tl
-
-
-def _ast_lens(phi, *, types, focus=lens()):
-    psi = focus.get(state=phi)
-    ret_lens = [focus] if type(psi) in types else []
-
-    if isinstance(psi, LinEq):
-        return ret_lens
-
-    child_lenses = list(_child_lens(psi, focus=focus))
-    ret_lens += [_ast_lens(phi, types=types, focus=cl) for cl in child_lenses]
-    return ret_lens