OLD | NEW |
(Empty) | |
| 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 |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #ifndef THIRD_PARTY_LIBADDRESSINPUT_CHROMIUM_TRIE_H_ |
| 6 #define THIRD_PARTY_LIBADDRESSINPUT_CHROMIUM_TRIE_H_ |
| 7 |
| 8 #include <stdint.h> |
| 9 #include <map> |
| 10 #include <set> |
| 11 #include <vector> |
| 12 |
| 13 namespace autofill { |
| 14 |
| 15 // A prefix search tree. Can return all objects whose keys start with a prefix |
| 16 // byte sequence. |
| 17 // |
| 18 // Maps keys to multiple objects. This property is useful when mapping region |
| 19 // names to region objects. For example, there's a "St. Petersburg" in Florida, |
| 20 // and there's a "St. Petersburg" in Russia. A lookup for "St. Petersburg" key |
| 21 // should return two distinct objects. |
| 22 template <typename T> |
| 23 class Trie { |
| 24 public: |
| 25 Trie(); |
| 26 ~Trie(); |
| 27 |
| 28 // Returns true if no data was added in AddDataForKey(). |
| 29 bool empty() const { return data_list_.empty() && sub_nodes_.empty(); } |
| 30 |
| 31 // Adds a mapping from the 0 terminated |key| to |data_item|. Can be called |
| 32 // with the same |key| multiple times. |
| 33 void AddDataForKey(const std::vector<uint8_t>& key, const T& data_item); |
| 34 |
| 35 // Adds all objects whose keys start with the 0 terminated |key_prefix| to the |
| 36 // |results| parameter. The |results| parameter should not be NULL. |
| 37 void FindDataForKeyPrefix(const std::vector<uint8_t>& key_prefix, |
| 38 std::set<T>* results) const; |
| 39 |
| 40 private: |
| 41 // All objects for this node in the Trie. This field is a collection to enable |
| 42 // mapping the same key to multiple objects. |
| 43 std::set<T> data_list_; |
| 44 |
| 45 // Trie sub nodes. The root Trie stores the objects for the empty key. The |
| 46 // children of the root Trie store the objects for the one-byte keys. The |
| 47 // grand-children of the root Trie store the objects for the two-byte keys, |
| 48 // and so on. |
| 49 std::map<uint8_t, Trie<T> > sub_nodes_; |
| 50 }; |
| 51 |
| 52 } // namespace autofill |
| 53 |
| 54 #endif // THIRD_PARTY_LIBADDRESSINPUT_CHROMIUM_TRIE_H_ |
OLD | NEW |