| Left: | ||
| Right: |
| OLD | NEW |
|---|---|
| (Empty) | |
| 1 #ifndef ADBLOCK_PLUS_STRING_MAP_H | |
| 2 #define ADBLOCK_PLUS_STRING_MAP_H | |
| 3 | |
| 4 #include <cstddef> | |
| 5 #include <cmath> | |
| 6 #include <initializer_list> | |
| 7 | |
| 8 #include "String.h" | |
| 9 #include "debug.h" | |
| 10 | |
| 11 template<class Entry> | |
|
René Jeschke
2016/02/02 11:11:22
Nit: The usage of `class` for declaring template p
Wladimir Palant
2016/02/02 17:55:18
Done.
| |
| 12 struct _HashContainerIterator | |
| 13 { | |
| 14 typedef Entry entry_type; | |
| 15 typedef _HashContainerIterator<Entry> iterator; | |
| 16 | |
| 17 const entry_type* mPos; | |
| 18 const entry_type* mEnd; | |
| 19 | |
| 20 explicit _HashContainerIterator(const entry_type* start, const entry_type* end ) | |
| 21 : mPos(start), mEnd(end) | |
| 22 { | |
| 23 if (mPos != mEnd && mPos->first.is_invalid()) | |
| 24 ++(*this); | |
| 25 } | |
| 26 | |
| 27 _HashContainerIterator() | |
| 28 { | |
| 29 } | |
| 30 | |
| 31 const entry_type& operator*() const | |
| 32 { | |
| 33 return *mPos; | |
| 34 } | |
| 35 | |
| 36 const entry_type* operator->() const | |
| 37 { | |
| 38 return mPos; | |
| 39 } | |
| 40 | |
| 41 iterator& operator++() | |
| 42 { | |
| 43 do { | |
| 44 ++mPos; | |
| 45 } while(mPos != mEnd && mPos->first.is_invalid()); | |
| 46 return *this; | |
| 47 } | |
| 48 | |
| 49 bool operator==(const iterator& it) const | |
| 50 { | |
| 51 return mPos == it.mPos; | |
| 52 } | |
| 53 | |
| 54 bool operator!=(const iterator& it) const | |
| 55 { | |
| 56 return mPos != it.mPos; | |
| 57 } | |
| 58 }; | |
| 59 | |
| 60 template<class Entry> | |
| 61 class _HashContainer | |
| 62 { | |
| 63 public: | |
| 64 typedef Entry entry_type; | |
| 65 typedef size_t size_type; | |
| 66 typedef _HashContainerIterator<Entry> iterator; | |
| 67 | |
| 68 private: | |
| 69 _HashContainer(const _HashContainer& other) {} | |
| 70 void operator=(const _HashContainer& other) {} | |
| 71 | |
| 72 protected: | |
| 73 static constexpr size_type MIN_BUCKETS = 1; | |
| 74 static constexpr double LOAD_FACTOR = 0.8; | |
| 75 entry_type* mBuckets; | |
| 76 size_type mBucketCount; | |
| 77 size_type mEntryCount; | |
| 78 | |
| 79 explicit _HashContainer(size_type expectedEntries = 0) | |
| 80 : mEntryCount(0) | |
| 81 { | |
| 82 expectedEntries = ceil(expectedEntries / LOAD_FACTOR); | |
| 83 mBucketCount = MIN_BUCKETS; | |
| 84 while (mBucketCount < expectedEntries) | |
| 85 mBucketCount <<= 1; | |
| 86 | |
| 87 mBuckets = new entry_type[mBucketCount]; | |
| 88 // Working around https://github.com/waywardmonkeys/emscripten-trace-collect or/issues/2 here | |
| 89 annotate_address(reinterpret_cast<size_type*>(mBuckets) - 1, "Hash table buf fer"); | |
| 90 } | |
| 91 | |
| 92 ~_HashContainer() | |
| 93 { | |
| 94 delete[] mBuckets; | |
| 95 } | |
| 96 | |
| 97 static size_type hash(const String& str) | |
| 98 { | |
| 99 // FNV-1a hash function | |
| 100 size_type result = 2166136261; | |
| 101 for (String::size_type i = 0; i < str.length(); i++) | |
| 102 result = (result ^ str[i]) * 16777619; | |
| 103 return result; | |
| 104 } | |
| 105 | |
| 106 entry_type* find_entry(const String& key) const | |
| 107 { | |
| 108 size_type h = hash(key); | |
| 109 | |
| 110 // This does quadratic probing, effectively the following formula is used: | |
| 111 // pos = (hash + 1/2 i + 1/2 i ^ 2) mod bucketCount | |
| 112 for (size_type i = 0; ; ++i) | |
| 113 { | |
| 114 entry_type* entry = mBuckets + h % mBucketCount; | |
| 115 if (entry->first.is_invalid() || entry->first.equals(key)) | |
| 116 return entry; | |
| 117 h += i; | |
| 118 } | |
| 119 } | |
| 120 | |
| 121 void resize(size_type bucketCount) | |
| 122 { | |
| 123 entry_type* oldBuckets = mBuckets; | |
| 124 size_type oldCount = mBucketCount; | |
| 125 | |
| 126 mEntryCount = 0; | |
| 127 mBucketCount = bucketCount; | |
| 128 mBuckets = new entry_type[mBucketCount]; | |
| 129 // Working around https://github.com/waywardmonkeys/emscripten-trace-collect or/issues/2 here | |
| 130 annotate_address(reinterpret_cast<size_type*>(mBuckets) - 1, "Hash table buf fer"); | |
| 131 | |
| 132 for (size_type i = 0; i < oldCount; i++) | |
| 133 { | |
| 134 entry_type& entry = oldBuckets[i]; | |
| 135 if (!entry.first.is_invalid()) | |
| 136 { | |
| 137 *find_entry(entry.first) = entry; | |
| 138 mEntryCount++; | |
| 139 } | |
| 140 } | |
| 141 | |
| 142 delete[] oldBuckets; | |
| 143 } | |
| 144 | |
| 145 entry_type* assign(entry_type* existing, const entry_type& entry) | |
| 146 { | |
| 147 bool added = existing->first.is_invalid(); | |
| 148 *existing = entry; | |
| 149 if (added) | |
| 150 { | |
| 151 mEntryCount++; | |
| 152 if (mEntryCount >= mBucketCount * LOAD_FACTOR) | |
| 153 { | |
| 154 resize(mBucketCount << 1); | |
| 155 existing = find_entry(entry.first); | |
| 156 } | |
| 157 } | |
| 158 return existing; | |
| 159 } | |
| 160 | |
| 161 public: | |
| 162 void insert(const entry_type& entry) | |
| 163 { | |
| 164 assign(find_entry(entry.first), entry); | |
| 165 } | |
| 166 | |
| 167 iterator find(const String& key) const | |
| 168 { | |
| 169 entry_type* entry = find_entry(key); | |
| 170 if (entry->first.is_invalid()) | |
| 171 return end(); | |
| 172 else | |
| 173 return iterator(entry, mBuckets + mBucketCount); | |
| 174 } | |
| 175 | |
| 176 iterator begin() const | |
| 177 { | |
| 178 return iterator(mBuckets, mBuckets + mBucketCount); | |
| 179 } | |
| 180 | |
| 181 iterator end() const | |
| 182 { | |
| 183 return iterator(mBuckets + mBucketCount, mBuckets + mBucketCount); | |
| 184 } | |
| 185 }; | |
| 186 | |
| 187 struct _StringSetEntry | |
| 188 { | |
| 189 _StringSetEntry() {} | |
| 190 _StringSetEntry(const String& key) | |
| 191 : first(key) | |
| 192 { | |
| 193 } | |
| 194 | |
| 195 String first; | |
| 196 }; | |
| 197 | |
| 198 class StringSet : public _HashContainer<_StringSetEntry> | |
| 199 { | |
| 200 }; | |
| 201 | |
| 202 template<class T> | |
| 203 struct _StringMapEntry | |
| 204 { | |
| 205 _StringMapEntry() {} | |
| 206 _StringMapEntry(const String& key) | |
| 207 : first(key) | |
| 208 { | |
| 209 } | |
| 210 _StringMapEntry(const String& key, T value) | |
| 211 : first(key), second(value) | |
| 212 { | |
| 213 } | |
| 214 | |
| 215 String first; | |
| 216 T second; | |
| 217 }; | |
| 218 | |
| 219 template<class T> | |
| 220 class StringMap : public _HashContainer<_StringMapEntry<T>> | |
| 221 { | |
| 222 public: | |
| 223 typedef _HashContainer<_StringMapEntry<T>> super; | |
| 224 typedef typename super::size_type size_type; | |
| 225 typedef typename super::entry_type entry_type; | |
| 226 | |
| 227 explicit StringMap(size_type expectedEntries = 0) | |
| 228 : super(expectedEntries) | |
| 229 { | |
| 230 } | |
| 231 | |
| 232 StringMap(std::initializer_list<entry_type> list) | |
| 233 : super(list.size()) | |
| 234 { | |
| 235 for (auto it = list.begin(); it != list.end(); ++it) | |
| 236 super::insert(*it); | |
| 237 } | |
| 238 | |
| 239 ~StringMap() | |
| 240 { | |
| 241 } | |
| 242 | |
| 243 T& operator[](const String& key) | |
| 244 { | |
| 245 entry_type* entry = super::find_entry(key); | |
| 246 if (entry->first.is_invalid()) | |
| 247 entry = super::assign(entry, key); | |
| 248 return entry->second; | |
| 249 } | |
| 250 }; | |
| 251 | |
| 252 #endif | |
| OLD | NEW |