| Left: | ||
| Right: |
| OLD | NEW |
|---|---|
| (Empty) | |
| 1 #pragma once | |
| 2 | |
| 3 #include <cstddef> | |
| 4 #include <cmath> | |
| 5 #include <initializer_list> | |
| 6 #include <memory> | |
| 7 | |
| 8 #include "String.h" | |
| 9 #include "debug.h" | |
| 10 | |
| 11 template<typename T> | |
| 12 class StringMap; | |
| 13 | |
| 14 namespace StringMap_internal | |
| 15 { | |
| 16 template<typename Entry> | |
| 17 struct HashContainerIterator | |
| 18 { | |
| 19 typedef Entry entry_type; | |
| 20 typedef HashContainerIterator<Entry> iterator; | |
| 21 | |
| 22 const entry_type* mPos; | |
| 23 const entry_type* mEnd; | |
| 24 | |
| 25 explicit HashContainerIterator(const entry_type* start, const entry_type* en d) | |
| 26 : mPos(start), mEnd(end) | |
| 27 { | |
| 28 if (mPos != mEnd && mPos->first.is_invalid()) | |
| 29 ++(*this); | |
| 30 } | |
| 31 | |
| 32 const entry_type& operator*() const | |
| 33 { | |
| 34 return *mPos; | |
| 35 } | |
| 36 | |
| 37 const entry_type* operator->() const | |
| 38 { | |
| 39 return mPos; | |
| 40 } | |
| 41 | |
| 42 iterator& operator++() | |
| 43 { | |
| 44 do { | |
| 45 ++mPos; | |
| 46 } while(mPos != mEnd && mPos->first.is_invalid()); | |
| 47 return *this; | |
| 48 } | |
| 49 | |
| 50 bool operator==(const iterator& it) const | |
| 51 { | |
| 52 return mPos == it.mPos; | |
| 53 } | |
| 54 | |
| 55 bool operator!=(const iterator& it) const | |
| 56 { | |
| 57 return mPos != it.mPos; | |
| 58 } | |
| 59 }; | |
| 60 | |
| 61 template<typename Entry> | |
| 62 struct HashContainerReference | |
| 63 { | |
| 64 typedef Entry entry_type; | |
| 65 | |
| 66 entry_type* mEntry; | |
| 67 | |
| 68 explicit HashContainerReference(entry_type* entry) | |
| 69 : mEntry(entry) | |
| 70 { | |
| 71 } | |
| 72 | |
| 73 const entry_type* operator->() const | |
| 74 { | |
| 75 return mEntry; | |
| 76 } | |
| 77 | |
| 78 operator bool() const | |
| 79 { | |
| 80 return !mEntry->first.is_invalid(); | |
| 81 } | |
| 82 }; | |
| 83 | |
| 84 template<typename Entry> | |
| 85 class HashContainer | |
| 86 { | |
| 87 public: | |
| 88 typedef Entry entry_type; | |
| 89 typedef size_t size_type; | |
| 90 typedef HashContainerIterator<Entry> const_iterator; | |
| 91 typedef HashContainerReference<const Entry> const_reference; | |
| 92 | |
| 93 private: | |
| 94 HashContainer(const HashContainer& other); | |
| 95 void operator=(const HashContainer& other); | |
| 96 | |
| 97 protected: | |
| 98 static constexpr size_type MIN_BUCKETS = 1; | |
| 99 static constexpr double LOAD_FACTOR = 0.8; | |
| 100 std::unique_ptr<entry_type[]> mBuckets; | |
| 101 size_type mBucketCount; | |
| 102 size_type mEntryCount; | |
| 103 | |
| 104 #if defined(DEBUG) | |
| 105 size_type mInsertCounter; | |
| 106 #endif | |
| 107 | |
| 108 explicit HashContainer(size_type expectedEntries = 0) | |
| 109 : mEntryCount(0) | |
| 110 { | |
| 111 expectedEntries = ceil(expectedEntries / LOAD_FACTOR); | |
| 112 mBucketCount = MIN_BUCKETS; | |
| 113 while (mBucketCount < expectedEntries) | |
| 114 mBucketCount <<= 1; | |
| 115 | |
| 116 mBuckets.reset(new entry_type[mBucketCount]); | |
| 117 // Working around https://github.com/waywardmonkeys/emscripten-trace-colle ctor/issues/2 here | |
| 118 annotate_address(reinterpret_cast<size_type*>(mBuckets.get()) - 1, "Hash t able buffer"); | |
| 119 } | |
| 120 | |
| 121 static size_type hash(const String& str) | |
| 122 { | |
| 123 // FNV-1a hash function | |
| 124 size_type result = 2166136261; | |
| 125 for (String::size_type i = 0; i < str.length(); i++) | |
| 126 result = (result ^ str[i]) * 16777619; | |
| 127 return result; | |
| 128 } | |
| 129 | |
| 130 entry_type* find_bucket(const String& key) const | |
| 131 { | |
| 132 size_type h = hash(key); | |
| 133 | |
| 134 // This does quadratic probing, effectively the following formula is used: | |
| 135 // pos = (hash + 1/2 i + 1/2 i ^ 2) mod bucketCount | |
| 136 for (size_type i = 0; ; ++i) | |
| 137 { | |
| 138 // mBucketCount is 2^n so (h & mBucketCount - 1) is equivalent to | |
| 139 // h % mBucketCount but significantly faster. | |
| 140 entry_type* entry = &mBuckets[h & mBucketCount - 1]; | |
|
sergei
2017/01/10 15:58:04
Could we put "h & (mBucketCount - 1)" to hide comp
Wladimir Palant
2017/03/13 17:42:23
Done.
| |
| 141 if (entry->first.is_invalid() || entry->first.equals(key)) | |
| 142 return entry; | |
| 143 h += i; | |
| 144 } | |
| 145 } | |
| 146 | |
| 147 void resize(size_type bucketCount) | |
| 148 { | |
| 149 std::unique_ptr<entry_type[]> oldBuckets(std::move(mBuckets)); | |
| 150 size_type oldCount = mBucketCount; | |
| 151 | |
| 152 mEntryCount = 0; | |
| 153 mBucketCount = bucketCount; | |
| 154 mBuckets.reset(new entry_type[mBucketCount]); | |
| 155 // Working around https://github.com/waywardmonkeys/emscripten-trace-colle ctor/issues/2 here | |
| 156 annotate_address(reinterpret_cast<size_type*>(mBuckets.get()) - 1, "Hash t able buffer"); | |
| 157 | |
| 158 // Copy old entries into the new buffer | |
| 159 for (size_type i = 0; i < oldCount; i++) | |
| 160 { | |
| 161 entry_type& entry = oldBuckets[i]; | |
| 162 if (!entry.first.is_invalid() && !entry.first.is_deleted()) | |
| 163 { | |
| 164 *find_bucket(entry.first) = entry; | |
| 165 mEntryCount++; | |
| 166 } | |
| 167 } | |
| 168 } | |
| 169 | |
| 170 entry_type* assign(entry_type* existing, const entry_type& entry) | |
| 171 { | |
| 172 if (existing->first.is_invalid()) | |
| 173 { | |
| 174 if (mEntryCount + 1 >= mBucketCount * LOAD_FACTOR) | |
| 175 { | |
| 176 resize(mBucketCount << 1); | |
| 177 existing = find_bucket(entry.first); | |
| 178 } | |
| 179 mEntryCount++; | |
| 180 #if defined(DEBUG) | |
| 181 mInsertCounter++; | |
| 182 #endif | |
| 183 } | |
| 184 *existing = entry; | |
| 185 return existing; | |
| 186 } | |
| 187 | |
| 188 public: | |
| 189 void insert(const entry_type& entry) | |
| 190 { | |
| 191 assign(find_bucket(entry.first), entry); | |
| 192 } | |
| 193 | |
| 194 bool erase(const String& key) | |
| 195 { | |
| 196 entry_type* entry = find_bucket(key); | |
| 197 if (entry->first.is_invalid()) | |
| 198 return false; | |
| 199 | |
| 200 entry->first.erase(); | |
| 201 return true; | |
| 202 } | |
| 203 | |
| 204 const_reference find(const String& key) const | |
| 205 { | |
| 206 return const_reference(find_bucket(key)); | |
| 207 } | |
| 208 | |
| 209 const_iterator begin() const | |
| 210 { | |
| 211 return const_iterator(&mBuckets[0], &mBuckets[mBucketCount]); | |
| 212 } | |
| 213 | |
| 214 const_iterator end() const | |
| 215 { | |
| 216 return const_iterator(&mBuckets[mBucketCount], &mBuckets[mBucketCount]); | |
| 217 } | |
| 218 }; | |
| 219 | |
| 220 struct StringSetEntry | |
| 221 { | |
| 222 StringSetEntry() {} | |
| 223 StringSetEntry(const String& key) | |
| 224 : first(key) | |
| 225 { | |
| 226 } | |
| 227 | |
| 228 DependentString first; | |
| 229 }; | |
| 230 | |
| 231 template<typename T> | |
| 232 struct StringMapEntry | |
| 233 { | |
| 234 StringMapEntry() {} | |
| 235 StringMapEntry(const String& key) | |
| 236 : first(key), second() | |
| 237 { | |
| 238 } | |
| 239 StringMapEntry(const String& key, T value) | |
| 240 : first(key), second(value) | |
| 241 { | |
| 242 } | |
| 243 | |
| 244 DependentString first; | |
| 245 T second; | |
| 246 }; | |
| 247 | |
| 248 template<typename T> | |
| 249 struct StringMapEntryReference : public HashContainerReference<StringMapEntry< T>> | |
| 250 { | |
| 251 typedef HashContainerReference<StringMapEntry<T>> super; | |
| 252 typedef typename super::entry_type entry_type; | |
| 253 typedef StringMap<T> map_type; | |
| 254 | |
| 255 map_type* mMap; | |
| 256 | |
| 257 #if defined(DEBUG) | |
| 258 typename map_type::size_type mInsertCounter; | |
| 259 typename map_type::size_type mHash; | |
| 260 #endif | |
| 261 | |
| 262 StringMapEntryReference(map_type* map, const String& key, entry_type* entry) | |
| 263 : super(entry), mMap(map) | |
| 264 { | |
| 265 #if defined(DEBUG) | |
| 266 mInsertCounter = mMap->mInsertCounter; | |
| 267 mHash = mMap->hash(key); | |
| 268 #endif | |
| 269 } | |
| 270 | |
| 271 void assign(const String& key, const T& value) | |
| 272 { | |
| 273 #if defined(DEBUG) | |
| 274 assert(mInsertCounter == mMap->mInsertCounter, | |
| 275 u"There should be no insert operations performed between map.find() an d assign()"_str); | |
| 276 assert(mHash == mMap->hash(key), | |
| 277 u"The keys used in map.find() and assign() should be identical"_str); | |
| 278 #endif | |
| 279 | |
| 280 mMap->assign(this->mEntry, entry_type(key, value)); | |
| 281 } | |
| 282 }; | |
| 283 } | |
| 284 | |
| 285 class StringSet | |
| 286 : public StringMap_internal::HashContainer<StringMap_internal::StringSetEntry> | |
| 287 { | |
| 288 }; | |
| 289 | |
| 290 template<typename T> | |
| 291 class StringMap | |
| 292 : public StringMap_internal::HashContainer<StringMap_internal::StringMapEntry< T>> | |
| 293 { | |
| 294 public: | |
| 295 typedef StringMap_internal::HashContainer<StringMap_internal::StringMapEntry<T >> super; | |
| 296 typedef typename super::size_type size_type; | |
| 297 typedef typename super::entry_type entry_type; | |
| 298 typedef typename super::const_reference const_reference; | |
| 299 typedef StringMap_internal::StringMapEntryReference<T> reference; | |
| 300 friend class StringMap_internal::StringMapEntryReference<T>; | |
| 301 | |
| 302 explicit StringMap(size_type expectedEntries = 0) | |
| 303 : super(expectedEntries) | |
| 304 { | |
| 305 } | |
| 306 | |
| 307 StringMap(std::initializer_list<entry_type> list) | |
| 308 : super(list.size()) | |
| 309 { | |
| 310 for (auto it = list.begin(); it != list.end(); ++it) | |
| 311 super::insert(*it); | |
| 312 } | |
| 313 | |
| 314 ~StringMap() | |
| 315 { | |
| 316 } | |
| 317 | |
| 318 T& operator[](const String& key) | |
| 319 { | |
| 320 entry_type* entry = super::find_bucket(key); | |
| 321 if (entry->first.is_invalid()) | |
| 322 entry = super::assign(entry, key); | |
| 323 return entry->second; | |
| 324 } | |
| 325 | |
| 326 const_reference find(const String& key) const | |
| 327 { | |
| 328 return super::find(key); | |
| 329 } | |
| 330 | |
| 331 reference find(const String& key) | |
| 332 { | |
| 333 return reference(this, key, super::find_bucket(key)); | |
| 334 } | |
| 335 }; | |
| OLD | NEW |