blob: 745be1e26cf9c1e848697cdaf55fb5cf5d8975fe (
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
|
// SPDX-License-Identifier: Apache-2.0
package firrtl.passes
import firrtl._
import firrtl.ir._
import firrtl.Mappers._
import firrtl.backends.experimental.smt.random.DefRandom
import firrtl.traversals.Foreachers._
object ResolveKinds extends Pass {
override def prerequisites = firrtl.stage.Forms.MinimalHighForm
override def invalidates(a: Transform) = false
type KindMap = collection.mutable.HashMap[String, Kind]
private def find_port(kinds: KindMap)(p: Port): Unit = {
kinds(p.name) = PortKind
}
def resolve_expr(kinds: KindMap)(e: Expression): Expression = e match {
case ex: WRef => ex.copy(kind = kinds(ex.name))
case _ => e.map(resolve_expr(kinds))
}
def resolve_stmt(kinds: KindMap)(s: Statement): Statement = {
s match {
case sx: DefWire => kinds(sx.name) = WireKind
case sx: DefNode => kinds(sx.name) = NodeKind
case sx: DefRegister => kinds(sx.name) = RegKind
case sx: WDefInstance => kinds(sx.name) = InstanceKind
case sx: DefMemory => kinds(sx.name) = MemKind
case sx: DefRandom => kinds(sx.name) = RandomKind
case _ =>
}
s.map(resolve_stmt(kinds))
.map(resolve_expr(kinds))
}
def resolve_kinds(m: DefModule): DefModule = {
val kinds = new KindMap
m.foreach(find_port(kinds))
m.map(resolve_stmt(kinds))
}
def run(c: Circuit): Circuit =
c.copy(modules = c.modules.map(resolve_kinds))
}
|
