refactor(prettyprinter): restore DJPP#writeHeaders that clients can o…

…verride (#1866)
INRIA · Feb 21, 2018 · dec66d3 · dec66d3
1 parent f15f058
commit dec66d3
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/main/java/spoon/reflect/visitor/DefaultJavaPrettyPrinter.java b/src/main/java/spoon/reflect/visitor/DefaultJavaPrettyPrinter.java
@@ -1961,6 +1961,15 @@ private void reset() {
 		}
 	}
 
+
+	/**
+	 * Write the compilation unit header.
+	 */
+	public DefaultJavaPrettyPrinter writeHeader(List<CtType<?>> types, Collection<CtImport> imports) {
+		elementPrinterHelper.writeHeader(types, imports);
+		return this;
+	}
+
 	@Override
 	public void calculate(CompilationUnit sourceCompilationUnit, List<CtType<?>> types) {
 		// reset the importsContext to avoid errors with multiple CU
@@ -1975,7 +1984,7 @@ public void calculate(CompilationUnit sourceCompilationUnit, List<CtType<?>> typ
 		for (CtType<?> t : types) {
 			imports.addAll(computeImports(t));
 		}
-		elementPrinterHelper.writeHeader(types, imports);
+		this.writeHeader(types, imports);
 		for (CtType<?> t : types) {
 			scan(t);
 			if (!env.isPreserveLineNumbers()) {