langtools/test/tools/apt/Compile/ClassDeclApf2.java
changeset 10 06bc494ca11e
child 5520 86e4b9a9da40
--- /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);
+            }
+        }
+    }
+}