Keep track of instances and update each instance on add_port()

magma/circuit.py

@@ -77,6 +77,9 @@ def __new__(metacls, name, bases, dct):
         else:
             dct["debug_info"] = None
 
+        # Keep track of instances of this Circuit.
+        dct["my_instances"] = set()
+
 
         # create a new circuit class
         cls = type.__new__(metacls, name, bases, dct)
@@ -185,6 +188,10 @@ def add_port(cls, name, typ):
         cls.IO.ports.update({name: typ})
         cls.interface.add_port(name, typ, defn=cls, add_to_decl=True)
         setattr(cls, name, cls.interface.ports[name])
+        for inst in cls.my_instances:
+            inst.IO.ports.update({name: typ})
+            inst.interface.add_port(name, typ, inst=inst, add_to_decl=False)
+            setattr(inst, name, inst.interface.ports[name])
 
 
 #
@@ -361,6 +368,10 @@ def __init__(self, *largs, **kwargs):
         if currentDefinition:
              currentDefinition.place(self)
 
+        defn = type(self)
+        assert self not in defn.my_instances
+        defn.my_instances.add(self)
+
     def __repr__(self):
         args = []
         for k, v in self.kwargs.items():