@Override public DefaultJavaPrettyPrinter writeHeader(List<CtType<?>> types, Collection<CtImport> imports) { //run compilation unit header using pretty printer. The sniper mode is not supported for header yet. runInContext(SourceFragmentContextPrettyPrint.INSTANCE, () -> super.writeHeader(types, imports)); return this; }
@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(); } }