Use staged macros to implement parts of Tuple

Author: nicolasstucki

library/src-scala3/scala/StagedTuple.scala

@@ -0,0 +1,84 @@
+package scala
+
+import scala.quoted._
+
+object StagedTuple {
+  import Tuple._
+
+  def toArrayStaged(tup: Expr[Tuple], size: Option[Int]): Expr[Array[Object]] = {
+    if (!specialize) '(dynamicToArray(~tup))
+    else size match {
+      case Some(0) =>
+        '($emptyArray)
+      case Some(1) =>
+        tup.as[Tuple1[Object]].bind(t => '(Array((~t)._1)))
+      case Some(2) =>
+        tup.as[Tuple2[Object, Object]].bind(t => '(Array((~t)._1, (~t)._2)))
+      case Some(3) =>
+        tup.as[Tuple3[Object, Object, Object]].bind(t => '(Array((~t)._1, (~t)._2, (~t)._3)))
+      case Some(4) =>
+        tup.as[Tuple4[Object, Object, Object, Object]].bind(t => '(Array((~t)._1, (~t)._2, (~t)._3, (~t)._4)))
+      case Some(n) if n <= $MaxSpecialized =>
+        '($toArray(~tup, ~n.toExpr))
+      case Some(n) =>
+        '((~tup.as[TupleXXL]).elems)
+      case None =>
+        '(dynamicToArray(~tup))
+    }
+  }
+
+  def fromArrayStaged[T <: Tuple : Type](xs: Expr[Array[Object]], size: Int): Expr[T] = {
+    if (!specialize) '(dynamicFromArray[T](~xs))
+    else {
+      val tup = size match {
+        case 0  => '()
+        case 1  => '(Tuple1( (~xs)(0)))
+        case 2  => '(Tuple2( (~xs)(0), (~xs)(1)))
+        case 3  => '(Tuple3( (~xs)(0), (~xs)(1), (~xs)(2)))
+        case 4  => '(Tuple4( (~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3)))
+        case 5  => '(Tuple5( (~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4)))
+        case 6  => '(Tuple6( (~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5)))
+        case 7  => '(Tuple7( (~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6)))
+        case 8  => '(Tuple8( (~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7)))
+        case 9  => '(Tuple9( (~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8)))
+        case 10 => '(Tuple10((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9)))
+        case 11 => '(Tuple11((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9), (~xs)(10)))
+        case 12 => '(Tuple12((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9), (~xs)(10), (~xs)(11)))
+        case 13 => '(Tuple13((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9), (~xs)(10), (~xs)(11), (~xs)(12)))
+        case 14 => '(Tuple14((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9), (~xs)(10), (~xs)(11), (~xs)(12), (~xs)(13)))
+        case 15 => '(Tuple15((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9), (~xs)(10), (~xs)(11), (~xs)(12), (~xs)(13), (~xs)(14)))
+        case 16 => '(Tuple16((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9), (~xs)(10), (~xs)(11), (~xs)(12), (~xs)(13), (~xs)(14), (~xs)(15)))
+        case 17 => '(Tuple17((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9), (~xs)(10), (~xs)(11), (~xs)(12), (~xs)(13), (~xs)(14), (~xs)(15), (~xs)(16)))
+        case 18 => '(Tuple18((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9), (~xs)(10), (~xs)(11), (~xs)(12), (~xs)(13), (~xs)(14), (~xs)(15), (~xs)(16), (~xs)(17)))
+        case 19 => '(Tuple19((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9), (~xs)(10), (~xs)(11), (~xs)(12), (~xs)(13), (~xs)(14), (~xs)(15), (~xs)(16), (~xs)(17), (~xs)(18)))
+        case 20 => '(Tuple20((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9), (~xs)(10), (~xs)(11), (~xs)(12), (~xs)(13), (~xs)(14), (~xs)(15), (~xs)(16), (~xs)(17), (~xs)(18), (~xs)(19)))
+        case 21 => '(Tuple21((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9), (~xs)(10), (~xs)(11), (~xs)(12), (~xs)(13), (~xs)(14), (~xs)(15), (~xs)(16), (~xs)(17), (~xs)(18), (~xs)(19), (~xs)(20)))
+        case 22 => '(Tuple22((~xs)(0), (~xs)(1), (~xs)(2), (~xs)(3), (~xs)(4), (~xs)(5), (~xs)(6), (~xs)(7), (~xs)(8), (~xs)(9), (~xs)(10), (~xs)(11), (~xs)(12), (~xs)(13), (~xs)(14), (~xs)(15), (~xs)(16), (~xs)(17), (~xs)(18), (~xs)(19), (~xs)(20), (~xs)(21)))
+        case _  => '(TupleXXL(~xs))
+      }
+      tup.as[T]
+    }
+  }
+
+  def sizeStaged[Res <: Int : Type](tup: Expr[Tuple], size: Option[Int]): Expr[Res] = {
+    val res =
+      if (!specialize) '(dynamicSize(~tup).asInstanceOf[Res])
+      else size match {
+        case Some(n) => n.toExpr
+        case None => '(dynamicSize(~tup).asInstanceOf[Res])
+      }
+    res.as[Res]
+  }
+
+  private implicit class ExprOps[U: Type](expr: Expr[U]) {
+
+    def as[T: Type]: Expr[T] = '{ (~expr).asInstanceOf[T] }
+
+    def bind[T](in: Expr[U] => Expr[T]): Expr[T] = '{
+      val t: U = (~expr)
+      ~(in('(t)))
+    }
+
+  }
+
+}

library/src-scala3/scala/Tuple.scala

@@ -4,32 +4,10 @@ import typelevel._
 
 sealed trait Tuple extends Any {
   import Tuple._
+  import StagedTuple._
 
   inline def toArray: Array[Object] =
-    /*if (specialize)
-      inline constValueOpt[BoundedSize[this.type]] match {
-      case Some(0) =>
-        $emptyArray
-      case Some(1) =>
-        val t = asInstanceOf[Tuple1[Object]]
-        Array(t._1)
-      case Some(2) =>
-        val t = asInstanceOf[Tuple2[Object, Object]]
-        Array(t._1, t._2)
-      case Some(3) =>
-        val t = asInstanceOf[Tuple3[Object, Object, Object]]
-        Array(t._1, t._2, t._3)
-      case Some(4) =>
-        val t = asInstanceOf[Tuple4[Object, Object, Object, Object]]
-        Array(t._1, t._2, t._3, t._4)
-      case Some(n) if n <= $MaxSpecialized =>
-        $toArray(this, n)
-      case Some(n) =>
-        asInstanceOf[TupleXXL].elems
-      case None =>
-        dynamicToArray(this)
-    }
-    else*/ dynamicToArray(this)
+    ~toArrayStaged('(this), constValueOpt[BoundedSize[this.type]])
 
   inline def *: [H] (x: H): H *: this.type =
     /*if (specialize) {
@@ -101,17 +79,13 @@ sealed trait Tuple extends Any {
     fromArray[T](xs.toArray ++ ys.toArray)
 
   inline def size: Size[this.type] =
-    /*if (specialize) {
-      type Result = Size[this.type]
-      inline constValueOpt[BoundedSize[this.type]] match {
-        case Some(n) => n.asInstanceOf[Result]
-        case _ => dynamicSize(this)
-      }
-    }
-    else*/ dynamicSize(this)
+    ~sizeStaged[Size[this.type]]('(this), constValueOpt[BoundedSize[this.type]])
+
 }
 
 object Tuple {
+  import StagedTuple._
+
   inline val $MaxSpecialized = 22
   inline private val XXL = $MaxSpecialized + 1
 
@@ -175,34 +149,7 @@ object Tuple {
   }
 
   inline def fromArray[T <: Tuple](xs: Array[Object]): T =
-    /*if (specialize)
-      inline constValue[BoundedSize[T]] match {
-        case 0  => ().asInstanceOf[T]
-        case 1  => Tuple1(xs(0)).asInstanceOf[T]
-        case 2  => Tuple2(xs(0), xs(1)).asInstanceOf[T]
-        case 3  => Tuple3(xs(0), xs(1), xs(2)).asInstanceOf[T]
-        case 4  => Tuple4(xs(0), xs(1), xs(2), xs(3)).asInstanceOf[T]
-        case 5  => Tuple5(xs(0), xs(1), xs(2), xs(3), xs(4)).asInstanceOf[T]
-        case 6  => Tuple6(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5)).asInstanceOf[T]
-        case 7  => Tuple7(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6)).asInstanceOf[T]
-        case 8  => Tuple8(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7)).asInstanceOf[T]
-        case 9  => Tuple9(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8)).asInstanceOf[T]
-        case 10 => Tuple10(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9)).asInstanceOf[T]
-        case 11 => Tuple11(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9), xs(10)).asInstanceOf[T]
-        case 12 => Tuple12(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9), xs(10), xs(11)).asInstanceOf[T]
-        case 13 => Tuple13(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9), xs(10), xs(11), xs(12)).asInstanceOf[T]
-        case 14 => Tuple14(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9), xs(10), xs(11), xs(12), xs(13)).asInstanceOf[T]
-        case 15 => Tuple15(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9), xs(10), xs(11), xs(12), xs(13), xs(14)).asInstanceOf[T]
-        case 16 => Tuple16(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9), xs(10), xs(11), xs(12), xs(13), xs(14), xs(15)).asInstanceOf[T]
-        case 17 => Tuple17(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9), xs(10), xs(11), xs(12), xs(13), xs(14), xs(15), xs(16)).asInstanceOf[T]
-        case 18 => Tuple18(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9), xs(10), xs(11), xs(12), xs(13), xs(14), xs(15), xs(16), xs(17)).asInstanceOf[T]
-        case 19 => Tuple19(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9), xs(10), xs(11), xs(12), xs(13), xs(14), xs(15), xs(16), xs(17), xs(18)).asInstanceOf[T]
-        case 20 => Tuple20(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9), xs(10), xs(11), xs(12), xs(13), xs(14), xs(15), xs(16), xs(17), xs(18), xs(19)).asInstanceOf[T]
-        case 21 => Tuple21(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9), xs(10), xs(11), xs(12), xs(13), xs(14), xs(15), xs(16), xs(17), xs(18), xs(19), xs(20)).asInstanceOf[T]
-        case 22 => Tuple22(xs(0), xs(1), xs(2), xs(3), xs(4), xs(5), xs(6), xs(7), xs(8), xs(9), xs(10), xs(11), xs(12), xs(13), xs(14), xs(15), xs(16), xs(17), xs(18), xs(19), xs(20), xs(21)).asInstanceOf[T]
-        case _ => TupleXXL(xs).asInstanceOf[T]
-      }
-    else */dynamicFromArray[T](xs)
+    ~fromArrayStaged[T]('(xs), constValue[BoundedSize[this.type]])
 
   def dynamicFromArray[T <: Tuple](xs: Array[Object]): T = xs.length match {
     case 0  => ().asInstanceOf[T]
@@ -433,7 +380,7 @@ object NonEmptyTuple {
       case self: Tuple2[_, _] => Tuple1(self._2)
       case self: Tuple3[_, _, _] => Tuple2(self._2, self._3)
       case self: Tuple4[_, _, _, _] => Tuple3(self._2, self._3, self._4)
-      case _ => dynamicFromArray[Result](self.toArray.tail)
+      case _ => dynamicFromArray[Result](dynamicToArray(self).tail)
     }
     res.asInstanceOf[Result]
   }