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

Use sympy for time variable

Browse source
This commit is contained in:
Marcell Vazquez-Chanlatte 2016-07-08 20:28:28 -07:00
parent 2d7e033df0
commit 7a73554525
2 changed files with 28 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

6
stl_parser.py
View file

@ -116,7 +116,7 @@ class STLVisitor(NodeVisitor):
(var_kind, iden), time_node = children
time_node = list(flatten(time_node))
time = time_node[0] if len(time_node) > 0 else 0
time = time_node[0] if len(time_node) > 0 else stl.t_sym
return stl.Var(var_kind, iden, time)
@ -124,13 +124,13 @@ class STLVisitor(NodeVisitor):
return children[3]* children[5]
def visit_prime(self, *_):
return -stl.dt
return -stl.dt_sym
def visit_const(self, const, children):
return float(const.text)
def visit_dt(self, *_):
return stl.dt
return stl.dt_sym
def visit_term(self, _, children):
coeffs, var = children