yomichan/ext/bg/js/translator.js

128 lines
4.3 KiB
JavaScript
Raw Normal View History

2016-03-20 17:52:27 +00:00
/*
* 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/>.
*/
2016-03-24 02:25:32 +00:00
2016-03-20 17:52:27 +00:00
class Translator {
2016-03-26 21:16:21 +00:00
constructor() {
2016-08-07 20:01:27 +00:00
this.loaded = false;
2016-12-18 02:45:19 +00:00
this.ruleMeta = null;
2016-11-07 16:29:21 +00:00
this.database = new Database();
2016-04-02 19:56:47 +00:00
this.deinflector = new Deinflector();
2016-03-26 21:16:21 +00:00
}
2016-11-13 19:58:23 +00:00
prepare() {
2016-04-04 05:22:38 +00:00
if (this.loaded) {
2016-09-11 19:40:45 +00:00
return Promise.resolve();
2016-03-26 21:16:21 +00:00
}
2016-03-20 17:52:27 +00:00
2016-11-13 19:58:23 +00:00
const promises = [
2016-12-18 03:08:28 +00:00
loadJsonInt('bg/data/deinflect.json'),
2016-11-13 19:58:23 +00:00
this.database.prepare()
];
2016-12-18 02:45:19 +00:00
return Promise.all(promises).then(([reasons]) => {
this.deinflector.setReasons(reasons);
2016-08-24 03:53:11 +00:00
this.loaded = true;
2016-08-21 20:32:36 +00:00
});
2016-03-20 17:52:27 +00:00
}
2016-03-21 01:45:37 +00:00
2016-11-14 05:53:56 +00:00
findTerm(text, dictionaries, enableSoftKatakanaSearch) {
2016-12-21 06:43:27 +00:00
const cache = {};
2016-12-23 23:44:16 +00:00
return this.findTermDeinflections(text, dictionaries, cache).then(deinfHiragana => {
const textHiragana = wanakana._katakanaToHiragana(text);
if (text !== textHiragana && enableSoftKatakanaSearch) {
2016-12-23 23:44:16 +00:00
return this.findTermDeinflections(textHiragana, dictionaries, cache).then(deinfHiragana => deinfHiragana.concat(deinfHiragana));
} else {
2016-12-23 23:44:16 +00:00
return deinfHiragana;
}
2016-12-23 23:44:16 +00:00
}).then(deinflections => {
let definitions = [];
for (const deinflection of deinflections) {
for (const definition of deinflection.definitions) {
definitions.push({
source: deinflection.source,
reasons: deinflection.reasons,
score: definition.score,
id: definition.id,
dictionary: definition.dictionary,
expression: definition.expression,
reading: definition.reading,
glossary: definition.glossary,
tags: sortTags(definition.tags.map(tag => buildTag(tag, definition.tagMeta)))
});
}
2016-05-08 01:24:31 +00:00
}
2016-12-23 23:44:16 +00:00
definitions = undupeTermDefs(definitions);
definitions = sortTermDefs(definitions);
2016-09-11 02:36:28 +00:00
let length = 0;
2016-12-23 23:44:16 +00:00
for (const definition of definitions) {
length = Math.max(length, definition.source.length);
2016-04-02 05:13:37 +00:00
}
2016-12-23 23:44:16 +00:00
return {length, definitions};
2016-04-02 05:13:37 +00:00
});
2016-03-21 01:45:37 +00:00
}
2016-11-13 03:29:30 +00:00
findKanji(text, dictionaries) {
2016-12-21 06:43:27 +00:00
const processed = {}, promises = [];
2016-08-10 04:23:05 +00:00
for (const c of text) {
2016-04-25 01:59:29 +00:00
if (!processed[c]) {
2016-11-13 03:29:30 +00:00
promises.push(this.database.findKanji(c, dictionaries));
2016-03-21 03:34:50 +00:00
processed[c] = true;
}
}
return Promise.all(promises).then(sets => this.processKanji(sets.reduce((a, b) => a.concat(b), [])));
2016-03-21 01:45:37 +00:00
}
2016-12-23 23:44:16 +00:00
findTermDeinflections(text, dictionaries, cache) {
2016-12-21 06:43:27 +00:00
const definer = term => {
if (cache.hasOwnProperty(term)) {
return Promise.resolve(cache[term]);
}
return this.database.findTerm(term, dictionaries).then(definitions => cache[term] = definitions);
};
2016-11-13 03:29:30 +00:00
2016-12-23 23:44:16 +00:00
const promises = [];
2016-11-13 03:29:30 +00:00
for (let i = text.length; i > 0; --i) {
2016-12-23 23:44:16 +00:00
promises.push(this.deinflector.deinflect(text.slice(0, i), definer));
2016-11-13 03:29:30 +00:00
}
2016-12-23 23:44:16 +00:00
return Promise.all(promises).then(results => {
let deinflections = [];
for (const result of results) {
deinflections = deinflections.concat(result);
2016-04-19 02:38:03 +00:00
}
2016-12-20 05:23:17 +00:00
2016-12-23 23:44:16 +00:00
return deinflections;
});
2016-03-21 03:34:50 +00:00
}
2016-09-11 19:40:45 +00:00
processKanji(definitions) {
for (const definition of definitions) {
2016-12-18 07:38:14 +00:00
const tags = definition.tags.map(tag => buildTag(tag, definition.tagMeta));
2016-12-18 05:26:46 +00:00
definition.tags = sortTags(tags);
2016-08-08 02:56:50 +00:00
}
2016-09-11 19:40:45 +00:00
return definitions;
2016-08-08 05:42:56 +00:00
}
2016-03-20 17:52:27 +00:00
}