blob: 190e85fa69eb8f0dbe2b75fe02a935f548e03696 (
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
71
|
// SPDX-License-Identifier: Apache-2.0
package firrtl.annotations
import firrtl._
import ir.{DefInstance, DefModule}
/** Building block to represent a [[Target]] of a FIRRTL component */
sealed trait TargetToken {
def keyword: String
def value: Any
/** Returns whether this token is one of the type of tokens whose keyword is passed as an argument
* @param keywords
* @return
*/
def is(keywords: String*): Boolean = {
keywords.map { kw =>
require(
TargetToken.keyword2targettoken.keySet.contains(kw),
s"Keyword $kw must be in set ${TargetToken.keyword2targettoken.keys}"
)
val lastClass = this.getClass
lastClass == TargetToken.keyword2targettoken(kw)("0").getClass
}.reduce(_ || _)
}
}
/** Object containing all [[TargetToken]] subclasses */
case object TargetToken {
case class Instance(value: String) extends TargetToken { override def keyword: String = "inst" }
case class OfModule(value: String) extends TargetToken { override def keyword: String = "of" }
case class Ref(value: String) extends TargetToken { override def keyword: String = "ref" }
case class Index(value: Int) extends TargetToken { override def keyword: String = "[]" }
case class Field(value: String) extends TargetToken { override def keyword: String = "." }
case object Clock extends TargetToken { override def keyword: String = "clock"; val value = "" }
case object Init extends TargetToken { override def keyword: String = "init"; val value = "" }
case object Reset extends TargetToken { override def keyword: String = "reset"; val value = "" }
implicit class fromStringToTargetToken(s: String) {
def Instance: Instance = new TargetToken.Instance(s)
def OfModule: OfModule = new TargetToken.OfModule(s)
def Ref: Ref = new TargetToken.Ref(s)
def Field: Field = new TargetToken.Field(s)
}
implicit class fromIntToTargetToken(i: Int) {
def Index: Index = new TargetToken.Index(i)
}
implicit class fromDefModuleToTargetToken(m: DefModule) {
def OfModule: OfModule = new TargetToken.OfModule(m.name)
}
implicit class fromDefInstanceToTargetToken(i: DefInstance) {
def Instance: Instance = new TargetToken.Instance(i.name)
def OfModule: OfModule = new TargetToken.OfModule(i.module)
def toTokens: (Instance, OfModule) = (new TargetToken.Instance(i.name), new TargetToken.OfModule(i.module))
}
val keyword2targettoken = Map(
"inst" -> ((value: String) => Instance(value)),
"of" -> ((value: String) => OfModule(value)),
"ref" -> ((value: String) => Ref(value)),
"[]" -> ((value: String) => Index(value.toInt)),
"." -> ((value: String) => Field(value)),
"clock" -> ((value: String) => Clock),
"init" -> ((value: String) => Init),
"reset" -> ((value: String) => Reset)
)
}
|
