mirror of
https://github.com/Chatterino/chatterino2.git
synced 2024-11-13 19:49:51 +01:00
74 lines
1.6 KiB
C++
74 lines
1.6 KiB
C++
#pragma once
|
|
|
|
#include <QString>
|
|
#include <functional>
|
|
#include <set>
|
|
#include <unordered_map>
|
|
|
|
namespace chatterino {
|
|
class Prefix
|
|
{
|
|
public:
|
|
Prefix(const QString &string);
|
|
bool operator==(const Prefix &other) const;
|
|
bool isStartOf(const QString &string) const;
|
|
|
|
private:
|
|
QChar first;
|
|
QChar second;
|
|
|
|
friend struct std::hash<Prefix>;
|
|
};
|
|
} // namespace chatterino
|
|
|
|
namespace std {
|
|
template <>
|
|
struct hash<chatterino::Prefix> {
|
|
size_t operator()(const chatterino::Prefix &prefix) const
|
|
{
|
|
return (size_t(prefix.first.unicode()) << 16) |
|
|
size_t(prefix.second.unicode());
|
|
}
|
|
};
|
|
} // namespace std
|
|
|
|
namespace chatterino {
|
|
class UsernameSet
|
|
{
|
|
public:
|
|
static constexpr int PrefixLength = 2;
|
|
|
|
using Iterator = std::set<QString>::iterator;
|
|
using ConstIterator = std::set<QString>::const_iterator;
|
|
|
|
class Range
|
|
{
|
|
public:
|
|
Range(ConstIterator start, ConstIterator end);
|
|
|
|
ConstIterator begin();
|
|
ConstIterator end();
|
|
|
|
private:
|
|
ConstIterator start_;
|
|
ConstIterator end_;
|
|
};
|
|
|
|
ConstIterator begin() const;
|
|
ConstIterator end() const;
|
|
Range subrange(const Prefix &prefix) const;
|
|
|
|
std::set<QString>::size_type size() const;
|
|
|
|
std::pair<Iterator, bool> insert(const QString &value);
|
|
std::pair<Iterator, bool> insert(QString &&value);
|
|
|
|
private:
|
|
void insertPrefix(const QString &string);
|
|
|
|
std::set<QString> items;
|
|
std::unordered_map<Prefix, QString> firstKeyForPrefix;
|
|
};
|
|
|
|
} // namespace chatterino
|