Add MultiSourceInformationStrategy

pkg/compiler/lib/src/io/code_output.dart

 // Copyright (c) 2012, 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 dart2js.code_output;
 
 import '../../compiler_new.dart';
 import 'source_information.dart';
 
 /// Listener interface for [CodeOutput] activity.
 abstract class CodeOutputListener {
   /// Called when [text] is added to the output.
   void onText(String text);
 
   /// Called when the output is closed with a final length of [length].
   void onDone(int length);
 }
 
 /// Interface for a mapping of target offsets to source locations.
 abstract class SourceLocations {
+  /// The name identifying this source mapping.
+  String get name;
+
   /// Adds a [sourceLocation] at the specified [targetOffset].
   void addSourceLocation(int targetOffset, SourceLocation sourcePosition);
 
   /// Applies [f] to every target offset and associated source location.
   void forEachSourceLocation(
       void f(int targetOffset, SourceLocation sourceLocation));
 }
 
-abstract class CodeOutput implements SourceLocations {
+class _SourceLocationsImpl implements SourceLocations {
+  final String name;
+  final AbstractCodeOutput codeOutput;
+  Map<int, List<SourceLocation>> markers = <int, List<SourceLocation>>{};
+
+  _SourceLocationsImpl(this.name, this.codeOutput);
+
+  @override
+  void addSourceLocation(int targetOffset, SourceLocation sourceLocation) {
+    assert(targetOffset <= codeOutput.length);
+    List<SourceLocation> sourceLocations =
+        markers.putIfAbsent(targetOffset, () => <SourceLocation>[]);
+    sourceLocations.add(sourceLocation);
+  }
+
+  @override
+  void forEachSourceLocation(void f(int targetOffset, var sourceLocation)) {
+    markers.forEach((int targetOffset, List<SourceLocation> sourceLocations) {
+      for (SourceLocation sourceLocation in sourceLocations) {
+        f(targetOffset, sourceLocation);
+      }
+    });
+  }
+
+  void _addSourceLocations(_SourceLocationsImpl other) {
+    assert(name == other.name);
+    if (other.markers.length > 0) {
+      other.markers
+          .forEach((int targetOffset, List<SourceLocation> sourceLocations) {
+        markers
+            .putIfAbsent(
+                codeOutput.length + targetOffset, () => <SourceLocation>[])
+            .addAll(sourceLocations);
+      });
+    }
+  }
+}
+
+abstract class SourceLocationsProvider {
+  /// Creates a [SourceLocations] mapping identified by [name] and associates
+  /// it with this code output.
+  SourceLocations createSourceLocations(String name);
+
+  /// Returns the source location mappings associated with this code output.
+  Iterable<SourceLocations> get sourceLocations;
+}
+
+abstract class CodeOutput implements SourceLocationsProvider {
   /// Write [text] to this output.
   ///
   /// If the output is closed, a [StateError] is thrown.
   void add(String text);
 
   /// Adds the content of [buffer] to the output and adds its markers to
   /// [markers].
   ///
   /// If the output is closed, a [StateError] is thrown.
   void addBuffer(CodeBuffer buffer);
 
   /// Returns the number of characters currently written to this output.
   int get length;
 
   /// Returns `true` if this output has been closed.
   bool get isClosed;
 
   /// Closes the output. Further writes will cause a [StateError].
   void close();
 }
 
 abstract class AbstractCodeOutput extends CodeOutput {
-  Map<int, List<SourceLocation>> markers = <int, List<SourceLocation>>{};
+  Map<String, _SourceLocationsImpl> sourceLocationsMap =
+      <String, _SourceLocationsImpl>{};
   bool isClosed = false;
 
   void _addInternal(String text);
 
   @override
   void add(String text) {
     if (isClosed) {
       throw new StateError("Code output is closed. Trying to write '$text'.");
     }
     _addInternal(text);
   }
 
   @override
   void addBuffer(CodeBuffer other) {
-    if (other.markers.length > 0) {
-      other.markers
-          .forEach((int targetOffset, List<SourceLocation> sourceLocations) {
-        markers
-            .putIfAbsent(length + targetOffset, () => <SourceLocation>[])
-            .addAll(sourceLocations);
-      });
-    }
+    other.sourceLocationsMap.forEach((String name, _SourceLocationsImpl other) {
+      createSourceLocations(name)._addSourceLocations(other);
+    });
     if (!other.isClosed) {
       other.close();
     }
     _addInternal(other.getText());
   }
 
-  void addSourceLocation(int targetOffset, SourceLocation sourceLocation) {
-    assert(targetOffset <= length);
-    List<SourceLocation> sourceLocations =
-        markers.putIfAbsent(targetOffset, () => <SourceLocation>[]);
-    sourceLocations.add(sourceLocation);
-  }
-
-  void forEachSourceLocation(void f(int targetOffset, var sourceLocation)) {
-    markers.forEach((int targetOffset, List<SourceLocation> sourceLocations) {
-      for (SourceLocation sourceLocation in sourceLocations) {
-        f(targetOffset, sourceLocation);
-      }
-    });
-  }
-
+  @override
   void close() {
     if (isClosed) {
       throw new StateError("Code output is already closed.");
     }
     isClosed = true;
   }
+
+  @override
+  Iterable<SourceLocations> get sourceLocations => sourceLocationsMap.values;
+
+  @override
+  _SourceLocationsImpl createSourceLocations(String name) {
+    return sourceLocationsMap.putIfAbsent(
+        name, () => new _SourceLocationsImpl(name, this));
+  }
 }
 
 abstract class BufferedCodeOutput {
   String getText();
 }
 
 /// [CodeOutput] using a [StringBuffer] as backend.
 class CodeBuffer extends AbstractCodeOutput implements BufferedCodeOutput {
   StringBuffer buffer = new StringBuffer();
 
@@ skipping 32 matching lines @@
   }
 
   void close() {
     output.close();
     super.close();
     if (_listeners != null) {
       _listeners.forEach((listener) => listener.onDone(length));
     }
   }
 }