Merge pull request #49 from freechipsproject/fix_Reference_ref

Add new parameters to Reference
freechipsproject · May 28, 2020 · 09c9075 · 09c9075
2 parents 7682b0e + 6f2b871
commit 09c9075
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/scala/dotvisualizer/transforms/MakeOneDiagram.scala b/src/main/scala/dotvisualizer/transforms/MakeOneDiagram.scala
@@ -237,7 +237,7 @@ class MakeOneDiagram extends Transform {
             muxNode.asRhs
           case WRef(name, _, _, _) =>
             resolveRef(getFirrtlName(name), expand(name))
-          case Reference(name, _) =>
+          case Reference(name, _, _, _) =>
             resolveRef(getFirrtlName(name), expand(name))
           case subfield: WSubField =>
             resolveRef(getFirrtlName(subfield.serialize), expand(subfield.serialize))
@@ -324,7 +324,7 @@ class MakeOneDiagram extends Transform {
       def getConnectInfo(expression: Expression): String = {
         val (fName, dotName) = expression match {
           case WRef(name, _, _, _) => (getFirrtlName(name), expand(name))
-          case Reference(name, _) => (getFirrtlName(name), expand(name))
+          case Reference(name, _, _, _) => (getFirrtlName(name), expand(name))
           case subfield: WSubField =>
             (getFirrtlName(subfield.serialize), expand(subfield.serialize))
           case subfield: SubField =>