test/tools/apt/Compile/ClassDeclApf2.java

Thu, 25 Aug 2011 17:18:25 -0700

author
schien
date
Thu, 25 Aug 2011 17:18:25 -0700
changeset 1067
f497fac86cf9
parent 554
9d9f26857129
permissions
-rw-r--r--

Added tag jdk8-b02 for changeset b3c059de2a61

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 the interaction of -XclassesAsDecls
duke@1 40 * with command line options handling.
duke@1 41 */
duke@1 42 public class ClassDeclApf2 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 public void process() {
duke@1 82 int size = env.getSpecifiedTypeDeclarations().size();
duke@1 83 Filer f = env.getFiler();
duke@1 84
duke@1 85 try {
duke@1 86 round++;
duke@1 87 switch (size) {
duke@1 88 case 3:
duke@1 89 if (round == 0) {
duke@1 90 drain(new FileInputStream("./tmp/classes/Round1Class.class"),
duke@1 91 f.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 f.createSourceFile("AhOne").println("public class AhOne {}");
duke@1 99 System.out.println("Before writing AndAhTwoClass");
duke@1 100 drain(new FileInputStream("./tmp/classes/AndAhTwoClass.class"),
duke@1 101 f.createClassFile("AndAhTwoClass"));
duke@1 102 System.out.println("After writing 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
duke@1 107 case 2:
duke@1 108 if (round != 2) {
duke@1 109 throw new RuntimeException("Got " + size + " decl's in round " + round);
duke@1 110 }
duke@1 111 break;
duke@1 112 default:
duke@1 113 throw new RuntimeException("Unexpected number of declarations:" + size +
duke@1 114 "\n Specified:" + env.getSpecifiedTypeDeclarations() +
duke@1 115 "\n Included:" + env.getTypeDeclarations() );
duke@1 116 }
duke@1 117
duke@1 118 } catch (java.io.IOException ioe) {
duke@1 119 throw new RuntimeException(ioe);
duke@1 120 }
duke@1 121 }
duke@1 122 }
duke@1 123 }

mercurial