From 7ce29a715f0fc3662217e5ef826492d0909abf20 Mon Sep 17 00:00:00 2001 From: Guillaume Martres Date: Fri, 6 Apr 2018 01:44:51 +0200 Subject: [PATCH] Fix TreeMap context for PackageDef and LambdaTypeTree No test, but this mirrors what we already do in TreeAccumulator --- compiler/src/dotty/tools/dotc/ast/Trees.scala | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/compiler/src/dotty/tools/dotc/ast/Trees.scala b/compiler/src/dotty/tools/dotc/ast/Trees.scala index 33e3a3642653..0011db002823 100644 --- a/compiler/src/dotty/tools/dotc/ast/Trees.scala +++ b/compiler/src/dotty/tools/dotc/ast/Trees.scala @@ -1184,6 +1184,7 @@ object Trees { case AppliedTypeTree(tpt, args) => cpy.AppliedTypeTree(tree)(transform(tpt), transform(args)) case LambdaTypeTree(tparams, body) => + implicit val ctx = localCtx cpy.LambdaTypeTree(tree)(transformSub(tparams), transform(body)) case ByNameTypeTree(result) => cpy.ByNameTypeTree(tree)(transform(result)) @@ -1213,7 +1214,7 @@ object Trees { case Import(expr, selectors) => cpy.Import(tree)(transform(expr), selectors) case PackageDef(pid, stats) => - cpy.PackageDef(tree)(transformSub(pid), transformStats(stats)) + cpy.PackageDef(tree)(transformSub(pid), transformStats(stats)(localCtx)) case Annotated(arg, annot) => cpy.Annotated(tree)(transform(arg), transform(annot)) case Thicket(trees) =>