OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "config.h" | 5 #include "config.h" |
6 #include "core/dom/DOMNodeIds.h" | 6 #include "core/dom/DOMNodeIds.h" |
7 | |
8 #if ENABLE(OILPAN) | |
9 #include "core/dom/Node.h" | 7 #include "core/dom/Node.h" |
10 #else | |
11 #include "core/dom/WeakNodeMap.h" | |
12 #endif | |
13 #include "platform/heap/Handle.h" | 8 #include "platform/heap/Handle.h" |
14 | 9 |
15 namespace blink { | 10 namespace blink { |
16 | 11 |
17 #if ENABLE(OILPAN) | 12 #if !ENABLE(OILPAN) |
18 typedef HeapHashMap<WeakMember<Node>, int> NodeToIdMap; | 13 void WeakIdentifierMapTraits<Node>::removedFromIdentifierMap(Node* node) |
19 typedef HeapHashMap<int, WeakMember<Node>> IdToNodeMap; | |
20 | |
21 static NodeToIdMap& nodeToIdMap() | |
22 { | 14 { |
23 DEFINE_STATIC_LOCAL(Persistent<NodeToIdMap>, nodeToIdMap, (new NodeToIdMap()
)); | 15 node->clearFlag(Node::HasWeakReferencesFlag); |
24 return *nodeToIdMap; | |
25 } | 16 } |
26 | 17 |
27 static IdToNodeMap& idToNodeMap() | 18 void WeakIdentifierMapTraits<Node>::addedToIdentifierMap(Node* node) |
28 { | 19 { |
29 DEFINE_STATIC_LOCAL(Persistent<IdToNodeMap>, idToNodeMap, (new IdToNodeMap()
)); | 20 node->setFlag(Node::HasWeakReferencesFlag); |
30 return *idToNodeMap; | 21 } |
| 22 #endif |
| 23 |
| 24 static WeakNodeMap& nodeIds() |
| 25 { |
| 26 DEFINE_STATIC_LOCAL(RawPtrWillBePersistent<WeakNodeMap>, self, (new WeakNode
Map())); |
| 27 return *self; |
31 } | 28 } |
32 | 29 |
33 int DOMNodeIds::idForNode(Node* node) | 30 int DOMNodeIds::idForNode(Node* node) |
34 { | 31 { |
35 static int s_nextNodeId = 1; | 32 return nodeIds().identifier(node); |
36 NodeToIdMap::iterator it = nodeToIdMap().find(node); | |
37 if (it != nodeToIdMap().end()) | |
38 return it->value; | |
39 int id = s_nextNodeId++; | |
40 nodeToIdMap().set(node, id); | |
41 ASSERT(idToNodeMap().find(id) == idToNodeMap().end()); | |
42 idToNodeMap().set(id, node); | |
43 return id; | |
44 } | |
45 | |
46 Node* DOMNodeIds::nodeForId(int id) | |
47 { | |
48 return idToNodeMap().get(id); | |
49 } | |
50 #else | |
51 static WeakNodeMap& nodeIds() | |
52 { | |
53 DEFINE_STATIC_LOCAL(WeakNodeMap, self, ()); | |
54 return self; | |
55 } | |
56 | |
57 int DOMNodeIds::idForNode(Node* node) | |
58 { | |
59 static int s_nextNodeId = 1; | |
60 WeakNodeMap& ids = nodeIds(); | |
61 int result = ids.identifier(node); | |
62 if (!result) { | |
63 result = s_nextNodeId++; | |
64 ids.put(node, result); | |
65 } | |
66 return result; | |
67 } | 33 } |
68 | 34 |
69 Node* DOMNodeIds::nodeForId(int id) | 35 Node* DOMNodeIds::nodeForId(int id) |
70 { | 36 { |
71 return nodeIds().lookup(id); | 37 return nodeIds().lookup(id); |
72 } | 38 } |
73 #endif | |
74 | 39 |
75 } | 40 } |
OLD | NEW |