From: whitequark <whitequark@whitequark.org>
Date: Tue, 9 Jul 2019 19:45:15 +0000 (+0000)
Subject: back.rtlil: add decodings to cases when switching on a signal.
X-Git-Tag: locally_working~80
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=37f363e338ef0bd95ca92e25384014dafeec3a8d;p=nmigen.git

back.rtlil: add decodings to cases when switching on a signal.

Fixes #134.
---

diff --git a/nmigen/back/rtlil.py b/nmigen/back/rtlil.py
index 1a83675..4f6592e 100644
--- a/nmigen/back/rtlil.py
+++ b/nmigen/back/rtlil.py
@@ -603,10 +603,10 @@ class _StatementCompiler(xfrm.StatementVisitor):
         self._has_rhs     = False
 
     @contextmanager
-    def case(self, switch, values, src=""):
+    def case(self, switch, values, attrs={}, src=""):
         try:
             old_case = self._case
-            with switch.case(*values, src=src) as self._case:
+            with switch.case(*values, attrs=attrs, src=src) as self._case:
                 yield
         finally:
             self._case = old_case
@@ -659,11 +659,18 @@ class _StatementCompiler(xfrm.StatementVisitor):
 
         with self._case.switch(test_sigspec, src=src(stmt.src_loc)) as switch:
             for values, stmts in stmt.cases.items():
+                case_attrs = {}
                 if values in stmt.case_src_locs:
-                    case_src = src(stmt.case_src_locs[values])
-                else:
-                    case_src = ""
-                with self.case(switch, values, src=case_src):
+                    case_attrs["src"] = src(stmt.case_src_locs[values])
+                if isinstance(stmt.test, ast.Signal) and stmt.test.decoder:
+                    decoded_values = []
+                    for value in values:
+                        if "-" in value:
+                            decoded_values.append("<multiple>")
+                        else:
+                            decoded_values.append(stmt.test.decoder(int(value, 2)))
+                    case_attrs["nmigen.decoding"] = "|".join(decoded_values)
+                with self.case(switch, values, attrs=case_attrs):
                     self.on_statements(stmts)
 
     def on_statement(self, stmt):