Thu, 15 Dec 2011 15:47:47 -0800
Added tag jdk8-b16 for changeset ec2c0973cc31
duke@1 | 1 | /* |
ohair@554 | 2 | * Copyright (c) 2004, 2007, Oracle and/or its affiliates. All rights reserved. |
duke@1 | 3 | * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
duke@1 | 4 | * |
duke@1 | 5 | * This code is free software; you can redistribute it and/or modify it |
duke@1 | 6 | * under the terms of the GNU General Public License version 2 only, as |
duke@1 | 7 | * published by the Free Software Foundation. |
duke@1 | 8 | * |
duke@1 | 9 | * This code is distributed in the hope that it will be useful, but WITHOUT |
duke@1 | 10 | * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
duke@1 | 11 | * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
duke@1 | 12 | * version 2 for more details (a copy is included in the LICENSE file that |
duke@1 | 13 | * accompanied this code). |
duke@1 | 14 | * |
duke@1 | 15 | * You should have received a copy of the GNU General Public License version |
duke@1 | 16 | * 2 along with this work; if not, write to the Free Software Foundation, |
duke@1 | 17 | * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
duke@1 | 18 | * |
ohair@554 | 19 | * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
ohair@554 | 20 | * or visit www.oracle.com if you need additional information or have any |
ohair@554 | 21 | * questions. |
duke@1 | 22 | */ |
duke@1 | 23 | |
duke@1 | 24 | |
duke@1 | 25 | import com.sun.mirror.apt.*; |
duke@1 | 26 | import com.sun.mirror.declaration.*; |
duke@1 | 27 | import com.sun.mirror.type.*; |
duke@1 | 28 | import com.sun.mirror.util.*; |
duke@1 | 29 | |
duke@1 | 30 | import java.util.Collection; |
duke@1 | 31 | import java.util.Set; |
duke@1 | 32 | import java.util.Arrays; |
duke@1 | 33 | |
duke@1 | 34 | import java.io.*; |
duke@1 | 35 | |
duke@1 | 36 | import static java.util.Collections.*; |
duke@1 | 37 | |
duke@1 | 38 | /* |
duke@1 | 39 | * This class is used to test the ability to store static state across |
duke@1 | 40 | * apt rounds. |
duke@1 | 41 | */ |
duke@1 | 42 | public class ClassDeclApf implements AnnotationProcessorFactory { |
duke@1 | 43 | static int round = -1; |
duke@1 | 44 | |
duke@1 | 45 | // Process any set of annotations |
duke@1 | 46 | private static final Collection<String> supportedAnnotations |
duke@1 | 47 | = unmodifiableCollection(Arrays.asList("*")); |
duke@1 | 48 | |
duke@1 | 49 | // No supported options |
duke@1 | 50 | private static final Collection<String> supportedOptions = emptySet(); |
duke@1 | 51 | |
duke@1 | 52 | public Collection<String> supportedAnnotationTypes() { |
duke@1 | 53 | return supportedAnnotations; |
duke@1 | 54 | } |
duke@1 | 55 | |
duke@1 | 56 | public Collection<String> supportedOptions() { |
duke@1 | 57 | return supportedOptions; |
duke@1 | 58 | } |
duke@1 | 59 | |
duke@1 | 60 | public AnnotationProcessor getProcessorFor(Set<AnnotationTypeDeclaration> atds, |
duke@1 | 61 | AnnotationProcessorEnvironment env) { |
duke@1 | 62 | return new ClassDeclAp(env); |
duke@1 | 63 | } |
duke@1 | 64 | |
duke@1 | 65 | private static class ClassDeclAp implements AnnotationProcessor { |
duke@1 | 66 | private final AnnotationProcessorEnvironment env; |
duke@1 | 67 | ClassDeclAp(AnnotationProcessorEnvironment env) { |
duke@1 | 68 | this.env = env; |
duke@1 | 69 | } |
duke@1 | 70 | |
duke@1 | 71 | // Simple inefficient drain |
duke@1 | 72 | void drain(InputStream is, OutputStream os) { |
duke@1 | 73 | try { |
duke@1 | 74 | while (is.available() > 0 ) |
duke@1 | 75 | os.write(is.read()); |
duke@1 | 76 | } catch (java.io.IOException ioe) { |
duke@1 | 77 | throw new RuntimeException(ioe); |
duke@1 | 78 | } |
duke@1 | 79 | |
duke@1 | 80 | } |
duke@1 | 81 | |
duke@1 | 82 | public void process() { |
duke@1 | 83 | int size = env.getSpecifiedTypeDeclarations().size(); |
duke@1 | 84 | |
duke@1 | 85 | try { |
duke@1 | 86 | round++; |
duke@1 | 87 | switch (size) { |
duke@1 | 88 | case 0: |
duke@1 | 89 | if (round == 0) { |
duke@1 | 90 | drain(new FileInputStream("./tmp/classes/Round1Class.class"), |
duke@1 | 91 | env.getFiler().createClassFile("Round1Class")); |
duke@1 | 92 | } else |
duke@1 | 93 | throw new RuntimeException("Got " + size + " decl's in round " + round); |
duke@1 | 94 | break; |
duke@1 | 95 | |
duke@1 | 96 | case 1: |
duke@1 | 97 | if (round == 1) { |
duke@1 | 98 | drain(new FileInputStream("./tmp/classes/AhOneClass.class"), |
duke@1 | 99 | env.getFiler().createClassFile("AhOneClass")); |
duke@1 | 100 | |
duke@1 | 101 | drain(new FileInputStream("./tmp/classes/AndAhTwoClass.class"), |
duke@1 | 102 | env.getFiler().createClassFile("AndAhTwoClass")); |
duke@1 | 103 | } else |
duke@1 | 104 | throw new RuntimeException("Got " + size + " decl's in round " + round); |
duke@1 | 105 | break; |
duke@1 | 106 | case 2: |
duke@1 | 107 | if (round != 2) { |
duke@1 | 108 | throw new RuntimeException("Got " + size + " decl's in round " + round); |
duke@1 | 109 | } |
duke@1 | 110 | break; |
duke@1 | 111 | } |
duke@1 | 112 | |
duke@1 | 113 | } catch (java.io.IOException ioe) { |
duke@1 | 114 | throw new RuntimeException(); |
duke@1 | 115 | } |
duke@1 | 116 | |
duke@1 | 117 | } |
duke@1 | 118 | |
duke@1 | 119 | } |
duke@1 | 120 | } |