src/share/classes/com/sun/tools/javac/processing/PrintingProcessor.java

Thu, 02 Oct 2008 19:58:40 -0700

author
xdono
date
Thu, 02 Oct 2008 19:58:40 -0700
changeset 117
24a47c3062fe
parent 104
5e89c4ca637c
child 224
dab918a1c907
permissions
-rw-r--r--

6754988: Update copyright year
Summary: Update for files that have been modified starting July 2008
Reviewed-by: ohair, tbell

     1 /*
     2  * Copyright 2005-2008 Sun Microsystems, Inc.  All Rights Reserved.
     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
     4  *
     5  * This code is free software; you can redistribute it and/or modify it
     6  * under the terms of the GNU General Public License version 2 only, as
     7  * published by the Free Software Foundation.  Sun designates this
     8  * particular file as subject to the "Classpath" exception as provided
     9  * by Sun in the LICENSE file that accompanied this code.
    10  *
    11  * This code is distributed in the hope that it will be useful, but WITHOUT
    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
    14  * version 2 for more details (a copy is included in the LICENSE file that
    15  * accompanied this code).
    16  *
    17  * You should have received a copy of the GNU General Public License version
    18  * 2 along with this work; if not, write to the Free Software Foundation,
    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
    20  *
    21  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
    22  * CA 95054 USA or visit www.sun.com if you need additional information or
    23  * have any questions.
    24  */
    26 package com.sun.tools.javac.processing;
    28 import javax.annotation.processing.*;
    29 import javax.lang.model.*;
    30 import javax.lang.model.element.*;
    31 import static javax.lang.model.element.ElementKind.*;
    32 import static javax.lang.model.element.NestingKind.*;
    33 import javax.lang.model.type.*;
    34 import javax.lang.model.util.*;
    36 import java.io.PrintWriter;
    37 import java.io.Writer;
    38 import java.util.*;
    40 /**
    41  * A processor which prints out elements.  Used to implement the
    42  * -Xprint option; the included visitor class is used to implement
    43  * Elements.printElements.
    44  *
    45  * <p><b>This is NOT part of any API supported by Sun Microsystems.
    46  * If you write code that depends on this, you do so at your own risk.
    47  * This code and its internal interfaces are subject to change or
    48  * deletion without notice.</b>
    49  */
    50 @SupportedAnnotationTypes("*")
    51 @SupportedSourceVersion(SourceVersion.RELEASE_6)
    52 public class PrintingProcessor extends AbstractProcessor {
    53     PrintWriter writer;
    55     public PrintingProcessor() {
    56         super();
    57         writer = new PrintWriter(System.out);
    58     }
    60     public void setWriter(Writer w) {
    61         writer = new PrintWriter(w);
    62     }
    64     @Override
    65     public boolean process(Set<? extends TypeElement> tes,
    66                            RoundEnvironment renv) {
    68         for(Element element : renv.getRootElements()) {
    69             print(element);
    70         }
    72         // Just print the elements, nothing more to do.
    73         return true;
    74     }
    76     void print(Element element) {
    77         new PrintingElementVisitor(writer, processingEnv.getElementUtils()).
    78             visit(element).flush();
    79     }
    81     /**
    82      * Used for the -Xprint option and called by Elements.printElements
    83      */
    84     public static class PrintingElementVisitor
    85         extends SimpleElementVisitor6<PrintingElementVisitor, Boolean> {
    86         int indentation; // Indentation level;
    87         final PrintWriter writer;
    88         final Elements elementUtils;
    90         public PrintingElementVisitor(Writer w, Elements elementUtils) {
    91             super();
    92             this.writer = new PrintWriter(w);
    93             this.elementUtils = elementUtils;
    94             indentation = 0;
    95         }
    97         @Override
    98         protected PrintingElementVisitor defaultAction(Element e, Boolean newLine) {
    99             if (newLine != null && newLine)
   100                 writer.println();
   101             printDocComment(e);
   102             printModifiers(e);
   103             return this;
   104         }
   106         @Override
   107         public PrintingElementVisitor visitExecutable(ExecutableElement e, Boolean p) {
   108             ElementKind kind = e.getKind();
   110             if (kind != STATIC_INIT &&
   111                 kind != INSTANCE_INIT) {
   112                 Element enclosing = e.getEnclosingElement();
   114                 // Don't print out the constructor of an anonymous class
   115                 if (kind == CONSTRUCTOR &&
   116                     enclosing != null &&
   117                     NestingKind.ANONYMOUS ==
   118                     // Use an anonymous class to determine anonymity!
   119                     (new SimpleElementVisitor6<NestingKind, Void>() {
   120                         @Override
   121                         public NestingKind visitType(TypeElement e, Void p) {
   122                             return e.getNestingKind();
   123                         }
   124                     }).visit(enclosing))
   125                     return this;
   127                 defaultAction(e, true);
   128                 printFormalTypeParameters(e);
   130                 switch(kind) {
   131                     case CONSTRUCTOR:
   132                     // Print out simple name of the class
   133                     writer.print(e.getEnclosingElement().getSimpleName());
   134                     break;
   136                     case METHOD:
   137                     writer.print(e.getReturnType().toString());
   138                     writer.print(" ");
   139                     writer.print(e.getSimpleName().toString());
   140                     break;
   141                 }
   143                 writer.print("(");
   144                 printParameters(e);
   145                 writer.print(")");
   146                 AnnotationValue defaultValue = e.getDefaultValue();
   147                 if (defaultValue != null)
   148                     writer.print(" default " + defaultValue);
   150                 printThrows(e);
   151                 writer.println(";");
   152             }
   153             return this;
   154         }
   157         @Override
   158         public PrintingElementVisitor visitType(TypeElement e, Boolean p) {
   159             ElementKind kind = e.getKind();
   160             NestingKind nestingKind = e.getNestingKind();
   162             if (NestingKind.ANONYMOUS == nestingKind) {
   163                 // Print out an anonymous class in the style of a
   164                 // class instance creation expression rather than a
   165                 // class declaration.
   166                 writer.print("new ");
   168                 // If the anonymous class implements an interface
   169                 // print that name, otherwise print the superclass.
   170                 List<? extends TypeMirror> interfaces = e.getInterfaces();
   171                 if (!interfaces.isEmpty())
   172                     writer.print(interfaces.get(0));
   173                 else
   174                     writer.print(e.getSuperclass());
   176                 writer.print("(");
   177                 // Anonymous classes that implement an interface can't
   178                 // have any constructor arguments.
   179                 if (interfaces.isEmpty()) {
   180                     // Print out the parameter list from the sole
   181                     // constructor.  For now, don't try to elide any
   182                     // synthetic parameters by determining if the
   183                     // anonymous class is in a static context, etc.
   184                     List<? extends ExecutableElement> constructors =
   185                         ElementFilter.constructorsIn(e.getEnclosedElements());
   187                     if (!constructors.isEmpty())
   188                         printParameters(constructors.get(0));
   189                 }
   190                 writer.print(")");
   191             } else {
   192                 if (nestingKind == TOP_LEVEL) {
   193                     PackageElement pkg = elementUtils.getPackageOf(e);
   194                     if (!pkg.isUnnamed())
   195                         writer.print("package " + pkg.getQualifiedName() + ";\n");
   196                 }
   198                 defaultAction(e, true);
   200                 switch(kind) {
   201                 case ANNOTATION_TYPE:
   202                     writer.print("@interface");
   203                     break;
   204                 default:
   205                     writer.print(kind.toString().toLowerCase());
   206                 }
   207                 writer.print(" ");
   208                 writer.print(e.getSimpleName());
   210                 printFormalTypeParameters(e);
   212                 // Print superclass information if informative
   213                 if (kind == CLASS) {
   214                     TypeMirror supertype = e.getSuperclass();
   215                     if (supertype.getKind() != TypeKind.NONE) {
   216                         TypeElement e2 = (TypeElement)
   217                             ((DeclaredType) supertype).asElement();
   218                         if (e2.getSuperclass().getKind() != TypeKind.NONE)
   219                             writer.print(" extends " + supertype);
   220                     }
   221                 }
   223                 printInterfaces(e);
   224             }
   225             writer.println(" {");
   226             indentation++;
   228             if (kind == ENUM) {
   229                 List<Element> enclosedElements =
   230                     new ArrayList<Element>(e.getEnclosedElements());
   231                 List<Element> enumConstants = new ArrayList<Element>();
   232                 for(Element element : enclosedElements) {
   233                     if (element.getKind() == ENUM_CONSTANT)
   234                         enumConstants.add(element);
   235                 }
   237                 int i;
   238                 for(i = 0; i < enumConstants.size()-1; i++) {
   239                     this.visit(enumConstants.get(i), true);
   240                     writer.print(",");
   241                 }
   242                 if (i >= 0 ) {
   243                     this.visit(enumConstants.get(i), true);
   244                     writer.print(";");
   245                 }
   247                 enclosedElements.removeAll(enumConstants);
   248                 for(Element element : enclosedElements)
   249                     this.visit(element);
   250             } else {
   251                 for(Element element : e.getEnclosedElements())
   252                     this.visit(element);
   253             }
   255             indentation--;
   256             indent();
   257             writer.println("}");
   258             return this;
   259         }
   261         @Override
   262         public PrintingElementVisitor visitVariable(VariableElement e, Boolean newLine) {
   263             ElementKind kind = e.getKind();
   264             defaultAction(e, newLine);
   266             if (kind == ENUM_CONSTANT)
   267                 writer.print(e.getSimpleName());
   268             else {
   269                 writer.print(e.asType().toString() + " " + e.getSimpleName() );
   270                 Object constantValue  = e.getConstantValue();
   271                 if (constantValue != null) {
   272                     writer.print(" = ");
   273                     writer.print(elementUtils.getConstantExpression(constantValue));
   274                 }
   275                 writer.println(";");
   276             }
   277             return this;
   278         }
   280         @Override
   281         public PrintingElementVisitor visitTypeParameter(TypeParameterElement e, Boolean p) {
   282             writer.print(e.getSimpleName());
   283             return this;
   284         }
   286         // Should we do more here?
   287         @Override
   288         public PrintingElementVisitor visitPackage(PackageElement e, Boolean p) {
   289             defaultAction(e, false);
   290             if (!e.isUnnamed())
   291                 writer.println("package " + e.getQualifiedName() + ";");
   292             else
   293                 writer.println("// Unnamed package");
   294             return this;
   295         }
   297         public void flush() {
   298             writer.flush();
   299         }
   301         private void printDocComment(Element e) {
   302             String docComment = elementUtils.getDocComment(e);
   304             if (docComment != null) {
   305                 // Break comment into lines
   306                 java.util.StringTokenizer st = new StringTokenizer(docComment,
   307                                                                   "\n\r");
   308                 indent();
   309                 writer.println("/**");
   311                 while(st.hasMoreTokens()) {
   312                     indent();
   313                     writer.print(" *");
   314                     writer.println(st.nextToken());
   315                 }
   317                 indent();
   318                 writer.println(" */");
   319             }
   320         }
   322         private void printModifiers(Element e) {
   323             ElementKind kind = e.getKind();
   324             if (kind == PARAMETER) {
   325                 printAnnotationsInline(e);
   326             } else {
   327                 printAnnotations(e);
   328                 indent();
   329             }
   331             if (kind == ENUM_CONSTANT)
   332                 return;
   334             Set<Modifier> modifiers = new LinkedHashSet<Modifier>();
   335             modifiers.addAll(e.getModifiers());
   337             switch (kind) {
   338             case ANNOTATION_TYPE:
   339             case INTERFACE:
   340                 modifiers.remove(Modifier.ABSTRACT);
   341                 break;
   343             case ENUM:
   344                 modifiers.remove(Modifier.FINAL);
   345                 modifiers.remove(Modifier.ABSTRACT);
   346                 break;
   348             case METHOD:
   349             case FIELD:
   350                 Element enclosingElement = e.getEnclosingElement();
   351                 if (enclosingElement != null &&
   352                     enclosingElement.getKind().isInterface()) {
   353                     modifiers.remove(Modifier.PUBLIC);
   354                     modifiers.remove(Modifier.ABSTRACT); // only for methods
   355                     modifiers.remove(Modifier.STATIC);   // only for fields
   356                     modifiers.remove(Modifier.FINAL);    // only for fields
   357                 }
   358                 break;
   360             }
   362             for(Modifier m: modifiers) {
   363                 writer.print(m.toString() + " ");
   364             }
   365         }
   367         private void printFormalTypeParameters(ExecutableElement executable) {
   368             printFormalTypeParameters(executable.getTypeParameters(), true);
   369         }
   371         private void printFormalTypeParameters(TypeElement type) {
   372             printFormalTypeParameters(type.getTypeParameters(), false);
   373         }
   375         private void printFormalTypeParameters(List<? extends TypeParameterElement> typeParams,
   376                                                boolean pad) {
   377             if (typeParams.size() > 0) {
   378                 writer.print("<");
   380                 boolean first = true;
   381                 for(TypeParameterElement tpe: typeParams) {
   382                     if (!first)
   383                         writer.print(", ");
   384                     writer.print(tpe.toString());
   385                     first = false;
   386                 }
   388                 writer.print(">");
   389                 if (pad)
   390                     writer.print(" ");
   391             }
   392         }
   394         private void printAnnotationsInline(Element e) {
   395             List<? extends AnnotationMirror> annots = e.getAnnotationMirrors();
   396             for(AnnotationMirror annotationMirror : annots) {
   397                 writer.print(annotationMirror);
   398                 writer.print(" ");
   399             }
   400         }
   402         private void printAnnotations(Element e) {
   403             List<? extends AnnotationMirror> annots = e.getAnnotationMirrors();
   404             for(AnnotationMirror annotationMirror : annots) {
   405                 indent();
   406                 writer.println(annotationMirror);
   407             }
   408         }
   410         // TODO: Refactor
   411         private void printParameters(ExecutableElement e) {
   412             List<? extends VariableElement> parameters = e.getParameters();
   413             int size = parameters.size();
   415             switch (size) {
   416             case 0:
   417                 break;
   419             case 1:
   420                 for(VariableElement parameter: parameters) {
   421                     printModifiers(parameter);
   423                     if (e.isVarArgs() ) {
   424                         TypeMirror tm = parameter.asType();
   425                         if (tm.getKind() != TypeKind.ARRAY)
   426                             throw new AssertionError("Var-args parameter is not an array type: " + tm);
   427                         writer.print((ArrayType.class.cast(tm)).getComponentType() );
   428                         writer.print("...");
   429                     } else
   430                         writer.print(parameter.asType());
   431                     writer.print(" " + parameter.getSimpleName());
   432                 }
   433                 break;
   435             default:
   436                 {
   437                     int i = 1;
   438                     for(VariableElement parameter: parameters) {
   439                         if (i == 2)
   440                             indentation++;
   442                         if (i > 1)
   443                             indent();
   445                         printModifiers(parameter);
   447                         if (i == size && e.isVarArgs() ) {
   448                             TypeMirror tm = parameter.asType();
   449                             if (tm.getKind() != TypeKind.ARRAY)
   450                                 throw new AssertionError("Var-args parameter is not an array type: " + tm);
   451                                     writer.print((ArrayType.class.cast(tm)).getComponentType() );
   453                             writer.print("...");
   454                         } else
   455                             writer.print(parameter.asType());
   456                         writer.print(" " + parameter.getSimpleName());
   458                         if (i < size)
   459                             writer.println(",");
   461                         i++;
   462                     }
   464                     if (parameters.size() >= 2)
   465                         indentation--;
   466                 }
   467                 break;
   468             }
   469         }
   471         private void printInterfaces(TypeElement e) {
   472             ElementKind kind = e.getKind();
   474             if(kind != ANNOTATION_TYPE) {
   475                 List<? extends TypeMirror> interfaces = e.getInterfaces();
   476                 if (interfaces.size() > 0) {
   477                     writer.print((kind.isClass() ? " implements" : " extends"));
   479                     boolean first = true;
   480                     for(TypeMirror interf: interfaces) {
   481                         if (!first)
   482                             writer.print(",");
   483                         writer.print(" ");
   484                         writer.print(interf.toString());
   485                         first = false;
   486                     }
   487                 }
   488             }
   489         }
   491         private void printThrows(ExecutableElement e) {
   492             List<? extends TypeMirror> thrownTypes = e.getThrownTypes();
   493             final int size = thrownTypes.size();
   494             if (size != 0) {
   495                 writer.print(" throws");
   497                 int i = 1;
   498                 for(TypeMirror thrownType: thrownTypes) {
   499                     if (i == 1)
   500                         writer.print(" ");
   502                     if (i == 2)
   503                         indentation++;
   505                     if (i >= 2)
   506                         indent();
   508                     writer.print(thrownType);
   510                     if (i != size)
   511                         writer.println(", ");
   513                     i++;
   514                 }
   516                 if (size >= 2)
   517                     indentation--;
   518             }
   519         }
   521         private static final String [] spaces = {
   522             "",
   523             "  ",
   524             "    ",
   525             "      ",
   526             "        ",
   527             "          ",
   528             "            ",
   529             "              ",
   530             "                ",
   531             "                  ",
   532             "                    "
   533         };
   535         private void indent() {
   536             int indentation = this.indentation;
   537             if (indentation < 0)
   538                 return;
   539             final int maxIndex = spaces.length - 1;
   541             while (indentation > maxIndex) {
   542                 writer.print(spaces[maxIndex]);
   543                 indentation -= maxIndex;
   544             }
   545             writer.print(spaces[indentation]);
   546         }
   548     }
   549 }

mercurial