add true and floor div to SelectableInt
authorLuke Kenneth Casson Leighton <lkcl@lkcl.net>
Fri, 19 Jun 2020 09:54:14 +0000 (10:54 +0100)
committerLuke Kenneth Casson Leighton <lkcl@lkcl.net>
Fri, 19 Jun 2020 09:54:14 +0000 (10:54 +0100)
src/soc/decoder/selectable_int.py

index 716ad682af3185885154e93e9c9031767c599eb1..72640909fd3ceff79945eab5e9e38b242d8f52d4 100644 (file)
@@ -1,8 +1,8 @@
 import unittest
 from copy import copy
 from soc.decoder.power_fields import BitRange
-from operator import (add, sub, mul, truediv, mod, or_, and_, xor, neg, inv,
-                      lshift, rshift)
+from operator import (add, sub, mul, floordiv, truediv, mod, or_, and_, xor,
+                      neg, inv, lshift, rshift)
 
 
 def check_extsign(a, b):
@@ -169,7 +169,9 @@ class SelectableInt:
         return self._op(sub, b)
     def __mul__(self, b):
         return self._op(mul, b)
-    def __div__(self, b):
+    def __floordiv__(self, b):
+        return self._op(floordiv, b)
+    def __truediv__(self, b):
         return self._op(truediv, b)
     def __mod__(self, b):
         return self._op(mod, b)