Hardware
/
litex
mirror of https://github.com/enjoy-digital/litex.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

pytholite/compiler: go to next state

This commit is contained in:
Sebastien Bourdeauducq 2012-11-09 20:12:15 +01:00
parent e1075a962c
commit c1b8492b61
1 changed files with 37 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
migen/pytholite/compiler.py
View File

@ -68,30 +68,51 @@ class _Compiler:
if isinstance(node, ast.Module) \ if isinstance(node, ast.Module) \
and len(node.body) == 1 \ and len(node.body) == 1 \
and isinstance(node.body[0], ast.FunctionDef): and isinstance(node.body[0], ast.FunctionDef):
return self.visit_block(node.body[0].body) states, exit_states = self.visit_block(node.body[0].body)
return states
else: else:
raise NotImplementedError raise NotImplementedError
# blocks and statements # blocks and statements
def visit_block(self, statements): def visit_block(self, statements):
states = [] states = []
exit_states = []
for statement in statements: for statement in statements:
if isinstance(statement, ast.Assign): n_states, n_exit_states = self.visit_statement(statement)
op = self.visit_assign(statement) if n_states:
if op: states += n_states
states.append(op) for exit_state in exit_states:
elif isinstance(statement, ast.If): exit_state.insert(0, _AbstractNextState(n_states[0]))
test = self.visit_expr(statement.test) exit_states = n_exit_states
states_t = self.visit_block(statement.body) return states, exit_states
states_f = self.visit_block(statement.orelse)
test_state_stmt = If(test, _AbstractNextState(states_t[0])) # entry state is first state returned
if states_f: def visit_statement(self, statement):
test_state_stmt.Else(_AbstractNextState(states_f[0])) states = []
states.append([test_state_stmt]) exit_states = []
states += states_t + states_f if isinstance(statement, ast.Assign):
op = self.visit_assign(statement)
if op:
states.append(op)
exit_states.append(op)
elif isinstance(statement, ast.If):
test = self.visit_expr(statement.test)
states_t, exit_states_t = self.visit_block(statement.body)
states_f, exit_states_f = self.visit_block(statement.orelse)
test_state_stmt = If(test, _AbstractNextState(states_t[0]))
test_state = [test_state_stmt]
if states_f:
test_state_stmt.Else(_AbstractNextState(states_f[0]))
else: else:
raise NotImplementedError exit_states.append(test_state)
return states
states.append(test_state)
states += states_t + states_f
exit_states += exit_states_t + exit_states_f
else:
raise NotImplementedError
return states, exit_states
def visit_assign(self, node): def visit_assign(self, node):
if isinstance(node.targets[0], ast.Name): if isinstance(node.targets[0], ast.Name):