Add counter for depth of when scope

1 files changed, 5 insertions, 0 deletions

diff --git a/chiselFrontend/src/main/scala/chisel3/internal/Builder.scala b/chiselFrontend/src/main/scala/chisel3/internal/Builder.scala
index d8202da6..9e389788 100644
--- a/chiselFrontend/src/main/scala/chisel3/internal/Builder.scala
+++ b/chiselFrontend/src/main/scala/chisel3/internal/Builder.scala
@@ -150,6 +150,7 @@ private[chisel3] class DynamicContext() {
   // Set by object Module.apply before calling class Module constructor
   // Used to distinguish between no Module() wrapping, multiple wrappings, and rewrapping
   var readyForModuleConstr: Boolean = false
+  var whenDepth: Int = 0 // Depth of when nesting
   val errors = new ErrorLog
   val namingStack = new internal.naming.NamingStack
 }
@@ -181,6 +182,10 @@ private[chisel3] object Builder {
   def readyForModuleConstr_=(target: Boolean): Unit = {
     dynamicContext.readyForModuleConstr = target
   }
+  def whenDepth: Int = dynamicContext.whenDepth
+  def whenDepth_=(target: Int): Unit = {
+    dynamicContext.whenDepth = target
+  }
 
   // TODO(twigg): Ideally, binding checks and new bindings would all occur here
   // However, rest of frontend can't support this yet.