/*
 * Copyright (c) 2006, 2010, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

/*
 * @test
 * @bug 6406164
 * @summary Test that ElementFilter iterable methods behave properly.
 * @author  Joseph D. Darcy
 * @library /tools/javac/lib
 * @build JavacTestingAbstractProcessor
 * @compile TestIterables.java
 * @compile -processor TestIterables -proc:only Foo1.java
 * @compile Foo1.java
 * @compile -processor TestIterables Foo1 TestIterables.java
 */

import java.util.Set;
import java.util.HashSet;
import java.util.Arrays;
import java.util.Iterator;
import javax.annotation.processing.*;
import javax.lang.model.SourceVersion;
import static javax.lang.model.SourceVersion.*;
import javax.lang.model.element.*;
import javax.lang.model.util.*;
import static javax.tools.Diagnostic.Kind.*;

import static javax.lang.model.util.ElementFilter.*;

/**
 * Verify that the {@code ElementFilter} methods taking iterables
 * behave correctly by comparing ExpectedElementCounts with actual
 * results.
 */
@SupportedAnnotationTypes("ExpectedElementCounts")
@ExpectedElementCounts(methods=2)
public class TestIterables extends JavacTestingAbstractProcessor {
    public boolean process(Set<? extends TypeElement> annotations,
                           RoundEnvironment roundEnv) {
        if (!roundEnv.processingOver()) {
            boolean error = false;
            int topLevelCount = 0;

            for (TypeElement type : typesIn(roundEnv.getRootElements())) {
                topLevelCount++;
                ExpectedElementCounts elementCounts = type.getAnnotation(ExpectedElementCounts.class);
                if (elementCounts == null)
                    throw new RuntimeException("Missing ExpectedElementCounts annotation!");

                Iterable<? extends Element> irritable = type.getEnclosedElements();

                int constructorCount = getCount(constructorsIn(irritable));
                int fieldCount       = getCount(fieldsIn(irritable));
                int methodCount      = getCount(methodsIn(irritable));
                int typeCount        = getCount(typesIn(irritable));

                System.out.println("\nType: " + type.getQualifiedName());
                System.out.format("Element      Actual\tExpected%n");
                System.out.format("constructors %d\t\t%d%n", constructorCount, elementCounts.constructors());
                System.out.format("fields       %d\t\t%d%n", fieldCount,       elementCounts.fields());
                System.out.format("methods      %d\t\t%d%n", methodCount,      elementCounts.methods());
                System.out.format("types        %d\t\t%d%n", typeCount,        elementCounts.types());

                if ((constructorCount != elementCounts.constructors()) ||
                    (fieldCount       != elementCounts.fields()) ||
                    (methodCount      != elementCounts.methods()) ||
                    (typeCount        != elementCounts.types()) )
                    error = true;
            }

            if (topLevelCount == 0)
                error = true;

            if (error)
                throw new RuntimeException("A count mismatch has occured.");
        }

        return true;
    }

    private int getCount(Iterable<? extends Element> iter) {
        int count1 = 0;
        int count2 = 0;

        Iterator<? extends Element> iterator = iter.iterator();
        while (iterator.hasNext() ) {
            count1++;
            iterator.next();
        }

        iterator = iter.iterator();
        while (iterator.hasNext() ) {
            count2++;
            iterator.next();
        }

        if (count1 != count2)
            throw new RuntimeException("Inconsistent counts from iterator.");

        return count1;
    }
}