Sumire-Desktop/BuiltInSystemLexicon.cpp at main · KazumaProject/Sumire-Desktop · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
#include "BuiltInSystemLexicon.h"

namespace
{
LexiconEntry MakeEntry(const std::wstring& reading, const std::wstring& surface, int rank)
{
    LexiconEntry entry;
    entry.reading = reading;
    entry.surface = surface;
    entry.wordCost = 1000 + (rank * 100);
    entry.dictionaryKind = DictionaryKind::System;
    entry.sourceEntryId = static_cast<std::uint32_t>(rank);
    return entry;
}
}

BuiltInSystemLexicon::BuiltInSystemLexicon()
{
    _dictionary[L"\u308f\u305f\u3057"] = { L"\u79c1" };
    _dictionary[L"\u306b\u307b\u3093"] = { L"\u65e5\u672c" };
    _dictionary[L"\u306b\u307b\u3093\u3054"] = { L"\u65e5\u672c\u8a9e" };
    _dictionary[L"\u304d\u3087\u3046"] = { L"\u4eca\u65e5", L"\u4eac\u90fd" };
    _dictionary[L"\u3042\u3057\u305f"] = { L"\u660e\u65e5" };
    _dictionary[L"\u3068\u3046\u304d\u3087\u3046"] = { L"\u6771\u4eac" };
    _dictionary[L"\u304b\u3093\u3058"] = { L"\u6f22\u5b57" };
    _dictionary[L"\u3078\u3093\u304b\u3093"] = { L"\u5909\u63db" };
    _dictionary[L"\u304b\u306a"] = { L"\u4eee\u540d", L"\u304b\u306a" };
    _dictionary[L"\u304c\u3063\u3053\u3046"] = { L"\u5b66\u6821" };
    _dictionary[L"\u305b\u3093\u305b\u3044"] = { L"\u5148\u751f" };
    _dictionary[L"\u3058\u304b\u3093"] = { L"\u6642\u9593" };
    _dictionary[L"\u3058\u3057\u3087"] = { L"\u8f9e\u66f8" };
    _dictionary[L"\u3066\u3093\u304d"] = { L"\u5929\u6c17" };
    _dictionary[L"\u3044\u3044"] = { L"\u826f\u3044", L"\u3044\u3044" };
    _dictionary[L"\u3072\u3089\u304c\u306a"] = { L"\u5e73\u4eee\u540d", L"\u3072\u3089\u304c\u306a" };
    _dictionary[L"\u304b\u305f\u304b\u306a"] = { L"\u7247\u4eee\u540d", L"\u30ab\u30bf\u30ab\u30ca" };
}

DictionaryKind BuiltInSystemLexicon::GetKind() const
{
    return DictionaryKind::System;
}

void BuiltInSystemLexicon::LookupPrefix(const std::wstring& reading, size_t start, std::vector<LexiconEntry>* out) const
{
    if (out == nullptr)
    {
        return;
    }

    out->clear();

    for (const auto& pair : _dictionary)
    {
        const std::wstring& key = pair.first;
        if (start + key.size() > reading.size())
        {
            continue;
        }

        if (reading.compare(start, key.size(), key) == 0)
        {
            AppendEntries(key, out);
        }
    }
}

void BuiltInSystemLexicon::LookupExact(const std::wstring& reading, std::vector<LexiconEntry>* out) const
{
    if (out == nullptr)
    {
        return;
    }

    out->clear();

    auto it = _dictionary.find(reading);
    if (it == _dictionary.end())
    {
        return;
    }

    AppendEntries(it->first, out);
}

void BuiltInSystemLexicon::AppendEntries(const std::wstring& key, std::vector<LexiconEntry>* out) const
{
    auto it = _dictionary.find(key);
    if (it == _dictionary.end())
    {
        return;
    }

    const std::vector<std::wstring>& values = it->second;
    for (size_t index = 0; index < values.size(); ++index)
    {
        out->push_back(MakeEntry(key, values[index], static_cast<int>(index)));
    }
}