yomichan/ext/bg/js/deinflector.js

96 lines
3.2 KiB
JavaScript
Raw Normal View History

2016-03-21 00:52:14 +00:00
/*
* Copyright (C) 2016-2020 Yomichan Authors
2016-03-21 00:52:14 +00:00
*
* 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
2020-01-01 17:00:31 +00:00
* along with this program. If not, see <https://www.gnu.org/licenses/>.
2016-03-21 00:52:14 +00:00
*/
2019-08-31 01:06:21 +00:00
class Deinflector {
constructor(reasons) {
this.reasons = Deinflector.normalizeReasons(reasons);
2016-03-21 00:52:14 +00:00
}
deinflect(source, rawSource) {
2019-08-31 01:06:21 +00:00
const results = [{
source,
rawSource,
2019-08-31 01:06:21 +00:00
term: source,
rules: 0,
reasons: [],
databaseDefinitions: []
2019-08-31 01:06:21 +00:00
}];
for (let i = 0; i < results.length; ++i) {
const {rules, term, reasons} = results[i];
for (const [reason, variants] of this.reasons) {
for (const [kanaIn, kanaOut, rulesIn, rulesOut] of variants) {
if (
(rules !== 0 && (rules & rulesIn) === 0) ||
!term.endsWith(kanaIn) ||
(term.length - kanaIn.length + kanaOut.length) <= 0
) {
2019-08-31 01:06:21 +00:00
continue;
}
2016-03-21 00:52:14 +00:00
2019-08-31 01:06:21 +00:00
results.push({
source,
rawSource,
term: term.substring(0, term.length - kanaIn.length) + kanaOut,
rules: rulesOut,
reasons: [reason, ...reasons],
databaseDefinitions: []
2019-08-31 01:06:21 +00:00
});
2016-03-21 00:52:14 +00:00
}
}
}
2016-12-21 05:54:01 +00:00
return results;
2016-03-21 00:52:14 +00:00
}
static normalizeReasons(reasons) {
const normalizedReasons = [];
2020-02-26 00:54:59 +00:00
for (const [reason, reasonInfo] of Object.entries(reasons)) {
const variants = [];
2020-02-26 00:54:59 +00:00
for (const {kanaIn, kanaOut, rulesIn, rulesOut} of reasonInfo) {
variants.push([
kanaIn,
kanaOut,
Deinflector.rulesToRuleFlags(rulesIn),
Deinflector.rulesToRuleFlags(rulesOut)
]);
}
normalizedReasons.push([reason, variants]);
}
return normalizedReasons;
}
static rulesToRuleFlags(rules) {
const ruleTypes = Deinflector.ruleTypes;
let value = 0;
for (const rule of rules) {
2020-02-02 01:51:30 +00:00
const ruleBits = ruleTypes.get(rule);
if (typeof ruleBits === 'undefined') { continue; }
value |= ruleBits;
}
return value;
}
2016-03-21 00:52:14 +00:00
}
2020-02-02 01:51:30 +00:00
Deinflector.ruleTypes = new Map([
['v1', 0b0000001], // Verb ichidan
['v5', 0b0000010], // Verb godan
['vs', 0b0000100], // Verb suru
['vk', 0b0001000], // Verb kuru
['adj-i', 0b0010000], // Adjective i
['iru', 0b0100000] // Intermediate -iru endings for progressive or perfect tense
]);