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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
package Chisel
import scala.util.DynamicVariable
import scala.collection.mutable.{ArrayBuffer, HashMap}
private class Namespace(parent: Option[Namespace], kws: Option[Set[String]]) {
private var i = 0L
private val names = collection.mutable.HashSet[String]()
private val keywords = kws.getOrElse(Set())
private def rename(n: String) = { i += 1; s"${n}_${i}" }
def contains(elem: String): Boolean = {
keywords.contains(elem) || names.contains(elem) ||
parent.map(_ contains elem).getOrElse(false)
}
def name(elem: String): String = {
val res = if(this contains elem) rename(elem) else elem
names += res
res
}
def child(kws: Option[Set[String]]): Namespace = new Namespace(Some(this), kws)
def child: Namespace = new Namespace(Some(this), None)
}
private class FIRRTLNamespace extends Namespace(None, Some(Set("mem", "node", "wire", "reg", "inst")))
private class IdGen {
private var counter = -1L
def next: Long = {
counter += 1
counter
}
}
private[Chisel] trait HasId {
private[Chisel] val _id = Builder.idGen.next
private[Chisel] def setRef() = Builder.globalRefMap.setRef(this, s"T_${_id}")
private[Chisel] def setRef(imm: Immediate) = Builder.globalRefMap.setRef(this, imm)
private[Chisel] def setRef(name: String) = Builder.globalRefMap.setRef(this, name)
private[Chisel] def setRef(parent: HasId, name: String) = Builder.globalRefMap.setField(parent, this, name)
private[Chisel] def setRef(parent: HasId, index: Int) = Builder.globalRefMap.setIndex(parent, this, index)
}
class RefMap {
private val _refmap = new HashMap[Long,Immediate]()
private[Chisel] def setRef(id: HasId, ref: Immediate): Unit =
_refmap(id._id) = ref
private[Chisel] def setRef(id: HasId, name: String): Unit =
if (!_refmap.contains(id._id)) setRef(id, Ref(name))
private[Chisel] def setField(parentid: HasId, id: HasId, name: String): Unit =
_refmap(id._id) = Slot(Alias(parentid), name)
private[Chisel] def setIndex(parentid: HasId, id: HasId, index: Int): Unit =
_refmap(id._id) = Index(Alias(parentid), index)
def apply(id: HasId): Immediate = _refmap(id._id)
}
private class DynamicContext {
val idGen = new IdGen
val globalNamespace = new FIRRTLNamespace
val globalRefMap = new RefMap
val components = ArrayBuffer[Component]()
var currentModule: Option[Module] = None
val parameterDump = new ParameterDump
}
private object Builder {
// All global mutable state must be referenced via dynamicContextVar!!
private val dynamicContextVar = new DynamicVariable[Option[DynamicContext]](None)
private val currentParamsVar = new DynamicVariable[Parameters](Parameters.empty)
def dynamicContext = dynamicContextVar.value.get
def idGen = dynamicContext.idGen
def globalNamespace = dynamicContext.globalNamespace
def globalRefMap = dynamicContext.globalRefMap
def components = dynamicContext.components
def parameterDump = dynamicContext.parameterDump
def pushCommand(c: Command) {
dynamicContext.currentModule.foreach(_._commands += c)
}
def pushOp[T <: Data](cmd: DefPrim[T]) = {
pushCommand(cmd)
cmd.id
}
def getParams: Parameters = currentParamsVar.value
def paramsScope[T](p: Parameters)(body: => T): T = {
currentParamsVar.withValue(p)(body)
}
def build[T <: Module](f: => T): Circuit = {
dynamicContextVar.withValue(Some(new DynamicContext)) {
val mod = f
mod.setRef(globalNamespace.name(mod.name))
Circuit(components.last.name, components, globalRefMap, parameterDump)
}
}
}
|
