#pragma once
#include <string_view>
#include "Common.hpp"
#include "Segments.hpp"
#include "UTF8Util.hpp"
namespace opencc {
class OPENCC_EXPORT DictEntry {
public:
virtual ~DictEntry() {}
virtual std::string_view KeyView() const = 0;
virtual std::string_view GetDefaultView() const = 0;
virtual std::string Key() const { return std::string(KeyView()); }
virtual std::vector<std::string> Values() const = 0;
virtual std::string GetDefault() const { return std::string(GetDefaultView()); }
virtual size_t NumValues() const = 0;
virtual std::string ToString() const = 0;
size_t KeyLength() const { return KeyView().length(); }
bool operator<(const DictEntry& that) const { return KeyView() < that.KeyView(); }
bool operator==(const DictEntry& that) const { return KeyView() == that.KeyView(); }
static bool UPtrLessThan(const std::unique_ptr<DictEntry>& a,
const std::unique_ptr<DictEntry>& b) {
return *a < *b;
}
};
class OPENCC_EXPORT NoValueDictEntry : public DictEntry {
public:
NoValueDictEntry(const std::string& _key) : key(_key) {}
virtual ~NoValueDictEntry() {}
std::string Key() const override { return key; }
std::string_view KeyView() const override { return key; }
std::string GetDefault() const override { return key; }
std::string_view GetDefaultView() const override { return key; }
std::vector<std::string> Values() const override {
return std::vector<std::string>();
}
size_t NumValues() const override { return 0; }
std::string ToString() const override { return key; }
private:
std::string key;
};
class OPENCC_EXPORT SingleValueDictEntry : public DictEntry {
public:
virtual std::string Value() const = 0;
virtual std::string_view ValueView() const = 0;
std::string_view GetDefaultView() const override { return ValueView(); }
std::vector<std::string> Values() const override {
return std::vector<std::string>{Value()};
}
size_t NumValues() const override { return 1; }
std::string ToString() const override {
return std::string(KeyView()) + "\t" + Value();
}
};
class OPENCC_EXPORT StrSingleValueDictEntry : public SingleValueDictEntry {
public:
StrSingleValueDictEntry(const std::string& _key, const std::string& _value)
: key(_key), value(_value) {}
virtual ~StrSingleValueDictEntry() {}
std::string Key() const override { return key; }
std::string_view KeyView() const override { return key; }
std::string Value() const override { return value; }
std::string GetDefault() const override { return value; }
std::string_view ValueView() const override { return value; }
private:
std::string key;
std::string value;
};
class OPENCC_EXPORT MultiValueDictEntry : public DictEntry {
public:
virtual std::string ToString() const;
};
class OPENCC_EXPORT StrMultiValueDictEntry : public MultiValueDictEntry {
public:
StrMultiValueDictEntry(const std::string& _key,
const std::vector<std::string>& _values)
: key(_key), values(_values) {}
virtual ~StrMultiValueDictEntry() {}
std::string Key() const override { return key; }
std::string_view KeyView() const override { return key; }
std::string GetDefault() const override {
return values.empty() ? key : values[0];
}
std::string_view GetDefaultView() const override {
return values.empty() ? std::string_view(key) : std::string_view(values[0]);
}
size_t NumValues() const override { return values.size(); }
std::vector<std::string> Values() const override { return values; }
private:
std::string key;
std::vector<std::string> values;
};
class OPENCC_EXPORT DictEntryFactory {
public:
static DictEntry* New(const std::string& key) {
return new NoValueDictEntry(key);
}
static DictEntry* New(const std::string& key, const std::string& value) {
return new StrSingleValueDictEntry(key, value);
}
static DictEntry* New(const std::string& key,
const std::vector<std::string>& values) {
if (values.size() == 0) {
return New(key);
} else if (values.size() == 1) {
return New(key, values.front());
}
return new StrMultiValueDictEntry(key, values);
}
static DictEntry* New(const DictEntry* entry) {
if (entry->NumValues() == 0) {
return new NoValueDictEntry(entry->Key());
} else if (entry->NumValues() == 1) {
return new StrSingleValueDictEntry(entry->Key(), entry->Values().front());
} else {
return new StrMultiValueDictEntry(entry->Key(), entry->Values());
}
}
};
}