Experimental Streams using quotes

compiler/src/dotty/tools/dotc/core/quoted/PickledQuotes.scala

@@ -66,13 +66,13 @@ object PickledQuotes {
   /** Unpickle the tree contained in the TastyExpr */
   private def unpickleExpr(expr: TastyExpr[_])(implicit ctx: Context): Tree = {
     val tastyBytes = TastyString.unpickle(expr.tasty)
-    unpickle(tastyBytes, expr.args)
+    unpickle(tastyBytes, expr.args, isType = false)
   }
 
   /** Unpickle the tree contained in the TastyType */
   private def unpickleType(ttpe: TastyType[_])(implicit ctx: Context): Tree = {
     val tastyBytes = TastyString.unpickle(ttpe.tasty)
-    unpickle(tastyBytes, ttpe.args)
+    unpickle(tastyBytes, ttpe.args, isType = true)
   }
 
   // TASTY picklingtests/pos/quoteTest.scala
@@ -85,28 +85,33 @@ object PickledQuotes {
     treePkl.compactify()
     pickler.addrOfTree = treePkl.buf.addrOfTree
     pickler.addrOfSym = treePkl.addrOfSym
-    // if (tree.pos.exists)
-    //   new PositionPickler(pickler, treePkl.buf.addrOfTree).picklePositions(tree :: Nil)
 
-    // other pickle sections go here.
+    if (pickling ne noPrinter)
+      println(i"**** pickling quote of \n${tree.show}")
+
     val pickled = pickler.assembleParts()
 
-    if (pickling ne noPrinter) {
-      println(i"**** pickled quote of \n${tree.show}")
+    if (pickling ne noPrinter)
       new TastyPrinter(pickled).printContents()
-    }
 
     pickled
   }
 
   /** Unpickle TASTY bytes into it's tree */
-  private def unpickle(bytes: Array[Byte], splices: Seq[Any])(implicit ctx: Context): Tree = {
+  private def unpickle(bytes: Array[Byte], splices: Seq[Any], isType: Boolean)(implicit ctx: Context): Tree = {
     val unpickler = new TastyUnpickler(bytes, splices)
-    val tree = unpickler.unpickleExpr()
     if (pickling ne noPrinter) {
-      println(i"**** unpickled quote for \n${tree.show}")
+      println(i"**** unpickling quote from TASTY")
       new TastyPrinter(bytes).printContents()
     }
+
+    val tree =
+      if (isType) unpickler.unpickleTypeTree()
+      else unpickler.unpickleExpr()
+
+    if (pickling ne noPrinter)
+      println(i"**** unpickle quote ${tree.show}")
+
     tree
   }
 

compiler/src/dotty/tools/dotc/core/tasty/DottyUnpickler.scala

@@ -49,6 +49,9 @@ class DottyUnpickler(bytes: Array[Byte]) extends ClassfileParser.Embedded with t
   def unpickleExpr()(implicit ctx: Context): Tree =
     treeUnpickler.unpickleExpr()
 
+  def unpickleTypeTree()(implicit ctx: Context): Tree =
+    treeUnpickler.unpickleTypeTree()
+
   protected def treeSectionUnpickler(posUnpicklerOpt: Option[PositionUnpickler]): TreeSectionUnpickler = {
     new TreeSectionUnpickler(posUnpicklerOpt)
   }

compiler/src/dotty/tools/dotc/core/tasty/TreeUnpickler.scala

@@ -85,6 +85,13 @@ class TreeUnpickler(reader: TastyReader,
     rdr.readTerm()
   }
 
+  def unpickleTypeTree()(implicit ctx: Context): Tree = {
+    this.roots = Set(ctx.owner)
+    val rdr = new TreeReader(reader).fork
+    ownerTree = new OwnerTree(NoAddr, 0, rdr.fork, reader.endAddr)
+    rdr.readTpt()
+  }
+
   /** The unpickled trees */
   def unpickle()(implicit ctx: Context): List[Tree] = {
     assert(roots != null, "unpickle without previous enterTopLevel")

compiler/test/dotty/tools/dotc/CompilationTests.scala

@@ -198,7 +198,8 @@ class CompilationTests extends ParallelTesting {
     implicit val testGroup: TestGroup = TestGroup("runAll")
     compileFilesInDir("tests/run", defaultOptions) +
     compileFilesInDir("tests/run-no-optimise", defaultOptions) +
-    compileFilesInDir("tests/run-with-compiler", defaultRunWithCompilerOptions)
+    compileFilesInDir("tests/run-with-compiler", defaultRunWithCompilerOptions) +
+    compileFile("tests/run-with-compiler-custom-args/staged-streams_1.scala", defaultRunWithCompilerOptions without "-Yno-deep-subtypes")
   }.checkRuns()
 
   // Generic java signatures tests ---------------------------------------------

tests/run-with-compiler-custom-args/staged-streams_1.check

@@ -0,0 +1,19 @@
+6
+
+12
+
+36
+
+2
+
+3
+
+7
+
+12
+
+15
+
+15
+
+72