Mercurial > jdk8-mips64-public > langtools / file revision

 /*

  * Copyright (c) 2006, Oracle and/or its affiliates. All rights reserved.

  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

  *

  * This code is free software; you can redistribute it and/or modify it

  * under the terms of the GNU General Public License version 2 only, as

  * published by the Free Software Foundation.

  *

  * This code is distributed in the hope that it will be useful, but WITHOUT

  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

  * version 2 for more details (a copy is included in the LICENSE file that

  * accompanied this code).

  *

  * You should have received a copy of the GNU General Public License version

  * 2 along with this work; if not, write to the Free Software Foundation,

  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

  *

  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

  * or visit www.oracle.com if you need additional information or have any

  * questions.

  */

 /*

  * @test

  * @bug 6346249 6392177

  * @summary new Trees API

  */

 import com.sun.source.tree.*;

 import com.sun.source.util.*;

 import java.io.*;

 import java.lang.annotation.*;

 import java.util.*;

 import javax.annotation.processing.*;

 import javax.lang.model.SourceVersion;

 import javax.lang.model.element.*;

 import javax.lang.model.type.*;

 import javax.tools.*;

 import com.sun.tools.javac.api.JavacTool;

 import com.sun.tools.javac.tree.JCTree;

 import com.sun.tools.javac.tree.TreeInfo;

 @Anno

 @SupportedAnnotationTypes("*")

 public class TestTrees extends AbstractProcessor {

     @Anno

     void annoMethod() { }

     @Anno

     int annoField;

     static final String testSrcDir = System.getProperty("test.src");

     static final String testClassDir = System.getProperty("test.classes");

     static final String self = TestTrees.class.getName();

     static PrintWriter out = new PrintWriter(System.err, true);

     public static void main(String[] args) throws IOException {

         new TestTrees().run();

     }

     void run() throws IOException {

         JavacTool tool = JavacTool.create();

         DiagnosticListener<JavaFileObject> dl = new DiagnosticListener<JavaFileObject>() {

                 public void report(Diagnostic d) {

                     error(d.toString());

                 }

             };

         StandardJavaFileManager fm = tool.getStandardFileManager(dl, null, null);

         Iterable<? extends JavaFileObject> files =

             fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrcDir, self + ".java")));

         Iterable<String> opts = Arrays.asList("-d", ".");

         System.err.println("simple compilation, no processing");

         JavacTask task = tool.getTask(out, fm, dl, opts, null, files);

         task.setTaskListener(new MyTaskListener(task));

         if (!task.call())

             throw new AssertionError("compilation failed");

         opts =  Arrays.asList("-d", ".", "-processorpath", testClassDir, "-processor", self);

         System.err.println();

         System.err.println("compilation with processing");

         task = tool.getTask(out, fm, dl,opts, null, files);

         if (!task.call())

             throw new AssertionError("compilation failed");

         if (errors > 0)

             throw new AssertionError(errors + " errors occurred");

     }

     void testElement(Trees trees, Element e) {

         trees.getClass();

         e.getClass();

         System.err.println("testElement: " + e);

         Tree tree = trees.getTree(e);

         //System.err.println(tree);

         if (TreeInfo.symbolFor((JCTree)tree) != e)

             error("bad result from getTree");

         TreePath path = trees.getPath(e);

         if (path == null) {

             error("getPath returned null");

             return;

         }

         if (path.getLeaf() != tree)

             error("bad result from getPath");

         Element e2 = trees.getElement(path);

         if (e2 == null) {

             error("getElement returned null");

             return;

         }

         if (e2 != e)

             error("bad result from getElement");

         // The TypeMirror is not available yet when annotation processing;

         // it is set up later during ANALYSE.

         TypeMirror t = trees.getTypeMirror(path);

         if (t != null && t.getKind() == TypeKind.DECLARED &&

                 ((DeclaredType)t).asElement() != e2)

             error("bad result from getTypeMirror");

         for (AnnotationMirror m: e.getAnnotationMirrors()) {

             testAnnotation(trees, e, m);

         }

     }

     void testAnnotation(Trees trees, Element e, AnnotationMirror a) {

         System.err.println("testAnnotation: " + e + " " + a);

         Tree tree = trees.getTree(e, a);

         if (tree.getKind() != Tree.Kind.ANNOTATION)

             error("bad result from getTree");

         TreePath path = trees.getPath(e, a);

         if (path.getLeaf() != tree)

             error("bad result from getPath");

     }

     void error(String msg) {

         if (messager != null)

             // annotation processing will happen in a separate instance/classloader

             // so pass the message back to the calling instance.

             messager.printMessage(Diagnostic.Kind.ERROR, msg);

         else {

             System.err.println(msg);

             errors++;

         }

     }

     Messager messager;

     int errors;

     public boolean process(Set<? extends TypeElement> annos, RoundEnvironment rEnv) {

         Trees trees = Trees.instance(processingEnv);

         messager = processingEnv.getMessager();

         for (Element e: rEnv.getRootElements()) {

             testElement(trees, e);

         }

         for (TypeElement anno: annos) {

             Set<? extends Element> elts = rEnv.getElementsAnnotatedWith(anno);

             System.err.println("anno: " + anno);

             System.err.println("elts: " + elts);

             if (elts != null) { // 6397298, should return empty set

                 for (Element e: rEnv.getElementsAnnotatedWith(anno))

                     testElement(trees, e);

             }

         }

         return true;

     }

     @Override

     public SourceVersion getSupportedSourceVersion() {

         return SourceVersion.latest();

     }

     class MyTaskListener implements TaskListener {

         MyTaskListener(JavacTask task) {

             this.task = task;

         }

         public void started(TaskEvent e) {

             System.err.println("started " + e);

         }

         public void finished(TaskEvent e) {

             //System.err.println("finished " + e);

             switch (e.getKind()) {

             case ANALYZE:

                 testElement(Trees.instance(task), e.getTypeElement());

                 break;

             }

         }

         private final JavacTask task;

     }

 }

 @Retention(RetentionPolicy.SOURCE)

 @interface Anno {

 }