test/tools/javac/lib/JavacTestingAbstractProcessor.java

changeset 699
d2aaaec153e8
child 932
edf03ca74991
     1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/test/tools/javac/lib/JavacTestingAbstractProcessor.java	Wed Sep 29 23:27:57 2010 -0700
     1.3 @@ -0,0 +1,100 @@
     1.4 +/*
     1.5 + * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
     1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
     1.7 + *
     1.8 + * This code is free software; you can redistribute it and/or modify it
     1.9 + * under the terms of the GNU General Public License version 2 only, as
    1.10 + * published by the Free Software Foundation.
    1.11 + *
    1.12 + * This code is distributed in the hope that it will be useful, but WITHOUT
    1.13 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
    1.14 + * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
    1.15 + * version 2 for more details (a copy is included in the LICENSE file that
    1.16 + * accompanied this code).
    1.17 + *
    1.18 + * You should have received a copy of the GNU General Public License version
    1.19 + * 2 along with this work; if not, write to the Free Software Foundation,
    1.20 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
    1.21 + *
    1.22 + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
    1.23 + * or visit www.oracle.com if you need additional information or have any
    1.24 + * questions.
    1.25 + */
    1.26 +
    1.27 +import java.util.*;
    1.28 +import javax.annotation.processing.*;
    1.29 +import javax.lang.model.SourceVersion;
    1.30 +import static javax.lang.model.SourceVersion.*;
    1.31 +import javax.lang.model.element.*;
    1.32 +import javax.lang.model.util.*;
    1.33 +
    1.34 +/**
    1.35 + * An abstract annotation processor tailored to javac regression testing.
    1.36 + */
    1.37 +public abstract class JavacTestingAbstractProcessor extends AbstractProcessor {
    1.38 +    private static final Set<String> allAnnotations;
    1.39 +
    1.40 +    static {
    1.41 +        Set<String> tmp = new HashSet<>();
    1.42 +        tmp.add("*");
    1.43 +        allAnnotations = Collections.unmodifiableSet(tmp);
    1.44 +    }
    1.45 +
    1.46 +    protected Elements eltUtils;
    1.47 +    protected Elements elements;
    1.48 +    protected Types    typeUtils;
    1.49 +    protected Types    types;
    1.50 +    protected Filer    filer;
    1.51 +    protected Messager messager;
    1.52 +    protected Map<String, String> options;
    1.53 +
    1.54 +    /**
    1.55 +     * Constructor for subclasses to call.
    1.56 +     */
    1.57 +    protected JavacTestingAbstractProcessor() {
    1.58 +        super();
    1.59 +    }
    1.60 +
    1.61 +    /**
    1.62 +     * Return the latest source version. Unless this method is
    1.63 +     * overridden, an {@code IllegalStateException} will be thrown if a
    1.64 +     * subclass has a {@code SupportedSourceVersion} annotation.
    1.65 +     */
    1.66 +    @Override
    1.67 +    public SourceVersion getSupportedSourceVersion() {
    1.68 +        SupportedSourceVersion ssv = this.getClass().getAnnotation(SupportedSourceVersion.class);
    1.69 +        if (ssv != null)
    1.70 +            throw new IllegalStateException("SupportedSourceVersion annotation not supported here.");
    1.71 +
    1.72 +        return SourceVersion.latest();
    1.73 +    }
    1.74 +
    1.75 +    /**
    1.76 +     * If the processor class is annotated with {@link
    1.77 +     * SupportedAnnotationTypes}, return an unmodifiable set with the
    1.78 +     * same set of strings as the annotation.  If the class is not so
    1.79 +     * annotated, a one-element set containing {@code "*"} is returned
    1.80 +     * to indicate all annotations are processed.
    1.81 +     *
    1.82 +     * @return the names of the annotation types supported by this
    1.83 +     * processor, or an empty set if none
    1.84 +     */
    1.85 +    @Override
    1.86 +    public Set<String> getSupportedAnnotationTypes() {
    1.87 +        SupportedAnnotationTypes sat = this.getClass().getAnnotation(SupportedAnnotationTypes.class);
    1.88 +        if (sat != null)
    1.89 +            return super.getSupportedAnnotationTypes();
    1.90 +        else
    1.91 +            return allAnnotations;
    1.92 +    }
    1.93 +
    1.94 +    @Override
    1.95 +    public void init(ProcessingEnvironment processingEnv) {
    1.96 +        super.init(processingEnv);
    1.97 +        elements = eltUtils  = processingEnv.getElementUtils();
    1.98 +        types = typeUtils = processingEnv.getTypeUtils();
    1.99 +        filer     = processingEnv.getFiler();
   1.100 +        messager  = processingEnv.getMessager();
   1.101 +        options   = processingEnv.getOptions();
   1.102 +    }
   1.103 +}

mercurial