Use ASTVisitor as in new AST.

compiler/java/com/google/dart/compiler/ast/DartUnit.java

 // Copyright (c) 2011, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 package com.google.dart.compiler.ast;
 
 import com.google.common.collect.Lists;
 import com.google.common.collect.Sets;
 import com.google.dart.compiler.DartSource;
-import com.google.dart.compiler.util.DefaultTextOutput;
 
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
 import java.util.Set;
 
 /**
  * Represents a Dart compilation unit.
  */
 public class DartUnit extends DartNode {
 
   private static final long serialVersionUID = -3407637869012712127L;
 
   private LibraryUnit library;
   private List<DartDirective> directives;
   private final List<DartNode> topLevelNodes = Lists.newArrayList();
   private final DartSource source;
   private final boolean isDiet;
   /** A list of comments. May be null. */
   private List<DartComment> comments;
-  private String dietParse;
 
   public DartUnit(DartSource source, boolean isDiet) {
     this.source = source;
     this.isDiet = isDiet;
   }
 
   public void addTopLevelNode(DartNode node) {
     topLevelNodes.add(becomeParentOf(node));
   }
 
@@ skipping 27 matching lines @@
 
   public LibraryUnit getLibrary() {
     return library;
   }
 
   public List<DartNode> getTopLevelNodes() {
     return topLevelNodes;
   }
 
   @Override
-  public void traverse(DartVisitor v, DartContext ctx) {
-    if (v.visit(this, ctx)) {
-      if (directives != null) {
-        v.acceptWithInsertRemove(this, directives);
-      }
-      v.acceptWithInsertRemove(this, topLevelNodes);
-      if (comments != null) {
-        v.acceptWithInsertRemove(this, comments);
-      }
-    }
-    v.endVisit(this, ctx);
-  }
-
-  @Override
-  public void visitChildren(DartPlainVisitor<?> visitor) {
+  public void visitChildren(ASTVisitor<?> visitor) {
     if (directives != null) {
       visitor.visit(directives);
     }
     visitor.visit(topLevelNodes);
     if (comments != null) {
       visitor.visit(comments);
     }
   }
 
   @Override
-  public <R> R accept(DartPlainVisitor<R> visitor) {
+  public <R> R accept(ASTVisitor<R> visitor) {
     return visitor.visitUnit(this);
   }
 
   /**
    * Whether this is an diet unit, meaning it contains no method bodies.
    */
   public boolean isDiet() {
     return isDiet;
   }
 
   /**
-   * Generates a diet version of this unit, which contains no method bodies.
-   */
-  public final String toDietSource() {
-    if (dietParse == null) {
-      DefaultTextOutput out = new DefaultTextOutput(false);
-      new DartToSourceVisitor(out, true).accept(this);
-      dietParse = out.toString();
-    }
-    return dietParse;
-  }
-
-  /**
    * Add the specified directive to the receiver's list of directives
    */
   public void addDirective(DartDirective directive) {
     if (directives == null) {
       directives = new ArrayList<DartDirective>();
     }
     directives.add(becomeParentOf(directive));
   }
 
   /**
@@ skipping 35 matching lines @@
       }
     }
     return topLevelSymbols;
   }
 
   /**
    * @return the {@link Set} of names of all declarations.
    */
   public Set<String> getDeclarationNames() {
     final Set<String> symbols = Sets.newHashSet();
-    accept(new DartNodeTraverser<Void>() {
+    accept(new ASTVisitor<Void>() {
       @Override
       public Void visitFunctionTypeAlias(DartFunctionTypeAlias node) {
         symbols.add(node.getName().getTargetName());
         return super.visitFunctionTypeAlias(node);
       }
 
       @Override
       public Void visitClass(DartClass node) {
         symbols.add(node.getClassName());
         return super.visitClass(node);
@@ skipping 27 matching lines @@
 
       @Override
       public Void visitVariable(DartVariable node) {
         symbols.add(node.getVariableName());
         return super.visitVariable(node);
       }
     });
     return symbols;
   }
 }