public abstract class StatementReplacingVisitorSupport extends ClassCodeVisitorSupport {
  private Statement replacement;

  /**
   * Visits the specified statement. If the statement's visit method calls
   * replaceVisitedMethodWith(), the statement will be replaced.
   */
  public Statement replace(Statement stat) {
    replacement = null;
    stat.visit(this);
    Statement result = replacement == null ? stat : replacement;