1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/test/tools/javac/processing/model/element/TestAnonSourceNames.java Wed Mar 03 16:05:34 2010 -0800
1.3 @@ -0,0 +1,92 @@
1.4 +/*
1.5 + * Copyright 2010 Sun Microsystems, Inc. All Rights Reserved.
1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
1.7 + *
1.8 + * This code is free software; you can redistribute it and/or modify it
1.9 + * under the terms of the GNU General Public License version 2 only, as
1.10 + * published by the Free Software Foundation.
1.11 + *
1.12 + * This code is distributed in the hope that it will be useful, but WITHOUT
1.13 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
1.14 + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
1.15 + * version 2 for more details (a copy is included in the LICENSE file that
1.16 + * accompanied this code).
1.17 + *
1.18 + * You should have received a copy of the GNU General Public License version
1.19 + * 2 along with this work; if not, write to the Free Software Foundation,
1.20 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
1.21 + *
1.22 + * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
1.23 + * CA 95054 USA or visit www.sun.com if you need additional information or
1.24 + * have any questions.
1.25 + */
1.26 +
1.27 +import java.io.*;
1.28 +import javax.annotation.processing.*;
1.29 +import javax.lang.model.*;
1.30 +import javax.lang.model.element.*;
1.31 +import javax.lang.model.type.*;
1.32 +import javax.lang.model.util.*;
1.33 +import java.util.*;
1.34 +import com.sun.source.tree.*;
1.35 +import com.sun.source.util.*;
1.36 +import static javax.tools.Diagnostic.Kind.*;
1.37 +
1.38 +/**
1.39 + * Using the tree API, retrieve element representations of anonymous
1.40 + * classes and verify their names are as specified.
1.41 + */
1.42 +@SupportedAnnotationTypes("*")
1.43 +public class TestAnonSourceNames extends AbstractProcessor {
1.44 +
1.45 + public boolean process(Set<? extends TypeElement> annotations,
1.46 + RoundEnvironment roundEnv) {
1.47 + if (!roundEnv.processingOver()) {
1.48 + Trees trees = Trees.instance(processingEnv);
1.49 +
1.50 + for(Element rootElement : roundEnv.getRootElements()) {
1.51 + TreePath treePath = trees.getPath(rootElement);
1.52 +
1.53 + (new ClassTreeScanner(trees)).
1.54 + scan(trees.getTree(rootElement),
1.55 + treePath.getCompilationUnit());
1.56 + }
1.57 + }
1.58 + return true;
1.59 + }
1.60 +
1.61 + class ClassTreeScanner extends TreeScanner<Void, CompilationUnitTree> {
1.62 + private Trees trees;
1.63 +
1.64 + public ClassTreeScanner(Trees trees) {
1.65 + super();
1.66 + this.trees = trees;
1.67 + }
1.68 + @Override
1.69 + public Void visitClass(ClassTree node, CompilationUnitTree cu) {
1.70 + Element element = trees.getElement(trees.getPath(cu, node));
1.71 + if (element == null) {
1.72 + processingEnv.getMessager().printMessage(ERROR,
1.73 + "No element retreived for node named ''" +
1.74 + node.getSimpleName() + "''.");
1.75 + } else {
1.76 +
1.77 + System.out.println("\nVisiting class ``" + element.getSimpleName() +
1.78 + "'' of kind " + element.getKind());
1.79 + if (element instanceof TypeElement) {
1.80 + TypeElement typeElement = (TypeElement) element;
1.81 + String s = typeElement.getQualifiedName().toString();
1.82 + System.out.println("\tqualified name:" + s);
1.83 + } else {
1.84 + throw new RuntimeException("TypeElement not gotten from ClassTree.");
1.85 + }
1.86 + }
1.87 + return super.visitClass(node, cu);
1.88 + }
1.89 + }
1.90 +
1.91 + @Override
1.92 + public SourceVersion getSupportedSourceVersion() {
1.93 + return SourceVersion.latest();
1.94 + }
1.95 +}
