import unittest
from copy import copy
+def check_extsign(a, b):
+ if b.bits != 256:
+ return b
+ return SelectableInt(b.value, a.bits)
+
class SelectableInt:
def __init__(self, value, bits):
def __add__(self, b):
if isinstance(b, int):
b = SelectableInt(b, self.bits)
+ b = check_extsign(self, b)
assert b.bits == self.bits
return SelectableInt(self.value + b.value, self.bits)
def __sub__(self, b):
if isinstance(b, int):
b = SelectableInt(b, self.bits)
+ b = check_extsign(self, b)
assert b.bits == self.bits
return SelectableInt(self.value - b.value, self.bits)
def __mul__(self, b):
+ b = check_extsign(self, b)
assert b.bits == self.bits
return SelectableInt(self.value * b.value, self.bits)
def __div__(self, b):
+ b = check_extsign(self, b)
assert b.bits == self.bits
return SelectableInt(self.value / b.value, self.bits)
def __mod__(self, b):
+ b = check_extsign(self, b)
assert b.bits == self.bits
return SelectableInt(self.value % b.value, self.bits)
def __or__(self, b):
+ b = check_extsign(self, b)
assert b.bits == self.bits
return SelectableInt(self.value | b.value, self.bits)
def __and__(self, b):
+ b = check_extsign(self, b)
assert b.bits == self.bits
return SelectableInt(self.value & b.value, self.bits)
def __xor__(self, b):
+ b = check_extsign(self, b)
assert b.bits == self.bits
return SelectableInt(self.value ^ b.value, self.bits)
def __ge__(self, other):
if isinstance(other, SelectableInt):
+ other = check_extsign(self, other)
assert other.bits == self.bits
other = other.value
if isinstance(other, int):
def __le__(self, other):
if isinstance(other, SelectableInt):
+ other = check_extsign(self, other)
assert other.bits == self.bits
other = other.value
if isinstance(other, int):
def __gt__(self, other):
if isinstance(other, SelectableInt):
+ other = check_extsign(self, other)
assert other.bits == self.bits
other = other.value
if isinstance(other, int):
def __lt__(self, other):
if isinstance(other, SelectableInt):
+ other = check_extsign(self, other)
assert other.bits == self.bits
other = other.value
if isinstance(other, int):
def __eq__(self, other):
if isinstance(other, SelectableInt):
+ other = check_extsign(self, other)
assert other.bits == self.bits
other = other.value
if isinstance(other, int):
a[0:4] = a[4:8]
self.assertEqual(a, 0x199)
+ def test_concat(self):
+ a = SelectableInt(0x1, 1)
+ c = selectconcat(a, repeat=8)
+ self.assertEqual(c, 0xff)
+ self.assertEqual(c.bits, 8)
+ a = SelectableInt(0x0, 1)
+ c = selectconcat(a, repeat=8)
+ self.assertEqual(c, 0x00)
+ self.assertEqual(c.bits, 8)
+
+
if __name__ == "__main__":
unittest.main()