From 6fc2b2887277d9aad3e8f38fbd8bcccbace8b8f1 Mon Sep 17 00:00:00 2001 From: Luke Kenneth Casson Leighton Date: Wed, 22 Jan 2020 13:42:13 +0000 Subject: [PATCH] add __xor__ and __add__ --- src/ieee754/part/partsig.py | 25 ++++++++++++++++++++++++- 1 file changed, 24 insertions(+), 1 deletion(-) 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 -- 2.30.2