From 6033b7549ca48ed5a6875910d8c597a31bd5ce33 Mon Sep 17 00:00:00 2001
From: Michael Nolan <mtnolan2640@gmail.com>
Date: Fri, 14 Feb 2020 09:22:59 -0500
Subject: [PATCH] Replace giant switch statement with mux chain

---
 src/ieee754/part_shift/part_shift_dynamic.py | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/src/ieee754/part_shift/part_shift_dynamic.py b/src/ieee754/part_shift/part_shift_dynamic.py
index a0dac157..41bf16e3 100644
--- a/src/ieee754/part_shift/part_shift_dynamic.py
+++ b/src/ieee754/part_shift/part_shift_dynamic.py
@@ -100,15 +100,11 @@ class PartitionedDynamicShift(Elaboratable):
             s, e = intervals[i]
             index = gates[:i]  # selects the 'i' least significant bits
                                # of gates
-            element = Signal(matrix[0][i].width, name="element%d" % i)
-            for index in range(1<<i):
-                print(index)
-                with m.Switch(gates[:i]):
-                    with m.Case(index):
-                        index = math.ceil(math.log2(index + 1))
-                        comb += element.eq(matrix[index][i])
+            element = matrix[0][i]
+            for index in range(i):
+                element = Mux(gates[index], matrix[index+1][i], element)
             print(keys[i-1])
-            temp = Signal(element.width, name="intermed%d" % i)
+            temp = Signal(matrix[0][i].width, name="intermed%d" % i)
             print(intermed[keys[0]:])
             # XXX bit of a mess here, but rather than select
             # element or (element | intermed), select between 0 or intermed
-- 
2.30.2