Namespace - only save suffix for temp names (#667)

This prevents collisions for one prefix (including temp) from
incrementing the suffix for other prefixes. Makes names more stable.

1 files changed, 12 insertions, 4 deletions

diff --git a/src/main/scala/firrtl/Namespace.scala b/src/main/scala/firrtl/Namespace.scala
index b3629211..cf8472a6 100644
--- a/src/main/scala/firrtl/Namespace.scala
+++ b/src/main/scala/firrtl/Namespace.scala
@@ -11,7 +11,7 @@ class Namespace private {
   private val tempNamePrefix: String = "_GEN"
   // Begin with a tempNamePrefix in namespace so we always have a number suffix
   private val namespace = mutable.HashSet[String](tempNamePrefix)
-  private var n = 0L
+  private var tempN = 0
 
   def tryName(value: String): Boolean = {
     val unused = !contains(value)
@@ -21,15 +21,23 @@ class Namespace private {
 
   def contains(value: String): Boolean = namespace.contains(value)
 
-  def newName(value: String): String = {
+  private def newNameIndex(value: String, idx: Int): (String, Int) = {
+    var n = idx
     var str = value
     while (!tryName(str)) {
       str = s"${value}_$n"
       n += 1
     }
-    str
+    (str, n)
+  }
+
+  def newName(value: String): String = newNameIndex(value, 0)._1
+
+  def newTemp: String = {
+    val (name, n) = newNameIndex(tempNamePrefix, tempN)
+    tempN = n
+    name
   }
-  def newTemp: String = newName(tempNamePrefix)
 }
 
 object Namespace {