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
|
package Chisel
private class Emitter(circuit: Circuit) {
override def toString = res.toString
def join(parts: Seq[String], sep: String): StringBuilder =
parts.tail.foldLeft(new StringBuilder(parts.head))((s, p) => s ++= sep ++= p)
def emitDir(e: Port, isTop: Boolean): String =
if (isTop) (if (e.id.isFlip) "input " else "output ")
else (if (e.id.isFlip) "flip " else "")
def emitPort(e: Port, isTop: Boolean): String =
s"${emitDir(e, isTop)}${circuit.refMap(e.id).name} : ${emitType(e.kind)}"
private def emitType(e: Kind): String = e match {
case e: UnknownType => "?"
case e: UIntType => s"UInt<${e.width}>"
case e: SIntType => s"SInt<${e.width}>"
case e: BundleType => s"{${join(e.ports.map(x => emitPort(x, false)), ", ")}}"
case e: VectorType => s"${emitType(e.kind)}[${e.size}]"
case e: ClockType => s"Clock"
}
private def emit(e: Command, ctx: Component): String = e match {
case e: DefPrim[_] => s"node ${e.name} = ${e.op.name}(${join(e.args.map(x => x.fullName(ctx)), ", ")})"
case e: DefWire => s"wire ${e.name} : ${emitType(e.kind)}"
case e: DefRegister => s"reg ${e.name} : ${emitType(e.kind)}, ${e.clock.fullName(ctx)}, ${e.reset.fullName(ctx)}"
case e: DefMemory => s"cmem ${e.name} : ${emitType(e.kind)}[${e.size}], ${e.clock.fullName(ctx)}";
case e: DefSeqMemory => s"smem ${e.name} : ${emitType(e.kind)}[${e.size}]";
case e: DefAccessor => s"infer accessor ${e.name} = ${e.source.fullName(ctx)}[${e.index.fullName(ctx)}]"
case e: Connect => s"${e.loc.fullName(ctx)} := ${e.exp.fullName(ctx)}"
case e: BulkConnect => s"${e.loc1.fullName(ctx)} <> ${e.loc2.fullName(ctx)}"
case e: ConnectInit => s"onreset ${e.loc.fullName(ctx)} := ${e.exp.fullName(ctx)}"
case e: DefInstance => {
val modName = moduleMap.getOrElse(e.id.name, e.id.name)
val res = new StringBuilder(s"inst ${e.name} of $modName")
res ++= newline
for (p <- e.ports; x <- initPort(p, INPUT, ctx))
res ++= newline + x
res.toString
}
case w: WhenBegin =>
indent()
s"when ${w.pred.fullName(ctx)} :"
case _: WhenElse =>
indent()
"else :"
case _: WhenEnd =>
unindent()
"skip"
}
private def initPort(p: Port, dir: Direction, ctx: Component) = {
for (x <- p.id.flatten; if x.dir == dir)
yield s"${circuit.refMap(x).fullName(ctx)} := ${x.makeLit(0).name}"
}
private def emitBody(m: Component) = {
val me = new StringBuilder
withIndent {
for (p <- m.ports)
me ++= newline + emitPort(p, true)
me ++= newline
for (p <- m.ports; x <- initPort(p, OUTPUT, m))
me ++= newline + x
me ++= newline
for (cmd <- m.commands)
me ++= newline + emit(cmd, m)
me ++= newline
}
me
}
private val bodyMap = collection.mutable.HashMap[StringBuilder, String]()
private val moduleMap = collection.mutable.HashMap[String, String]()
private def emit(m: Component): String = {
val body = emitBody(m)
bodyMap get body match {
case Some(name) =>
moduleMap(m.name) = name
""
case None =>
bodyMap(body) = m.name
newline + s"module ${m.name} : " + body
}
}
private var indentLevel = 0
private def newline = "\n" + (" " * indentLevel)
private def indent(): Unit = indentLevel += 1
private def unindent() { require(indentLevel > 0); indentLevel -= 1 }
private def withIndent(f: => Unit) { indent(); f; unindent() }
private val res = new StringBuilder(s"circuit ${circuit.name} : ")
withIndent { circuit.components.foreach(c => res ++= emit(c)) }
res ++= newline
}
|
