Use IncrementalCompilationUnitElementBuilder in BuildCompilationUnitElementTask.

pkg/analyzer/lib/src/task/dart.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.
 
 library analyzer.src.task.dart;
 
 import 'dart:collection';
 import 'dart:math' as math;
 
 import 'package:analyzer/src/generated/ast.dart';
 import 'package:analyzer/src/generated/element.dart';
 import 'package:analyzer/src/generated/engine.dart' hide AnalysisTask;
 import 'package:analyzer/src/generated/error.dart';
 import 'package:analyzer/src/generated/error_verifier.dart';
 import 'package:analyzer/src/generated/java_engine.dart';
 import 'package:analyzer/src/generated/parser.dart';
 import 'package:analyzer/src/generated/resolver.dart';
 import 'package:analyzer/src/generated/scanner.dart';
 import 'package:analyzer/src/generated/sdk.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:analyzer/src/task/general.dart';
 import 'package:analyzer/task/dart.dart';
 import 'package:analyzer/task/general.dart';
 import 'package:analyzer/task/model.dart';
+import 'package:analyzer/src/task/incremental_element_builder.dart';
 
 /**
  * The errors produced while resolving a library directives.
  *
  * The list will be empty if there were no errors, but will not be `null`.
  *
  * The result is only available for targets representing a Dart library.
  */
 final ResultDescriptor<List<AnalysisError>> BUILD_DIRECTIVES_ERRORS =
     new ResultDescriptor<List<AnalysisError>>(
@@ skipping 485 matching lines @@
       }
       for (int i = argumentCount; i < parameterCount; i++) {
         types[i] = dynamic;
       }
     }
     return types;
   }
 }
 
 /**
+ * The memento for [BuildCompilationUnitElementTask].
+ */
+class BuildCompilationUnitElementMemento {
+  final List<ClassElement> classElements;
+  final CompilationUnit unit;
+  final CompilationUnitElement unitElement;
+  BuildCompilationUnitElementMemento(
+      this.classElements, this.unit, this.unitElement);
+}
+
+/**
  * A task that builds a compilation unit element for a single compilation unit.
  */
 class BuildCompilationUnitElementTask extends SourceBasedAnalysisTask {
   /**
-   * The name of the input whose value is the line information for the
-   * compilation unit.
-   */
-  static const String LINE_INFO_INPUT_NAME = 'LINE_INFO_INPUT_NAME';
-
-  /**
    * The name of the input whose value is the AST for the compilation unit.
    */
   static const String PARSED_UNIT_INPUT_NAME = 'PARSED_UNIT_INPUT_NAME';
 
   /**
    * The task descriptor describing this kind of task.
    */
   static final TaskDescriptor DESCRIPTOR = new TaskDescriptor(
       'BuildCompilationUnitElementTask', createTask, buildInputs,
       <ResultDescriptor>[
@@ skipping 14 matching lines @@
   TaskDescriptor get descriptor => DESCRIPTOR;
 
   @override
   void internalPerform() {
     //
     // Prepare inputs.
     //
     Source source = getRequiredSource();
     CompilationUnit unit = getRequiredInput(PARSED_UNIT_INPUT_NAME);
     //
+    // Use memento.
+    //
+    if (inputMemento is BuildCompilationUnitElementMemento) {
+      BuildCompilationUnitElementMemento memento = inputMemento;
+      unit = AstCloner.clone(unit);
+      new IncrementalCompilationUnitElementBuilder(memento.unit, unit).build();
+      CompilationUnitElement element = unit.element;
+      outputs[CLASS_ELEMENTS] = element.types;
+      outputs[COMPILATION_UNIT_ELEMENT] = element;
+      outputs[RESOLVED_UNIT1] = unit;
+      outputMemento =
+          new BuildCompilationUnitElementMemento(element.types, unit, element);
+      return;
+    }
+    //
     // Process inputs.
     //
     unit = AstCloner.clone(unit);
     CompilationUnitBuilder builder = new CompilationUnitBuilder();
     CompilationUnitElement element = builder.buildCompilationUnit(source, unit);
     //
     // Record outputs.
     //
     outputs[CLASS_ELEMENTS] = element.types;
     outputs[COMPILATION_UNIT_ELEMENT] = element;
     outputs[RESOLVED_UNIT1] = unit;
+    outputMemento =
+        new BuildCompilationUnitElementMemento(element.types, unit, element);
   }
 
   /**
    * Return a map from the names of the inputs of this kind of task to the task
    * input descriptors describing those inputs for a task with the given
    * [target].
    */
   static Map<String, TaskInput> buildInputs(LibrarySpecificUnit target) {
     return <String, TaskInput>{
       PARSED_UNIT_INPUT_NAME: PARSED_UNIT.of(target.unit)
@@ skipping 1316 matching lines @@
 class ParseDartMemento {
   final Token inputTokenStream;
   final List<Source> explicitlyImportedLibraries;
   final List<Source> exportedLibraries;
   final List<Source> importedLibraries;
   final List<Source> includedParts;
   final List<AnalysisError> parseErrors;
   final CompilationUnit parsedUnit;
   final SourceKind sourceKind;
   final List<Source> units;
-  ParseDartMemento(this.inputTokenStream, this.explicitlyImportedLibraries, this.exportedLibraries,
-      this.importedLibraries, this.includedParts, this.parseErrors,
-      this.parsedUnit, this.sourceKind, this.units);
+  ParseDartMemento(this.inputTokenStream, this.explicitlyImportedLibraries,
+      this.exportedLibraries, this.importedLibraries, this.includedParts,
+      this.parseErrors, this.parsedUnit, this.sourceKind, this.units);
 }
 
 /**
  * A task that parses the content of a Dart file, producing an AST structure.
  */
 class ParseDartTask extends SourceBasedAnalysisTask {
   /**
    * The name of the input whose value is the line information produced for the
    * file.
    */
@@ skipping 32 matching lines @@
   @override
   void internalPerform() {
     Source source = getRequiredSource();
     LineInfo lineInfo = getRequiredInput(LINE_INFO_INPUT_NAME);
     Token tokenStream = getRequiredInput(TOKEN_STREAM_INPUT_NAME);
 
     if (inputMemento is ParseDartMemento) {
       ParseDartMemento memento = inputMemento;
       if (identical(memento.inputTokenStream, tokenStream)) {
         outputMemento = memento;
-        outputs[EXPLICITLY_IMPORTED_LIBRARIES] = memento.explicitlyImportedLibraries;
+        outputs[EXPLICITLY_IMPORTED_LIBRARIES] =
+            memento.explicitlyImportedLibraries;
         outputs[EXPORTED_LIBRARIES] = memento.exportedLibraries;
         outputs[IMPORTED_LIBRARIES] = memento.importedLibraries;
         outputs[INCLUDED_PARTS] = memento.includedParts;
         outputs[PARSE_ERRORS] = memento.parseErrors;
         outputs[PARSED_UNIT] = memento.parsedUnit;
         outputs[SOURCE_KIND] = memento.sourceKind;
         outputs[UNITS] = memento.units;
         return;
       }
     }
@@ skipping 31 matching lines @@
               throw new AnalysisException(
                   '$runtimeType failed to handle a ${directive.runtimeType}');
             }
           }
         }
       }
     }
     //
     // Always include "dart:core" source.
     //
-    HashSet<Source> importedSourceSet = new HashSet.from(explicitlyImportedSourceSet);
+    HashSet<Source> importedSourceSet =
+        new HashSet.from(explicitlyImportedSourceSet);
     Source coreLibrarySource = context.sourceFactory.forUri(DartSdk.DART_CORE);
     importedSourceSet.add(coreLibrarySource);
     //
     // Compute kind.
     //
     SourceKind sourceKind = SourceKind.LIBRARY;
     if (!hasNonPartOfDirective && hasPartOfDirective) {
       sourceKind = SourceKind.PART;
     }
     //
     // Record outputs.
     //
-    List<Source> explicitlyImportedSources = explicitlyImportedSourceSet.toList();
+    List<Source> explicitlyImportedSources =
+        explicitlyImportedSourceSet.toList();
     List<Source> exportedSources = exportedSourceSet.toList();
     List<Source> importedSources = importedSourceSet.toList();
     List<Source> includedSources = includedSourceSet.toList();
     List<AnalysisError> parseErrors = errorListener.errors;
     List<Source> unitSources = <Source>[source]..addAll(includedSourceSet);
     outputs[EXPLICITLY_IMPORTED_LIBRARIES] = explicitlyImportedSources;
     outputs[EXPORTED_LIBRARIES] = exportedSources;
     outputs[IMPORTED_LIBRARIES] = importedSources;
     outputs[INCLUDED_PARTS] = includedSources;
     outputs[PARSE_ERRORS] = parseErrors;
     outputs[PARSED_UNIT] = unit;
     outputs[SOURCE_KIND] = sourceKind;
     outputs[UNITS] = unitSources;
-    outputMemento = new ParseDartMemento(tokenStream, explicitlyImportedSources, exportedSources,
-        importedSources, includedSources, parseErrors, unit, sourceKind,
-        unitSources);
+    outputMemento = new ParseDartMemento(tokenStream, explicitlyImportedSources,
+        exportedSources, importedSources, includedSources, parseErrors, unit,
+        sourceKind, unitSources);
   }
 
   /**
    * Return a map from the names of the inputs of this kind of task to the task
    * input descriptors describing those inputs for a task with the given
    * [source].
    */
   static Map<String, TaskInput> buildInputs(Source source) {
     return <String, TaskInput>{
       LINE_INFO_INPUT_NAME: LINE_INFO.of(source),
@@ skipping 612 matching lines @@
   @override
   bool moveNext() {
     if (_newSources.isEmpty) {
       return false;
     }
     currentTarget = _newSources.first;
     _newSources.remove(currentTarget);
     return true;
   }
 }