0N/A/*
553N/A * Copyright (c) 2004, 2007, Oracle and/or its affiliates. All rights reserved.
0N/A * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
0N/A *
0N/A * This code is free software; you can redistribute it and/or modify it
0N/A * under the terms of the GNU General Public License version 2 only, as
0N/A * published by the Free Software Foundation.
0N/A *
0N/A * This code is distributed in the hope that it will be useful, but WITHOUT
0N/A * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
0N/A * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
0N/A * version 2 for more details (a copy is included in the LICENSE file that
0N/A * accompanied this code).
0N/A *
0N/A * You should have received a copy of the GNU General Public License version
0N/A * 2 along with this work; if not, write to the Free Software Foundation,
0N/A * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
0N/A *
553N/A * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
553N/A * or visit www.oracle.com if you need additional information or have any
553N/A * questions.
0N/A */
0N/A
0N/A
0N/Aimport com.sun.mirror.apt.*;
0N/Aimport com.sun.mirror.declaration.*;
0N/Aimport com.sun.mirror.type.*;
0N/Aimport com.sun.mirror.util.*;
0N/A
0N/Aimport java.util.Collection;
0N/Aimport java.util.Set;
0N/Aimport java.util.Arrays;
0N/A
0N/Aimport java.io.*;
0N/A
0N/Aimport static java.util.Collections.*;
0N/A
0N/A/*
0N/A * This class is used to test the the interaction of -XclassesAsDecls
0N/A * with command line options handling.
0N/A */
0N/Apublic class ClassDeclApf2 implements AnnotationProcessorFactory {
0N/A static int round = -1;
0N/A
0N/A // Process any set of annotations
0N/A private static final Collection<String> supportedAnnotations
0N/A = unmodifiableCollection(Arrays.asList("*"));
0N/A
0N/A // No supported options
0N/A private static final Collection<String> supportedOptions = emptySet();
0N/A
0N/A public Collection<String> supportedAnnotationTypes() {
0N/A return supportedAnnotations;
0N/A }
0N/A
0N/A public Collection<String> supportedOptions() {
0N/A return supportedOptions;
0N/A }
0N/A
0N/A public AnnotationProcessor getProcessorFor(Set<AnnotationTypeDeclaration> atds,
0N/A AnnotationProcessorEnvironment env) {
0N/A return new ClassDeclAp(env);
0N/A }
0N/A
0N/A private static class ClassDeclAp implements AnnotationProcessor {
0N/A private final AnnotationProcessorEnvironment env;
0N/A ClassDeclAp(AnnotationProcessorEnvironment env) {
0N/A this.env = env;
0N/A }
0N/A
0N/A // Simple inefficient drain
0N/A void drain(InputStream is, OutputStream os) {
0N/A try {
0N/A while (is.available() > 0 )
0N/A os.write(is.read());
0N/A } catch (java.io.IOException ioe) {
0N/A throw new RuntimeException(ioe);
0N/A }
0N/A }
0N/A
0N/A public void process() {
0N/A int size = env.getSpecifiedTypeDeclarations().size();
0N/A Filer f = env.getFiler();
0N/A
0N/A try {
0N/A round++;
0N/A switch (size) {
0N/A case 3:
0N/A if (round == 0) {
0N/A drain(new FileInputStream("./tmp/classes/Round1Class.class"),
0N/A f.createClassFile("Round1Class"));
0N/A } else
0N/A throw new RuntimeException("Got " + size + " decl's in round " + round);
0N/A break;
0N/A
0N/A case 1:
0N/A if (round == 1) {
0N/A f.createSourceFile("AhOne").println("public class AhOne {}");
0N/A System.out.println("Before writing AndAhTwoClass");
0N/A drain(new FileInputStream("./tmp/classes/AndAhTwoClass.class"),
0N/A f.createClassFile("AndAhTwoClass"));
0N/A System.out.println("After writing AndAhTwoClass");
0N/A } else
0N/A throw new RuntimeException("Got " + size + " decl's in round " + round);
0N/A break;
0N/A
0N/A case 2:
0N/A if (round != 2) {
0N/A throw new RuntimeException("Got " + size + " decl's in round " + round);
0N/A }
0N/A break;
0N/A default:
0N/A throw new RuntimeException("Unexpected number of declarations:" + size +
0N/A "\n Specified:" + env.getSpecifiedTypeDeclarations() +
0N/A "\n Included:" + env.getTypeDeclarations() );
0N/A }
0N/A
0N/A } catch (java.io.IOException ioe) {
0N/A throw new RuntimeException(ioe);
0N/A }
0N/A }
0N/A }
0N/A}