forked from M-Labs/artiq
1
0
Fork 0

compiler/interleave: use eval_constant from tools

This commit is contained in:
Sebastien Bourdeauducq 2014-08-13 17:58:06 +08:00
parent 5481baac1d
commit 70fbc6ad56
1 changed files with 8 additions and 11 deletions

View File

@ -1,12 +1,12 @@
import ast, types import ast, types
from artiq.language import units from artiq.language import units
from artiq.compiler.tools import eval_ast from artiq.compiler.tools import *
# -1 statement duration could not be pre-determined # -1 statement duration could not be pre-determined
# 0 statement has no effect on timeline # 0 statement has no effect on timeline
# >0 statement is a static delay that advances the timeline # >0 statement is a static delay that advances the timeline
# by the given amount (in base_s_unit) # by the given amount (in seconds)
def _get_duration(stmt): def _get_duration(stmt):
if isinstance(stmt, (ast.Expr, ast.Assign)): if isinstance(stmt, (ast.Expr, ast.Assign)):
return _get_duration(stmt.value) return _get_duration(stmt.value)
@ -18,16 +18,13 @@ def _get_duration(stmt):
elif isinstance(stmt, ast.Call) and isinstance(stmt.func, ast.Name): elif isinstance(stmt, ast.Call) and isinstance(stmt.func, ast.Name):
name = stmt.func.id name = stmt.func.id
if name == "delay": if name == "delay":
da = stmt.args[0] try:
if isinstance(da, ast.Call) \ da = eval_constant(stmt.args[0])
and isinstance(da.func, ast.Name) \ except NotConstant:
and da.func.id == "Quantity" \
and isinstance(da.args[0], ast.Num):
if not isinstance(da.args[1], ast.Name) or da.args[1].id != "base_s_unit":
raise units.DimensionError("Delay not expressed in seconds")
return da.args[0].n
else:
return -1 return -1
if da.unit != units.s_unit:
raise units.DimensionError("Delay not expressed in seconds")
return da.amount
else: else:
return 0 return 0
else: else: