<change id="WorkingCopy-resolveRewriteTarget">

             <api name="general"/>

             <summary>Added <code>WorkingCopy.resolveRewriteTarget</code>.</summary>

             <version major="0" minor="101"/>

             <date day="9" month="4" year="2012"/>

             <author login="jlahoda"/>

            <compatibility addition="yes" binary="compatible" deletion="no" deprecation="no" modification="no" semantic="compatible" source="compatible"/>

             <description>

                 Added a method<code>resolveRewriteTarget</code> to class <code>WorkingCopy</code>,

                 that returns the tree into which the given tree has been rewritten.

             </description>

             <class package="org.netbeans.api.java.source" name="WorkingCopy"/>

             <issue number="209375"/>

        </change>

    /**Returns the tree into which the given tree was rewritten using the

     * {@link #rewrite(com.sun.source.tree.Tree, com.sun.source.tree.Tree) } method,

     * transitively.

     * Will return the input tree if the input tree was never passed as the first

     * parameter of the {@link #rewrite(com.sun.source.tree.Tree, com.sun.source.tree.Tree) }

     * method.

     *

     * <p>Note that the returned tree will be exactly equivalent to a tree passed as

     * the second parameter to {@link #rewrite(com.sun.source.tree.Tree, com.sun.source.tree.Tree) }.

     * No attribution or other information will be added (or removed) to (or from) the tree.

     *

     * @param in the tree to inspect

     * @return tree into which the given tree was rewritten using the

     * {@link #rewrite(com.sun.source.tree.Tree, com.sun.source.tree.Tree) } method,

     * transitively

     * @since 0.101

     */

    public synchronized @NonNull Tree resolveRewriteTarget(@NonNull Tree in) {

        Map<Tree, Tree> localChanges = new IdentityHashMap<Tree, Tree>(changes);

        while (localChanges.containsKey(in)) {

            in = localChanges.remove(in);

        }

        return in;

    }

    public void testResolveRewriteTarget() throws Exception {

        File f = new File(getWorkDir(), "TestClass.java");

        String code = "package foo;" +

                      "public class TestClass{" +

                      "   public void foo() {" +

                      "   }" +

                      "}";

        TestUtilities.copyStringToFile(f, code);

        FileObject fo = FileUtil.toFileObject(f);

        JavaSource javaSource = JavaSource.forFileObject(fo);

        javaSource.runModificationTask(new Task<WorkingCopy>() {

            public void run(WorkingCopy copy) throws Exception {

                copy.toPhase(Phase.RESOLVED);

                TreeMaker maker = copy.getTreeMaker();

                ClassTree classTree = (ClassTree)copy.getCompilationUnit().getTypeDecls().get(0);

                TypeElement serializableElement = copy.getElements().getTypeElement("java.io.Serializable");

                ExpressionTree serializableTree = maker.QualIdent(serializableElement);

                ClassTree newClassTree = maker.addClassImplementsClause(classTree, serializableTree);

                copy.rewrite(classTree, newClassTree);

                assertSame(newClassTree, copy.resolveRewriteTarget(classTree));

                ClassTree finalClassTree = maker.removeClassImplementsClause(newClassTree, 0);

                copy.rewrite(newClassTree, finalClassTree);

                assertSame(finalClassTree, copy.resolveRewriteTarget(classTree));

                // remove the following to make the test pass

                copy.toPhase(Phase.RESOLVED);

            }

        }).commit();

        assertEquals(code, fo.asText());

    }