Move compiler and compiler tests to compiler dir

1 files changed, 34 insertions, 0 deletions

diff --git a/compiler/src/dotty/tools/dotc/transform/GetClass.scala b/compiler/src/dotty/tools/dotc/transform/GetClass.scala
new file mode 100644
index 000000000..6a9a5fda2
--- /dev/null
+++ b/compiler/src/dotty/tools/dotc/transform/GetClass.scala
@@ -0,0 +1,34 @@
+package dotty.tools.dotc
+package transform
+
+import ast.tpd
+import core.Contexts.Context
+import core.StdNames.nme
+import core.Phases.Phase
+import TypeUtils._
+import TreeTransforms.{MiniPhaseTransform, TransformerInfo}
+
+/** Rewrite `getClass` calls as follow:
+  *
+  *  For every instance of primitive class C whose boxed class is called B:
+  *    instanceC.getClass    -> B.TYPE
+  *  For every instance of non-primitive class D:
+  *    instanceD.getClass    -> instanceD.getClass
+  */
+class GetClass extends MiniPhaseTransform {
+  import tpd._
+
+  override def phaseName: String = "getClass"
+
+  // getClass transformation should be applied to specialized methods
+  override def runsAfter: Set[Class[_ <: Phase]] = Set(classOf[Erasure], classOf[FunctionalInterfaces])
+
+  override def transformApply(tree: Apply)(implicit ctx: Context, info: TransformerInfo): Tree = {
+    import ast.Trees._
+    tree match {
+      case Apply(Select(qual, nme.getClass_), Nil) if qual.tpe.widen.isPrimitiveValueType =>
+        clsOf(qual.tpe.widen).withPos(tree.pos)
+      case _ => tree
+    }
+  }
+}