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

Tue, 24 Aug 2010 15:09:21 -0700

author
jjg
date
Tue, 24 Aug 2010 15:09:21 -0700
changeset 655
f3323b1c65ee
parent 620
2cf925ad67ab
child 700
7b413ac1a720
permissions
-rw-r--r--

6929404: Filer.getResource(SOURCE_PATH, ...) does not work when -sourcepath contains >1 entry
Reviewed-by: darcy

duke@1 1 /*
ohair@554 2 * Copyright (c) 2005, 2008, 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
ohair@554 7 * published by the Free Software Foundation. Oracle designates this
duke@1 8 * particular file as subject to the "Classpath" exception as provided
ohair@554 9 * by Oracle 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 *
ohair@554 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
ohair@554 22 * or visit www.oracle.com if you need additional information or have any
ohair@554 23 * questions.
duke@1 24 */
duke@1 25
duke@1 26 package com.sun.tools.javac.processing;
duke@1 27
duke@1 28 import com.sun.tools.javac.util.*;
duke@1 29 import javax.annotation.processing.*;
duke@1 30 import javax.lang.model.SourceVersion;
duke@1 31 import javax.lang.model.element.NestingKind;
duke@1 32 import javax.lang.model.element.Modifier;
duke@1 33 import javax.lang.model.element.Element;
duke@1 34 import java.util.*;
duke@1 35
duke@1 36 import java.io.Closeable;
jjg@655 37 import java.io.FileNotFoundException;
duke@1 38 import java.io.InputStream;
duke@1 39 import java.io.OutputStream;
duke@1 40 import java.io.FilterOutputStream;
duke@1 41 import java.io.Reader;
duke@1 42 import java.io.Writer;
duke@1 43 import java.io.FilterWriter;
duke@1 44 import java.io.PrintWriter;
duke@1 45 import java.io.IOException;
duke@1 46
duke@1 47 import javax.tools.*;
duke@1 48 import static java.util.Collections.*;
duke@1 49
duke@1 50 import javax.tools.JavaFileManager.Location;
duke@1 51 import static javax.tools.StandardLocation.SOURCE_OUTPUT;
duke@1 52 import static javax.tools.StandardLocation.CLASS_OUTPUT;
duke@1 53
duke@1 54 /**
duke@1 55 * The FilerImplementation class must maintain a number of
duke@1 56 * constraints. First, multiple attempts to open the same path within
duke@1 57 * the same invocation of the tool results in an IOException being
duke@1 58 * thrown. For example, trying to open the same source file twice:
duke@1 59 *
duke@1 60 * <pre>
duke@1 61 * createSourceFile("foo.Bar")
duke@1 62 * ...
duke@1 63 * createSourceFile("foo.Bar")
duke@1 64 * </pre>
duke@1 65 *
duke@1 66 * is disallowed as is opening a text file that happens to have
duke@1 67 * the same name as a source file:
duke@1 68 *
duke@1 69 * <pre>
duke@1 70 * createSourceFile("foo.Bar")
duke@1 71 * ...
duke@1 72 * createTextFile(SOURCE_TREE, "foo", new File("Bar"), null)
duke@1 73 * </pre>
duke@1 74 *
duke@1 75 * <p>Additionally, creating a source file that corresponds to an
duke@1 76 * already created class file (or vice versa) also results in an
duke@1 77 * IOException since each type can only be created once. However, if
duke@1 78 * the Filer is used to create a text file named *.java that happens
duke@1 79 * to correspond to an existing class file, a warning is *not*
duke@1 80 * generated. Similarly, a warning is not generated for a binary file
duke@1 81 * named *.class and an existing source file.
duke@1 82 *
duke@1 83 * <p>The reason for this difference is that source files and class
duke@1 84 * files are registered with the tool and can get passed on as
duke@1 85 * declarations to the next round of processing. Files that are just
duke@1 86 * named *.java and *.class are not processed in that manner; although
duke@1 87 * having extra source files and class files on the source path and
duke@1 88 * class path can alter the behavior of the tool and any final
duke@1 89 * compile.
duke@1 90 *
jjg@581 91 * <p><b>This is NOT part of any supported API.
duke@1 92 * If you write code that depends on this, you do so at your own risk.
duke@1 93 * This code and its internal interfaces are subject to change or
duke@1 94 * deletion without notice.</b>
duke@1 95 */
duke@1 96 public class JavacFiler implements Filer, Closeable {
duke@1 97 // TODO: Implement different transaction model for updating the
duke@1 98 // Filer's record keeping on file close.
duke@1 99
duke@1 100 private static final String ALREADY_OPENED =
duke@1 101 "Output stream or writer has already been opened.";
duke@1 102 private static final String NOT_FOR_READING =
duke@1 103 "FileObject was not opened for reading.";
duke@1 104 private static final String NOT_FOR_WRITING =
duke@1 105 "FileObject was not opened for writing.";
duke@1 106
duke@1 107 /**
duke@1 108 * Wrap a JavaFileObject to manage writing by the Filer.
duke@1 109 */
duke@1 110 private class FilerOutputFileObject extends ForwardingFileObject<FileObject> {
duke@1 111 private boolean opened = false;
duke@1 112 private String name;
duke@1 113
duke@1 114 FilerOutputFileObject(String name, FileObject fileObject) {
duke@1 115 super(fileObject);
duke@1 116 this.name = name;
duke@1 117 }
duke@1 118
duke@1 119 @Override
duke@1 120 public synchronized OutputStream openOutputStream() throws IOException {
duke@1 121 if (opened)
duke@1 122 throw new IOException(ALREADY_OPENED);
duke@1 123 opened = true;
duke@1 124 return new FilerOutputStream(name, fileObject);
duke@1 125 }
duke@1 126
duke@1 127 @Override
duke@1 128 public synchronized Writer openWriter() throws IOException {
duke@1 129 if (opened)
duke@1 130 throw new IOException(ALREADY_OPENED);
duke@1 131 opened = true;
duke@1 132 return new FilerWriter(name, fileObject);
duke@1 133 }
duke@1 134
duke@1 135 // Three anti-literacy methods
duke@1 136 @Override
duke@1 137 public InputStream openInputStream() throws IOException {
duke@1 138 throw new IllegalStateException(NOT_FOR_READING);
duke@1 139 }
duke@1 140
duke@1 141 @Override
duke@1 142 public Reader openReader(boolean ignoreEncodingErrors) throws IOException {
duke@1 143 throw new IllegalStateException(NOT_FOR_READING);
duke@1 144 }
duke@1 145
duke@1 146 @Override
duke@1 147 public CharSequence getCharContent(boolean ignoreEncodingErrors) throws IOException {
duke@1 148 throw new IllegalStateException(NOT_FOR_READING);
duke@1 149 }
duke@1 150
duke@1 151 @Override
duke@1 152 public boolean delete() {
duke@1 153 return false;
duke@1 154 }
duke@1 155 }
duke@1 156
duke@1 157 private class FilerOutputJavaFileObject extends FilerOutputFileObject implements JavaFileObject {
duke@1 158 private final JavaFileObject javaFileObject;
duke@1 159 FilerOutputJavaFileObject(String name, JavaFileObject javaFileObject) {
duke@1 160 super(name, javaFileObject);
duke@1 161 this.javaFileObject = javaFileObject;
duke@1 162 }
duke@1 163
duke@1 164 public JavaFileObject.Kind getKind() {
duke@1 165 return javaFileObject.getKind();
duke@1 166 }
duke@1 167
duke@1 168 public boolean isNameCompatible(String simpleName,
duke@1 169 JavaFileObject.Kind kind) {
duke@1 170 return javaFileObject.isNameCompatible(simpleName, kind);
duke@1 171 }
duke@1 172
duke@1 173 public NestingKind getNestingKind() {
duke@1 174 return javaFileObject.getNestingKind();
duke@1 175 }
duke@1 176
duke@1 177 public Modifier getAccessLevel() {
duke@1 178 return javaFileObject.getAccessLevel();
duke@1 179 }
duke@1 180 }
duke@1 181
duke@1 182 /**
duke@1 183 * Wrap a JavaFileObject to manage reading by the Filer.
duke@1 184 */
duke@1 185 private class FilerInputFileObject extends ForwardingFileObject<FileObject> {
duke@1 186 FilerInputFileObject(FileObject fileObject) {
duke@1 187 super(fileObject);
duke@1 188 }
duke@1 189
duke@1 190 @Override
duke@1 191 public OutputStream openOutputStream() throws IOException {
duke@1 192 throw new IllegalStateException(NOT_FOR_WRITING);
duke@1 193 }
duke@1 194
duke@1 195 @Override
duke@1 196 public Writer openWriter() throws IOException {
duke@1 197 throw new IllegalStateException(NOT_FOR_WRITING);
duke@1 198 }
duke@1 199
duke@1 200 @Override
duke@1 201 public boolean delete() {
duke@1 202 return false;
duke@1 203 }
duke@1 204 }
duke@1 205
duke@1 206 private class FilerInputJavaFileObject extends FilerInputFileObject implements JavaFileObject {
duke@1 207 private final JavaFileObject javaFileObject;
duke@1 208 FilerInputJavaFileObject(JavaFileObject javaFileObject) {
duke@1 209 super(javaFileObject);
duke@1 210 this.javaFileObject = javaFileObject;
duke@1 211 }
duke@1 212
duke@1 213 public JavaFileObject.Kind getKind() {
duke@1 214 return javaFileObject.getKind();
duke@1 215 }
duke@1 216
duke@1 217 public boolean isNameCompatible(String simpleName,
duke@1 218 JavaFileObject.Kind kind) {
duke@1 219 return javaFileObject.isNameCompatible(simpleName, kind);
duke@1 220 }
duke@1 221
duke@1 222 public NestingKind getNestingKind() {
duke@1 223 return javaFileObject.getNestingKind();
duke@1 224 }
duke@1 225
duke@1 226 public Modifier getAccessLevel() {
duke@1 227 return javaFileObject.getAccessLevel();
duke@1 228 }
duke@1 229 }
duke@1 230
duke@1 231
duke@1 232 /**
duke@1 233 * Wrap a {@code OutputStream} returned from the {@code
duke@1 234 * JavaFileManager} to properly register source or class files
duke@1 235 * when they are closed.
duke@1 236 */
duke@1 237 private class FilerOutputStream extends FilterOutputStream {
duke@1 238 String typeName;
duke@1 239 FileObject fileObject;
duke@1 240 boolean closed = false;
duke@1 241
duke@1 242 /**
duke@1 243 * @param typeName name of class or {@code null} if just a
duke@1 244 * binary file
duke@1 245 */
duke@1 246 FilerOutputStream(String typeName, FileObject fileObject) throws IOException {
duke@1 247 super(fileObject.openOutputStream());
duke@1 248 this.typeName = typeName;
duke@1 249 this.fileObject = fileObject;
duke@1 250 }
duke@1 251
duke@1 252 public synchronized void close() throws IOException {
duke@1 253 if (!closed) {
duke@1 254 closed = true;
duke@1 255 /*
duke@1 256 * If an IOException occurs when closing the underlying
duke@1 257 * stream, still try to process the file.
duke@1 258 */
duke@1 259
duke@1 260 closeFileObject(typeName, fileObject);
duke@1 261 out.close();
duke@1 262 }
duke@1 263 }
duke@1 264 }
duke@1 265
duke@1 266 /**
duke@1 267 * Wrap a {@code Writer} returned from the {@code JavaFileManager}
duke@1 268 * to properly register source or class files when they are
duke@1 269 * closed.
duke@1 270 */
duke@1 271 private class FilerWriter extends FilterWriter {
duke@1 272 String typeName;
duke@1 273 FileObject fileObject;
duke@1 274 boolean closed = false;
duke@1 275
duke@1 276 /**
duke@1 277 * @param fileObject the fileObject to be written to
duke@1 278 * @param typeName name of source file or {@code null} if just a
duke@1 279 * text file
duke@1 280 */
duke@1 281 FilerWriter(String typeName, FileObject fileObject) throws IOException {
duke@1 282 super(fileObject.openWriter());
duke@1 283 this.typeName = typeName;
duke@1 284 this.fileObject = fileObject;
duke@1 285 }
duke@1 286
duke@1 287 public synchronized void close() throws IOException {
duke@1 288 if (!closed) {
duke@1 289 closed = true;
duke@1 290 /*
duke@1 291 * If an IOException occurs when closing the underlying
duke@1 292 * Writer, still try to process the file.
duke@1 293 */
duke@1 294
duke@1 295 closeFileObject(typeName, fileObject);
duke@1 296 out.close();
duke@1 297 }
duke@1 298 }
duke@1 299 }
duke@1 300
duke@1 301 JavaFileManager fileManager;
duke@1 302 Log log;
duke@1 303 Context context;
duke@1 304 boolean lastRound;
duke@1 305
duke@1 306 private final boolean lint;
duke@1 307
duke@1 308 /**
duke@1 309 * Logical names of all created files. This set must be
duke@1 310 * synchronized.
duke@1 311 */
duke@1 312 private final Set<FileObject> fileObjectHistory;
duke@1 313
duke@1 314 /**
duke@1 315 * Names of types that have had files created but not closed.
duke@1 316 */
duke@1 317 private final Set<String> openTypeNames;
duke@1 318
duke@1 319 /**
duke@1 320 * Names of source files closed in this round. This set must be
duke@1 321 * synchronized. Its iterators should preserve insertion order.
duke@1 322 */
duke@1 323 private Set<String> generatedSourceNames;
duke@1 324
duke@1 325 /**
duke@1 326 * Names and class files of the class files closed in this round.
duke@1 327 * This set must be synchronized. Its iterators should preserve
duke@1 328 * insertion order.
duke@1 329 */
duke@1 330 private final Map<String, JavaFileObject> generatedClasses;
duke@1 331
duke@1 332 /**
duke@1 333 * JavaFileObjects for source files closed in this round. This
duke@1 334 * set must be synchronized. Its iterators should preserve
duke@1 335 * insertion order.
duke@1 336 */
duke@1 337 private Set<JavaFileObject> generatedSourceFileObjects;
duke@1 338
duke@1 339 /**
duke@1 340 * Names of all created source files. Its iterators should
duke@1 341 * preserve insertion order.
duke@1 342 */
duke@1 343 private final Set<String> aggregateGeneratedSourceNames;
duke@1 344
duke@1 345 /**
duke@1 346 * Names of all created class files. Its iterators should
duke@1 347 * preserve insertion order.
duke@1 348 */
duke@1 349 private final Set<String> aggregateGeneratedClassNames;
duke@1 350
duke@1 351
duke@1 352 JavacFiler(Context context) {
duke@1 353 this.context = context;
duke@1 354 fileManager = context.get(JavaFileManager.class);
duke@1 355
duke@1 356 log = Log.instance(context);
duke@1 357
duke@1 358 fileObjectHistory = synchronizedSet(new LinkedHashSet<FileObject>());
duke@1 359 generatedSourceNames = synchronizedSet(new LinkedHashSet<String>());
duke@1 360 generatedSourceFileObjects = synchronizedSet(new LinkedHashSet<JavaFileObject>());
duke@1 361
duke@1 362 generatedClasses = synchronizedMap(new LinkedHashMap<String, JavaFileObject>());
duke@1 363
duke@1 364 openTypeNames = synchronizedSet(new LinkedHashSet<String>());
duke@1 365
duke@1 366 aggregateGeneratedSourceNames = new LinkedHashSet<String>();
duke@1 367 aggregateGeneratedClassNames = new LinkedHashSet<String>();
duke@1 368
duke@1 369 lint = (Options.instance(context)).lint("processing");
duke@1 370 }
duke@1 371
duke@1 372 public JavaFileObject createSourceFile(CharSequence name,
duke@1 373 Element... originatingElements) throws IOException {
duke@1 374 return createSourceOrClassFile(true, name.toString());
duke@1 375 }
duke@1 376
duke@1 377 public JavaFileObject createClassFile(CharSequence name,
duke@1 378 Element... originatingElements) throws IOException {
duke@1 379 return createSourceOrClassFile(false, name.toString());
duke@1 380 }
duke@1 381
duke@1 382 private JavaFileObject createSourceOrClassFile(boolean isSourceFile, String name) throws IOException {
jjg@618 383 if (lint) {
jjg@618 384 int periodIndex = name.lastIndexOf(".");
jjg@618 385 if (periodIndex != -1) {
jjg@618 386 String base = name.substring(periodIndex);
jjg@618 387 String extn = (isSourceFile ? ".java" : ".class");
jjg@618 388 if (base.equals(extn))
jjg@618 389 log.warning("proc.suspicious.class.name", name, extn);
jjg@618 390 }
jjg@618 391 }
duke@1 392 checkNameAndExistence(name, isSourceFile);
duke@1 393 Location loc = (isSourceFile ? SOURCE_OUTPUT : CLASS_OUTPUT);
duke@1 394 JavaFileObject.Kind kind = (isSourceFile ?
duke@1 395 JavaFileObject.Kind.SOURCE :
duke@1 396 JavaFileObject.Kind.CLASS);
duke@1 397
duke@1 398 JavaFileObject fileObject =
duke@1 399 fileManager.getJavaFileForOutput(loc, name, kind, null);
duke@1 400 checkFileReopening(fileObject, true);
duke@1 401
duke@1 402 if (lastRound)
duke@1 403 log.warning("proc.file.create.last.round", name);
duke@1 404
duke@1 405 if (isSourceFile)
duke@1 406 aggregateGeneratedSourceNames.add(name);
duke@1 407 else
duke@1 408 aggregateGeneratedClassNames.add(name);
duke@1 409 openTypeNames.add(name);
duke@1 410
duke@1 411 return new FilerOutputJavaFileObject(name, fileObject);
duke@1 412 }
duke@1 413
duke@1 414 public FileObject createResource(JavaFileManager.Location location,
duke@1 415 CharSequence pkg,
duke@1 416 CharSequence relativeName,
duke@1 417 Element... originatingElements) throws IOException {
duke@1 418 locationCheck(location);
duke@1 419
duke@1 420 String strPkg = pkg.toString();
duke@1 421 if (strPkg.length() > 0)
duke@1 422 checkName(strPkg);
duke@1 423
duke@1 424 FileObject fileObject =
duke@1 425 fileManager.getFileForOutput(location, strPkg,
duke@1 426 relativeName.toString(), null);
duke@1 427 checkFileReopening(fileObject, true);
duke@1 428
duke@1 429 if (fileObject instanceof JavaFileObject)
duke@1 430 return new FilerOutputJavaFileObject(null, (JavaFileObject)fileObject);
duke@1 431 else
duke@1 432 return new FilerOutputFileObject(null, fileObject);
duke@1 433 }
duke@1 434
duke@1 435 private void locationCheck(JavaFileManager.Location location) {
duke@1 436 if (location instanceof StandardLocation) {
duke@1 437 StandardLocation stdLoc = (StandardLocation) location;
duke@1 438 if (!stdLoc.isOutputLocation())
duke@1 439 throw new IllegalArgumentException("Resource creation not supported in location " +
duke@1 440 stdLoc);
duke@1 441 }
duke@1 442 }
duke@1 443
duke@1 444 public FileObject getResource(JavaFileManager.Location location,
duke@1 445 CharSequence pkg,
duke@1 446 CharSequence relativeName) throws IOException {
duke@1 447 String strPkg = pkg.toString();
duke@1 448 if (strPkg.length() > 0)
duke@1 449 checkName(strPkg);
duke@1 450
duke@1 451 // TODO: Only support reading resources in selected output
duke@1 452 // locations? Only allow reading of non-source, non-class
duke@1 453 // files from the supported input locations?
jjg@655 454 FileObject fileObject = fileManager.getFileForInput(location,
jjg@655 455 pkg.toString(),
jjg@655 456 relativeName.toString());
jjg@655 457 if (fileObject == null) {
jjg@655 458 String name = (pkg.length() == 0)
jjg@655 459 ? relativeName.toString() : (pkg + "/" + relativeName);
jjg@655 460 throw new FileNotFoundException(name);
jjg@655 461 }
jjg@655 462
duke@1 463 // If the path was already opened for writing, throw an exception.
duke@1 464 checkFileReopening(fileObject, false);
duke@1 465 return new FilerInputFileObject(fileObject);
duke@1 466 }
duke@1 467
duke@1 468 private void checkName(String name) throws FilerException {
duke@1 469 checkName(name, false);
duke@1 470 }
duke@1 471
duke@1 472 private void checkName(String name, boolean allowUnnamedPackageInfo) throws FilerException {
duke@1 473 if (!SourceVersion.isName(name) && !isPackageInfo(name, allowUnnamedPackageInfo)) {
duke@1 474 if (lint)
duke@1 475 log.warning("proc.illegal.file.name", name);
duke@1 476 throw new FilerException("Illegal name " + name);
duke@1 477 }
duke@1 478 }
duke@1 479
duke@1 480 private boolean isPackageInfo(String name, boolean allowUnnamedPackageInfo) {
duke@1 481 // Is the name of the form "package-info" or
duke@1 482 // "foo.bar.package-info"?
duke@1 483 final String PKG_INFO = "package-info";
duke@1 484 int periodIndex = name.lastIndexOf(".");
duke@1 485 if (periodIndex == -1) {
duke@1 486 return allowUnnamedPackageInfo ? name.equals(PKG_INFO) : false;
duke@1 487 } else {
duke@1 488 // "foo.bar.package-info." illegal
duke@1 489 String prefix = name.substring(0, periodIndex);
duke@1 490 String simple = name.substring(periodIndex+1);
duke@1 491 return SourceVersion.isName(prefix) && simple.equals(PKG_INFO);
duke@1 492 }
duke@1 493 }
duke@1 494
duke@1 495 private void checkNameAndExistence(String typename, boolean allowUnnamedPackageInfo) throws FilerException {
duke@1 496 // TODO: Check if type already exists on source or class path?
duke@1 497 // If so, use warning message key proc.type.already.exists
duke@1 498 checkName(typename, allowUnnamedPackageInfo);
duke@1 499 if (aggregateGeneratedSourceNames.contains(typename) ||
duke@1 500 aggregateGeneratedClassNames.contains(typename)) {
duke@1 501 if (lint)
duke@1 502 log.warning("proc.type.recreate", typename);
duke@1 503 throw new FilerException("Attempt to recreate a file for type " + typename);
duke@1 504 }
duke@1 505 }
duke@1 506
duke@1 507 /**
duke@1 508 * Check to see if the file has already been opened; if so, throw
duke@1 509 * an exception, otherwise add it to the set of files.
duke@1 510 */
duke@1 511 private void checkFileReopening(FileObject fileObject, boolean addToHistory) throws FilerException {
duke@1 512 for(FileObject veteran : fileObjectHistory) {
duke@1 513 if (fileManager.isSameFile(veteran, fileObject)) {
duke@1 514 if (lint)
duke@1 515 log.warning("proc.file.reopening", fileObject.getName());
duke@1 516 throw new FilerException("Attempt to reopen a file for path " + fileObject.getName());
duke@1 517 }
duke@1 518 }
duke@1 519 if (addToHistory)
duke@1 520 fileObjectHistory.add(fileObject);
duke@1 521 }
duke@1 522
duke@1 523 public boolean newFiles() {
duke@1 524 return (!generatedSourceNames.isEmpty())
duke@1 525 || (!generatedClasses.isEmpty());
duke@1 526 }
duke@1 527
duke@1 528 public Set<String> getGeneratedSourceNames() {
duke@1 529 return generatedSourceNames;
duke@1 530 }
duke@1 531
duke@1 532 public Set<JavaFileObject> getGeneratedSourceFileObjects() {
duke@1 533 return generatedSourceFileObjects;
duke@1 534 }
duke@1 535
duke@1 536 public Map<String, JavaFileObject> getGeneratedClasses() {
duke@1 537 return generatedClasses;
duke@1 538 }
duke@1 539
duke@1 540 public void warnIfUnclosedFiles() {
duke@1 541 if (!openTypeNames.isEmpty())
duke@1 542 log.warning("proc.unclosed.type.files", openTypeNames.toString());
duke@1 543 }
duke@1 544
duke@1 545 /**
duke@1 546 * Update internal state for a new round.
duke@1 547 */
jjg@620 548 public void newRound(Context context) {
duke@1 549 this.context = context;
duke@1 550 this.log = Log.instance(context);
jjg@620 551 clearRoundState();
jjg@620 552 }
jjg@620 553
jjg@620 554 void setLastRound(boolean lastRound) {
duke@1 555 this.lastRound = lastRound;
duke@1 556 }
duke@1 557
duke@1 558 public void close() {
duke@1 559 clearRoundState();
duke@1 560 // Cross-round state
duke@1 561 fileObjectHistory.clear();
duke@1 562 openTypeNames.clear();
duke@1 563 aggregateGeneratedSourceNames.clear();
duke@1 564 aggregateGeneratedClassNames.clear();
duke@1 565 }
duke@1 566
duke@1 567 private void clearRoundState() {
duke@1 568 generatedSourceNames.clear();
duke@1 569 generatedSourceFileObjects.clear();
duke@1 570 generatedClasses.clear();
duke@1 571 }
duke@1 572
duke@1 573 /**
duke@1 574 * Debugging function to display internal state.
duke@1 575 */
duke@1 576 public void displayState() {
duke@1 577 PrintWriter xout = context.get(Log.outKey);
duke@1 578 xout.println("File Object History : " + fileObjectHistory);
duke@1 579 xout.println("Open Type Names : " + openTypeNames);
duke@1 580 xout.println("Gen. Src Names : " + generatedSourceNames);
duke@1 581 xout.println("Gen. Cls Names : " + generatedClasses.keySet());
duke@1 582 xout.println("Agg. Gen. Src Names : " + aggregateGeneratedSourceNames);
duke@1 583 xout.println("Agg. Gen. Cls Names : " + aggregateGeneratedClassNames);
duke@1 584 }
duke@1 585
duke@1 586 public String toString() {
duke@1 587 return "javac Filer";
duke@1 588 }
duke@1 589
duke@1 590 /**
duke@1 591 * Upon close, register files opened by create{Source, Class}File
duke@1 592 * for annotation processing.
duke@1 593 */
duke@1 594 private void closeFileObject(String typeName, FileObject fileObject) {
duke@1 595 /*
duke@1 596 * If typeName is non-null, the file object was opened as a
duke@1 597 * source or class file by the user. If a file was opened as
duke@1 598 * a resource, typeName will be null and the file is *not*
duke@1 599 * subject to annotation processing.
duke@1 600 */
duke@1 601 if ((typeName != null)) {
duke@1 602 if (!(fileObject instanceof JavaFileObject))
duke@1 603 throw new AssertionError("JavaFileOject not found for " + fileObject);
duke@1 604 JavaFileObject javaFileObject = (JavaFileObject)fileObject;
duke@1 605 switch(javaFileObject.getKind()) {
duke@1 606 case SOURCE:
duke@1 607 generatedSourceNames.add(typeName);
duke@1 608 generatedSourceFileObjects.add(javaFileObject);
duke@1 609 openTypeNames.remove(typeName);
duke@1 610 break;
duke@1 611
duke@1 612 case CLASS:
duke@1 613 generatedClasses.put(typeName, javaFileObject);
duke@1 614 openTypeNames.remove(typeName);
duke@1 615 break;
duke@1 616
duke@1 617 default:
duke@1 618 break;
duke@1 619 }
duke@1 620 }
duke@1 621 }
duke@1 622
duke@1 623 }

mercurial