@Override public void calculate(CtCompilationUnit sourceCompilationUnit, List<CtType<?>> types) { // reset the importsContext to avoid errors with multiple CU reset(); this.sourceCompilationUnit = sourceCompilationUnit; this.imports = new HashSet<>(); if (sourceCompilationUnit != null) { this.importsContext.initWithImports(sourceCompilationUnit.getImports()); } for (CtType<?> t : types) { imports.addAll(computeImports(t)); } this.writeHeader(types, imports); printTypes(types); }
public void calculate(CompilationUnit sourceCompilationUnit, List<CtType<?>> types) { this.sourceCompilationUnit = sourceCompilationUnit; Collection<CtTypeReference<?>> imports = Collections.emptyList(); for (CtType<?> t : types) { imports = computeImports(t); } writeHeader(types, imports); for (CtType<?> t : types) { scan(t); writeln().writeln().writeTabs(); } }
@Override public String toString() { DefaultJavaPrettyPrinter printer = new DefaultJavaPrettyPrinter(getFactory().getEnvironment()); String errorMessage = ""; try { printer.computeImports(this); printer.scan(this); } catch (ParentNotInitializedException ignore) { errorMessage = ERROR_MESSAGE_TO_STRING; } return printer.toString() + errorMessage; }