2007-12-01 00:00:00 +00:00

105 lines
3.6 KiB
Java

/*
* 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();
}
}
}
}