Add toExprOfTuple method for tuples

Author: anatoliykmetyuk

compiler/test/dotc/pos-test-pickling.blacklist

@@ -25,3 +25,6 @@ i7087.scala
 
 # Opaque type
 i5720.scala
+
+# Tuples
+toexproftuple.scala

library/src-bootstrapped/scala/quoted/package.scala

@@ -95,6 +95,12 @@ package object quoted {
           '{ Tuple.fromIArray(IArray(${seq.toExprOfSeq}: _*)) }
       }
     }
-  }
 
+    /** Given a tuple of the form `(Expr[A1], ..., Expr[An])`, outputs a tuple `Expr[(A1, ..., An)]`. */
+    def (tup: T) toExprOfTuple[T <: Tuple: Tuple.IsMappedBy[Expr]: Type] given (ctx: QuoteContext): Expr[Tuple.InverseMap[T, Expr]] = {
+      import ctx.tasty._
+      tup.asInstanceOf[Product].productIterator.toSeq.asInstanceOf[Seq[Expr[_]]].toExprOfTuple
+        .cast[Tuple.InverseMap[T, Expr]]
+    }
+  }
 }

library/src/scala/Tuple.scala

@@ -74,6 +74,23 @@ object Tuple {
     case h *: t => F[h] *: Map[t, F]
   }
 
+  /** Converts a tuple `(F[T1], ..., F[Tn])` to `(T1,  ... Tn)` */
+  type InverseMap[X <: Tuple, F[_]] <: Tuple = X match {
+    case F[x] *: t => x *: InverseMap[t, F]
+    case Unit => Unit
+  }
+
+  /** Implicit evidence. IsMappedBy[F][X] is present in the implicit scope iff
+   *  X is a tuple for which each element's type is constructed via `F`. E.g.
+   *  (F[A1], ..., F[An]), but not `(F[A1], B2, ..., F[An])` where A2 does not
+   *  have the shape of `F[A]`.
+   */
+  type IsMappedBy[F[_]] = [X <: Tuple] =>> X <:< IsMappedByImpl[X, F]
+  type IsMappedByImpl[X <: Tuple, F[_]] <: Tuple = X match {
+    case F[x] *: t => F[x] *: IsMappedByImpl[t, F]
+    case Unit => Unit
+  }
+
   /** Convert an array into a tuple of unknown arity and types */
   def fromArray[T](xs: Array[T]): Tuple = {
     val xs2 = xs match {

tests/neg/toexproftuple.scala

@@ -0,0 +1,11 @@
+import scala.quoted._, scala.deriving._
+import given scala.quoted._
+
+inline def mcr: Any = ${mcrImpl}
+def mcrImpl given (ctx: QuoteContext): Expr[Any] = {
+  val tpl: (Expr[1], Expr[2], Expr[3]) = ('{1}, '{2}, '{3})
+  '{val res: (1, 3, 3) = ${tpl.toExprOfTuple}; res}  // error
+
+  val tpl2: (Expr[1], 2, Expr[3]) = ('{1}, 2, '{3})
+  '{val res = ${tpl2.toExprOfTuple}; res}  // error
+}

tests/pos/toexproftuple.scala

@@ -0,0 +1,8 @@
+import scala.quoted._, scala.deriving._
+import given scala.quoted._
+
+inline def mcr: Any = ${mcrImpl}
+def mcrImpl given (ctx: QuoteContext): Expr[Any] = {
+  val tpl: (Expr[1], Expr[2], Expr[3]) = ('{1}, '{2}, '{3})
+  '{val res: (1, 2, 3) = ${tpl.toExprOfTuple}; res}
+}