summaryrefslogtreecommitdiff
path: root/ext/bg/dictionary.js
blob: a68c2daf85dae8ec8c3a48a114c721ca8fa78ba6 (plain)
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
/*
 * Copyright (C) 2016  Alex Yatskov <alex@foosoft.net>
 * Author: Alex Yatskov <alex@foosoft.net>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */


class Dictionary {
    constructor() {
        this.terms       = [];
        this.termIndices = {};

        this.kanji        = [];
        this.kanjiIndices = {};
    }

    addTermData(terms) {
        let index = this.terms.length;
        for (const [e, r, g, t] of terms) {
            this.storeIndex(this.termIndices, e, index);
            this.storeIndex(this.termIndices, r, index++);
            this.terms.push([e, r, g, t]);
        }
    }

    addKanjiData(kanji) {
        let index = this.kanji.length;
        for (const [c, k, o, g] of kanji) {
            this.storeIndex(this.kanjiIndices, c, index++);
            this.kanji.push([c, k, o, g]);
        }
    }

    findTerm(term) {
        return (this.termIndices[term] || []).map(index => {
            const [e, r, g, t] = this.terms[index];
            return {id: index, expression: e, reading: r, glossary: g, tags: t.split(' ')};
        });
    }

    findKanji(kanji) {
        return (this.kanjiIndices[kanji] || []).map(index => {
            const [c, k, o, g] = def;
            return {id: index, character: c, kunyomi: k, onyomi: o, glossary: g};
        });
    }

    storeIndex(indices, term, index) {
        if (term.length > 0) {
            const indices = this.termIndices[term] || [];
            indices.push(index);
            this.termIndices[term] = indices;
        }
    }
}