mirror of
https://github.com/d0zingcat/rime-ice.git
synced 2026-06-04 23:26:50 +00:00
refactor: pin_cand_filter.lua 匹配逻辑改为无空格的 cand.preedit #586
This commit is contained in:
Dvel
@@ -1,7 +1,7 @@
|
||||
-- 置顶候选项
|
||||
-- Powered By ChatGPT
|
||||
|
||||
local function isInList(list, str)
|
||||
local function is_in_list(list, str)
|
||||
for i, v in ipairs(list) do
|
||||
if v == str then
|
||||
return true, i
|
||||
@@ -10,21 +10,24 @@ local function isInList(list, str)
|
||||
return false, 0
|
||||
end
|
||||
|
||||
local function starts_with(a, b)
|
||||
return string.sub(a, 1, string.len(b)) == b
|
||||
end
|
||||
|
||||
local M = {}
|
||||
|
||||
function M.init(env)
|
||||
env.name_space = env.name_space:gsub("^*", "")
|
||||
local list = env.engine.schema.config:get_list(env.name_space)
|
||||
|
||||
-- 如果定义了 'da zhuan' 或 'da zhong' ，会自动生成 'da z' 和 'da zh'。
|
||||
-- 然而，如果明确定义了 'da z' 或 'da zh'，则会优先使用这些明确自定义的简码，用 set 来做判断。
|
||||
if not list then return end -- no configuration found -> stop
|
||||
|
||||
-- 如果定义了 'da zhuan' 或 'da zhong' ，会自动生成 'daz' 和 'dazh' 的键。
|
||||
-- 然而，如果明确定义了 'da z' 或 'da zh'，则会优先使用这些明确自定义的简码，用 set 来做判断。
|
||||
local set = {}
|
||||
for i = 0, list.size - 1 do
|
||||
local preedit, texts = list:get_value_at(i).value:match("([^\t]+)\t(.+)")
|
||||
-- use #text to match both nil and empty value
|
||||
if #preedit > 0 and #texts > 0 then
|
||||
set[preedit] = true
|
||||
set[preedit:gsub(" ", "")] = true
|
||||
end
|
||||
end
|
||||
|
||||
@@ -32,43 +35,46 @@ function M.init(env)
|
||||
-- 'l 了 啦' → M.pin_cands["l"] = {"了", "啦"}
|
||||
-- 'ta 他 她 它' → M.pin_cands["ta"] = {"他", "她", "它"}
|
||||
--
|
||||
-- 对于词汇（preedit 包含空格），同时生成简码的拼写（最后一个空格后的首字母），如：
|
||||
-- 'ni hao 你好 拟好' → M.pin_cands["ni hao"] = {"你好", "拟好"}
|
||||
-- → M.pin_cands["ni h"] = {"你好", "拟好"}
|
||||
-- 无空格词汇，如 `nihao 你好` → M.pin_cands["nihao"] = {"你好"}
|
||||
--
|
||||
-- 包含空格的词汇，同时生成简码的拼写（最后一个空格后的首字母），如：
|
||||
-- 'ni hao 你好 拟好' → M.pin_cands["nihao"] = {"你好", "拟好"}
|
||||
-- → M.pin_cands["nih"] = {"你好", "拟好"}
|
||||
--
|
||||
-- 如果最后一个空格后以 zh ch sh 开头，额外再生成 zh, ch, sh 的拼写，如：
|
||||
-- 'zhi chi 支持' → M.pin_cands["zhi chi"] = {"支持"}
|
||||
-- → M.pin_cands["zhi c"] = {"支持"}
|
||||
-- → M.pin_cands["zhi ch"] = {"支持"}
|
||||
-- 'zhi chi 支持' → M.pin_cands["zhichi"] = {"支持"}
|
||||
-- → M.pin_cands["zhic"] = {"支持"}
|
||||
-- → M.pin_cands["zhich"] = {"支持"}
|
||||
--
|
||||
-- 如果同时定义了 'da zhuan 大专' 'da zhong 大众'，会生成：
|
||||
-- M.pin_cands["da zhuan"] = {"大专"}
|
||||
-- M.pin_cands["da zhong"] = {"大众"}
|
||||
-- M.pin_cands["da z"] = {"大专", "大众"} -- 先写的排在前面
|
||||
-- M.pin_cands["da zh"] = {"大专", "大众"} -- 先写的排在前面
|
||||
-- M.pin_cands["dazhuan"] = {"大专"}
|
||||
-- M.pin_cands["dazhong"] = {"大众"}
|
||||
-- M.pin_cands["daz"] = {"大专", "大众"} -- 先写的排在前面
|
||||
-- M.pin_cands["dazh"] = {"大专", "大众"} -- 先写的排在前面
|
||||
--
|
||||
-- 如果同时定义了 'da zhuan 大专' 'da zhong 大众' 且明确定义了简码形式 'da z 打字'，会生成：
|
||||
-- M.pin_cands["da zhuan"] = {"大专"}
|
||||
-- M.pin_cands["da zhong"] = {"大众"}
|
||||
-- M.pin_cands["da z"] = {"打字"} -- 明确定义的优先级更高
|
||||
-- M.pin_cands["da zh"] = {"大专", "大众"} -- 没明确定义的，仍然按上面的方式生成
|
||||
-- M.pin_cands["dazhuan"] = {"大专"}
|
||||
-- M.pin_cands["dazhong"] = {"大众"}
|
||||
-- M.pin_cands["daz"] = {"打字"} -- 明确定义的优先级更高
|
||||
-- M.pin_cands["dazh"] = {"大专", "大众"} -- 没明确定义的，仍然按上面的方式生成
|
||||
|
||||
M.pin_cands = {}
|
||||
for i = 0, list.size - 1 do
|
||||
local preedit, texts = list:get_value_at(i).value:match("([^\t]+)\t(.+)")
|
||||
-- use #text to match both nil and empty value
|
||||
if #preedit > 0 and #texts > 0 then
|
||||
M.pin_cands[preedit] = {}
|
||||
local preedit_no_spaces = preedit:gsub(" ", "")
|
||||
M.pin_cands[preedit_no_spaces] = {}
|
||||
-- 按照配置生成完整的拼写
|
||||
for text in texts:gmatch("%S+") do
|
||||
table.insert(M.pin_cands[preedit], text)
|
||||
table.insert(M.pin_cands[preedit_no_spaces], text)
|
||||
end
|
||||
-- 额外处理包含空格的 preedit，增加最后一个拼音的首字母和 zh, ch, sh 的简码
|
||||
if preedit:find(" ") then
|
||||
local preceding_part, last_part = preedit:match("^(.+)%s(%S+)$")
|
||||
if #last_part > 0 then
|
||||
-- 生成最后一个拼音的简码拼写（最后一个空格后的首字母），如 ni hao 生成 ni h
|
||||
local p1 = preceding_part .. " " .. last_part:sub(1, 1)
|
||||
-- 生成最后一个拼音的简码拼写（最后一个空格后的首字母），如 ni hao 生成 nih
|
||||
local p1 = preceding_part:gsub(" ", "") .. last_part:sub(1, 1)
|
||||
-- 只在没有明确定义此简码时才生成，已有的追加，没有的直接赋值
|
||||
if not set[p1] then
|
||||
if M.pin_cands[p1] ~= nil then
|
||||
@@ -76,12 +82,12 @@ function M.init(env)
|
||||
table.insert(M.pin_cands[p1], text)
|
||||
end
|
||||
else
|
||||
M.pin_cands[p1] = M.pin_cands[preedit]
|
||||
M.pin_cands[p1] = M.pin_cands[preedit_no_spaces]
|
||||
end
|
||||
end
|
||||
-- 生成最后一个拼音的 zh, ch, sh 的简码拼写（最后一个空格后以 zh ch sh 开头），如 zhi chi 生成 zhi ch
|
||||
-- 生成最后一个拼音的 zh, ch, sh 的简码拼写（最后一个空格后以 zh ch sh 开头），如 zhi chi 生成 zhich
|
||||
if last_part:match("^[zcs]h") then
|
||||
local p2 = preceding_part .. " " .. last_part:sub(1, 2)
|
||||
local p2 = preceding_part:gsub(" ", "") .. last_part:sub(1, 2)
|
||||
-- 只在没有明确定义此简码时才生成，已有的追加，没有的直接赋值
|
||||
if not set[p2] then
|
||||
if M.pin_cands[p2] ~= nil then
|
||||
@@ -89,7 +95,7 @@ function M.init(env)
|
||||
table.insert(M.pin_cands[p2], text)
|
||||
end
|
||||
else
|
||||
M.pin_cands[p2] = M.pin_cands[preedit]
|
||||
M.pin_cands[p2] = M.pin_cands[preedit_no_spaces]
|
||||
end
|
||||
end
|
||||
end
|
||||
@@ -99,26 +105,27 @@ function M.init(env)
|
||||
end
|
||||
end
|
||||
|
||||
function M.func(input)
|
||||
function M.func(input, env)
|
||||
-- If there is no configuration, no filtering will be performed
|
||||
if not M.pin_cands then
|
||||
for cand in input:iter() do yield(cand) end
|
||||
return
|
||||
end
|
||||
|
||||
local pined = {} -- 提升的候选项
|
||||
local others = {} -- 其余候选项
|
||||
local pined_count = 0
|
||||
|
||||
for cand in input:iter() do
|
||||
local texts = M.pin_cands[cand.preedit]
|
||||
local candtext = cand.text
|
||||
if cand:get_dynamic_type() == "Shadow" then
|
||||
-- handle cands converted by simplifier
|
||||
local originalCand = cand:get_genuine()
|
||||
if #originalCand.text == #candtext then
|
||||
-- 笑|😄 candtext = 😄; 麼|么 candtext = 麼;
|
||||
candtext = originalCand.text
|
||||
end
|
||||
local preedit_no_spaces = cand.preedit:gsub(" ", "")
|
||||
|
||||
-- 跳过不需要处理的部分，这样期望置顶的字词在句子开头及输入到一半时也会被置顶
|
||||
if M.pin_cands[preedit_no_spaces] == nil then
|
||||
yield(cand)
|
||||
goto continue
|
||||
end
|
||||
|
||||
local texts = M.pin_cands[preedit_no_spaces]
|
||||
if texts then
|
||||
-- 给 pined 几个空字符串占位元素，后面直接 pined[idx] = cand 确保 pined 与 texts 顺序一致
|
||||
if #pined < #texts then
|
||||
@@ -126,8 +133,18 @@ function M.func(input)
|
||||
table.insert(pined, '')
|
||||
end
|
||||
end
|
||||
-- 处理简繁转换后的问题
|
||||
local candtext = cand.text
|
||||
if cand:get_dynamic_type() == "Shadow" then
|
||||
-- handle cands converted by simplifier
|
||||
local originalCand = cand:get_genuine()
|
||||
if #originalCand.text == #candtext then
|
||||
-- 笑|😄 candtext = 😄; 麼|么 candtext = 麼;
|
||||
candtext = originalCand.text
|
||||
end
|
||||
end
|
||||
-- 要置顶的放到 pined 中，其余的放到 others
|
||||
local ok, idx = isInList(texts, candtext)
|
||||
local ok, idx = is_in_list(texts, candtext)
|
||||
if ok then
|
||||
pined[idx] = cand
|
||||
pined_count = pined_count + 1
|
||||
@@ -142,6 +159,8 @@ function M.func(input)
|
||||
table.insert(others, cand)
|
||||
break
|
||||
end
|
||||
|
||||
::continue::
|
||||
end
|
||||
|
||||
-- yield pined others 及后续的候选项
|
||||
|
||||
Reference in New Issue
Block a user