blob: fda1b2a793b3f7f5255adc52f2643698a814c70f (
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
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
106
107
108
109
110
111
112
113
|
// SPDX-License-Identifier: Apache-2.0
package firrtl.traversals
import firrtl.ir._
import language.implicitConversions
/** Enables FIRRTL IR nodes to use foreach to traverse children IR nodes
*/
object Foreachers {
/** Statement Foreachers */
trait StmtForMagnet {
def foreach(stmt: Statement): Unit
}
object StmtForMagnet {
implicit def forStmt(f: Statement => Unit): StmtForMagnet = new StmtForMagnet {
def foreach(stmt: Statement): Unit = stmt.foreachStmt(f)
}
implicit def forExp(f: Expression => Unit): StmtForMagnet = new StmtForMagnet {
def foreach(stmt: Statement): Unit = stmt.foreachExpr(f)
}
implicit def forType(f: Type => Unit): StmtForMagnet = new StmtForMagnet {
def foreach(stmt: Statement): Unit = stmt.foreachType(f)
}
implicit def forString(f: String => Unit): StmtForMagnet = new StmtForMagnet {
def foreach(stmt: Statement): Unit = stmt.foreachString(f)
}
implicit def forInfo(f: Info => Unit): StmtForMagnet = new StmtForMagnet {
def foreach(stmt: Statement): Unit = stmt.foreachInfo(f)
}
}
implicit class StmtForeach(val _stmt: Statement) extends AnyVal {
// Using implicit types to allow overloading of function type to foreach, see StmtForMagnet above
def foreach[T](f: T => Unit)(implicit magnet: (T => Unit) => StmtForMagnet): Unit = magnet(f).foreach(_stmt)
}
/** Expression Foreachers */
trait ExprForMagnet {
def foreach(expr: Expression): Unit
}
object ExprForMagnet {
implicit def forExpr(f: Expression => Unit): ExprForMagnet = new ExprForMagnet {
def foreach(expr: Expression): Unit = expr.foreachExpr(f)
}
implicit def forType(f: Type => Unit): ExprForMagnet = new ExprForMagnet {
def foreach(expr: Expression): Unit = expr.foreachType(f)
}
implicit def forWidth(f: Width => Unit): ExprForMagnet = new ExprForMagnet {
def foreach(expr: Expression): Unit = expr.foreachWidth(f)
}
}
implicit class ExprForeach(val _expr: Expression) extends AnyVal {
def foreach[T](f: T => Unit)(implicit magnet: (T => Unit) => ExprForMagnet): Unit = magnet(f).foreach(_expr)
}
/** Type Foreachers */
trait TypeForMagnet {
def foreach(tpe: Type): Unit
}
object TypeForMagnet {
implicit def forType(f: Type => Unit): TypeForMagnet = new TypeForMagnet {
def foreach(tpe: Type): Unit = tpe.foreachType(f)
}
implicit def forWidth(f: Width => Unit): TypeForMagnet = new TypeForMagnet {
def foreach(tpe: Type): Unit = tpe.foreachWidth(f)
}
}
implicit class TypeForeach(val _tpe: Type) extends AnyVal {
def foreach[T](f: T => Unit)(implicit magnet: (T => Unit) => TypeForMagnet): Unit = magnet(f).foreach(_tpe)
}
/** Module Foreachers */
trait ModuleForMagnet {
def foreach(module: DefModule): Unit
}
object ModuleForMagnet {
implicit def forStmt(f: Statement => Unit): ModuleForMagnet = new ModuleForMagnet {
def foreach(module: DefModule): Unit = module.foreachStmt(f)
}
implicit def forPorts(f: Port => Unit): ModuleForMagnet = new ModuleForMagnet {
def foreach(module: DefModule): Unit = module.foreachPort(f)
}
implicit def forString(f: String => Unit): ModuleForMagnet = new ModuleForMagnet {
def foreach(module: DefModule): Unit = module.foreachString(f)
}
implicit def forInfo(f: Info => Unit): ModuleForMagnet = new ModuleForMagnet {
def foreach(module: DefModule): Unit = module.foreachInfo(f)
}
}
implicit class ModuleForeach(val _module: DefModule) extends AnyVal {
def foreach[T](f: T => Unit)(implicit magnet: (T => Unit) => ModuleForMagnet): Unit = magnet(f).foreach(_module)
}
/** Circuit Foreachers */
trait CircuitForMagnet {
def foreach(module: Circuit): Unit
}
object CircuitForMagnet {
implicit def forModules(f: DefModule => Unit): CircuitForMagnet = new CircuitForMagnet {
def foreach(circuit: Circuit): Unit = circuit.foreachModule(f)
}
implicit def forString(f: String => Unit): CircuitForMagnet = new CircuitForMagnet {
def foreach(circuit: Circuit): Unit = circuit.foreachString(f)
}
implicit def forInfo(f: Info => Unit): CircuitForMagnet = new CircuitForMagnet {
def foreach(circuit: Circuit): Unit = circuit.foreachInfo(f)
}
}
implicit class CircuitForeach(val _circuit: Circuit) extends AnyVal {
def foreach[T](f: T => Unit)(implicit magnet: (T => Unit) => CircuitForMagnet): Unit = magnet(f).foreach(_circuit)
}
}
|
