test/tools/apt/Scanners/Counter.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

     1 /*
     2  * Copyright (c) 2004, 2007, Oracle and/or its affiliates. All rights reserved.
     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
     4  *
     5  * This code is free software; you can redistribute it and/or modify it
     6  * under the terms of the GNU General Public License version 2 only, as
     7  * published by the Free Software Foundation.
     8  *
     9  * This code is distributed in the hope that it will be useful, but WITHOUT
    10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
    11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
    12  * version 2 for more details (a copy is included in the LICENSE file that
    13  * accompanied this code).
    14  *
    15  * You should have received a copy of the GNU General Public License version
    16  * 2 along with this work; if not, write to the Free Software Foundation,
    17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
    18  *
    19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
    20  * or visit www.oracle.com if you need additional information or have any
    21  * questions.
    22  */
    25 import com.sun.mirror.apt.*;
    26 import com.sun.mirror.declaration.*;
    27 import com.sun.mirror.type.*;
    28 import com.sun.mirror.util.*;
    30 import java.util.Collection;
    31 import java.util.Set;
    32 import java.util.Arrays;
    34 import static java.util.Collections.*;
    35 import static com.sun.mirror.util.DeclarationVisitors.*;
    37 /*
    38  * Used to verify counts from different kinds of declaration scanners.
    39  */
    40 public class Counter implements AnnotationProcessorFactory {
    41     static class CounterProc implements AnnotationProcessor {
    42         static class CountingVisitor extends SimpleDeclarationVisitor {
    43             int count;
    44             int count() {
    45                 return count;
    46             }
    48             CountingVisitor() {
    49                 count = 0;
    50             }
    52             public void visitDeclaration(Declaration d) {
    53                 count++;
    54                 System.out.println(d.getSimpleName());
    55             }
    56         }
    58         AnnotationProcessorEnvironment env;
    59         CounterProc(AnnotationProcessorEnvironment env) {
    60             this.env = env;
    61         }
    63         public void process() {
    64             for(TypeDeclaration td: env.getSpecifiedTypeDeclarations() ) {
    65                 CountingVisitor sourceOrder = new CountingVisitor();
    66                 CountingVisitor someOrder = new CountingVisitor();
    68                 System.out.println("Source Order Scanner");
    69                 td.accept(getSourceOrderDeclarationScanner(sourceOrder,
    70                                                            NO_OP));
    72                 System.out.println("\nSome Order Scanner");
    73                 td.accept(getDeclarationScanner(someOrder,
    74                                                 NO_OP));
    76                 if (sourceOrder.count() != someOrder.count() )
    77                     throw new RuntimeException("Counts from different scanners don't agree");
    78             }
    80         }
    81     }
    83     static Collection<String> supportedTypes;
    84     static {
    85         String types[] = {"*"};
    86         supportedTypes = unmodifiableCollection(Arrays.asList(types));
    87     }
    89     static Collection<String> supportedOptions;
    90     static {
    91         String options[] = {""};
    92         supportedOptions = unmodifiableCollection(Arrays.asList(options));
    93     }
    95     public Collection<String> supportedOptions() {
    96         return supportedOptions;
    97     }
    99     public Collection<String> supportedAnnotationTypes() {
   100         return supportedTypes;
   101     }
   103     public AnnotationProcessor getProcessorFor(Set<AnnotationTypeDeclaration> atds,
   104                                                AnnotationProcessorEnvironment env) {
   105         return new CounterProc(env);
   106     }
   108 }

mercurial