import unittest
import struct
from copy import copy
-from openpower.decoder.power_fields import BitRange
+import functools
+from collections import OrderedDict
from operator import (add, sub, mul, floordiv, truediv, mod, or_, and_, xor,
- neg, inv, lshift, rshift)
+ neg, inv, lshift, rshift, lt, eq)
from openpower.util import log
return SelectableInt(b.value, a.bits)
+class BitRange(OrderedDict):
+ """BitRange: remaps from straight indices (0,1,2..) to bit numbers
+ """
+
+ def __getitem__(self, subscript):
+ if isinstance(subscript, slice):
+ return list(self.values())[subscript]
+ else:
+ return OrderedDict.__getitem__(self, subscript)
+
+
+@functools.total_ordering
class FieldSelectableInt:
"""FieldSelectableInt: allows bit-range selection onto another target
"""
def __init__(self, si, br):
- self.si = si # target selectable int
+ if not isinstance(si, (FieldSelectableInt, SelectableInt)):
+ raise ValueError(si)
+
if isinstance(br, (list, tuple, range)):
_br = BitRange()
for i, v in enumerate(br):
_br[i] = v
br = _br
- self.br = br # map of indices.
+
+ if isinstance(si, FieldSelectableInt):
+ fsi = si
+ if len(br) > len(fsi.br):
+ raise OverflowError(br)
+ _br = BitRange()
+ for (i, v) in br.items():
+ _br[i] = fsi.br[v]
+ br = _br
+ si = fsi.si
+
+ self.si = si # target selectable int
+ self.br = br # map of indices
def eq(self, b):
if isinstance(b, int):
vi = op(vi)
return self.merge(vi)
+ def __len__(self):
+ return len(self.br)
+
def __getitem__(self, key):
log("getitem", key, self.br)
if isinstance(key, SelectableInt):
key = key.value
+
if isinstance(key, int):
key = self.br[key] # don't do POWER 1.3.4 bit-inversion
return self.si[key]
- if isinstance(key, slice):
+ elif isinstance(key, slice):
key = self.br[key]
return selectconcat(*[self.si[x] for x in key])
+ elif isinstance(key, (tuple, list, range)):
+ return FieldSelectableInt(si=self, br=key)
+ else:
+ raise ValueError(key)
def __setitem__(self, key, value):
if isinstance(key, SelectableInt):
self.si[k] = value[i]
def __negate__(self):
- return self._op1(negate)
+ return self._op1(neg)
def __invert__(self):
return self._op1(inv)
def __xor__(self, b):
return self._op(xor, b)
+ def __lt__(self, b):
+ vi = self.get_range()
+ return onebit(lt(vi, b))
+
+ def __eq__(self, b):
+ vi = self.get_range()
+ return onebit(eq(vi, b))
+
def get_range(self):
vi = SelectableInt(0, len(self.br))
for k, v in self.br.items():
def __repr__(self):
return f"{self.__class__.__name__}(si={self.si}, br={self.br})"
+ def __bool__(self):
+ for key in self.br.values():
+ bit = self.si[key].value
+ if bit:
+ return True
+ return False
+
+ def __int__(self):
+ return self.asint(msb0=True)
+
def asint(self, msb0=False):
res = 0
brlen = len(self.br)
for i, key in self.br.items():
+ log("asint", i, key, self.si[key])
bit = self.si[key].value
#log("asint", i, key, bit)
res |= bit << ((brlen-i-1) if msb0 else i)
def __init__(self, value, bits=None):
if isinstance(value, SelectableInt):
+ if bits is not None:
+ # check if the bitlength is different. TODO, allow override?
+ if bits != value.bits:
+ 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)
+ if bits is None:
+ raise ValueError(bits)
mask = (1 << bits) - 1
self.value = value & mask
self.bits = bits
return self
def __rsub__(self, b):
+ log("rsub", b, self.value)
if isinstance(b, int):
- b = SelectableInt(b, self.bits)
- b = check_extsign(self, b)
- assert b.bits == self.bits
- return SelectableInt(b.value - self.value, self.bits)
+ b = SelectableInt(b, 256) # max extent
+ #b = check_extsign(self, b)
+ #assert b.bits == self.bits
+ return SelectableInt(b.value - self.value, b.bits)
def __radd__(self, b):
if isinstance(b, int):
return SelectableInt(self.value >> b.value, self.bits)
def __getitem__(self, key):
+ log ("SelectableInt.__getitem__", self, key, type(key))
if isinstance(key, SelectableInt):
key = key.value
if isinstance(key, int):
start = self.bits - key.stop
bits = stop - start
- #log ("__getitem__ slice num bits", start, stop, bits)
+ log ("__getitem__ slice num bits", start, stop, bits)
mask = (1 << bits) - 1
value = (self.value >> start) & mask
log("getitem", stop, start, self.bits, hex(self.value), value)
return SelectableInt(value, bits)
+ else:
+ bits = []
+ key = tuple(key)
+ for bit in key:
+ if not isinstance(bit, (int, SelectableInt)):
+ raise ValueError(key)
+ bits.append(self[bit])
+ return selectconcat(*bits)
def __setitem__(self, key, value):
if isinstance(key, SelectableInt):
mask = 1 << key
self.value = (self.value & ~mask) | (value & mask)
elif isinstance(key, slice):
- assert key.step is None or key.step == 1
- assert key.start < key.stop
- assert key.start >= 0
- assert key.stop <= self.bits, \
- "key stop %d bits %d" % (key.stop, self.bits)
-
- stop = self.bits - key.start
- start = self.bits - key.stop
+ kstart, kstop, kstep = key.start, key.stop, key.step
+ if isinstance(kstart, SelectableInt): kstart = kstart.asint()
+ if isinstance(kstop, SelectableInt): kstop = kstop.asint()
+ if isinstance(kstep, SelectableInt): kstep = kstep.asint()
+ log ("__setitem__ slice ", kstart, kstop, kstep)
+ assert kstep is None or kstep == 1
+ assert kstart < kstop
+ assert kstart >= 0
+ assert kstop <= self.bits, \
+ "key stop %d bits %d" % (kstop, self.bits)
+
+ stop = self.bits - kstart
+ start = self.bits - kstop
bits = stop - start
#log ("__setitem__ slice num bits", bits)
mask = ((1 << bits) - 1) << start
value = value << start
self.value = (self.value & ~mask) | (value & mask)
+ else:
+ raise ValueError(key)
def __ge__(self, other):
if isinstance(other, FieldSelectableInt):
return self.value != 0
def __repr__(self):
- value = f"value=0x{self.value:x}, bits={self.bits}"
+ value = f"value={hex(self.value)}, bits={self.bits}"
return f"{self.__class__.__name__}({value})"
def __len__(self):
def asint(self):
return self.value
+ def __int__(self):
+ return self.asint()
+
def __float__(self):
"""convert to double-precision float. TODO, properly convert
rather than a hack-job: must actually support Power IEEE754 FP
return struct.unpack('<d', data)[0]
-class SelectableIntMapping(dict):
- def __init__(self, si, fields=None):
- def field(item):
- (key, value) = item
- if isinstance(value, dict):
- value = dict(map(field, value.items()))
- else:
- value = FieldSelectableInt(si, value)
- return (key, value)
-
- return super().__init__(map(field, fields.items()))
-
- def __getattr__(self, attr):
- try:
- return self[attr]
- except KeyError as error:
- raise AttributeError from error
-
-
def onebit(bit):
return SelectableInt(1 if bit else 0, 1)