From 5d733713904af89383cf72c85da448e45b5645ce Mon Sep 17 00:00:00 2001
From: Adriaan Moors <adriaan.moors@typesafe.com>
Date: Sun, 11 Oct 2015 17:15:54 +0200
Subject: LazyVals looks for local vals in all term trees

Discovered while refactoring the way fields are emitted in traits.

TODO: a lot more cleanup
---
 .../scala/tools/nsc/transform/LazyVals.scala       | 23 +++++++++-------------
 1 file changed, 9 insertions(+), 14 deletions(-)

(limited to 'src')

diff --git a/src/compiler/scala/tools/nsc/transform/LazyVals.scala b/src/compiler/scala/tools/nsc/transform/LazyVals.scala
index b6695efb0b..93e41dde7b 100644
--- a/src/compiler/scala/tools/nsc/transform/LazyVals.scala
+++ b/src/compiler/scala/tools/nsc/transform/LazyVals.scala
@@ -38,6 +38,7 @@ abstract class LazyVals extends Transform with TypingTransformers with ast.TreeD
 
           case ClassDef(_, _, _, _) | DefDef(_, _, _, _, _, _) | ModuleDef(_, _, _) =>
 
+          // Avoid adding bitmaps when they are fully overshadowed by those that are added inside loops
           case LabelDef(name, _, _) if nme.isLoopHeaderLabel(name) =>
 
           case _ =>
@@ -108,22 +109,16 @@ abstract class LazyVals extends Transform with TypingTransformers with ast.TreeD
         }
 
         case Template(_, _, body) => atOwner(currentOwner) {
-          val body1 = super.transformTrees(body)
+          // TODO: shady business... can this logic be encapsulated in LocalLazyValFinder?
           var added = false
-          val stats =
-            for (stat <- body1) yield stat match {
-              case Block(_, _) | Apply(_, _) | If(_, _, _) | Try(_, _, _) if !added =>
-                // Avoid adding bitmaps when they are fully overshadowed by those
-                // that are added inside loops
-                if (LocalLazyValFinder.find(stat)) {
-                  added = true
-                  typed(addBitmapDefs(sym, stat))
-                } else stat
-              case ValDef(_, _, _, _) =>
-                typed(deriveValDef(stat)(addBitmapDefs(stat.symbol, _)))
-              case _ =>
-                stat
+          val stats = super.transformTrees(body) mapConserve {
+              case stat: ValDef => typed(deriveValDef(stat)(addBitmapDefs(stat.symbol, _)))
+              case stat: TermTree if !added && (LocalLazyValFinder find stat) =>
+                added = true
+                typed(addBitmapDefs(sym, stat))
+              case stat => stat
             }
+
           val innerClassBitmaps = if (!added && currentOwner.isClass && bitmaps.contains(currentOwner)) {
               // add bitmap to inner class if necessary
                 val toAdd0 = bitmaps(currentOwner).map(s => typed(ValDef(s, ZERO)))
-- 
cgit v1.2.3