From 5e13f1efdcea7a6ea664448f440ec9b19fd14127 Mon Sep 17 00:00:00 2001
From: Jacob Lifshay <programmerjake@gmail.com>
Date: Wed, 27 Oct 2021 21:06:15 -0700
Subject: [PATCH] add initial SimdShape.__add__

---
 src/ieee754/part/partsig.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/src/ieee754/part/partsig.py b/src/ieee754/part/partsig.py
index d12c4398..d85566dc 100644
--- a/src/ieee754/part/partsig.py
+++ b/src/ieee754/part/partsig.py
@@ -193,6 +193,25 @@ class SimdShape(Shape):
     def __rmul__(self, other):
         return self.__mul__(other)
 
+    def __add__(self, other):
+        if isinstance(other, int):
+            lane_shapes = {k: v + other for k, v in self.lane_shapes}
+            return SimdShape(self.scope, lane_shapes, signed=self.signed)
+        elif isinstance(other, SimdShape):
+            assert other.scope is self.scope, "scope mismatch"
+            o = other.lane_shapes
+            lane_shapes = {k: v + o[k] for k, v in self.lane_shapes}
+            # XXX not correct, we need a width-hint, not an overwrite
+            # lane_shapes argument...
+            return SimdShape(self.scope, lane_shapes, signed=self.signed,
+                             fixed_width=self.width + other.width)
+        else:
+            raise NotImplementedError(
+                f"Adding a SimdShape to {type(other)} isn't implemented")
+
+    def __radd__(self, other):
+        return self.__add__(other)
+
 
 class SimdSignal(UserValue):
     # XXX ################################################### XXX
-- 
2.30.2