sim: truncate case test value

m-labs · Oct 19, 2015 · ac5271e · ac5271e
1 parent ee28357
commit ac5271e
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/migen/sim/core.py b/migen/sim/core.py
@@ -5,6 +5,7 @@
 from migen.fhdl.structure import (_Value, _Statement,
                                   _Operator, _Slice, _ArrayProxy,
                                   _Assign, _Fragment)
+from migen.fhdl.bitcontainer import value_bits_sign
 from migen.fhdl.tools import list_signals, list_targets, insert_resets
 from migen.fhdl.simplify import MemoryToArray
 from migen.fhdl.specials import _MemoryLocation
@@ -75,6 +76,13 @@ def tick(self):
 }
 
 
+def _truncate(value, nbits, signed):
+    value = value & (2**nbits - 1)
+    if signed and (value & 2**(nbits - 1)):
+        value -= 2**nbits
+    return value
+
+
 class Evaluator:
     def __init__(self, clock_domains, replaced_memories):
         self.clock_domains = clock_domains
@@ -152,10 +160,8 @@ def eval(self, node, postcommit=False):
     def assign(self, node, value):
         if isinstance(node, Signal):
             assert not node.variable
-            value = value & (2**node.nbits - 1)
-            if node.signed and (value & 2**(node.nbits - 1)):
-                value -= 2**node.nbits
-            self.modifications[node] = value
+            self.modifications[node] = _truncate(value,
+                                                 node.nbits, node.signed)
         elif isinstance(node, Cat):
             for element in node.l:
                 nbits = len(element)
@@ -187,7 +193,8 @@ def execute(self, statements):
                 else:
                     self.execute(s.f)
             elif isinstance(s, Case):
-                test = self.eval(s.test)
+                nbits, signed = value_bits_sign(s.test)
+                test = _truncate(self.eval(s.test), nbits, signed)
                 for k, v in s.cases.items():
                     if isinstance(k, Constant) and k.value == test:
                         self.execute(v)