trying a different way of specifying conditional defaults

Author: mdko

pyrtl/conditional.py

@@ -67,6 +67,13 @@ def currently_under_condition():
 # instances (hopefully the only and unchanging instances) of the following two types.
 
 class _ConditionalAssignment(object):
+    def __init__(self):
+        self.defaults = {}
+
+    def __call__(self, defaults):
+        self.defaults = defaults
+        return self
+
     """ Context providing funcitionality of "conditional_assignment". """
     def __enter__(self):
         global _depth
@@ -75,7 +82,7 @@ def __enter__(self):
 
     def __exit__(self, *exc_info):
         try:
-            _finalize()
+            _finalize(self.defaults)
         finally:
             # even if the above finalization throws an error we need to
             # reset the state to prevent errors from bleeding over
@@ -181,7 +188,7 @@ def _pred_sets_are_in_conflict(pred_set_a, pred_set_b):
     return True
 
 
-def _finalize():
+def _finalize(defaults):
     """Build the required muxes and call back to WireVector to finalize the wirevector build."""
     from .memory import MemBlock
     from pyrtl.corecircuits import select
@@ -203,13 +210,13 @@ def _finalize():
         # handle wirevector and register assignments
         else:
             if isinstance(lhs, Register):
-                if hasattr(lhs, 'condition_default'):
-                    result = lhs.condition_default
+                if lhs in defaults:
+                    result = defaults[lhs]
                 else:
                     result = lhs  # default for registers is "self"
             elif isinstance(lhs, WireVector):
-                if hasattr(lhs, 'condition_default'):
-                    result = lhs.condition_default
+                if lhs in defaults:
+                    result = defaults[lhs]
                 else:
                     result = 0  # default for wire is "0"
             else:

tests/test_conditional.py

@@ -121,6 +121,37 @@ def test_two_signals_under_default_condition(self):
                 r2.next |= 3
         self.check_trace(' i 01230123\nr1 01222344\nr2 00013334\n')
 
+    def test_default_value_for_wires(self):
+        i = pyrtl.Register(bitwidth=2, name='i')
+        i.next <<= i + 1
+        r1 = pyrtl.Register(bitwidth=3, name='r1')
+        r2 = pyrtl.Register(bitwidth=3, name='r2')
+        r3 = pyrtl.Register(bitwidth=3, name='r3')
+        o = pyrtl.Output(bitwidth=3, name='o')
+        with pyrtl.conditional_assignment(defaults={r1: r1 + 2, r2: 6, o: 3}):
+            with i < 2:
+                r1.next |= r1 + 1
+                # r2 will be updated to 6
+                # r3 remains the same as previous cycle
+                o |= i
+            with i < 3:
+                # r1 will be updated to r1 + 2
+                r2.next |= r2 + 1
+                # r3 remains the same as previous cycle
+                # o will be updated to 3
+            with pyrtl.otherwise:
+                # r1 will be updated to r1 + 2
+                # r2 will be updated to 6
+                r3.next |= 2
+                o |= 7
+        self.check_trace(
+            ' i 01230123\n'
+            ' o 01370137\n'
+            'r1 01246702\n'
+            'r2 06676667\n'
+            'r3 00002222\n'
+        )
+
     def test_error_on_unconditioned_update_in_under_conditional(self):
         i = pyrtl.Register(bitwidth=2, name='i')
         with pyrtl.conditional_assignment: