From e985d47312458459e9ebe42fe99b5a063c08e637 Mon Sep 17 00:00:00 2001
From: azidar
Date: Sun, 31 Jan 2016 12:59:31 -0800
Subject: Changed stanza output of UInt/SInt to include widths. Made tests
 match accordingly

---
 src/main/scala/firrtl/Visitor.scala | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'src/main/scala/firrtl')

diff --git a/src/main/scala/firrtl/Visitor.scala b/src/main/scala/firrtl/Visitor.scala
index e9bc633c..56d3bbe7 100644
--- a/src/main/scala/firrtl/Visitor.scala
+++ b/src/main/scala/firrtl/Visitor.scala
@@ -164,14 +164,20 @@ class Visitor(val fullFilename: String) extends FIRRTLBaseVisitor[AST]
           val (width, value) = 
             if (ctx.getChildCount > 4) 
               (IntWidth(string2BigInt(ctx.IntLit(0).getText)), string2BigInt(ctx.IntLit(1).getText))
-            else (UnknownWidth(), string2BigInt(ctx.IntLit(0).getText))
+            else {
+               val bigint = string2BigInt(ctx.IntLit(0).getText)
+               (IntWidth(BigInt(bigint.bitLength)),bigint)
+            }
           UIntValue(value, width)
         }
         case "SInt" => {
           val (width, value) = 
             if (ctx.getChildCount > 4) 
               (IntWidth(string2BigInt(ctx.IntLit(0).getText)), string2BigInt(ctx.IntLit(1).getText))
-            else (UnknownWidth(), string2BigInt(ctx.IntLit(0).getText))
+            else {
+               val bigint = string2BigInt(ctx.IntLit(0).getText)
+               (IntWidth(BigInt(bigint.bitLength + 1)),bigint)
+            }
           SIntValue(value, width)
         }
         case "validif(" => ValidIf(visitExp(ctx.exp(0)), visitExp(ctx.exp(1)), UnknownType())
-- 
cgit v1.2.3