upgrades typingTransform

typingTransform and typingTransform's atOwner now work both with
solitary trees and Tree+Symbol couples.

1 files changed, 9 insertions, 0 deletions

diff --git a/src/reflect/scala/reflect/macros/Internals.scala b/src/reflect/scala/reflect/macros/Internals.scala
index 843644b7e3..75164344da 100644
--- a/src/reflect/scala/reflect/macros/Internals.scala
+++ b/src/reflect/scala/reflect/macros/Internals.scala
@@ -51,6 +51,10 @@ trait Internals {
        */
       def atOwner[T](owner: Symbol)(op: => T): T
 
+      /** Temporarily pushes the given tree onto the recursion stack, and then calls `atOwner(symbol)(trans)`.
+       */
+      def atOwner[T](tree: Tree, owner: Symbol)(op: => T): T
+
       /** Returns the symbol currently on the top of the owner stack.
        *  If we're not inside any `atOwner` call, then macro application's context owner will be used.
        */
@@ -66,5 +70,10 @@ trait Internals {
      *  @see [[TypingTransformApi]]
      */
     def typingTransform(tree: Tree)(transformer: (Tree, TypingTransformApi) => Tree): Tree
+
+    /** Transforms a given tree at a given owner using the provided function.
+     *  @see [[TypingTransformApi]]
+     */
+    def typingTransform(tree: Tree, owner: Symbol)(transformer: (Tree, TypingTransformApi) => Tree): Tree
   }
 }