From 661147d84d8c27a5b4f051ced12ebf7efecb40dc Mon Sep 17 00:00:00 2001
From: Jack Koenig
Date: Fri, 14 Jul 2017 15:44:39 -0700
Subject: Fix bug in DiGraph.reverse on an graph with one vertex, no edges
 (#628)

---
 src/main/scala/firrtl/graph/DiGraph.scala | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'src/main')

diff --git a/src/main/scala/firrtl/graph/DiGraph.scala b/src/main/scala/firrtl/graph/DiGraph.scala
index e28e53e5..ee00789e 100644
--- a/src/main/scala/firrtl/graph/DiGraph.scala
+++ b/src/main/scala/firrtl/graph/DiGraph.scala
@@ -295,7 +295,10 @@ class DiGraph[T] (val edges: Map[T, Set[T]]) extends DiGraphLike[T] {
   /** Returns a graph with all edges reversed */
   def reverse: DiGraph[T] = {
     val mdg = new MutableDiGraph[T]
-    edges foreach { case (u,edges) => edges.foreach({ v => mdg.addEdge(v,u) }) }
+    edges.foreach { case (u, edges) =>
+      mdg.addVertex(u)
+      edges.foreach(v => mdg.addEdge(v,u))
+    }
     DiGraph(mdg)
   }
 
-- 
cgit v1.2.3