fhdl/module: avoid flushing self._submodules and create do_exit.

m-labs · Mar 9, 2015 · ee1091f · sbourdeauducq · Mar 9, 2015 · ee1091f
1 parent efc5f22
commit ee1091f
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/migen/fhdl/module.py b/migen/fhdl/module.py
@@ -142,8 +142,10 @@ def __setattr__(self, name, value):
 			object.__setattr__(self, name, value)
 
 	def _collect_submodules(self):
-		r = [(name, submodule.get_fragment()) for name, submodule in self._submodules]
-		self._submodules = []
+		r = []
+		for name, submodule in self._submodules:
+			if not submodule._get_fragment_called:
+				r += [(name, submodule.get_fragment())]
 		return r
 
 	def finalize(self, *args, **kwargs):
@@ -176,3 +178,7 @@ def finalize(self, *args, **kwargs):
 
 	def do_finalize(self):
 		pass
+
+	def do_exit(self, *args, **kwargs):
+		for name, submodule in self._submodules:
+			submodule.do_exit(*args, **kwargs)