kakoune/src/word_db.hh

65 lines
1.5 KiB
C++
Raw Normal View History

#ifndef word_db_hh_INCLUDED
#define word_db_hh_INCLUDED
#include "buffer.hh"
#include "interned_string.hh"
#include "unordered_map.hh"
#include "vector.hh"
2014-10-28 20:23:02 +01:00
#include <bitset>
namespace Kakoune
{
2014-12-23 20:32:42 +01:00
using UsedLetters = std::bitset<64>;
UsedLetters used_letters(StringView str);
// maintain a database of words available in a buffer
class WordDB
{
public:
WordDB(const Buffer& buffer);
2014-12-22 21:08:53 +01:00
WordDB(const WordDB&) = delete;
2014-10-28 20:23:02 +01:00
WordDB(WordDB&&) = default;
2014-12-23 20:32:42 +01:00
using WordList = std::vector<InternedString>;
template<typename MatchFunc>
WordList find_matching(StringView str, MatchFunc match)
{
update_db();
const UsedLetters letters = used_letters(str);
std::vector<InternedString> res;
for (auto&& word : m_words)
{
if ((letters & word.second.letters) == letters and
match(word.first, str))
res.push_back(word.first);
}
return res;
}
int get_word_occurences(StringView word) const;
2014-10-28 20:23:02 +01:00
struct WordInfo
{
2014-12-23 20:32:42 +01:00
UsedLetters letters;
2014-10-28 20:23:02 +01:00
int refcount;
};
using WordToInfo = UnorderedMap<InternedString, WordInfo, MemoryDomain::WordDB>;
private:
using LineToWords = Vector<WordList, MemoryDomain::WordDB>;
void update_db();
2014-12-23 20:32:42 +01:00
void add_words(const WordList& words);
void remove_words(const WordList& words);
safe_ptr<const Buffer> m_buffer;
size_t m_timestamp;
2014-12-23 20:32:42 +01:00
WordToInfo m_words;
LineToWords m_line_to_words;
};
}
#endif // word_db_hh_INCLUDED