src/share/classes/com/sun/tools/apt/main/JavaCompiler.java

Mon, 16 Jun 2008 13:28:00 -0700

author
jjg
date
Mon, 16 Jun 2008 13:28:00 -0700
changeset 50
b9bcea8bbe24
parent 1
9a66ca7c79fa
child 54
eaf608c64fec
permissions
-rw-r--r--

6714364: refactor javac File handling code into new javac.file package
Reviewed-by: mcimadamore

duke@1 1 /*
duke@1 2 * Copyright 2004-2006 Sun Microsystems, Inc. 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. Sun designates this
duke@1 8 * particular file as subject to the "Classpath" exception as provided
duke@1 9 * by Sun in the LICENSE file that accompanied this code.
duke@1 10 *
duke@1 11 * This code is distributed in the hope that it will be useful, but WITHOUT
duke@1 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
duke@1 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
duke@1 14 * version 2 for more details (a copy is included in the LICENSE file that
duke@1 15 * accompanied this code).
duke@1 16 *
duke@1 17 * You should have received a copy of the GNU General Public License version
duke@1 18 * 2 along with this work; if not, write to the Free Software Foundation,
duke@1 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
duke@1 20 *
duke@1 21 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
duke@1 22 * CA 95054 USA or visit www.sun.com if you need additional information or
duke@1 23 * have any questions.
duke@1 24 */
duke@1 25
duke@1 26 package com.sun.tools.apt.main;
duke@1 27
duke@1 28 import java.io.*;
duke@1 29 import java.util.Map;
duke@1 30
duke@1 31 import javax.tools.JavaFileManager;
duke@1 32 import javax.tools.JavaFileObject;
duke@1 33
jjg@50 34 import com.sun.tools.javac.file.JavacFileManager;
duke@1 35 import com.sun.tools.javac.util.*;
duke@1 36 import com.sun.tools.javac.code.*;
duke@1 37 import com.sun.tools.javac.jvm.*;
duke@1 38
duke@1 39 import com.sun.tools.javac.code.Symbol.*;
duke@1 40 import com.sun.tools.javac.tree.JCTree.*;
duke@1 41
duke@1 42 import com.sun.tools.apt.comp.*;
duke@1 43 import com.sun.tools.apt.util.Bark;
duke@1 44 import com.sun.mirror.apt.AnnotationProcessorFactory;
duke@1 45 import com.sun.tools.javac.parser.DocCommentScanner;
duke@1 46
duke@1 47 /**
duke@1 48 * <p><b>This is NOT part of any API supported by Sun Microsystems.
duke@1 49 * If you write code that depends on this, you do so at your own
duke@1 50 * risk. This code and its internal interfaces are subject to change
duke@1 51 * or deletion without notice.</b>
duke@1 52 */
duke@1 53 public class JavaCompiler extends com.sun.tools.javac.main.JavaCompiler {
duke@1 54 /** The context key for the compiler. */
duke@1 55 protected static final Context.Key<JavaCompiler> compilerKey =
duke@1 56 new Context.Key<JavaCompiler>();
duke@1 57
duke@1 58 /** Get the JavaCompiler instance for this context. */
duke@1 59 public static JavaCompiler instance(Context context) {
duke@1 60 JavaCompiler instance = context.get(compilerKey);
duke@1 61 if (instance == null)
duke@1 62 instance = new JavaCompiler(context);
duke@1 63 return instance;
duke@1 64 }
duke@1 65
duke@1 66
duke@1 67 java.util.Set<String> genSourceFileNames;
duke@1 68 java.util.Set<String> genClassFileNames;
duke@1 69
duke@1 70 public java.util.Set<String> getSourceFileNames() {
duke@1 71 return genSourceFileNames;
duke@1 72 }
duke@1 73
duke@1 74 /** List of names of generated class files.
duke@1 75 */
duke@1 76 public java.util.Set<String> getClassFileNames() {
duke@1 77 return genClassFileNames;
duke@1 78 }
duke@1 79
duke@1 80 java.util.Set<java.io.File> aggregateGenFiles = java.util.Collections.emptySet();
duke@1 81
duke@1 82 public java.util.Set<java.io.File> getAggregateGenFiles() {
duke@1 83 return aggregateGenFiles;
duke@1 84 }
duke@1 85
duke@1 86 /** The bark to be used for error reporting.
duke@1 87 */
duke@1 88 Bark bark;
duke@1 89
duke@1 90 /** The log to be used for error reporting.
duke@1 91 */
duke@1 92 Log log;
duke@1 93
duke@1 94 /** The annotation framework
duke@1 95 */
duke@1 96 Apt apt;
duke@1 97
duke@1 98 private static Context preRegister(Context context) {
duke@1 99 Bark.preRegister(context);
duke@1 100
duke@1 101 // force the use of the scanner that captures Javadoc comments
duke@1 102 DocCommentScanner.Factory.preRegister(context);
duke@1 103
duke@1 104 if (context.get(JavaFileManager.class) == null)
duke@1 105 JavacFileManager.preRegister(context);
duke@1 106
duke@1 107 return context;
duke@1 108 }
duke@1 109
duke@1 110 /** Construct a new compiler from a shared context.
duke@1 111 */
duke@1 112 public JavaCompiler(Context context) {
duke@1 113 super(preRegister(context));
duke@1 114
duke@1 115 context.put(compilerKey, this);
duke@1 116 apt = Apt.instance(context);
duke@1 117
duke@1 118 ClassReader classReader = ClassReader.instance(context);
duke@1 119 classReader.preferSource = true;
duke@1 120
duke@1 121 // TEMPORARY NOTE: bark==log, but while refactoring, we maintain their
duke@1 122 // original identities, to remember the original intent.
duke@1 123 log = Log.instance(context);
duke@1 124 bark = Bark.instance(context);
duke@1 125
duke@1 126 Options options = Options.instance(context);
duke@1 127 classOutput = options.get("-retrofit") == null;
duke@1 128 nocompile = options.get("-nocompile") != null;
duke@1 129 print = options.get("-print") != null;
duke@1 130 classesAsDecls= options.get("-XclassesAsDecls") != null;
duke@1 131
duke@1 132 genSourceFileNames = new java.util.LinkedHashSet<String>();
duke@1 133 genClassFileNames = new java.util.LinkedHashSet<String>();
duke@1 134
duke@1 135 // this forces a copy of the line map to be kept in the tree,
duke@1 136 // for use by com.sun.mirror.util.SourcePosition.
duke@1 137 lineDebugInfo = true;
duke@1 138 }
duke@1 139
duke@1 140 /* Switches:
duke@1 141 */
duke@1 142
duke@1 143 /** Emit class files. This switch is always set, except for the first
duke@1 144 * phase of retrofitting, where signatures are parsed.
duke@1 145 */
duke@1 146 public boolean classOutput;
duke@1 147
duke@1 148 /** The internal printing annotation processor should be used.
duke@1 149 */
duke@1 150 public boolean print;
duke@1 151
duke@1 152 /** Compilation should not be done after annotation processing.
duke@1 153 */
duke@1 154 public boolean nocompile;
duke@1 155
duke@1 156 /** Are class files being treated as declarations
duke@1 157 */
duke@1 158 public boolean classesAsDecls;
duke@1 159
duke@1 160 /** Try to open input stream with given name.
duke@1 161 * Report an error if this fails.
duke@1 162 * @param filename The file name of the input stream to be opened.
duke@1 163 */
duke@1 164 // PROVIDED FOR EXTREME BACKWARDS COMPATIBILITY
duke@1 165 // There are some very obscure errors that can arise while translating
duke@1 166 // the contents of a file from bytes to characters. In Tiger, these
duke@1 167 // diagnostics were ignored. This method provides compatibility with
duke@1 168 // that behavior. It would be better to honor those diagnostics, in which
duke@1 169 // case, this method can be deleted.
duke@1 170 @Override
duke@1 171 public CharSequence readSource(JavaFileObject filename) {
duke@1 172 try {
duke@1 173 inputFiles.add(filename);
duke@1 174 boolean prev = bark.setDiagnosticsIgnored(true);
duke@1 175 try {
duke@1 176 return filename.getCharContent(false);
duke@1 177 }
duke@1 178 finally {
duke@1 179 bark.setDiagnosticsIgnored(prev);
duke@1 180 }
duke@1 181 } catch (IOException e) {
duke@1 182 bark.error(Position.NOPOS, "cant.read.file", filename);
duke@1 183 return null;
duke@1 184 }
duke@1 185 }
duke@1 186
duke@1 187 /** Parse contents of input stream.
duke@1 188 * @param filename The name of the file from which input stream comes.
duke@1 189 * @param input The input stream to be parsed.
duke@1 190 */
duke@1 191 // PROVIDED FOR BACKWARDS COMPATIBILITY
duke@1 192 // In Tiger, diagnostics from the scanner and parser were ignored.
duke@1 193 // This method provides compatibility with that behavior.
duke@1 194 // It would be better to honor those diagnostics, in which
duke@1 195 // case, this method can be deleted.
duke@1 196 @Override
duke@1 197 protected JCCompilationUnit parse(JavaFileObject filename, CharSequence content) {
duke@1 198 boolean prev = bark.setDiagnosticsIgnored(true);
duke@1 199 try {
duke@1 200 return super.parse(filename, content);
duke@1 201 }
duke@1 202 finally {
duke@1 203 bark.setDiagnosticsIgnored(prev);
duke@1 204 }
duke@1 205 }
duke@1 206
duke@1 207 @Override
duke@1 208 protected boolean keepComments() {
duke@1 209 return true; // make doc comments available to mirror API impl.
duke@1 210 }
duke@1 211
duke@1 212 /** Track when the JavaCompiler has been used to compile something. */
duke@1 213 private boolean hasBeenUsed = false;
duke@1 214
duke@1 215 /** Main method: compile a list of files, return all compiled classes
duke@1 216 * @param filenames The names of all files to be compiled.
duke@1 217 */
duke@1 218 public List<ClassSymbol> compile(List<String> filenames,
duke@1 219 Map<String, String> origOptions,
duke@1 220 ClassLoader aptCL,
duke@1 221 AnnotationProcessorFactory providedFactory,
duke@1 222 java.util.Set<Class<? extends AnnotationProcessorFactory> > productiveFactories,
duke@1 223 java.util.Set<java.io.File> aggregateGenFiles)
duke@1 224 throws Throwable {
duke@1 225 // as a JavaCompiler can only be used once, throw an exception if
duke@1 226 // it has been used before.
duke@1 227 assert !hasBeenUsed : "attempt to reuse JavaCompiler";
duke@1 228 hasBeenUsed = true;
duke@1 229
duke@1 230 this.aggregateGenFiles = aggregateGenFiles;
duke@1 231
duke@1 232 long msec = System.currentTimeMillis();
duke@1 233
duke@1 234 ListBuffer<ClassSymbol> classes = new ListBuffer<ClassSymbol>();
duke@1 235 try {
duke@1 236 //parse all files
duke@1 237 ListBuffer<JCCompilationUnit> trees = new ListBuffer<JCCompilationUnit>();
duke@1 238 for (List<String> l = filenames; l.nonEmpty(); l = l.tail) {
duke@1 239 if (classesAsDecls) {
duke@1 240 if (! l.head.endsWith(".java") ) { // process as class file
duke@1 241 ClassSymbol cs = reader.enterClass(names.fromString(l.head));
duke@1 242 try {
duke@1 243 cs.complete();
duke@1 244 } catch(Symbol.CompletionFailure cf) {
duke@1 245 bark.aptError("CantFindClass", l);
duke@1 246 continue;
duke@1 247 }
duke@1 248
duke@1 249 classes.append(cs); // add to list of classes
duke@1 250 continue;
duke@1 251 }
duke@1 252 }
duke@1 253 trees.append(parse(l.head));
duke@1 254 }
duke@1 255
duke@1 256 //enter symbols for all files
duke@1 257 List<JCCompilationUnit> roots = trees.toList();
duke@1 258
duke@1 259 if (errorCount() == 0) {
duke@1 260 boolean prev = bark.setDiagnosticsIgnored(true);
duke@1 261 try {
duke@1 262 enter.main(roots);
duke@1 263 }
duke@1 264 finally {
duke@1 265 bark.setDiagnosticsIgnored(prev);
duke@1 266 }
duke@1 267 }
duke@1 268
duke@1 269 if (errorCount() == 0) {
duke@1 270 apt.main(roots,
duke@1 271 classes,
duke@1 272 origOptions, aptCL,
duke@1 273 providedFactory,
duke@1 274 productiveFactories);
duke@1 275 genSourceFileNames.addAll(apt.getSourceFileNames());
duke@1 276 genClassFileNames.addAll(apt.getClassFileNames());
duke@1 277 }
duke@1 278
duke@1 279 } catch (Abort ex) {
duke@1 280 }
duke@1 281
duke@1 282 if (verbose)
duke@1 283 printVerbose("total", Long.toString(System.currentTimeMillis() - msec));
duke@1 284
duke@1 285 chk.reportDeferredDiagnostics();
duke@1 286
duke@1 287 printCount("error", errorCount());
duke@1 288 printCount("warn", warningCount());
duke@1 289
duke@1 290 return classes.toList();
duke@1 291 }
duke@1 292 }

mercurial