blob: 9e1cae67e15d88fabb66095dfeba20cd7fac4061 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
// See LICENSE for license details.
package firrtl
import scala.collection.mutable
import scala.collection.mutable.HashSet
import firrtl.ir._
import Mappers._
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 tempN = 0
def tryName(value: String): Boolean = {
val unused = !contains(value)
if (unused) namespace += value
unused
}
def contains(value: String): Boolean = namespace.contains(value)
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, n)
}
def newName(value: String): String = newNameIndex(value, 0)._1
def newTemp: String = {
val (name, n) = newNameIndex(tempNamePrefix, tempN)
tempN = n
name
}
}
/* TODO(azidar): Make Namespace return unique names that will not conflict with expanded
* names after LowerTypes expands names (like the Uniquify pass).
*/
object Namespace {
// Initializes a namespace from a Module
def apply(m: DefModule): Namespace = {
val namespace = new Namespace
def buildNamespaceStmt(s: Statement): Seq[String] = s match {
case s: IsDeclaration => Seq(s.name)
case s: Conditionally => buildNamespaceStmt(s.conseq) ++ buildNamespaceStmt(s.alt)
case s: Block => s.stmts flatMap buildNamespaceStmt
case _ => Nil
}
namespace.namespace ++= m.ports map (_.name)
m match {
case in: Module =>
namespace.namespace ++= buildNamespaceStmt(in.body)
case _ => // Do nothing
}
namespace
}
/** Initializes a [[Namespace]] for [[ir.Module]] names in a [[ir.Circuit]] */
def apply(c: Circuit): Namespace = {
val namespace = new Namespace
namespace.namespace ++= c.modules map (_.name)
namespace
}
/** Initializes a [[Namespace]] from arbitrary strings **/
def apply(names: Seq[String] = Nil): Namespace = {
val namespace = new Namespace
namespace.namespace ++= names
namespace
}
}
|
