diff -r 000000000000 -r 9a66ca7c79fa test/tools/apt/Scanners/Counter.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/test/tools/apt/Scanners/Counter.java Sat Dec 01 00:00:00 2007 +0000 @@ -0,0 +1,108 @@ +/* + * Copyright 2004-2007 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. + */ + + +import com.sun.mirror.apt.*; +import com.sun.mirror.declaration.*; +import com.sun.mirror.type.*; +import com.sun.mirror.util.*; + +import java.util.Collection; +import java.util.Set; +import java.util.Arrays; + +import static java.util.Collections.*; +import static com.sun.mirror.util.DeclarationVisitors.*; + +/* + * Used to verify counts from different kinds of declaration scanners. + */ +public class Counter implements AnnotationProcessorFactory { + static class CounterProc implements AnnotationProcessor { + static class CountingVisitor extends SimpleDeclarationVisitor { + int count; + int count() { + return count; + } + + CountingVisitor() { + count = 0; + } + + public void visitDeclaration(Declaration d) { + count++; + System.out.println(d.getSimpleName()); + } + } + + AnnotationProcessorEnvironment env; + CounterProc(AnnotationProcessorEnvironment env) { + this.env = env; + } + + public void process() { + for(TypeDeclaration td: env.getSpecifiedTypeDeclarations() ) { + CountingVisitor sourceOrder = new CountingVisitor(); + CountingVisitor someOrder = new CountingVisitor(); + + System.out.println("Source Order Scanner"); + td.accept(getSourceOrderDeclarationScanner(sourceOrder, + NO_OP)); + + System.out.println("\nSome Order Scanner"); + td.accept(getDeclarationScanner(someOrder, + NO_OP)); + + if (sourceOrder.count() != someOrder.count() ) + throw new RuntimeException("Counts from different scanners don't agree"); + } + + } + } + + static Collection supportedTypes; + static { + String types[] = {"*"}; + supportedTypes = unmodifiableCollection(Arrays.asList(types)); + } + + static Collection supportedOptions; + static { + String options[] = {""}; + supportedOptions = unmodifiableCollection(Arrays.asList(options)); + } + + public Collection supportedOptions() { + return supportedOptions; + } + + public Collection supportedAnnotationTypes() { + return supportedTypes; + } + + public AnnotationProcessor getProcessorFor(Set atds, + AnnotationProcessorEnvironment env) { + return new CounterProc(env); + } + +}