| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009 Apple Inc. All rights reserv
ed. | 2 * Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009 Apple Inc. All rights reserv
ed. |
| 3 * | 3 * |
| 4 * Redistribution and use in source and binary forms, with or without | 4 * Redistribution and use in source and binary forms, with or without |
| 5 * modification, are permitted provided that the following conditions are | 5 * modification, are permitted provided that the following conditions are |
| 6 * met: | 6 * met: |
| 7 * | 7 * |
| 8 * * Redistributions of source code must retain the above copyright | 8 * * Redistributions of source code must retain the above copyright |
| 9 * notice, this list of conditions and the following disclaimer. | 9 * notice, this list of conditions and the following disclaimer. |
| 10 * * Redistributions in binary form must reproduce the above | 10 * * Redistributions in binary form must reproduce the above |
| (...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 64 void DocumentOrderedMap::clear() | 64 void DocumentOrderedMap::clear() |
| 65 { | 65 { |
| 66 m_map.clear(); | 66 m_map.clear(); |
| 67 } | 67 } |
| 68 | 68 |
| 69 void DocumentOrderedMap::add(StringImpl* key, Element* element) | 69 void DocumentOrderedMap::add(StringImpl* key, Element* element) |
| 70 { | 70 { |
| 71 ASSERT(key); | 71 ASSERT(key); |
| 72 ASSERT(element); | 72 ASSERT(element); |
| 73 | 73 |
| 74 Map::AddResult addResult = m_map.add(key, MapEntry(element)); | 74 Map::AddResult addResult = m_map.add(key, adoptPtr(new MapEntry(element))); |
| 75 if (addResult.isNewEntry) | 75 if (addResult.isNewEntry) |
| 76 return; | 76 return; |
| 77 | 77 |
| 78 MapEntry& entry = addResult.iterator->value; | 78 OwnPtr<MapEntry>& entry = addResult.iterator->value; |
| 79 ASSERT(entry.count); | 79 ASSERT(entry->count); |
| 80 entry.element = 0; | 80 entry->element = 0; |
| 81 entry.count++; | 81 entry->count++; |
| 82 entry.orderedList.clear(); | 82 entry->orderedList.clear(); |
| 83 } | 83 } |
| 84 | 84 |
| 85 void DocumentOrderedMap::remove(StringImpl* key, Element* element) | 85 void DocumentOrderedMap::remove(StringImpl* key, Element* element) |
| 86 { | 86 { |
| 87 ASSERT(key); | 87 ASSERT(key); |
| 88 ASSERT(element); | 88 ASSERT(element); |
| 89 | 89 |
| 90 Map::iterator it = m_map.find(key); | 90 Map::iterator it = m_map.find(key); |
| 91 if (it == m_map.end()) | 91 if (it == m_map.end()) |
| 92 return; | 92 return; |
| 93 | 93 |
| 94 MapEntry& entry = it->value; | 94 OwnPtr<MapEntry>& entry = it->value; |
| 95 ASSERT(entry.count); | 95 ASSERT(entry->count); |
| 96 if (entry.count == 1) { | 96 if (entry->count == 1) { |
| 97 ASSERT(!entry.element || entry.element == element); | 97 ASSERT(!entry->element || entry->element == element); |
| 98 m_map.remove(it); | 98 m_map.remove(it); |
| 99 } else { | 99 } else { |
| 100 if (entry.element == element) { | 100 if (entry->element == element) { |
| 101 ASSERT(entry.orderedList.isEmpty() || entry.orderedList.first() == e
lement); | 101 ASSERT(entry->orderedList.isEmpty() || entry->orderedList.first() ==
element); |
| 102 entry.element = entry.orderedList.size() > 1 ? entry.orderedList[1]
: 0; | 102 entry->element = entry->orderedList.size() > 1 ? entry->orderedList[
1] : 0; |
| 103 } | 103 } |
| 104 entry.count--; | 104 entry->count--; |
| 105 entry.orderedList.clear(); | 105 entry->orderedList.clear(); |
| 106 } | 106 } |
| 107 } | 107 } |
| 108 | 108 |
| 109 template<bool keyMatches(StringImpl*, Element*)> | 109 template<bool keyMatches(StringImpl*, Element*)> |
| 110 inline Element* DocumentOrderedMap::get(StringImpl* key, const TreeScope* scope)
const | 110 inline Element* DocumentOrderedMap::get(StringImpl* key, const TreeScope* scope)
const |
| 111 { | 111 { |
| 112 ASSERT(key); | 112 ASSERT(key); |
| 113 ASSERT(scope); | 113 ASSERT(scope); |
| 114 | 114 |
| 115 Map::iterator it = m_map.find(key); | 115 MapEntry* entry = m_map.get(key); |
| 116 if (it == m_map.end()) | 116 if (!entry) |
| 117 return 0; | 117 return 0; |
| 118 | 118 |
| 119 MapEntry& entry = it->value; | 119 ASSERT(entry->count); |
| 120 ASSERT(entry.count); | 120 if (entry->element) |
| 121 if (entry.element) | 121 return entry->element; |
| 122 return entry.element; | |
| 123 | 122 |
| 124 // We know there's at least one node that matches; iterate to find the first
one. | 123 // We know there's at least one node that matches; iterate to find the first
one. |
| 125 for (Element* element = ElementTraversal::firstWithin(*scope->rootNode()); e
lement; element = ElementTraversal::next(*element)) { | 124 for (Element* element = ElementTraversal::firstWithin(*scope->rootNode()); e
lement; element = ElementTraversal::next(*element)) { |
| 126 if (!keyMatches(key, element)) | 125 if (!keyMatches(key, element)) |
| 127 continue; | 126 continue; |
| 128 entry.element = element; | 127 entry->element = element; |
| 129 return element; | 128 return element; |
| 130 } | 129 } |
| 131 ASSERT_NOT_REACHED(); | 130 ASSERT_NOT_REACHED(); |
| 132 return 0; | 131 return 0; |
| 133 } | 132 } |
| 134 | 133 |
| 135 Element* DocumentOrderedMap::getElementById(StringImpl* key, const TreeScope* sc
ope) const | 134 Element* DocumentOrderedMap::getElementById(StringImpl* key, const TreeScope* sc
ope) const |
| 136 { | 135 { |
| 137 return get<keyMatchesId>(key, scope); | 136 return get<keyMatchesId>(key, scope); |
| 138 } | 137 } |
| 139 | 138 |
| 140 const Vector<Element*>& DocumentOrderedMap::getAllElementsById(StringImpl* key,
const TreeScope* scope) const | 139 const Vector<Element*>& DocumentOrderedMap::getAllElementsById(StringImpl* key,
const TreeScope* scope) const |
| 141 { | 140 { |
| 142 ASSERT(key); | 141 ASSERT(key); |
| 143 ASSERT(scope); | 142 ASSERT(scope); |
| 144 DEFINE_STATIC_LOCAL(Vector<Element*>, emptyVector, ()); | 143 DEFINE_STATIC_LOCAL(Vector<Element*>, emptyVector, ()); |
| 145 | 144 |
| 146 Map::iterator it = m_map.find(key); | 145 Map::iterator it = m_map.find(key); |
| 147 if (it == m_map.end()) | 146 if (it == m_map.end()) |
| 148 return emptyVector; | 147 return emptyVector; |
| 149 | 148 |
| 150 MapEntry& entry = it->value; | 149 OwnPtr<MapEntry>& entry = it->value; |
| 151 ASSERT(entry.count); | 150 ASSERT(entry->count); |
| 152 | 151 |
| 153 if (entry.orderedList.isEmpty()) { | 152 if (entry->orderedList.isEmpty()) { |
| 154 entry.orderedList.reserveCapacity(entry.count); | 153 entry->orderedList.reserveCapacity(entry->count); |
| 155 for (Element* element = entry.element ? entry.element : ElementTraversal
::firstWithin(*scope->rootNode()); entry.orderedList.size() < entry.count; eleme
nt = ElementTraversal::next(*element)) { | 154 for (Element* element = entry->element ? entry->element : ElementTravers
al::firstWithin(*scope->rootNode()); entry->orderedList.size() < entry->count; e
lement = ElementTraversal::next(*element)) { |
| 156 ASSERT(element); | 155 ASSERT(element); |
| 157 if (!keyMatchesId(key, element)) | 156 if (!keyMatchesId(key, element)) |
| 158 continue; | 157 continue; |
| 159 entry.orderedList.uncheckedAppend(element); | 158 entry->orderedList.uncheckedAppend(element); |
| 160 } | 159 } |
| 161 if (!entry.element) | 160 if (!entry->element) |
| 162 entry.element = entry.orderedList.first(); | 161 entry->element = entry->orderedList.first(); |
| 163 } | 162 } |
| 164 | 163 |
| 165 return entry.orderedList; | 164 return entry->orderedList; |
| 166 } | 165 } |
| 167 | 166 |
| 168 Element* DocumentOrderedMap::getElementByMapName(StringImpl* key, const TreeScop
e* scope) const | 167 Element* DocumentOrderedMap::getElementByMapName(StringImpl* key, const TreeScop
e* scope) const |
| 169 { | 168 { |
| 170 return get<keyMatchesMapName>(key, scope); | 169 return get<keyMatchesMapName>(key, scope); |
| 171 } | 170 } |
| 172 | 171 |
| 173 Element* DocumentOrderedMap::getElementByLowercasedMapName(StringImpl* key, cons
t TreeScope* scope) const | 172 Element* DocumentOrderedMap::getElementByLowercasedMapName(StringImpl* key, cons
t TreeScope* scope) const |
| 174 { | 173 { |
| 175 return get<keyMatchesLowercasedMapName>(key, scope); | 174 return get<keyMatchesLowercasedMapName>(key, scope); |
| 176 } | 175 } |
| 177 | 176 |
| 178 Element* DocumentOrderedMap::getElementByLabelForAttribute(StringImpl* key, cons
t TreeScope* scope) const | 177 Element* DocumentOrderedMap::getElementByLabelForAttribute(StringImpl* key, cons
t TreeScope* scope) const |
| 179 { | 178 { |
| 180 return get<keyMatchesLabelForAttribute>(key, scope); | 179 return get<keyMatchesLabelForAttribute>(key, scope); |
| 181 } | 180 } |
| 182 | 181 |
| 183 } // namespace WebCore | 182 } // namespace WebCore |
| OLD | NEW |