blob: ac72383195c6845b364b8a17220d83f2796c0361 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
/* NSC -- new Scala compiler
* Copyright 2005-2009 LAMP/EPFL
* @author Martin Odersky
*/
// $Id$
package scala.tools.nsc
package transform
import util.Position
import scala.collection.mutable.{Map, HashMap}
/** A base class for transforms.
* A transform contains a compiler phase which applies a tree transformer.
*/
trait TypingTransformers {
val global: Global
import global._
abstract class TypingTransformer(unit: CompilationUnit) extends Transformer {
var localTyper: analyzer.Typer = analyzer.newTyper(
analyzer.rootContext(unit, EmptyTree, true))
protected var curTree: Tree = _
protected def typedPos(pos: Position)(tree: Tree) = localTyper typed { atPos(pos)(tree) }
/** a typer for each enclosing class */
var typers: Map[Symbol, analyzer.Typer] = new HashMap
override def atOwner[A](owner: Symbol)(trans: => A): A = atOwner(curTree, owner)(trans)
def atOwner[A](tree: Tree, owner: Symbol)(trans: => A): A = {
val savedLocalTyper = localTyper
// println("ttransformer atOwner: " + owner + " isPackage? " + owner.isPackage)
localTyper = localTyper.atOwner(tree, if (owner.isModule) owner.moduleClass else owner)
typers += Pair(owner, localTyper)
val result = super.atOwner(owner)(trans)
localTyper = savedLocalTyper
typers -= owner
result
}
override def transform(tree: Tree): Tree = {
curTree = tree
tree match {
case Template(_, _, _) =>
// enter template into context chain
atOwner(currentOwner) { super.transform(tree) }
case PackageDef(_, _) =>
atOwner(tree.symbol) { super.transform(tree) }
case _ =>
super.transform(tree)
}
}
}
}
|