Don't use Element.computeNode() in tasks graph generator.

pkg/analyzer/tool/task_dependency_graph/generate.dart

 // Copyright (c) 2015, 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.
 
 /**
  * This file contains code to output a description of tasks and their
  * dependencies in ".dot" format.  Prior to running, the user should run "pub
  * get" in the analyzer directory to ensure that a "packages" folder exists.
  *
  * TODO(paulberry):
@@ skipping 171 matching lines @@
         setupSource(path.join('lib', 'src', 'plugin', 'engine_plugin.dart'));
     CompilationUnitElement modelElement = getUnit(modelSource).element;
     InterfaceType analysisTaskType = modelElement.getType('AnalysisTask').type;
     DartType dynamicType = context.typeProvider.dynamicType;
     resultDescriptorType = modelElement
         .getType('ResultDescriptor')
         .type
         .instantiate([dynamicType]);
     listOfResultDescriptorType =
         context.typeProvider.listType.instantiate([resultDescriptorType]);
-    CompilationUnitElement enginePluginUnitElement =
-        getUnit(enginePluginSource).element;
-    enginePluginClass = enginePluginUnitElement.getType('EnginePlugin');
+    CompilationUnit enginePluginUnit = getUnit(enginePluginSource);
+    enginePluginClass = enginePluginUnit.element.getType('EnginePlugin');
     extensionPointIdType =
-        enginePluginUnitElement.getType('ExtensionPointId').type;
+        enginePluginUnit.element.getType('ExtensionPointId').type;
     CompilationUnit dartDartUnit = getUnit(dartDartSource);
-    CompilationUnitElement dartDartUnitElement = dartDartUnit.element;
     CompilationUnit taskUnit = getUnit(taskSource);
     taskUnitElement = taskUnit.element;
     Set<String> results = new Set<String>();
     Set<String> resultLists = new Set<String>();
-    for (ClassElement cls in dartDartUnitElement.types) {
-      if (!cls.isAbstract && cls.type.isSubtypeOf(analysisTaskType)) {
-        String task = cls.name;
-        AstNode buildInputsAst = cls.getMethod('buildInputs').computeNode();
-        findResultDescriptors(buildInputsAst, (String input) {
-          results.add(input);
-          lines.add('  $input -> $task');
-        });
-        findResultDescriptorLists(buildInputsAst, (String input) {
-          resultLists.add(input);
-          lines.add('  $input -> $task');
-        });
-        findResultDescriptors(cls.getField('DESCRIPTOR').computeNode(),
-            (String out) {
-          results.add(out);
-          lines.add('  $task -> $out');
-        });
+    for (CompilationUnitMember dartUnitMember in dartDartUnit.declarations) {
+      if (dartUnitMember is ClassDeclaration) {
+        ClassDeclaration clazz = dartUnitMember;
+        if (!clazz.isAbstract &&
+            clazz.element.type.isSubtypeOf(analysisTaskType)) {
+          String task = clazz.name.name;
+
+          MethodDeclaration buildInputsAst;
+          VariableDeclaration descriptorField;
+          for (ClassMember classMember in clazz.members) {
+            if (classMember is MethodDeclaration &&
+                classMember.name.name == 'buildInputs') {
+              buildInputsAst = classMember;
+            }
+            if (classMember is FieldDeclaration) {
+              for (VariableDeclaration field in classMember.fields.variables) {
+                if (field.name.name == 'DESCRIPTOR') {
+                  descriptorField = field;
+                }
+              }
+            }
+          }
+
+          findResultDescriptors(buildInputsAst, (String input) {
+            results.add(input);
+            lines.add('  $input -> $task');
+          });
+          findResultDescriptorLists(buildInputsAst, (String input) {
+            resultLists.add(input);
+            lines.add('  $input -> $task');
+          });
+          findResultDescriptors(descriptorField, (String out) {
+            results.add(out);
+            lines.add('  $task -> $out');
+          });
+        }
       }
     }
-    AstNode enginePluginAst = enginePluginUnitElement.computeNode();
     for (String resultList in resultLists) {
       lines.add('  $resultList [shape=hexagon]');
       TopLevelVariableElement extensionIdVariable = _getExtensionId(resultList);
-      findExtensions(enginePluginAst, extensionIdVariable, (String extension) {
+      findExtensions(enginePluginUnit, extensionIdVariable, (String extension) {
         results.add(extension);
         lines.add('  $extension -> $resultList');
       });
     }
     for (String result in results) {
       lines.add('  $result [shape=box]');
     }
     lines.sort();
     return lines.join('\n');
   }
@@ skipping 121 matching lines @@
   @override
   visitIdentifier(Identifier node) {
     Element element = node.staticElement;
     if (element is PropertyAccessorElement &&
         element.isGetter &&
         element.returnType.isSubtypeOf(type)) {
       callback(element);
     }
   }
 }