34 lines
757 B
Python
34 lines
757 B
Python
import ast
|
|
|
|
from migen.genlib.fsm import FSM, NextState
|
|
|
|
def id_next_state(l):
|
|
return NextState(id(l))
|
|
|
|
# entry state is first state returned
|
|
class StateAssembler:
|
|
def __init__(self):
|
|
self.states = []
|
|
self.exit_states = []
|
|
|
|
def assemble(self, n_states, n_exit_states):
|
|
self.states += n_states
|
|
for exit_state in self.exit_states:
|
|
exit_state.insert(0, id_next_state(n_states[0]))
|
|
self.exit_states = n_exit_states
|
|
|
|
def ret(self):
|
|
return self.states, self.exit_states
|
|
|
|
def implement_fsm(states):
|
|
fsm = FSM()
|
|
for state in states:
|
|
fsm.act(id(state), state)
|
|
return fsm
|
|
|
|
def eval_ast(expr, symdict):
|
|
if not isinstance(expr, ast.Expression):
|
|
expr = ast.Expression(expr)
|
|
code = compile(expr, "<ast>", "eval")
|
|
return eval(code, symdict)
|