OLD | NEW |
| (Empty) |
1 // Copyright (c) 2015, the Dart project authors. Please see the AUTHORS file | |
2 // for details. All rights reserved. Use of this source code is governed by a | |
3 // BSD-style license that can be found in the LICENSE file. | |
4 | |
5 import 'dart:collection' show HashMap; | |
6 | |
7 import 'package:analyzer/dart/ast/ast.dart'; | |
8 import 'package:analyzer/dart/element/element.dart'; | |
9 import 'package:func/func.dart'; | |
10 import '../js_ast/js_ast.dart' as JS; | |
11 | |
12 /// Helper that tracks order of elements visited by the compiler, detecting | |
13 /// if the top level item can be loaded eagerly or not. | |
14 class ElementLoader { | |
15 final HashMap<Element, AstNode> _declarationNodes; | |
16 | |
17 /// The stack of currently emitting elements, if generating top-level code | |
18 /// for them. This is not used when inside method bodies, because order does | |
19 /// not matter for those. | |
20 final _topLevelElements = new List<Element>(); | |
21 | |
22 /// The current element being loaded. | |
23 /// We can use this to determine if we're loading top-level code or not: | |
24 /// | |
25 /// _currentElements.last == _topLevelElements.last | |
26 final _currentElements = new List<Element>(); | |
27 | |
28 bool _checkReferences; | |
29 | |
30 ElementLoader(this._declarationNodes) { | |
31 assert(!_declarationNodes.containsKey(null)); | |
32 } | |
33 | |
34 Element get currentElement => _currentElements.last; | |
35 | |
36 bool isLoaded(Element e) => !_declarationNodes.containsKey(e); | |
37 | |
38 /// True if the element is currently being loaded. | |
39 bool _isLoading(Element e) => _currentElements.contains(e); | |
40 | |
41 /// Start generating top-level code for the element [e]. | |
42 /// | |
43 /// Subsequent [emitDeclaration] calls will cause those elements to be | |
44 /// generated before this one, until [finishTopLevel] is called. | |
45 void startTopLevel(Element e) { | |
46 assert(identical(e, currentElement)); | |
47 _topLevelElements.add(e); | |
48 } | |
49 | |
50 /// Finishes the top-level code for the element [e]. | |
51 void finishTopLevel(Element e) { | |
52 var last = _topLevelElements.removeLast(); | |
53 assert(identical(e, last)); | |
54 } | |
55 | |
56 /// Starts recording calls to [declareBeforeUse], until | |
57 /// [finishCheckingReferences] is called. | |
58 void startCheckingReferences() { | |
59 // This function should not be reentrant, and we should not current be | |
60 // emitting top-level code. | |
61 assert(_checkReferences == null); | |
62 assert(_topLevelElements.isEmpty || | |
63 !identical(currentElement, _topLevelElements.last)); | |
64 // Assume true until proven otherwise | |
65 _checkReferences = true; | |
66 } | |
67 | |
68 /// Finishes recording references, and returns `true` if all referenced | |
69 /// items were loaded (or if no items were referenced). | |
70 bool finishCheckingReferences() { | |
71 var result = _checkReferences; | |
72 _checkReferences = null; | |
73 return result; | |
74 } | |
75 | |
76 /// Ensures a top-level declaration is generated, and returns `true` if it | |
77 /// is part of the current module. | |
78 /*=T*/ emitDeclaration/*<T extends JS.Node>*/( | |
79 Element e, Func1<AstNode, JS.Node/*=T*/ > visit) { | |
80 var node = _declarationNodes.remove(e); | |
81 if (node == null) return null; // not from this module or already loaded. | |
82 | |
83 _currentElements.add(e); | |
84 | |
85 var result = visit(node); | |
86 | |
87 var last = _currentElements.removeLast(); | |
88 assert(identical(e, last)); | |
89 | |
90 return result; | |
91 } | |
92 | |
93 /// To emit top-level module items, we sometimes need to reorder them. | |
94 /// | |
95 /// This function takes care of that, and also detects cases where reordering | |
96 /// failed, and we need to resort to lazy loading, by marking the element as | |
97 /// lazy. All elements need to be aware of this possibility and generate code | |
98 /// accordingly. | |
99 /// | |
100 /// If we are not emitting top-level code, this does nothing, because all | |
101 /// declarations are assumed to be available before we start execution. | |
102 /// See [startTopLevel]. | |
103 void declareBeforeUse(Element e, VoidFunc1<Element> emit) { | |
104 if (e == null) return; | |
105 | |
106 if (_checkReferences != null) { | |
107 _checkReferences = _checkReferences && isLoaded(e) && !_isLoading(e); | |
108 return; | |
109 } | |
110 | |
111 var topLevel = _topLevelElements; | |
112 if (topLevel.isNotEmpty && identical(currentElement, topLevel.last)) { | |
113 // If the item is from our library, try to emit it now. | |
114 emit(e); | |
115 } | |
116 } | |
117 } | |
OLD | NEW |