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
|
// SPDX-License-Identifier: Apache-2.0
package chiselTests.experimental
import chisel3._
import chisel3.util.Valid
import chisel3.stage.ChiselStage
import chisel3.experimental.DataMirror
import chiselTests.ChiselFlatSpec
object DataMirrorSpec {
import org.scalatest.matchers.should.Matchers._
class GrandChild(parent: RawModule) extends Module {
DataMirror.getParent(this) should be(Some(parent))
}
class Child(parent: RawModule) extends Module {
val inst = Module(new GrandChild(this))
DataMirror.getParent(inst) should be(Some(this))
DataMirror.getParent(this) should be(Some(parent))
}
class Parent extends Module {
val inst = Module(new Child(this))
DataMirror.getParent(inst) should be(Some(this))
DataMirror.getParent(this) should be(None)
}
}
class DataMirrorSpec extends ChiselFlatSpec {
import DataMirrorSpec._
behavior.of("DataMirror")
def assertBinding(x: Data, io: Boolean, wire: Boolean, reg: Boolean) = {
DataMirror.isIO(x) should be(io)
DataMirror.isWire(x) should be(wire)
DataMirror.isReg(x) should be(reg)
}
def assertIO(x: Data) = assertBinding(x, true, false, false)
def assertWire(x: Data) = assertBinding(x, false, true, false)
def assertReg(x: Data) = assertBinding(x, false, false, true)
def assertNone(x: Data) = assertBinding(x, false, false, false)
it should "validate bindings" in {
class MyModule extends Module {
val typ = UInt(4.W)
val vectyp = Vec(8, UInt(4.W))
val io = IO(new Bundle {
val in = Input(UInt(4.W))
val vec = Input(vectyp)
val out = Output(UInt(4.W))
})
val vec = Wire(vectyp)
val regvec = Reg(vectyp)
val wire = Wire(UInt(4.W))
val reg = RegNext(wire)
assertIO(io)
assertIO(io.in)
assertIO(io.out)
assertIO(io.vec(1))
assertIO(io.vec)
assertWire(vec)
assertWire(vec(0))
assertWire(wire)
assertReg(reg)
assertReg(regvec)
assertReg(regvec(2))
assertNone(typ)
assertNone(vectyp)
}
ChiselStage.elaborate(new MyModule)
}
it should "support getParent for normal modules" in {
ChiselStage.elaborate(new Parent)
}
it should "support getParent for normal modules even when used in a D/I context" in {
import chisel3.experimental.hierarchy._
class Top extends Module {
val defn = Definition(new Parent)
val inst = Instance(defn)
DataMirror.getParent(this) should be(None)
}
ChiselStage.elaborate(new Top)
}
}
|
