Refactor word_db, use an unordered map

1 files changed, 10 insertions, 1 deletions

diff --git a/src/word_db.hh b/src/word_db.hh
index ee4e6ab5..999d03b0 100644
--- a/src/word_db.hh
+++ b/src/word_db.hh
@@ -5,6 +5,7 @@
 #include "interned_string.hh"
 
 #include <map>
+#include <bitset>
 
 namespace Kakoune
 {
@@ -14,12 +15,20 @@ class WordDB
 {
 public:
     WordDB(const Buffer& buffer);
+    WordDB(const WordDB&) { kak_assert(false); }
+    WordDB(WordDB&&) = default;
 
     std::vector<InternedString> find_prefix(StringView prefix);
     std::vector<InternedString> find_subsequence(StringView subsequence);
     int get_word_occurences(StringView word) const;
 
-    using WordList = std::map<InternedString, int>;
+    using UsedChars = std::bitset<64>;
+    struct WordInfo
+    {
+        UsedChars letters;
+        int refcount;
+    };
+    using WordList = std::unordered_map<InternedString, WordInfo>;
 private:
     using LineToWords = std::vector<std::vector<InternedString>>;