kakoune/src/word_db.hh

#ifndef word_db_hh_INCLUDED
#define word_db_hh_INCLUDED

#include "buffer.hh"
#include "shared_string.hh"
#include "unordered_map.hh"
#include "vector.hh"

#include <bitset>

namespace Kakoune
{

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);
    WordDB(const WordDB&) = delete;
    WordDB(WordDB&&) = default;

    struct RankedWord
    {
        StringView word;
        int rank;
    };
    using RankedWordList = Vector<RankedWord>;

    RankedWordList find_matching(StringView str);

    int get_word_occurences(StringView word) const;
private:
    void update_db();
    void add_words(const SharedString& line);
    void remove_words(const SharedString& line);

    struct WordInfo
    {
        UsedLetters letters;
        int refcount;
    };
    using WordToInfo = UnorderedMap<SharedString, WordInfo, MemoryDomain::WordDB>;
    using Lines = Vector<StringDataPtr, MemoryDomain::WordDB>;

    SafePtr<const Buffer> m_buffer;
    size_t m_timestamp;
    WordToInfo m_words;
    Lines m_lines;
};

}

#endif // word_db_hh_INCLUDED