--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/langtools/test/tools/apt/Compile/ClassDeclApf2.java Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,123 @@
+/*
+ * 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 java.io.*;
+
+import static java.util.Collections.*;
+
+/*
+ * This class is used to test the the interaction of -XclassesAsDecls
+ * with command line options handling.
+ */
+public class ClassDeclApf2 implements AnnotationProcessorFactory {
+ static int round = -1;
+
+ // Process any set of annotations
+ private static final Collection<String> supportedAnnotations
+ = unmodifiableCollection(Arrays.asList("*"));
+
+ // No supported options
+ private static final Collection<String> supportedOptions = emptySet();
+
+ public Collection<String> supportedAnnotationTypes() {
+ return supportedAnnotations;
+ }
+
+ public Collection<String> supportedOptions() {
+ return supportedOptions;
+ }
+
+ public AnnotationProcessor getProcessorFor(Set<AnnotationTypeDeclaration> atds,
+ AnnotationProcessorEnvironment env) {
+ return new ClassDeclAp(env);
+ }
+
+ private static class ClassDeclAp implements AnnotationProcessor {
+ private final AnnotationProcessorEnvironment env;
+ ClassDeclAp(AnnotationProcessorEnvironment env) {
+ this.env = env;
+ }
+
+ // Simple inefficient drain
+ void drain(InputStream is, OutputStream os) {
+ try {
+ while (is.available() > 0 )
+ os.write(is.read());
+ } catch (java.io.IOException ioe) {
+ throw new RuntimeException(ioe);
+ }
+ }
+
+ public void process() {
+ int size = env.getSpecifiedTypeDeclarations().size();
+ Filer f = env.getFiler();
+
+ try {
+ round++;
+ switch (size) {
+ case 3:
+ if (round == 0) {
+ drain(new FileInputStream("./tmp/classes/Round1Class.class"),
+ f.createClassFile("Round1Class"));
+ } else
+ throw new RuntimeException("Got " + size + " decl's in round " + round);
+ break;
+
+ case 1:
+ if (round == 1) {
+ f.createSourceFile("AhOne").println("public class AhOne {}");
+ System.out.println("Before writing AndAhTwoClass");
+ drain(new FileInputStream("./tmp/classes/AndAhTwoClass.class"),
+ f.createClassFile("AndAhTwoClass"));
+ System.out.println("After writing AndAhTwoClass");
+ } else
+ throw new RuntimeException("Got " + size + " decl's in round " + round);
+ break;
+
+ case 2:
+ if (round != 2) {
+ throw new RuntimeException("Got " + size + " decl's in round " + round);
+ }
+ break;
+ default:
+ throw new RuntimeException("Unexpected number of declarations:" + size +
+ "\n Specified:" + env.getSpecifiedTypeDeclarations() +
+ "\n Included:" + env.getTypeDeclarations() );
+ }
+
+ } catch (java.io.IOException ioe) {
+ throw new RuntimeException(ioe);
+ }
+ }
+ }
+}