Added Circuit mappers

1 files changed, 2 insertions, 0 deletions

diff --git a/src/main/scala/firrtl/ir/IR.scala b/src/main/scala/firrtl/ir/IR.scala
index 938b5848..aafa17ca 100644
--- a/src/main/scala/firrtl/ir/IR.scala
+++ b/src/main/scala/firrtl/ir/IR.scala
@@ -496,4 +496,6 @@ case class Circuit(info: Info, modules: Seq[DefModule], main: String) extends Fi
   def serialize: String =
     s"circuit $main :" + info.serialize +
     (modules map ("\n" + _.serialize) map indent mkString "\n") + "\n"
+  def mapModule(f: DefModule => DefModule): Circuit = this.copy(modules = modules map f)
+  def mapString(f: String => String): Circuit = this.copy(main = f(main))
 }