/*

 * 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<String> supportedTypes;

    static {

        String types[] = {"*"};

        supportedTypes = unmodifiableCollection(Arrays.asList(types));

    }

    static Collection<String> supportedOptions;

    static {

        String options[] = {""};

        supportedOptions = unmodifiableCollection(Arrays.asList(options));

    }

    public Collection<String> supportedOptions() {

        return supportedOptions;

    }

    public Collection<String> supportedAnnotationTypes() {

        return supportedTypes;

    }

    public AnnotationProcessor getProcessorFor(Set<AnnotationTypeDeclaration> atds,

                                               AnnotationProcessorEnvironment env) {

        return new CounterProc(env);

    }

}