Pickle quoted constants to ValueExpr

* Avoid const of unpickling
* Smaller bytecode

Author: nicolasstucki

compiler/src/dotty/tools/dotc/core/Definitions.scala

@@ -628,6 +628,12 @@ class Definitions {
     lazy val QuotedExpr_runR = QuotedExprClass.requiredMethodRef(nme.run)
     def QuotedExpr_run(implicit ctx: Context) = QuotedExpr_runR.symbol
 
+  lazy val QuotedExprsModule = ctx.requiredModule("scala.quoted.Exprs")
+  def QuotedExprsClass(implicit ctx: Context) = QuotedExprsModule.symbol.asClass
+
+    lazy val QuotedExprs_valueExprR = QuotedExprsModule.requiredMethod("valueExpr")
+    def QuotedExprs_valueExpr(implicit ctx: Context) = QuotedExprs_valueExprR.symbol
+
   lazy val QuotedTypeType = ctx.requiredClassRef("scala.quoted.Type")
   def QuotedTypeClass(implicit ctx: Context) = QuotedTypeType.symbol.asClass
 

compiler/src/dotty/tools/dotc/quoted/Toolbox.scala

@@ -4,6 +4,7 @@ import dotty.tools.dotc.ast.Trees._
 import dotty.tools.dotc.ast.tpd
 import dotty.tools.dotc.core.Constants._
 import dotty.tools.dotc.printing.RefinedPrinter
+import dotty.tools.dotc.transform.ReifyQuotes
 
 import scala.quoted.Expr
 import scala.runtime.BoxedUnit

compiler/src/dotty/tools/dotc/transform/ReifyQuotes.scala

@@ -12,6 +12,7 @@ import tasty.TreePickler.Hole
 import MegaPhase.MiniPhase
 import SymUtils._
 import NameKinds._
+import dotty.tools.dotc.ast.tpd.Tree
 import typer.Implicits.SearchFailureType
 
 import scala.collection.mutable
@@ -290,12 +291,25 @@ class ReifyQuotes extends MacroTransformWithImplicits {
       }
       else {
         val (body1, splices) = nested(isQuote = true).split(body)
+        pickledQuote(body1, splices, isType).withPos(quote.pos)
+      }
+    }
+
+    private def pickledQuote(body: Tree, splices: List[Tree], isType: Boolean)(implicit ctx: Context) = {
+      def pickleAsValue[T](value: T) =
+        ref(defn.QuotedExprs_valueExpr).appliedToType(body.tpe.widen).appliedTo(Literal(Constant(value)))
+      def pickleAsTasty() = {
         val meth =
-          if (isType) ref(defn.Unpickler_unpickleType).appliedToType(body1.tpe)
-          else ref(defn.Unpickler_unpickleExpr).appliedToType(body1.tpe.widen)
+          if (isType) ref(defn.Unpickler_unpickleType).appliedToType(body.tpe)
+          else ref(defn.Unpickler_unpickleExpr).appliedToType(body.tpe.widen)
         meth.appliedTo(
-            liftList(PickledQuotes.pickleQuote(body1).map(x => Literal(Constant(x))), defn.StringType),
-            liftList(splices, defn.AnyType)).withPos(quote.pos)
+          liftList(PickledQuotes.pickleQuote(body).map(x => Literal(Constant(x))), defn.StringType),
+          liftList(splices, defn.AnyType))
+      }
+      if (splices.nonEmpty) pickleAsTasty()
+      else ReifyQuotes.toValue(body) match {
+        case Some(value) => pickleAsValue(value)
+        case _ => pickleAsTasty()
       }
     }
 
@@ -473,4 +487,13 @@ class ReifyQuotes extends MacroTransformWithImplicits {
       }
     }
   }
-}
+}
+
+object ReifyQuotes {
+  def toValue(tree: Tree): Option[Any] = tree match {
+    case Literal(Constant(c)) => Some(c)
+    case Block(Nil, e) => toValue(e)
+    case Inlined(_, Nil, e) => toValue(e)
+    case _ => None
+  }
+}

library/src/scala/quoted/Expr.scala

@@ -35,6 +35,8 @@ object Exprs {
     override def toString: String = s"Expr($value)"
   }
 
+  def valueExpr[T](value: T): ValueExpr[T] = new ValueExpr[T](value)
+
   /** An Expr backed by a tree. Only the current compiler trees are allowed. */
   final class TreeExpr[Tree](val tree: Tree) extends quoted.Expr[Any] {
     override def toString: String = s"Expr(<raw>)"

tests/run/quote-compile-constants.check

@@ -0,0 +1,12 @@
+class scala.quoted.Exprs$ValueExpr
+class scala.quoted.Exprs$ValueExpr
+class scala.quoted.Exprs$ValueExpr
+class scala.quoted.Exprs$ValueExpr
+class scala.quoted.Exprs$ValueExpr
+class scala.quoted.Exprs$ValueExpr
+class scala.quoted.Exprs$ValueExpr
+class scala.quoted.Exprs$ValueExpr
+class scala.quoted.Exprs$ValueExpr
+class scala.quoted.Exprs$ValueExpr
+class scala.quoted.Exprs$ValueExpr
+class scala.quoted.Exprs$ValueExpr

tests/run/quote-compile-constants.scala

@@ -0,0 +1,18 @@
+import scala.quoted._
+
+object Test {
+  def main(args: Array[String]): Unit = {
+    println(('(true)).getClass)
+    println(('{ 'a' }).getClass)
+    println(('{ '\n' }).getClass)
+    println(('{ '"' }).getClass)
+    println(('{ '\'' }).getClass)
+    println(('{ '\\' }).getClass)
+    println(('(1)).getClass)
+    println(('( { { 2 } } )).getClass)
+    println(('(3L)).getClass)
+    println(('(4.0f)).getClass)
+    println(('(5.0d)).getClass)
+    println(('("xyz")).getClass)
+  }
+}