Merge pull request #304 from ucb-bar/gsdt-adddirmethod

Add Data dir method to Chisel compatibility layer.

1 files changed, 10 insertions, 0 deletions

diff --git a/src/main/scala/chisel3/compatibility.scala b/src/main/scala/chisel3/compatibility.scala
index aad15f60..d13fcb06 100644
--- a/src/main/scala/chisel3/compatibility.scala
+++ b/src/main/scala/chisel3/compatibility.scala
@@ -12,6 +12,16 @@ package object Chisel {     // scalastyle:ignore package.object.name
   object Flipped {
     def apply[T<:Data](target: T): T = chisel3.core.Flipped[T](target)
   }
+  // TODO: Possibly move the AddDirectionToData class here?
+  implicit class AddDirMethodToData[T<:Data](val target: T) extends AnyVal {
+    def dir: Direction = {
+      target match {
+        case e: Element => e.dir
+        case _ => chisel3.core.Direction.Unspecified
+      }
+    }
+  }
+
   type ChiselException = chisel3.internal.ChiselException
 
   type Data = chisel3.core.Data