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

Sat, 01 Dec 2007 00:00:00 +0000

author
duke
date
Sat, 01 Dec 2007 00:00:00 +0000
changeset 1
9a66ca7c79fa
child 104
5e89c4ca637c
permissions
-rw-r--r--

Initial load

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

mercurial