Simplify Module internal data structures

1 files changed, 4 insertions, 0 deletions

diff --git a/src/main/scala/Chisel/Builder.scala b/src/main/scala/Chisel/Builder.scala
index 66d2df84..68d996ad 100644
--- a/src/main/scala/Chisel/Builder.scala
+++ b/src/main/scala/Chisel/Builder.scala
@@ -35,6 +35,10 @@ private class IdGen {
 }
 
 private[Chisel] trait HasId {
+  private[Chisel] def _onModuleClose {}
+  private[Chisel] val _parent = Builder.dynamicContext.currentModule
+  _parent.foreach(_.addId(this))
+
   private[Chisel] val _refMap = Builder.globalRefMap
   private[Chisel] val _id = Builder.idGen.next
   private[Chisel] def setRef(imm: Immediate) = _refMap.setRef(this, imm)