Re-add -DwriteVcd=1 feature (#110)

ucb-bar · Jan 15, 2020 · 2f1cbae · 2f1cbae
1 parent 2b6cae0
commit 2f1cbae
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/main/scala/chiseltest/ChiselScalatestTester.scala b/src/main/scala/chiseltest/ChiselScalatestTester.scala
@@ -8,6 +8,7 @@ import chisel3.MultiIOModule
 import firrtl.AnnotationSeq
 import org.scalatest._
 import org.scalatest.exceptions.TestFailedException
+import internal.WriteVcdAnnotation
 
 import scala.util.DynamicVariable
 
@@ -18,7 +19,10 @@ trait ChiselScalatestTester extends Assertions with TestSuiteMixin with TestEnvI
     }
 
     def apply(testFn: T => Unit): Unit = {
-      val newAnnos = addDefaultTargetDir(getTestName, annotationSeq)
+      var newAnnos = addDefaultTargetDir(getTestName, annotationSeq)
+      if (scalaTestContext.value.get.configMap.contains("writeVcd")) {
+        newAnnos = newAnnos ++ Seq(WriteVcdAnnotation)
+      }
       runTest(defaults.createDefaultTester(dutGen, newAnnos))(testFn)
     }
     // TODO: in the future, allow reset and re-use of a compiled design to avoid recompilation cost per test