testtrees.java

来自「是一款用JAVA 编写的编译器 具有很强的编译功能」· Java 代码 · 共 219 行

/*
 * Copyright 2006 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.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 { 
}