From: Luke Kenneth Casson Leighton <lkcl@lkcl.net>
Date: Wed, 22 Jan 2020 13:42:13 +0000 (+0000)
Subject: add __xor__ and __add__
X-Git-Tag: ls180-24jan2020~350
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=6fc2b2887277d9aad3e8f38fbd8bcccbace8b8f1;p=ieee754fpu.git

add __xor__ and __add__
---

diff --git a/src/ieee754/part/partsig.py b/src/ieee754/part/partsig.py
index 252beb4d..47a7625c 100644
--- a/src/ieee754/part/partsig.py
+++ b/src/ieee754/part/partsig.py
@@ -10,6 +10,7 @@ http://bugs.libre-riscv.org/show_bug.cgi?id=132
 
 from nmigen import (Module, Signal, Elaboratable,
                     )
+from ieee754.part_mul_add.adder import PartitionedAdder
 
 class PartitionedSignal(Elaboratable):
     def __init__(self, partition_points, *args, **kwargs)
@@ -17,9 +18,31 @@ class PartitionedSignal(Elaboratable):
                  attrs=None, decoder=None, src_loc_at=0):
         self.partpoints = partition_points
         self.sig = Signal(*args, **kwargs)
+        self.modnames = {}
+        for name in ['add']:
+            self.modnames[name] = 0
 
     def elaboratable(self, platform):
         self.m = m = Module()
         return m
 
-    
+    def get_modname(self, category):
+        self.modnames[category] += 1
+        return "%s%d" % (category, self.modnames[category])
+
+    def __xor__(self, other):
+        if isinstance(other, PartitionedSignal):
+            return self.sig ^ other.sig
+        return self.sig ^ other
+
+    def __add__(self, other):
+        shape = self.sig.shape()
+        pa = PartitionedAdder(shape[0], self.partpoints)
+        setattr(self.m.submodules, self.get_modname('add'), pa)
+        comb = self.m.d.comb
+        comb += pa.a.eq(self.sig)
+        if isinstance(other, PartitionedSignal):
+            comb += pa.b.eq(other.sig)
+        else:
+            comb += pa.b.eq(other)
+        return pa.output