test/tools/apt/Scanners/Scanner.java

Thu, 25 Aug 2011 17:18:25 -0700

author
schien
date
Thu, 25 Aug 2011 17:18:25 -0700
changeset 1067
f497fac86cf9
parent 554
9d9f26857129
permissions
-rw-r--r--

Added tag jdk8-b02 for changeset b3c059de2a61

duke@1 1 /*
ohair@554 2 * Copyright (c) 2004, 2007, Oracle and/or its affiliates. All rights reserved.
duke@1 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
duke@1 4 *
duke@1 5 * This code is free software; you can redistribute it and/or modify it
duke@1 6 * under the terms of the GNU General Public License version 2 only, as
duke@1 7 * published by the Free Software Foundation.
duke@1 8 *
duke@1 9 * This code is distributed in the hope that it will be useful, but WITHOUT
duke@1 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
duke@1 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
duke@1 12 * version 2 for more details (a copy is included in the LICENSE file that
duke@1 13 * accompanied this code).
duke@1 14 *
duke@1 15 * You should have received a copy of the GNU General Public License version
duke@1 16 * 2 along with this work; if not, write to the Free Software Foundation,
duke@1 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
duke@1 18 *
ohair@554 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@554 20 * or visit www.oracle.com if you need additional information or have any
ohair@554 21 * questions.
duke@1 22 */
duke@1 23
duke@1 24
duke@1 25 import com.sun.mirror.apt.*;
duke@1 26 import com.sun.mirror.declaration.*;
duke@1 27 import com.sun.mirror.type.*;
duke@1 28 import com.sun.mirror.util.*;
duke@1 29
duke@1 30 import java.util.Collection;
duke@1 31 import java.util.Set;
duke@1 32 import java.util.Map;
duke@1 33 import java.util.Arrays;
duke@1 34 import java.util.Collections;
duke@1 35
duke@1 36
duke@1 37 public class Scanner implements AnnotationProcessorFactory {
duke@1 38 static class ScannerProc implements AnnotationProcessor {
duke@1 39 AnnotationProcessorEnvironment env;
duke@1 40 ScannerProc(AnnotationProcessorEnvironment env) {
duke@1 41 this.env = env;
duke@1 42 }
duke@1 43
duke@1 44 static class CountingVisitor extends SimpleDeclarationVisitor {
duke@1 45 int count;
duke@1 46 CountingVisitor() {
duke@1 47 count = 0;
duke@1 48 }
duke@1 49
duke@1 50 public void visitDeclaration(Declaration d) {
duke@1 51 count++;
duke@1 52
duke@1 53 Collection<AnnotationMirror> ams = d.getAnnotationMirrors();
duke@1 54 if (ams == null)
duke@1 55 throw new RuntimeException("Declaration " + d +
duke@1 56 " not annotated with visit order.");
duke@1 57 else {
duke@1 58 if (ams.size() != 1)
duke@1 59 throw new RuntimeException("Declaration " + d +
duke@1 60 " has wrong number of declarations.");
duke@1 61 else {
duke@1 62 for(AnnotationMirror am: ams) {
duke@1 63 Map<AnnotationTypeElementDeclaration,AnnotationValue> elementValues = am.getElementValues();
duke@1 64 for(AnnotationTypeElementDeclaration atmd: elementValues.keySet()) {
duke@1 65 if (!atmd.getDeclaringType().toString().equals("VisitOrder"))
duke@1 66 throw new RuntimeException("Annotation " + atmd +
duke@1 67 " is the wrong type.");
duke@1 68 else {
duke@1 69 AnnotationValue av =
duke@1 70 elementValues.get(atmd);
duke@1 71 Integer value = (Integer) av.getValue();
duke@1 72 if (value.intValue() != count)
duke@1 73 throw new RuntimeException("Expected declaration " + d +
duke@1 74 " to be in position " + count +
duke@1 75 " instead of " + value.intValue());
duke@1 76
duke@1 77 System.out.println("Declaration " + d +
duke@1 78 ": visit order " + value.intValue());
duke@1 79 }
duke@1 80 }
duke@1 81
duke@1 82 }
duke@1 83 }
duke@1 84 }
duke@1 85
duke@1 86 }
duke@1 87 }
duke@1 88
duke@1 89 public void process() {
duke@1 90 for(TypeDeclaration td: env.getSpecifiedTypeDeclarations() ) {
duke@1 91 td.accept(DeclarationVisitors.getSourceOrderDeclarationScanner(new CountingVisitor(),
duke@1 92 DeclarationVisitors.NO_OP));
duke@1 93 }
duke@1 94 }
duke@1 95 }
duke@1 96
duke@1 97
duke@1 98 static Collection<String> supportedTypes;
duke@1 99 static {
duke@1 100 String types[] = {"*"};
duke@1 101 supportedTypes = Collections.unmodifiableCollection(Arrays.asList(types));
duke@1 102 }
duke@1 103
duke@1 104 static Collection<String> supportedOptions;
duke@1 105 static {
duke@1 106 String options[] = {""};
duke@1 107 supportedOptions = Collections.unmodifiableCollection(Arrays.asList(options));
duke@1 108 }
duke@1 109
duke@1 110 public Collection<String> supportedOptions() {
duke@1 111 return supportedOptions;
duke@1 112 }
duke@1 113
duke@1 114 public Collection<String> supportedAnnotationTypes() {
duke@1 115 return supportedTypes;
duke@1 116 }
duke@1 117
duke@1 118 public AnnotationProcessor getProcessorFor(Set<AnnotationTypeDeclaration> atds,
duke@1 119 AnnotationProcessorEnvironment env) {
duke@1 120 return new ScannerProc(env);
duke@1 121 }
duke@1 122
duke@1 123 }

mercurial