Move getElementAt() to CompilationUnitElement.

pkg/analyzer/lib/src/generated/element.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.
 
 // This code was auto-generated, is not intended to be edited, and is subject to
 // significant change. Please see the README file for more information.
 
 library engine.element;
 
 import 'dart:collection';
@@ skipping 1195 matching lines @@
    */
   List<TopLevelVariableElement> get topLevelVariables;
 
   /**
    * Return a list containing all of the classes contained in this compilation
    * unit.
    */
   List<ClassElement> get types;
 
   /**
+   * Return the element at the given [offset], maybe `null` if no such element.
+   */
+  Element getElementAt(int offset);
+
+  /**
    * Return the enum defined in this compilation unit that has the given [name],
    * or `null` if this compilation unit does not define an enum with the given
    * name.
    */
   ClassElement getEnum(String name);
 
   /**
    * Return the class defined in this compilation unit that has the given
    * [name], or `null` if this compilation unit does not define a class with the
    * given name.
@@ skipping 47 matching lines @@
    */
   List<ClassElement> _types = ClassElementImpl.EMPTY_ARRAY;
 
   /**
    * A list containing all of the variables contained in this compilation unit.
    */
   List<TopLevelVariableElement> _variables =
       TopLevelVariableElementImpl.EMPTY_ARRAY;
 
   /**
+   * A map from offsets to elements of this unit at these offsets.
+   */
+  final Map<int, Element> _offsetToElementMap = new HashMap<int, Element>();
+
+  /**
    * Initialize a newly created compilation unit element to have the given
    * [name].
    */
   CompilationUnitElementImpl(String name) : super(name, -1);
 
   @override
   List<PropertyAccessorElement> get accessors => _accessors;
 
   /**
    * Set the top-level accessors (getters and setters) contained in this
@@ skipping 100 matching lines @@
     this._types = types;
   }
 
   @override
   bool operator ==(Object object) =>
       object is CompilationUnitElementImpl && source == object.source;
 
   @override
   accept(ElementVisitor visitor) => visitor.visitCompilationUnitElement(this);
 
+  /**
+   * This method is invoked after this unit was incrementally resolved.
+   */
+  void afterIncrementalResolution() {
+    _offsetToElementMap.clear();
+  }
+
   @override
   void appendTo(StringBuffer buffer) {
     if (source == null) {
       buffer.write("{compilation unit}");
     } else {
       buffer.write(source.fullName);
     }
   }
 
   @override
@@ skipping 31 matching lines @@
     }
     for (ClassElement type in _enums) {
       if ((type as ClassElementImpl).identifier == identifier) {
         return type as ClassElementImpl;
       }
     }
     return null;
   }
 
   @override
+  Element getElementAt(int offset) {
+    if (_offsetToElementMap.isEmpty) {
+      accept(new _BuildOffsetToElementMap(_offsetToElementMap));
+    }
+    return _offsetToElementMap[offset];
+  }
+
+  @override
   ClassElement getEnum(String enumName) {
     for (ClassElement enumDeclaration in _enums) {
       if (enumDeclaration.name == enumName) {
         return enumDeclaration;
       }
     }
     return null;
   }
 
   @override
@@ skipping 5295 matching lines @@
    * the `part` directive.
    */
   List<CompilationUnitElement> get units;
 
   /**
    * Return a list containing all directly and indirectly imported libraries.
    */
   List<LibraryElement> get visibleLibraries;
 
   /**
-   * Return the element at the given [offset], maybe `null` if no such element.
-   */
-  Element getElementAt(int offset);
-
-  /**
    * Return a list containing all of the imports that share the given [prefix],
    * or an empty array if there are no such imports.
    */
   List<ImportElement> getImportsWithPrefix(PrefixElement prefix);
 
   /**
    * Return the class defined in this library that has the given [name], or
    * `null` if this library does not define a class with the given name.
    */
   ClassElement getType(String className);
@@ skipping 49 matching lines @@
    */
   List<CompilationUnitElement> _parts = CompilationUnitElementImpl.EMPTY_ARRAY;
 
   /**
    * The element representing the synthetic function `loadLibrary` that is
    * defined for this library, or `null` if the element has not yet been created.
    */
   FunctionElement _loadLibraryFunction;
 
   /**
-   * A map from offsets to elements of this library at these offsets.
-   */
-  final Map<int, Element> _offsetToElementMap = new HashMap<int, Element>();
-
-  /**
    * The export [Namespace] of this library, `null` if it has not been
    * computed yet.
    */
   @override
   Namespace exportNamespace;
 
   /**
    * The public [Namespace] of this library, `null` if it has not been
    * computed yet.
    */
@@ skipping 254 matching lines @@
     return new List.from(visibleLibraries);
   }
 
   @override
   bool operator ==(Object object) => object is LibraryElementImpl &&
       _definingCompilationUnit == object.definingCompilationUnit;
 
   @override
   accept(ElementVisitor visitor) => visitor.visitLibraryElement(this);
 
-  /**
-   * This method is invoked after this library was incrementally resolved.
-   */
-  void afterIncrementalResolution() {
-    _offsetToElementMap.clear();
-  }
-
   @override
   ElementImpl getChild(String identifier) {
     if ((_definingCompilationUnit as CompilationUnitElementImpl).identifier ==
         identifier) {
       return _definingCompilationUnit as CompilationUnitElementImpl;
     }
     for (CompilationUnitElement part in _parts) {
       if ((part as CompilationUnitElementImpl).identifier == identifier) {
         return part as CompilationUnitElementImpl;
       }
     }
     for (ImportElement importElement in _imports) {
       if ((importElement as ImportElementImpl).identifier == identifier) {
         return importElement as ImportElementImpl;
       }
     }
     for (ExportElement exportElement in _exports) {
       if ((exportElement as ExportElementImpl).identifier == identifier) {
         return exportElement as ExportElementImpl;
       }
     }
     return null;
   }
 
   @override
-  Element getElementAt(int offset) {
-    if (_offsetToElementMap.isEmpty) {
-      accept(new _BuildOffsetToElementMap(_offsetToElementMap));
-    }
-    return _offsetToElementMap[offset];
-  }
-
-  @override
   List<ImportElement> getImportsWithPrefix(PrefixElement prefixElement) {
     int count = _imports.length;
     List<ImportElement> importList = new List<ImportElement>();
     for (int i = 0; i < count; i++) {
       if (identical(_imports[i].prefix, prefixElement)) {
         importList.add(_imports[i]);
       }
     }
     return importList;
   }
@@ skipping 2969 matching lines @@
 
   @override
   void visitElement(Element element) {
     int offset = element.nameOffset;
     if (offset != -1) {
       map[offset] = element;
     }
     super.visitElement(element);
   }
 }