| 1 | // | 
|---|
| 2 | // HashTable.h | 
|---|
| 3 | // | 
|---|
| 4 | // Library: Foundation | 
|---|
| 5 | // Package: Hashing | 
|---|
| 6 | // Module:  HashTable | 
|---|
| 7 | // | 
|---|
| 8 | // Definition of the HashTable class. | 
|---|
| 9 | // | 
|---|
| 10 | // Copyright (c) 2006, Applied Informatics Software Engineering GmbH. | 
|---|
| 11 | // and Contributors. | 
|---|
| 12 | // | 
|---|
| 13 | // SPDX-License-Identifier:	BSL-1.0 | 
|---|
| 14 | // | 
|---|
| 15 |  | 
|---|
| 16 |  | 
|---|
| 17 | #ifndef Foundation_HashTable_INCLUDED | 
|---|
| 18 | #define Foundation_HashTable_INCLUDED | 
|---|
| 19 |  | 
|---|
| 20 |  | 
|---|
| 21 | #include "Poco/Foundation.h" | 
|---|
| 22 | #include "Poco/Exception.h" | 
|---|
| 23 | #include "Poco/HashFunction.h" | 
|---|
| 24 | #include "Poco/HashStatistic.h" | 
|---|
| 25 | #include <vector> | 
|---|
| 26 | #include <map> | 
|---|
| 27 | #include <cstddef> | 
|---|
| 28 | #include <cstring> | 
|---|
| 29 |  | 
|---|
| 30 |  | 
|---|
| 31 | namespace Poco { | 
|---|
| 32 |  | 
|---|
| 33 |  | 
|---|
| 34 | //@ deprecated | 
|---|
| 35 | template <class Key, class Value, class KeyHashFunction = HashFunction<Key> > | 
|---|
| 36 | class HashTable | 
|---|
| 37 | /// A HashTable stores a key value pair that can be looked up via a hashed key. | 
|---|
| 38 | /// | 
|---|
| 39 | /// Collision handling is done via overflow maps(!). With small hash tables performance of this | 
|---|
| 40 | /// data struct will be closer to that a map than a hash table, i.e. slower. On the plus side, | 
|---|
| 41 | /// this class offers remove operations. Also HashTable full errors are not possible. If a fast | 
|---|
| 42 | /// HashTable implementation is needed and the remove operation is not required, use SimpleHashTable | 
|---|
| 43 | /// instead. | 
|---|
| 44 | /// | 
|---|
| 45 | /// This class is NOT thread safe. | 
|---|
| 46 | { | 
|---|
| 47 | public: | 
|---|
| 48 | typedef std::map<Key, Value> HashEntryMap; | 
|---|
| 49 | typedef HashEntryMap**       HashTableVector; | 
|---|
| 50 |  | 
|---|
| 51 | typedef typename HashEntryMap::const_iterator ConstIterator; | 
|---|
| 52 | typedef typename HashEntryMap::iterator Iterator; | 
|---|
| 53 |  | 
|---|
| 54 | HashTable(UInt32 initialSize = 251): | 
|---|
| 55 | _entries(0), | 
|---|
| 56 | _size(0), | 
|---|
| 57 | _maxCapacity(initialSize) | 
|---|
| 58 | /// Creates the HashTable. | 
|---|
| 59 | { | 
|---|
| 60 | _entries = new HashEntryMap*[initialSize]; | 
|---|
| 61 | memset(_entries, '\0', sizeof(HashEntryMap*)*initialSize); | 
|---|
| 62 | } | 
|---|
| 63 |  | 
|---|
| 64 | HashTable(const HashTable& ht): | 
|---|
| 65 | _entries(new HashEntryMap*[ht._maxCapacity]), | 
|---|
| 66 | _size(ht._size), | 
|---|
| 67 | _maxCapacity(ht._maxCapacity) | 
|---|
| 68 | { | 
|---|
| 69 | for (UInt32 i = 0; i < _maxCapacity; ++i) | 
|---|
| 70 | { | 
|---|
| 71 | if (ht._entries[i]) | 
|---|
| 72 | _entries[i] = new HashEntryMap(ht._entries[i]->begin(), ht._entries[i]->end()); | 
|---|
| 73 | else | 
|---|
| 74 | _entries[i] = 0; | 
|---|
| 75 | } | 
|---|
| 76 | } | 
|---|
| 77 |  | 
|---|
| 78 | ~HashTable() | 
|---|
| 79 | /// Destroys the HashTable. | 
|---|
| 80 | { | 
|---|
| 81 | clear(); | 
|---|
| 82 | } | 
|---|
| 83 |  | 
|---|
| 84 | HashTable& operator = (const HashTable& ht) | 
|---|
| 85 | { | 
|---|
| 86 | if (this != &ht) | 
|---|
| 87 | { | 
|---|
| 88 | clear(); | 
|---|
| 89 | _maxCapacity = ht._maxCapacity; | 
|---|
| 90 | poco_assert_dbg (_entries == 0); | 
|---|
| 91 | _entries = new HashEntryMap*[_maxCapacity]; | 
|---|
| 92 | _size = ht._size; | 
|---|
| 93 |  | 
|---|
| 94 | for (UInt32 i = 0; i < _maxCapacity; ++i) | 
|---|
| 95 | { | 
|---|
| 96 | if (ht._entries[i]) | 
|---|
| 97 | _entries[i] = new HashEntryMap(ht._entries[i]->begin(), ht._entries[i]->end()); | 
|---|
| 98 | else | 
|---|
| 99 | _entries[i] = 0; | 
|---|
| 100 | } | 
|---|
| 101 | } | 
|---|
| 102 | return *this; | 
|---|
| 103 | } | 
|---|
| 104 |  | 
|---|
| 105 | void clear() | 
|---|
| 106 | { | 
|---|
| 107 | if (!_entries) | 
|---|
| 108 | return; | 
|---|
| 109 | for (UInt32 i = 0; i < _maxCapacity; ++i) | 
|---|
| 110 | { | 
|---|
| 111 | delete _entries[i]; | 
|---|
| 112 | } | 
|---|
| 113 | delete[] _entries; | 
|---|
| 114 | _entries     = 0; | 
|---|
| 115 | _size        = 0; | 
|---|
| 116 | _maxCapacity = 0; | 
|---|
| 117 | } | 
|---|
| 118 |  | 
|---|
| 119 | UInt32 insert(const Key& key, const Value& value) | 
|---|
| 120 | /// Returns the hash value of the inserted item. | 
|---|
| 121 | /// Throws an exception if the entry was already inserted | 
|---|
| 122 | { | 
|---|
| 123 | UInt32 hsh = hash(key); | 
|---|
| 124 | insertRaw(key, hsh, value); | 
|---|
| 125 | return hsh; | 
|---|
| 126 | } | 
|---|
| 127 |  | 
|---|
| 128 | Value& insertRaw(const Key& key, UInt32 hsh, const Value& value) | 
|---|
| 129 | /// Returns the hash value of the inserted item. | 
|---|
| 130 | /// Throws an exception if the entry was already inserted | 
|---|
| 131 | { | 
|---|
| 132 | if (!_entries[hsh]) | 
|---|
| 133 | _entries[hsh] = new HashEntryMap(); | 
|---|
| 134 | std::pair<typename HashEntryMap::iterator, bool> res(_entries[hsh]->insert(std::make_pair(key, value))); | 
|---|
| 135 | if (!res.second) | 
|---|
| 136 | throw InvalidArgumentException( "HashTable::insert, key already exists."); | 
|---|
| 137 | _size++; | 
|---|
| 138 | return res.first->second; | 
|---|
| 139 | } | 
|---|
| 140 |  | 
|---|
| 141 | UInt32 update(const Key& key, const Value& value) | 
|---|
| 142 | /// Returns the hash value of the inserted item. | 
|---|
| 143 | /// Replaces an existing entry if it finds one | 
|---|
| 144 | { | 
|---|
| 145 | UInt32 hsh = hash(key); | 
|---|
| 146 | updateRaw(key, hsh, value); | 
|---|
| 147 | return hsh; | 
|---|
| 148 | } | 
|---|
| 149 |  | 
|---|
| 150 | void updateRaw(const Key& key, UInt32 hsh, const Value& value) | 
|---|
| 151 | /// Returns the hash value of the inserted item. | 
|---|
| 152 | /// Replaces an existing entry if it finds one | 
|---|
| 153 | { | 
|---|
| 154 | if (!_entries[hsh]) | 
|---|
| 155 | _entries[hsh] = new HashEntryMap(); | 
|---|
| 156 | std::pair<Iterator, bool> res = _entries[hsh]->insert(std::make_pair(key, value)); | 
|---|
| 157 | if (res.second == false) | 
|---|
| 158 | res.first->second = value; | 
|---|
| 159 | else | 
|---|
| 160 | _size++; | 
|---|
| 161 | } | 
|---|
| 162 |  | 
|---|
| 163 | void remove(const Key& key) | 
|---|
| 164 | { | 
|---|
| 165 | UInt32 hsh = hash(key); | 
|---|
| 166 | removeRaw(key, hsh); | 
|---|
| 167 | } | 
|---|
| 168 |  | 
|---|
| 169 | void removeRaw(const Key& key, UInt32 hsh) | 
|---|
| 170 | /// Performance version, allows to specify the hash value | 
|---|
| 171 | { | 
|---|
| 172 | if (_entries[hsh]) | 
|---|
| 173 | { | 
|---|
| 174 | _size -= _entries[hsh]->erase(key); | 
|---|
| 175 | } | 
|---|
| 176 | } | 
|---|
| 177 |  | 
|---|
| 178 | UInt32 hash(const Key& key) const | 
|---|
| 179 | { | 
|---|
| 180 | return _hash(key, _maxCapacity); | 
|---|
| 181 | } | 
|---|
| 182 |  | 
|---|
| 183 | const Value& get(const Key& key) const | 
|---|
| 184 | /// Throws an exception if the value does not exist | 
|---|
| 185 | { | 
|---|
| 186 | UInt32 hsh = hash(key); | 
|---|
| 187 | return getRaw(key, hsh); | 
|---|
| 188 | } | 
|---|
| 189 |  | 
|---|
| 190 | const Value& getRaw(const Key& key, UInt32 hsh) const | 
|---|
| 191 | /// Throws an exception if the value does not exist | 
|---|
| 192 | { | 
|---|
| 193 | if (!_entries[hsh]) | 
|---|
| 194 | throw InvalidArgumentException( "key not found"); | 
|---|
| 195 |  | 
|---|
| 196 | ConstIterator it = _entries[hsh]->find(key); | 
|---|
| 197 | if (it == _entries[hsh]->end()) | 
|---|
| 198 | throw InvalidArgumentException( "key not found"); | 
|---|
| 199 |  | 
|---|
| 200 | return it->second; | 
|---|
| 201 | } | 
|---|
| 202 |  | 
|---|
| 203 | Value& get(const Key& key) | 
|---|
| 204 | /// Throws an exception if the value does not exist | 
|---|
| 205 | { | 
|---|
| 206 | UInt32 hsh = hash(key); | 
|---|
| 207 | return const_cast<Value&>(getRaw(key, hsh)); | 
|---|
| 208 | } | 
|---|
| 209 |  | 
|---|
| 210 | const Value& operator [] (const Key& key) const | 
|---|
| 211 | { | 
|---|
| 212 | return get(key); | 
|---|
| 213 | } | 
|---|
| 214 |  | 
|---|
| 215 | Value& operator [] (const Key& key) | 
|---|
| 216 | { | 
|---|
| 217 | UInt32 hsh = hash(key); | 
|---|
| 218 |  | 
|---|
| 219 | if (!_entries[hsh]) | 
|---|
| 220 | return insertRaw(key, hsh, Value()); | 
|---|
| 221 |  | 
|---|
| 222 | ConstIterator it = _entries[hsh]->find(key); | 
|---|
| 223 | if (it == _entries[hsh]->end()) | 
|---|
| 224 | return insertRaw(key, hsh, Value()); | 
|---|
| 225 |  | 
|---|
| 226 | return it->second; | 
|---|
| 227 | } | 
|---|
| 228 |  | 
|---|
| 229 | const Key& getKeyRaw(const Key& key, UInt32 hsh) | 
|---|
| 230 | /// Throws an exception if the key does not exist. returns a reference to the internally | 
|---|
| 231 | /// stored key. Useful when someone does an insert and wants for performance reason only to store | 
|---|
| 232 | /// a pointer to the key in another collection | 
|---|
| 233 | { | 
|---|
| 234 | if (!_entries[hsh]) | 
|---|
| 235 | throw InvalidArgumentException( "key not found"); | 
|---|
| 236 | ConstIterator it = _entries[hsh]->find(key); | 
|---|
| 237 | if (it == _entries[hsh]->end()) | 
|---|
| 238 | throw InvalidArgumentException( "key not found"); | 
|---|
| 239 | return it->first; | 
|---|
| 240 | } | 
|---|
| 241 |  | 
|---|
| 242 | bool get(const Key& key, Value& v) const | 
|---|
| 243 | /// Sets v to the found value, returns false if no value was found | 
|---|
| 244 | { | 
|---|
| 245 | UInt32 hsh = hash(key); | 
|---|
| 246 | return getRaw(key, hsh, v); | 
|---|
| 247 | } | 
|---|
| 248 |  | 
|---|
| 249 | bool getRaw(const Key& key, UInt32 hsh, Value& v) const | 
|---|
| 250 | /// Sets v to the found value, returns false if no value was found | 
|---|
| 251 | { | 
|---|
| 252 | if (!_entries[hsh]) | 
|---|
| 253 | return false; | 
|---|
| 254 |  | 
|---|
| 255 | ConstIterator it = _entries[hsh]->find(key); | 
|---|
| 256 | if (it == _entries[hsh]->end()) | 
|---|
| 257 | return false; | 
|---|
| 258 |  | 
|---|
| 259 | v = it->second; | 
|---|
| 260 | return true; | 
|---|
| 261 | } | 
|---|
| 262 |  | 
|---|
| 263 | bool exists(const Key& key) | 
|---|
| 264 | { | 
|---|
| 265 | UInt32 hsh = hash(key); | 
|---|
| 266 | return existsRaw(key, hsh); | 
|---|
| 267 | } | 
|---|
| 268 |  | 
|---|
| 269 | bool existsRaw(const Key& key, UInt32 hsh) | 
|---|
| 270 | { | 
|---|
| 271 | return _entries[hsh] && (_entries[hsh]->end() != _entries[hsh]->find(key)); | 
|---|
| 272 | } | 
|---|
| 273 |  | 
|---|
| 274 | std::size_t size() const | 
|---|
| 275 | /// Returns the number of elements already inserted into the HashTable | 
|---|
| 276 | { | 
|---|
| 277 | return _size; | 
|---|
| 278 | } | 
|---|
| 279 |  | 
|---|
| 280 | UInt32 maxCapacity() const | 
|---|
| 281 | { | 
|---|
| 282 | return _maxCapacity; | 
|---|
| 283 | } | 
|---|
| 284 |  | 
|---|
| 285 | void resize(UInt32 newSize) | 
|---|
| 286 | /// Resizes the hashtable, rehashes all existing entries. Expensive! | 
|---|
| 287 | { | 
|---|
| 288 | if (_maxCapacity != newSize) | 
|---|
| 289 | { | 
|---|
| 290 | HashTableVector cpy = _entries; | 
|---|
| 291 | _entries = 0; | 
|---|
| 292 | UInt32 oldSize = _maxCapacity; | 
|---|
| 293 | _maxCapacity = newSize; | 
|---|
| 294 | _entries = new HashEntryMap*[_maxCapacity]; | 
|---|
| 295 | memset(_entries, '\0', sizeof(HashEntryMap*)*_maxCapacity); | 
|---|
| 296 |  | 
|---|
| 297 | if (_size == 0) | 
|---|
| 298 | { | 
|---|
| 299 | // no data was yet inserted | 
|---|
| 300 | delete[] cpy; | 
|---|
| 301 | return; | 
|---|
| 302 | } | 
|---|
| 303 | _size = 0; | 
|---|
| 304 | for (UInt32 i = 0; i < oldSize; ++i) | 
|---|
| 305 | { | 
|---|
| 306 | if (cpy[i]) | 
|---|
| 307 | { | 
|---|
| 308 | ConstIterator it = cpy[i]->begin(); | 
|---|
| 309 | ConstIterator itEnd = cpy[i]->end(); | 
|---|
| 310 | for (; it != itEnd; ++it) | 
|---|
| 311 | { | 
|---|
| 312 | insert(it->first, it->second); | 
|---|
| 313 | } | 
|---|
| 314 | delete cpy[i]; | 
|---|
| 315 | } | 
|---|
| 316 | } | 
|---|
| 317 | delete[] cpy; | 
|---|
| 318 | } | 
|---|
| 319 | } | 
|---|
| 320 |  | 
|---|
| 321 | HashStatistic currentState(bool details = false) const | 
|---|
| 322 | /// Returns the current internal state | 
|---|
| 323 | { | 
|---|
| 324 | UInt32 numberOfEntries = (UInt32)_size; | 
|---|
| 325 | UInt32 numZeroEntries = 0; | 
|---|
| 326 | UInt32 maxEntriesPerHash = 0; | 
|---|
| 327 | std::vector<UInt32> detailedEntriesPerHash; | 
|---|
| 328 | #ifdef _DEBUG | 
|---|
| 329 | UInt32 totalSize = 0; | 
|---|
| 330 | #endif | 
|---|
| 331 | for (UInt32 i = 0; i < _maxCapacity; ++i) | 
|---|
| 332 | { | 
|---|
| 333 | if (_entries[i]) | 
|---|
| 334 | { | 
|---|
| 335 | UInt32 size = (UInt32)_entries[i]->size(); | 
|---|
| 336 | poco_assert_dbg(size != 0); | 
|---|
| 337 | if (size > maxEntriesPerHash) | 
|---|
| 338 | maxEntriesPerHash = size; | 
|---|
| 339 | if (details) | 
|---|
| 340 | detailedEntriesPerHash.push_back(size); | 
|---|
| 341 | #ifdef _DEBUG | 
|---|
| 342 | totalSize += size; | 
|---|
| 343 | #endif | 
|---|
| 344 | } | 
|---|
| 345 | else | 
|---|
| 346 | { | 
|---|
| 347 | numZeroEntries++; | 
|---|
| 348 | if (details) | 
|---|
| 349 | detailedEntriesPerHash.push_back(0); | 
|---|
| 350 | } | 
|---|
| 351 | } | 
|---|
| 352 | #ifdef _DEBUG | 
|---|
| 353 | poco_assert_dbg(totalSize == numberOfEntries); | 
|---|
| 354 | #endif | 
|---|
| 355 | return HashStatistic(_maxCapacity, numberOfEntries, numZeroEntries, maxEntriesPerHash, detailedEntriesPerHash); | 
|---|
| 356 | } | 
|---|
| 357 |  | 
|---|
| 358 | private: | 
|---|
| 359 | HashTableVector _entries; | 
|---|
| 360 | std::size_t     _size; | 
|---|
| 361 | UInt32          _maxCapacity; | 
|---|
| 362 | KeyHashFunction _hash; | 
|---|
| 363 | }; | 
|---|
| 364 |  | 
|---|
| 365 |  | 
|---|
| 366 | } // namespace Poco | 
|---|
| 367 |  | 
|---|
| 368 |  | 
|---|
| 369 | #endif // Foundation_HashTable_INCLUDED | 
|---|
| 370 |  | 
|---|