fix some broken FieldSelectableInt handling
authorJacob Lifshay <programmerjake@gmail.com>
Tue, 9 May 2023 06:10:29 +0000 (23:10 -0700)
committerJacob Lifshay <programmerjake@gmail.com>
Tue, 9 May 2023 06:10:29 +0000 (23:10 -0700)
src/openpower/decoder/helpers.py
src/openpower/decoder/selectable_int.py

index c1ecd9d19287d43726c1ffcb42b2779f723fca86..9211e5ab34ed88ece16e7e772176f62d2ef49868 100644 (file)
@@ -2,7 +2,7 @@ import unittest
 import struct
 import sys
 from openpower.decoder.selectable_int import (SelectableInt, onebit,
-                                              selectconcat)
+                                              selectconcat, FieldSelectableInt)
 from nmutil.divmod import trunc_divs, trunc_rems
 from operator import floordiv, mod
 from openpower.decoder.selectable_int import selectltu as ltu
@@ -83,7 +83,7 @@ def copy_assign_rhs(inp):
     """
     if isinstance(inp, (str, int)):
         return inp
-    if isinstance(inp, SelectableInt):
+    if isinstance(inp, (SelectableInt, FieldSelectableInt)):
         return SelectableInt(inp)
     if isinstance(inp, tuple):
         return tuple(map(copy_assign_rhs, inp))
index 37592fd0d2804725a81b3b3fb0c3fa06654955be..ee50e08142ee6313c4331f5a8b3a2d427c67bc7d 100644 (file)
@@ -58,18 +58,10 @@ class FieldSelectableInt:
         self.br = br  # map of indices
 
     def eq(self, b):
-        if isinstance(b, int):
-            # convert integer to same SelectableInt of same bitlength as range
-            blen = len(self.br)
-            b = SelectableInt(b, blen)
-            for i in range(b.bits):
-                self[i] = b[i]
-        elif isinstance(b, SelectableInt):
-            for i in range(b.bits):
-                self[i] = b[i]
-        else:
-            self.si = copy(b.si)
-            self.br = copy(b.br)
+        if not isinstance(b, SelectableInt):
+            b = SelectableInt(b, len(self.br))
+        for i in range(b.bits):
+            self[i] = b[i]
 
     def _op(self, op, b):
         vi = self.get_range()
@@ -239,6 +231,8 @@ class SelectableInt:
     """
 
     def __init__(self, value, bits=None):
+        if isinstance(value, FieldSelectableInt):
+            value = value.get_range()
         if isinstance(value, SelectableInt):
             if bits is not None:
                 # check if the bitlength is different. TODO, allow override?
@@ -246,11 +240,6 @@ class SelectableInt:
                     raise ValueError(value)
             bits = value.bits
             value = value.value
-        elif isinstance(value, FieldSelectableInt):
-            if bits is not None:
-                raise ValueError(value)
-            bits = len(value.br)
-            value = value.si.value
         else:
             if not isinstance(value, int):
                 raise ValueError(value)
@@ -575,7 +564,7 @@ def selectconcat(*args, repeat=1):
     res = copy(args[0])
     for i in args[1:]:
         if isinstance(i, FieldSelectableInt):
-            i = i.si
+            i = i.get_range()
         assert isinstance(i, SelectableInt), "can only concat SIs, sorry"
         res.bits += i.bits
         res.value = (res.value << i.bits) | i.value