duke@1: /*
ohair@554:  * Copyright (c) 2004, 2007, Oracle and/or its affiliates. All rights reserved.
duke@1:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
duke@1:  *
duke@1:  * This code is free software; you can redistribute it and/or modify it
duke@1:  * under the terms of the GNU General Public License version 2 only, as
duke@1:  * published by the Free Software Foundation.
duke@1:  *
duke@1:  * This code is distributed in the hope that it will be useful, but WITHOUT
duke@1:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
duke@1:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
duke@1:  * version 2 for more details (a copy is included in the LICENSE file that
duke@1:  * accompanied this code).
duke@1:  *
duke@1:  * You should have received a copy of the GNU General Public License version
duke@1:  * 2 along with this work; if not, write to the Free Software Foundation,
duke@1:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
duke@1:  *
ohair@554:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@554:  * or visit www.oracle.com if you need additional information or have any
ohair@554:  * questions.
duke@1:  */
duke@1: 
duke@1: 
duke@1: import com.sun.mirror.apt.*;
duke@1: import com.sun.mirror.declaration.*;
duke@1: import com.sun.mirror.type.*;
duke@1: import com.sun.mirror.util.*;
duke@1: 
duke@1: import java.util.Collection;
duke@1: import java.util.Set;
duke@1: import java.util.Map;
duke@1: import java.util.Arrays;
duke@1: import java.util.Collections;
duke@1: 
duke@1: 
duke@1: public class Scanner implements AnnotationProcessorFactory {
duke@1:     static class ScannerProc implements AnnotationProcessor {
duke@1:         AnnotationProcessorEnvironment env;
duke@1:         ScannerProc(AnnotationProcessorEnvironment env) {
duke@1:             this.env = env;
duke@1:         }
duke@1: 
duke@1:         static class CountingVisitor extends SimpleDeclarationVisitor {
duke@1:             int count;
duke@1:             CountingVisitor() {
duke@1:                 count = 0;
duke@1:             }
duke@1: 
duke@1:             public void visitDeclaration(Declaration d) {
duke@1:                 count++;
duke@1: 
duke@1:                 Collection<AnnotationMirror> ams = d.getAnnotationMirrors();
duke@1:                 if (ams == null)
duke@1:                     throw new RuntimeException("Declaration " + d +
duke@1:                                                " not annotated with visit order.");
duke@1:                 else {
duke@1:                     if (ams.size() != 1)
duke@1:                         throw new RuntimeException("Declaration " + d +
duke@1:                                                    " has wrong number of declarations.");
duke@1:                     else {
duke@1:                         for(AnnotationMirror am: ams) {
duke@1:                             Map<AnnotationTypeElementDeclaration,AnnotationValue> elementValues = am.getElementValues();
duke@1:                             for(AnnotationTypeElementDeclaration atmd: elementValues.keySet()) {
duke@1:                                 if (!atmd.getDeclaringType().toString().equals("VisitOrder"))
duke@1:                                     throw new RuntimeException("Annotation " + atmd +
duke@1:                                                                " is the wrong type.");
duke@1:                                 else {
duke@1:                                     AnnotationValue av =
duke@1:                                         elementValues.get(atmd);
duke@1:                                     Integer value = (Integer) av.getValue();
duke@1:                                     if (value.intValue() != count)
duke@1:                                         throw new RuntimeException("Expected declaration " + d +
duke@1:                                                                    " to be in position " + count +
duke@1:                                                                    " instead of " + value.intValue());
duke@1: 
duke@1:                                     System.out.println("Declaration " + d +
duke@1:                                                        ": visit order " + value.intValue());
duke@1:                                 }
duke@1:                             }
duke@1: 
duke@1:                         }
duke@1:                     }
duke@1:                 }
duke@1: 
duke@1:             }
duke@1:         }
duke@1: 
duke@1:         public void process() {
duke@1:             for(TypeDeclaration td: env.getSpecifiedTypeDeclarations() ) {
duke@1:                 td.accept(DeclarationVisitors.getSourceOrderDeclarationScanner(new CountingVisitor(),
duke@1:                                                                                DeclarationVisitors.NO_OP));
duke@1:             }
duke@1:         }
duke@1:     }
duke@1: 
duke@1: 
duke@1:     static Collection<String> supportedTypes;
duke@1:     static {
duke@1:         String types[] = {"*"};
duke@1:         supportedTypes = Collections.unmodifiableCollection(Arrays.asList(types));
duke@1:     }
duke@1: 
duke@1:     static Collection<String> supportedOptions;
duke@1:     static {
duke@1:         String options[] = {""};
duke@1:         supportedOptions = Collections.unmodifiableCollection(Arrays.asList(options));
duke@1:     }
duke@1: 
duke@1:     public Collection<String> supportedOptions() {
duke@1:         return supportedOptions;
duke@1:     }
duke@1: 
duke@1:     public Collection<String> supportedAnnotationTypes() {
duke@1:         return supportedTypes;
duke@1:     }
duke@1: 
duke@1:     public AnnotationProcessor getProcessorFor(Set<AnnotationTypeDeclaration> atds,
duke@1:                                         AnnotationProcessorEnvironment env) {
duke@1:         return new ScannerProc(env);
duke@1:     }
duke@1: 
duke@1: }