From 59eb7c24ee8896aa832a28ee86412a6db5f3ca82 Mon Sep 17 00:00:00 2001
From: Jacob Chang <jchang@.sifive.com>
Date: Thu, 12 Jan 2017 15:20:32 -0800
Subject: [PATCH] Add iterator function to LazyModule to iterate over all nodes

---
 src/main/scala/diplomacy/LazyModule.scala | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/main/scala/diplomacy/LazyModule.scala b/src/main/scala/diplomacy/LazyModule.scala
index d97bbf7f..5633523e 100644
--- a/src/main/scala/diplomacy/LazyModule.scala
+++ b/src/main/scala/diplomacy/LazyModule.scala
@@ -85,6 +85,11 @@ abstract class LazyModule()(implicit val p: Parameters)
     } }
     children.filter(!_.omitGraphML).foreach { c => c.edgesGraphML(buf, pad) }
   }
+  
+  def nodeIterator(iterfunc: (LazyModule) => Unit): Unit = {
+    iterfunc(this)
+    children.foreach( _.nodeIterator(iterfunc) )
+  }
 }
 
 object LazyModule