Move compiler and compiler tests to compiler dir

1 files changed, 30 insertions, 0 deletions

diff --git a/compiler/src/dotty/tools/dotc/transform/ClassOf.scala b/compiler/src/dotty/tools/dotc/transform/ClassOf.scala
new file mode 100644
index 000000000..e7b6977c7
--- /dev/null
+++ b/compiler/src/dotty/tools/dotc/transform/ClassOf.scala
@@ -0,0 +1,30 @@
+package dotty.tools.dotc
+package transform
+
+import ast.tpd
+import core.Constants.Constant
+import core.Contexts.Context
+import core.StdNames.nme
+import core.Symbols.{defn,TermSymbol}
+import core.TypeErasure
+import TreeTransforms.{MiniPhaseTransform, TransformerInfo, TreeTransform}
+
+/** Rewrite `classOf` calls as follow:
+ *
+ *  For every primitive class C whose boxed class is called B:
+ *    classOf[C]    -> B.TYPE
+ *  For every non-primitive class D:
+ *    classOf[D]    -> Literal(Constant(erasure(D)))
+ */
+class ClassOf extends MiniPhaseTransform {
+  import tpd._
+
+  override def phaseName: String = "classOf"
+
+  override def transformTypeApply(tree: TypeApply)(implicit ctx: Context, info: TransformerInfo): Tree =
+    if (tree.symbol eq defn.Predef_classOf) {
+      val targ = tree.args.head.tpe
+      clsOf(targ).ensureConforms(tree.tpe).withPos(tree.pos)
+    }
+    else tree
+}