core: add FSM support (and example)

enjoy-digital · Jul 20, 2018 · bfd06f8 · bfd06f8
1 parent 2ca58e4
commit bfd06f8
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/example_designs/targets/simple.py b/example_designs/targets/simple.py
@@ -59,6 +59,19 @@ def __init__(self, platform):
             self.cpu_or_bridge.wishbone
         ]
 
+        # fsm group
+        fsm = FSM(reset_state="STATE1")
+        self.submodules += fsm
+        fsm.act("STATE1",
+            NextState("STATE2")
+        )
+        fsm.act("STATE2",
+            NextState("STATE1")
+        )
+        analyzer_groups[2] = [
+            fsm
+        ]
+
         # analyzer
         self.submodules.analyzer = LiteScopeAnalyzer(analyzer_groups, 512)
 

diff --git a/litescope/core.py b/litescope/core.py
@@ -274,6 +274,10 @@ def format_groups(self, groups):
             for s in signals:
                 if isinstance(s, Record):
                     split_signals.extend(s.flatten())
+                elif isinstance(s, FSM):
+                    s.do_finalize()
+                    s.finalized = True
+                    split_signals.append(s.state)
                 else:
                     split_signals.append(s)
             new_groups[n] = split_signals