Integrity checker for CPS and Tree IR.

pkg/compiler/lib/src/tree_ir/tree_ir_nodes.dart

 // Copyright (c) 2014, 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.
 
 library tree_ir_nodes;
 
 import '../constants/expressions.dart';
 import '../constants/values.dart' as values;
 import '../dart_types.dart' show DartType, GenericType;
 import '../elements/elements.dart';
@@ skipping 101 matching lines @@
   /// Number of places where this variable occurs as:
   /// - left-hand of an [Assign]
   /// - left-hand of a [FunctionDeclaration]
   /// - parameter in a [FunctionDefinition]
   /// - catch parameter in a [Try]
   int writeCount = 0;
 
   Variable(this.host, this.element) {
     assert(host != null);
   }
+
+  String toString() => element == null ? 'Variable' : element.toString();
 }
 
 /// Read the value of a variable.
 class VariableUse extends Expression {
   Variable variable;
 
   /// Creates a use of [variable] and updates its `readCount`.
   VariableUse(this.variable) {
     variable.readCount++;
   }
@@ skipping 711 matching lines @@
    S visitStatement(Statement s) => s.accept(this);
 }
 
 abstract class Visitor1<S, E, A> implements ExpressionVisitor1<E, A>,
                                             StatementVisitor1<S, A> {
    E visitExpression(Expression e, A arg) => e.accept1(this, arg);
    S visitStatement(Statement s, A arg) => s.accept1(this, arg);
 }
 
 class RecursiveVisitor extends Visitor {
+  // TODO(asgerf): Clean up the tree visitor.
+  
+  visitExecutableDefinition(ExecutableDefinition node) {
+    if (node is ConstructorDefinition) return visitConstructorDefinition(node);
+    if (node is FunctionDefinition) return visitFunctionDefinition(node);
+    if (node is FieldDefinition) return visitFieldDefinition(node);
+    throw 'Unexpected ExecutableDefinition: $node';
+  }
+
   visitFunctionDefinition(FunctionDefinition node) {
     node.parameters.forEach(visitVariable);
+    if (node.body != null) visitStatement(node.body);
+  }
+
+  visitConstructorDefinition(ConstructorDefinition node) {
+    if (node.initializers != null) node.initializers.forEach(visitInitializer);
+    visitFunctionDefinition(node);
+  }
+
+  visitFieldDefinition(FieldDefinition node) {
+    if (node.body != null) {
+      visitStatement(node.body);
+    }
+  }
+
+  visitInitializer(Initializer node) {
+    if (node is FieldInitializer) {
+      return visitFieldInitializer(node);
+    } else {
+      return visitSuperInitializer(node);
+    }
+  }
+
+  visitFieldInitializer(FieldInitializer node) {
     visitStatement(node.body);
   }
 
+  visitSuperInitializer(SuperInitializer node) {
+    node.arguments.forEach(visitStatement);
+  }
+
   visitVariable(Variable node) {}
 
   visitVariableUse(VariableUse node) {
     visitVariable(node.variable);
   }
 
   visitInvokeStatic(InvokeStatic node) {
     node.arguments.forEach(visitExpression);
   }
 
@@ skipping 98 matching lines @@
   visitExpressionStatement(ExpressionStatement node) {
     visitExpression(node.expression);
     visitStatement(node.next);
   }
 
   visitTry(Try node) {
     visitStatement(node.tryBody);
     visitStatement(node.catchBody);
   }
 
-  visitFieldInitializer(FieldInitializer node) {
-    visitStatement(node.body);
-  }
-
-  visitSuperInitializer(SuperInitializer node) {
-    node.arguments.forEach(visitStatement);
-  }
-
   visitGetField(GetField node) {
     visitExpression(node.object);
   }
 
   visitSetField(SetField node) {
     visitExpression(node.object);
     visitExpression(node.value);
     visitStatement(node.next);
   }
 
   visitCreateBox(CreateBox node) {
   }
 
   visitCreateInstance(CreateInstance node) {
     node.arguments.forEach(visitExpression);
   }
 }