blob: b36292117e8539655a5debb02236347d8d96deb2 (
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
|
// 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 n = 0L
def tryName(value: String): Boolean = {
val unused = !contains(value)
if (unused) namespace += value
unused
}
def contains(value: String): Boolean = namespace.contains(value)
def newName(value: String): String = {
var str = value
while (!tryName(str)) {
str = s"${value}_$n"
n += 1
}
str
}
def newTemp: String = newName(tempNamePrefix)
}
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
}
}
|
