add __iter__ to several classes, add global shape() function use in FIFOControl

[ieee754fpu.git] / src / add / fpcommon / pack.py

diff --git a/src/add/fpcommon/pack.py b/src/add/fpcommon/pack.py
index 512be00008d3872968cf22c5983c07955dc16224..0f75d46ce2f29eb62b90fce7256293978d16dea8 100644 (file)
--- a/src/add/fpcommon/pack.py
+++ b/src/add/fpcommon/pack.py
@@ -16,11 +16,15 @@ class FPPackData:
         self.z = Signal(width, reset_less=True)
         self.mid = Signal(id_wid, reset_less=True)
 
+    def __iter__(self):
+        yield self.z
+        yield self.mid
+
     def eq(self, i):
         return [self.z.eq(i.z), self.mid.eq(i.mid)]
 
     def ports(self):
-        return [self.z, self.mid]
+        return list(self)
 
 
 class FPPackMod: