1 files changed, 50 insertions, 0 deletions
diff --git a/src/main/scala/firrtl/transforms/RenameModules.scala b/src/main/scala/firrtl/transforms/RenameModules.scala
new file mode 100644
index 00000000..af17dda9
--- /dev/null
+++ b/src/main/scala/firrtl/transforms/RenameModules.scala
@@ -0,0 +1,50 @@
+// See LICENSE for license details.
+
+package firrtl.transforms
+
+import firrtl.analyses.{InstanceGraph, ModuleNamespaceAnnotation}
+import firrtl.ir._
+import firrtl._
+
+import scala.collection.mutable
+
+/** Rename Modules
+  *
+  * using namespace created by [[analyses.GetNamespace]], create unique names for modules
+  */
+class RenameModules extends Transform {
+  def inputForm: LowForm.type = LowForm
+  def outputForm: LowForm.type = LowForm
+
+  def collectNameMapping(namespace: Namespace, moduleNameMap: mutable.HashMap[String, String])(mod: DefModule): Unit = {
+    val newName = namespace.newName(mod.name)
+    moduleNameMap.put(mod.name, newName)
+  }
+
+  def onStmt(moduleNameMap: mutable.HashMap[String, String])(stmt: Statement): Statement = stmt match {
+    case inst: WDefInstance if moduleNameMap.contains(inst.module) => inst.copy(module = moduleNameMap(inst.module))
+    case other => other.mapStmt(onStmt(moduleNameMap))
+  }
+
+  def execute(state: CircuitState): CircuitState = {
+    val namespace = state.annotations.collectFirst {
+      case m: ModuleNamespaceAnnotation => m
+    }.map(_.namespace)
+
+    if (namespace.isEmpty) {
+      logger.warn("Skipping Rename Modules")
+      state
+    } else {
+      val moduleOrder = new InstanceGraph(state.circuit).moduleOrder.reverse
+      val nameMappings = new mutable.HashMap[String, String]()
+      moduleOrder.foreach(collectNameMapping(namespace.get, nameMappings))
+
+      val modulesx = state.circuit.modules.map {
+        case mod: Module => mod.mapStmt(onStmt(nameMappings)).mapString(nameMappings)
+        case ext: ExtModule => ext
+      }
+
+      state.copy(circuit = state.circuit.copy(modules = modulesx, main = nameMappings(state.circuit.main)))
+    }
+  }
+}