1 files changed, 13 insertions, 0 deletions
diff --git a/src/main/scala/firrtl/Mappers.scala b/src/main/scala/firrtl/Mappers.scala
index e8283d93..b30b4518 100644
--- a/src/main/scala/firrtl/Mappers.scala
+++ b/src/main/scala/firrtl/Mappers.scala
@@ -6,6 +6,19 @@ import firrtl.ir._
 
 // TODO: Implement remaining mappers and recursive mappers
 object Mappers {
+  // ********** Port Mappers **********
+  private trait PortMagnet {
+    def map(p: Port): Port
+  }
+  private object PortMagnet {
+    implicit def forType(f: Type => Type): PortMagnet = new PortMagnet {
+      override def map(port: Port): Port = port mapType 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 {