aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/firrtl/Mappers.scala
diff options
context:
space:
mode:
authorchick2020-08-14 19:47:53 -0700
committerJack Koenig2020-08-14 19:47:53 -0700
commit6fc742bfaf5ee508a34189400a1a7dbffe3f1cac (patch)
tree2ed103ee80b0fba613c88a66af854ae9952610ce /src/main/scala/firrtl/Mappers.scala
parentb516293f703c4de86397862fee1897aded2ae140 (diff)
All of src/ formatted with scalafmt
Diffstat (limited to 'src/main/scala/firrtl/Mappers.scala')
-rw-r--r--src/main/scala/firrtl/Mappers.scala45
1 files changed, 22 insertions, 23 deletions
diff --git a/src/main/scala/firrtl/Mappers.scala b/src/main/scala/firrtl/Mappers.scala
index 3bf89885..e9a698ae 100644
--- a/src/main/scala/firrtl/Mappers.scala
+++ b/src/main/scala/firrtl/Mappers.scala
@@ -12,36 +12,35 @@ object Mappers {
}
private object PortMagnet {
implicit def forType(f: Type => Type): PortMagnet = new PortMagnet {
- override def map(port: Port): Port = port mapType f
+ override def map(port: Port): Port = port.mapType(f)
}
implicit def forString(f: String => String): PortMagnet = new PortMagnet {
- override def map(port: Port): Port = port mapString f
+ override def map(port: Port): Port = port.mapString(f)
}
}
implicit class PortMap(val _port: Port) extends AnyVal {
def map[T](f: T => T)(implicit magnet: (T => T) => PortMagnet): Port = magnet(f).map(_port)
}
-
// ********** Stmt Mappers **********
private trait StmtMagnet {
def map(stmt: Statement): Statement
}
private object StmtMagnet {
implicit def forStmt(f: Statement => Statement): StmtMagnet = new StmtMagnet {
- override def map(stmt: Statement): Statement = stmt mapStmt f
+ override def map(stmt: Statement): Statement = stmt.mapStmt(f)
}
implicit def forExp(f: Expression => Expression): StmtMagnet = new StmtMagnet {
- override def map(stmt: Statement): Statement = stmt mapExpr f
+ override def map(stmt: Statement): Statement = stmt.mapExpr(f)
}
implicit def forType(f: Type => Type): StmtMagnet = new StmtMagnet {
- override def map(stmt: Statement) : Statement = stmt mapType f
+ override def map(stmt: Statement): Statement = stmt.mapType(f)
}
implicit def forString(f: String => String): StmtMagnet = new StmtMagnet {
- override def map(stmt: Statement): Statement = stmt mapString f
+ override def map(stmt: Statement): Statement = stmt.mapString(f)
}
implicit def forInfo(f: Info => Info): StmtMagnet = new StmtMagnet {
- override def map(stmt: Statement): Statement = stmt mapInfo f
+ override def map(stmt: Statement): Statement = stmt.mapInfo(f)
}
}
implicit class StmtMap(val _stmt: Statement) extends AnyVal {
@@ -55,13 +54,13 @@ object Mappers {
}
private object ExprMagnet {
implicit def forExpr(f: Expression => Expression): ExprMagnet = new ExprMagnet {
- override def map(expr: Expression): Expression = expr mapExpr f
+ override def map(expr: Expression): Expression = expr.mapExpr(f)
}
implicit def forType(f: Type => Type): ExprMagnet = new ExprMagnet {
- override def map(expr: Expression): Expression = expr mapType f
+ override def map(expr: Expression): Expression = expr.mapType(f)
}
implicit def forWidth(f: Width => Width): ExprMagnet = new ExprMagnet {
- override def map(expr: Expression): Expression = expr mapWidth f
+ override def map(expr: Expression): Expression = expr.mapWidth(f)
}
}
implicit class ExprMap(val _expr: Expression) extends AnyVal {
@@ -74,10 +73,10 @@ object Mappers {
}
private object TypeMagnet {
implicit def forType(f: Type => Type): TypeMagnet = new TypeMagnet {
- override def map(tpe: Type): Type = tpe mapType f
+ override def map(tpe: Type): Type = tpe.mapType(f)
}
implicit def forWidth(f: Width => Width): TypeMagnet = new TypeMagnet {
- override def map(tpe: Type): Type = tpe mapWidth f
+ override def map(tpe: Type): Type = tpe.mapWidth(f)
}
}
implicit class TypeMap(val _tpe: Type) extends AnyVal {
@@ -91,7 +90,7 @@ object Mappers {
private object WidthMagnet {
implicit def forWidth(f: Width => Width): WidthMagnet = new WidthMagnet {
override def map(width: Width): Width = width match {
- case mapable: HasMapWidth => mapable mapWidth f // WIR
+ case mapable: HasMapWidth => mapable.mapWidth(f) // WIR
case other => other // Standard IR nodes
}
}
@@ -106,21 +105,21 @@ object Mappers {
}
private object ModuleMagnet {
implicit def forStmt(f: Statement => Statement): ModuleMagnet = new ModuleMagnet {
- override def map(module: DefModule): DefModule = module mapStmt f
+ override def map(module: DefModule): DefModule = module.mapStmt(f)
}
implicit def forPorts(f: Port => Port): ModuleMagnet = new ModuleMagnet {
- override def map(module: DefModule): DefModule = module mapPort f
+ override def map(module: DefModule): DefModule = module.mapPort(f)
}
implicit def forString(f: String => String): ModuleMagnet = new ModuleMagnet {
- override def map(module: DefModule): DefModule = module mapString f
+ override def map(module: DefModule): DefModule = module.mapString(f)
}
implicit def forInfo(f: Info => Info): ModuleMagnet = new ModuleMagnet {
- override def map(module: DefModule): DefModule = module mapInfo f
+ override def map(module: DefModule): DefModule = module.mapInfo(f)
}
}
implicit class ModuleMap(val _module: DefModule) extends AnyVal {
def map[T](f: T => T)(implicit magnet: (T => T) => ModuleMagnet): DefModule = magnet(f).map(_module)
- }
+ }
// ********** Circuit Mappers **********
private trait CircuitMagnet {
@@ -128,16 +127,16 @@ object Mappers {
}
private object CircuitMagnet {
implicit def forModules(f: DefModule => DefModule): CircuitMagnet = new CircuitMagnet {
- override def map(circuit: Circuit): Circuit = circuit mapModule f
+ override def map(circuit: Circuit): Circuit = circuit.mapModule(f)
}
implicit def forString(f: String => String): CircuitMagnet = new CircuitMagnet {
- override def map(circuit: Circuit): Circuit = circuit mapString f
+ override def map(circuit: Circuit): Circuit = circuit.mapString(f)
}
implicit def forInfo(f: Info => Info): CircuitMagnet = new CircuitMagnet {
- override def map(circuit: Circuit): Circuit = circuit mapInfo f
+ override def map(circuit: Circuit): Circuit = circuit.mapInfo(f)
}
}
implicit class CircuitMap(val _circuit: Circuit) extends AnyVal {
def map[T](f: T => T)(implicit magnet: (T => T) => CircuitMagnet): Circuit = magnet(f).map(_circuit)
- }
+ }
}