langtools/test/tools/apt/Compile/StaticApf.java
changeset 10 06bc494ca11e
child 5520 86e4b9a9da40
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/langtools/test/tools/apt/Compile/StaticApf.java	Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,104 @@
+/*
+ * 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.*;
+
+/*
+ * This class is used to test the ability to store static state across
+ * apt rounds.
+ */
+public class StaticApf 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 StaticAp(env);
+    }
+
+    private static class StaticAp implements AnnotationProcessor {
+        private final AnnotationProcessorEnvironment env;
+        StaticAp(AnnotationProcessorEnvironment env) {
+            this.env = env;
+        }
+
+        public void process() {
+            int size = env.getSpecifiedTypeDeclarations().size();
+
+            try {
+                round++;
+                switch (size) {
+                case 0:
+                    if (round == 0) {
+                        env.getFiler().createSourceFile("Round1").print("class Round1 {}");
+                    } else
+                        throw new RuntimeException("Got " + size + " decl's in round " + round);
+                    break;
+
+                case 1:
+                    if (round == 1) {
+                        env.getFiler().createSourceFile("AhOne").print("class AhOne {}");
+                        env.getFiler().createSourceFile("AndAhTwo").print("class AndAhTwo {}");
+                        env.getFiler().createClassFile("Foo");
+                    } 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;
+                }
+
+            } catch (java.io.IOException ioe) {
+                    throw new RuntimeException();
+                }
+
+            }
+
+    }
+}