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

fsm.py: set reset_state

This commit is contained in:
Robert Jördens 2013-07-22 02:09:12 -06:00 committed by Sebastien Bourdeauducq
parent 0cef98373f
commit 5bc9a0b383
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
migen/genlib/fsm.py
View File

@ -30,10 +30,10 @@ class _LowerNextState(NodeTransformer):
return node return node
class FSM(Module): class FSM(Module):
def __init__(self): def __init__(self, reset_state=None):
self.actions = OrderedDict() self.actions = OrderedDict()
self.state_aliases = dict() self.state_aliases = dict()
self.reset_state = None self.reset_state = reset_state
def act(self, state, *statements): def act(self, state, *statements):
if self.finalized: if self.finalized:
@ -62,6 +62,8 @@ class FSM(Module):
self.encoding = dict((s, n) for n, s in enumerate(self.actions.keys())) self.encoding = dict((s, n) for n, s in enumerate(self.actions.keys()))
self.state = Signal(max=nstates) self.state = Signal(max=nstates)
if self.reset_state is not None:
self.state.reset = self.encoding[self.reset_state]
self.next_state = Signal(max=nstates) self.next_state = Signal(max=nstates)
lns = _LowerNextState(self.next_state, self.encoding, self.state_aliases) lns = _LowerNextState(self.next_state, self.encoding, self.state_aliases)