From 502424684ac04a69480973af4fa82286f71a6d1c Mon Sep 17 00:00:00 2001
From: Jason Zaugg <jzaugg@gmail.com>
Date: Tue, 5 Nov 2013 10:04:26 +0100
Subject: Optimize the pickler phase

 - move "erroneous type" diagnostic into a crash recovery handler,
    rather than running it proactively
 - move the "unexpanded macros" check into refchecks.

 Cuts this phase in half, from about 1% of compile time to 0.5%.
---
 .../scala/tools/nsc/symtab/classfile/Pickler.scala | 33 ++++++++++++----------
 .../scala/tools/nsc/typechecker/RefChecks.scala    |  7 +++++
 2 files changed, 25 insertions(+), 15 deletions(-)

diff --git a/src/compiler/scala/tools/nsc/symtab/classfile/Pickler.scala b/src/compiler/scala/tools/nsc/symtab/classfile/Pickler.scala
index 90c15bca61..55f45257dc 100644
--- a/src/compiler/scala/tools/nsc/symtab/classfile/Pickler.scala
+++ b/src/compiler/scala/tools/nsc/symtab/classfile/Pickler.scala
@@ -55,23 +55,26 @@ abstract class Pickler extends SubComponent {
           case _ =>
         }
       }
-      // If there are any erroneous types in the tree, then we will crash
-      // when we pickle it: so let's report an error instead.  We know next
-      // to nothing about what happened, but our supposition is a lot better
-      // than "bad type: <error>" in terms of explanatory power.
-      for (t <- unit.body) {
-        if (t.isErroneous) {
-          unit.error(t.pos, "erroneous or inaccessible type")
-          return
-        }
 
-        if (!t.isDef && t.hasSymbolField && t.symbol.isTermMacro) {
-          unit.error(t.pos, "macro has not been expanded")
-          return
-        }
+      try {
+        pickle(unit.body)
+      } catch {
+        case e: FatalError =>
+          for (t <- unit.body) {
+            // If there are any erroneous types in the tree, then we will crash
+            // when we pickle it: so let's report an error instead.  We know next
+            // to nothing about what happened, but our supposition is a lot better
+            // than "bad type: <error>" in terms of explanatory power.
+            //
+            // OPT: do this only as a recovery after fatal error. Checking in advance was expensive.
+            if (t.isErroneous) {
+              if (settings.debug) e.printStackTrace()
+              unit.error(t.pos, "erroneous or inaccessible type")
+              return
+            }
+          }
+          throw e
       }
-
-      pickle(unit.body)
     }
   }
 
diff --git a/src/compiler/scala/tools/nsc/typechecker/RefChecks.scala b/src/compiler/scala/tools/nsc/typechecker/RefChecks.scala
index 68d724b6fc..f7684b93af 100644
--- a/src/compiler/scala/tools/nsc/typechecker/RefChecks.scala
+++ b/src/compiler/scala/tools/nsc/typechecker/RefChecks.scala
@@ -1596,6 +1596,10 @@ abstract class RefChecks extends InfoTransform with scala.reflect.internal.trans
       }
     }
 
+    private def checkUnexpandedMacro(t: Tree) =
+      if (!t.isDef && t.hasSymbolField && t.symbol.isTermMacro)
+        unit.error(t.pos, "macro has not been expanded")
+
     override def transform(tree: Tree): Tree = {
       val savedLocalTyper = localTyper
       val savedCurrentApplication = currentApplication
@@ -1755,6 +1759,9 @@ abstract class RefChecks extends InfoTransform with scala.reflect.internal.trans
             varianceValidator.traverse(tt.original) // See SI-7872
           case _ =>
         }
+
+        checkUnexpandedMacro(result)
+
         result
       } catch {
         case ex: TypeError =>
-- 
cgit v1.2.3