add time index to atomic predicates

This commit is contained in:
Marcell Vazquez-Chanlatte 2017-01-12 15:28:00 -08:00
parent 652fdde4d4
commit 395c87cbe6
2 changed files with 3 additions and 3 deletions

View file

@ -30,9 +30,9 @@ class AST(object):
class AtomicPred(namedtuple("AP", ["id"]), AST):
class AtomicPred(namedtuple("AP", ["id", "time"]), AST):
def __repr__(self):
return "{}".format(self.id)
return f"{self.id}[{self.time}]"
def children(self):
return []

View file

@ -157,7 +157,7 @@ class STLVisitor(NodeVisitor):
return Number(1) if node.text == "+" else Number(-1)
def visit_AP(self, node, _):
return ast.AtomicPred(node.text)
return ast.AtomicPred(node.text, ast.t_sym)
def visit_neg(self, _, children):
return ast.Neg(children[1])