add call to enter and exit

INRIA · Sep 18, 2017 · c256281 · c256281
1 parent a84bfce
commit c256281
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/src/main/java/spoon/reflect/visitor/DefaultJavaPrettyPrinter.java b/src/main/java/spoon/reflect/visitor/DefaultJavaPrettyPrinter.java
@@ -299,6 +299,7 @@ public void computeImports(CtElement element) {
 	@Override
 	public void scan(CtElement e) {
 		if (e != null) {
+			enter(e);
 			context.elementStack.push(e);
 			if (env.isPreserveLineNumbers()) {
 				if (!(e instanceof CtNamedElement)) {
@@ -318,6 +319,7 @@ public void scan(CtElement e) {
 				throw new SpoonException("Printing of " + elementInfo + "failed", ex);
 			}
 			context.elementStack.pop();
+			exit(e);
 		}
 	}
 
@@ -344,16 +346,6 @@ private static void addParentPath(StringBuilder sb, CtElement ele) {
 		}
 	}
 
-	/**
-	 * The generic scan method for a reference.
-	 */
-	public DefaultJavaPrettyPrinter scan(CtReference ref) {
-		if (ref != null) {
-			ref.accept(this);
-		}
-		return this;
-	}
-
 	private boolean shouldSetBracket(CtExpression<?> e) {
 		if (e.getTypeCasts().size() != 0) {
 			return true;