package rime import ( "bufio" "fmt" mapset "github.com/deckarep/golang-set/v2" "log" "os" "path/filepath" "regexp" "strings" "time" "unicode" ) // 多音字,手动选择注音 var polyphones = map[string]string{ "Eul的神圣法杖 > 的": "de", "艾AA > 艾": "ai", "大V > 大": "da", "QQ音乐 > 乐": "yue", "QQ会员 > 会": "hui", "QQ会员 > 员": "yuan", "阿Q精神 > 阿": "a", "G胖 > 胖": "pang", "阿Q > 阿": "a", "阿Q正传 > 阿": "a", "阿Q正传 > 传": "zhuan", "单边z变换 > 单": "dan", "卡拉OK > 卡": "ka", "IP地址 > 地": "di", "IP卡 > 卡": "ka", "SIM卡 > 卡": "ka", "UIM卡 > 卡": "ka", "USIM卡 > 卡": "ka", "X染色体 > 色": "se", "Y染色体 > 色": "se", "蒙奇·D·路飞 > 奇": "qi", "蒙奇·D·龙 > 奇": "qi", "马歇尔·D·蒂奇 > 奇": "qi", "蒙奇·D·卡普 > 奇": "qi", "蒙奇·D·卡普 > 卡": "ka", "波特卡斯·D·艾斯 > 卡": "ka", "波特卡斯·D·艾斯 > 艾": "ai", "A壳 > 壳": "ke", "B壳 > 壳": "ke", "C壳 > 壳": "ke", "D壳 > 壳": "ke", "芭比Q了 > 了": "le", "江南Style > 南": "nan", "三无Marblue > 无": "wu", "V字仇杀队 > 仇": "chou", "Q弹 > 弹": "tan", "M系列 > 系": "xi", "阿Sir > 阿": "a", "MAC地址 > 地": "di", "OK了 > 了": "le", "OK了吗 > 了": "le", "圈X > 圈": "quan", "A型血 > 血": "xue", "A血型 > 血": "xue", "B型血 > 血": "xue", "B血型 > 血": "xue", "AB型血 > 血": "xue", "AB血型 > 血": "xue", "O型血 > 血": "xue", "O血型 > 血": "xue", "没bug > 没": "mei", "没有bug > 没": "mei", "卡bug > 卡": "ka", "查bug > 查": "cha", "提bug > 提": "ti", "CT检查 > 查": "cha", "N卡 > 卡": "ka", "A卡 > 卡": "ka", "A区 > 区": "qu", "B区 > 区": "qu", "C区 > 区": "qu", "D区 > 区": "qu", "E区 > 区": "qu", "F区 > 区": "qu", "IT行业 > 行": "hang", "TF卡 > 卡": "ka", "A屏 > 屏": "ping", "A和B > 和": "he", "X和Y > 和": "he", "查IP > 查": "cha", "VIP卡 > 卡": "ka", "VIP会员 > 会": "hui", "VIP会员 > 员": "yuan", "Chromium系 > 系": "xi", "Chrome系 > 系": "xi", "QQ游戏大厅 > 大": "da", "QQ飞车 > 车": "che", } var digitMap = map[string]string{ "0": "零", "1": "一", "2": "二", "3": "三", "4": "四", "5": "五", "6": "六", "7": "七", "8": "八", "9": "九", } type schema struct { name string desc string combinationType string path string mapping map[string]string additionalMapping map[string]string excludingMapping map[string]string file *os.File } var doublePinyin = schema{ name: "cn_en_double_pinyin", desc: "自然码双拼", combinationType: "unique", path: filepath.Join(RimeDir, "en_dicts/cn_en_double_pinyin.txt"), mapping: map[string]string{ // 零声母 "-a-": "aa", "-e-": "ee", "-o-": "oo", "-ai-": "ai", "-ei-": "ei", "-ou-": "ou", "-an-": "an", "-en-": "en", "-ang-": "ah", "-eng-": "eg", "-ao-": "ao", "-er-": "er", // zh ch sh "zh": "v", "ch": "i", "sh": "u", // 韵母 "iu": "q", "ia": "w", "ua": "w", "uan": "r", "ue": "t", "ve": "t", "ing": "y", "uai": "y", "uo": "o", "un": "p", "iong": "s", "ong": "s", "iang": "d", "uang": "d", "en": "f", "eng": "g", "ang": "h", "an": "j", "ao": "k", "ai": "l", "ei": "z", "ie": "x", "iao": "c", "ui": "v", "ou": "b", "in": "n", "ian": "m", }, } var doublePinyinFlypy = schema{ name: "cn_en_flypy", desc: "小鹤双拼", combinationType: "unique", path: filepath.Join(RimeDir, "en_dicts/cn_en_flypy.txt"), mapping: map[string]string{ // 零声母 "-a-": "aa", "-e-": "ee", "-o-": "oo", "-ai-": "ai", "-ei-": "ei", "-ou-": "ou", "-an-": "an", "-en-": "en", "-ang-": "ah", "-eng-": "eg", "-ao-": "ao", "-er-": "er", // zh ch sh "zh": "v", "ch": "i", "sh": "u", // 韵母 "iu": "q", "ei": "w", "uan": "r", "ue": "t", "ve": "t", "un": "y", "uo": "o", "ie": "p", "iong": "s", "ong": "s", "ai": "d", "en": "f", "eng": "g", "ang": "h", "an": "j", "ing": "k", "uai": "k", "iang": "l", "uang": "l", "ou": "z", "ia": "x", "ua": "x", "ao": "c", "ui": "v", "in": "b", "iao": "n", "ian": "m", }, } var doublePinyinMSPY = schema{ name: "cn_en_mspy", desc: "微软双拼", combinationType: "unique", path: filepath.Join(RimeDir, "en_dicts/cn_en_mspy.txt"), mapping: map[string]string{ // 零声母 "-a-": "oa", "-e-": "oe", "-o-": "oo", "-ai-": "ol", "-ei-": "oz", "-ou-": "ob", "-an-": "oj", "-en-": "of", "-ang-": "oh", "-eng-": "og", "-ao-": "ok", "-er-": "or", // zh ch sh "zh": "v", "ch": "i", "sh": "u", // 韵母 "iu": "q", "ia": "w", "ua": "w", "er": "r", "uan": "r", "ue": "t", "uai": "y", "uo": "o", "un": "p", "iong": "s", "ong": "s", "iang": "d", "uang": "d", "en": "f", "eng": "g", "ang": "h", "an": "j", "ao": "k", "ai": "l", "ing": ";", "ei": "z", "ie": "x", "iao": "c", "ui": "v", "ve": "v", "ou": "b", "in": "n", "ian": "m", }, } var doublePinyinSogou = schema{ name: "cn_en_sogou", desc: "搜狗双拼", combinationType: "unique", path: filepath.Join(RimeDir, "en_dicts/cn_en_sogou.txt"), mapping: map[string]string{ // 零声母 "-a-": "oa", "-e-": "oe", "-o-": "oo", "-ai-": "ol", "-ei-": "oz", "-ou-": "ob", "-an-": "oj", "-en-": "of", "-ang-": "oh", "-eng-": "og", "-ao-": "ok", "-er-": "or", // zh ch sh "zh": "v", "ch": "i", "sh": "u", // 韵母 "iu": "q", "ia": "w", "ua": "w", "er": "r", "uan": "r", "ue": "t", "uai": "y", "uo": "o", "un": "p", "iong": "s", "ong": "s", "iang": "d", "uang": "d", "en": "f", "eng": "g", "ang": "h", "an": "j", "ao": "k", "ai": "l", "ing": ";", "ei": "z", "ie": "x", "iao": "c", "ui": "v", "ve": "t", "ou": "b", "in": "n", "ian": "m", }, } var doublePinyinZiGuang = schema{ name: "cn_en_ziguang", desc: "紫光双拼", combinationType: "unique", path: filepath.Join(RimeDir, "en_dicts/cn_en_ziguang.txt"), mapping: map[string]string{ // 零声母 "-a-": "oa", "-e-": "oe", "-o-": "oo", "-ai-": "op", "-ei-": "ok", "-ou-": "oz", "-an-": "or", "-en-": "ow", "-ang-": "os", "-eng-": "ot", "-ao-": "oq", "-er-": "oj", // zh ch sh "zh": "u", "ch": "a", "sh": "i", // 韵母 "ao": "q", "en": "w", "an": "r", "eng": "t", "in": "y", "uai": "y", "uo": "o", "ai": "p", "ang": "s", "ie": "d", "ian": "f", "iang": "g", "uang": "g", "iong": "h", "ong": "h", "er": "j", "iu": "j", "ei": "k", "uan": "l", "ing": ";", "ou": "z", "ia": "x", "ua": "x", "iao": "b", "ue": "n", "ui": "n", "un": "m", }, } var doublePinyinABC = schema{ name: "cn_en_abc", desc: "智能 ABC 双拼", combinationType: "unique", path: filepath.Join(RimeDir, "en_dicts/cn_en_abc.txt"), mapping: map[string]string{ // 零声母 "-a-": "oa", "-e-": "oe", "-o-": "oo", "-ai-": "ol", "-ei-": "oq", "-ou-": "ob", "-an-": "oj", "-en-": "of", "-ang-": "oh", "-eng-": "og", "-ao-": "ok", "-er-": "or", // zh ch sh "zh": "a", "ch": "e", "sh": "v", // 韵母 "ao": "k", "en": "f", "an": "j", "eng": "g", "in": "c", "uai": "c", "uo": "o", "ai": "l", "ang": "h", "ie": "x", "ian": "w", "iang": "t", "uang": "t", "iong": "s", "ong": "s", "er": "r", "iu": "r", "ei": "q", "uan": "p", "ing": "y", "ou": "b", "ia": "d", "ua": "d", "iao": "z", "ue": "m", "ui": "m", "un": "n", }, } // CnEn 从 others/cn_en.txt 生成全拼和各个双拼的中英混输词库 func CnEn() { // 控制台输出 defer printlnTimeCost("更新中英混输 ", time.Now()) cnEnTXT, err := os.Open(filepath.Join(RimeDir, "others/cn_en.txt")) if err != nil { log.Fatalln(err) } defer cnEnTXT.Close() schemas := []schema{ {name: "cn_en", desc: "全拼", combinationType: "unique", path: filepath.Join(RimeDir, "en_dicts/cn_en.txt")}, doublePinyin, doublePinyinFlypy, doublePinyinMSPY, doublePinyinSogou, doublePinyinZiGuang, doublePinyinABC, } // 写入前缀内容 for i := range schemas { schemas[i].file, err = os.OpenFile(schemas[i].path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644) if err != nil { log.Fatalln(err) } writePrefix(schemas[i]) } // 转换注音并写入,顺便查重 uniq := mapset.NewSet[string]() sc := bufio.NewScanner(cnEnTXT) for sc.Scan() { line := sc.Text() if line == "" || strings.HasPrefix(line, "#") { continue } if strings.TrimSpace(line) != line { fmt.Println("❌ 前后有空格", line) } if uniq.Contains(line) { fmt.Println("❌ 重复", line) continue } uniq.Add(line) for _, schema := range schemas { if schema.combinationType != "multi" { code := textToPinyin(line, schema) _, err := schema.file.WriteString(line + "\t" + code + "\n") if err != nil { log.Fatalln(err) } lowerCode := strings.ToLower(code) if code != lowerCode { _, err := schema.file.WriteString(line + "\t" + lowerCode + "\n") if err != nil { log.Fatalln(err) } } } else { codes := textToPinyinMulti(line, schema) for _, code := range codes { _, err := schema.file.WriteString(line + "\t" + code + "\n") if err != nil { log.Fatalln(err) } lowerCode := strings.ToLower(code) if code != lowerCode { _, err := schema.file.WriteString(line + "\t" + lowerCode + "\n") if err != nil { log.Fatalln(err) } } } } } } for i := range schemas { schemas[i].file.Close() } } // 写入前缀内容 func writePrefix(s schema) { content := fmt.Sprintf(`# Rime table # coding: utf-8 #@/db_name %s.txt #@/db_type tabledb # # https://github.com/iDvel/rime-ice # ------- 中英混输词库 for %s ------- # 由 others/cn_en.txt 自动生成 # # 此行之后不能写注释 `, s.name, s.desc) _, err := s.file.WriteString(content) if err != nil { log.Fatalln(err) } } // 生成编码,返回原大小写 func textToPinyin(text string, s schema) string { var code string parts := splitMixedWords(text) for _, part := range parts { if digit, ok := digitMap[part]; ok { // 数字 if s.desc == "全拼" { code += hanPinyin[digit][0] } else { code += convertToDoublePinyin(hanPinyin[digit][0], s) } } else if len(hanPinyin[part]) == 0 { // 不在字典的就是英文,返回原大小写 code += part } else if len(hanPinyin[part]) > 1 { // 多音字,按字典指定的读音 if value, ok := polyphones[text+" > "+part]; ok { if s.desc == "全拼" { code += value } else { code += convertToDoublePinyin(value, s) } } else { log.Fatalln("❌ 多音字未指定读音", text, part) } } else { // 其他,按唯一的读音 if s.desc == "全拼" { code += hanPinyin[part][0] } else { code += convertToDoublePinyin(hanPinyin[part][0], s) } } } return code } func textToPinyinMulti(text string, s schema) []string { parts := splitMixedWords(text) map4DoublePinyins := make(map[int][]string) for index, part := range parts { if digit, ok := digitMap[part]; ok { // 数字 map4DoublePinyins[index] = convertToDoublePinyinMulti(hanPinyin[digit][0], s) } else if len(hanPinyin[part]) > 1 { // 多音字,按字典指定的读音 if value, ok := polyphones[text+" > "+part]; ok { map4DoublePinyins[index] = convertToDoublePinyinMulti(value, s) } else { log.Fatalln("❌ 多音字未指定读音", text, part) } } else if len(hanPinyin[part]) == 1 { // 非多音字汉字,按唯一的读音 map4DoublePinyins[index] = convertToDoublePinyinMulti(hanPinyin[part][0], s) } } var result = make([]string, 0) return stepFurther(parts, 0, "", map4DoublePinyins, result) } func stepFurther(parts []string, index int, arranged string, map4DoublePinyins map[int][]string, result []string) []string { if index >= len(parts) { result = append(result, arranged) return result } if combinations, ok := map4DoublePinyins[index]; ok { // 数字或汉字 for _, combination := range combinations { result = stepFurther(parts, index+1, arranged+combination, map4DoublePinyins, result) } } else { // 英文字母 result = stepFurther(parts, index+1, arranged+parts[index], map4DoublePinyins, result) } return result } // 中英文分割,去掉间隔号和横杠 // "哆啦A梦" → ["哆", "啦", "A", "梦"] // "QQ号" → ["QQ", "号"] // "Wi-Fi密码" → ["WiFi", "密", "码"] // "特拉法尔加·D·瓦铁尔·罗" → ["特", "拉", "法", "尔", "加", "D", "瓦", "铁", "尔", "罗"] // "A4纸" → ["A", "4", "纸"] func splitMixedWords(input string) []string { var result []string word := "" for _, r := range input { if string(r) == "·" || string(r) == "-" { continue } else if unicode.Is(unicode.Latin, r) { word += string(r) } else { if word != "" { result = append(result, word) word = "" } result = append(result, string(r)) } } if word != "" { result = append(result, word) } return result } // 将全拼 code 转为双拼 code func convertToDoublePinyin(code string, s schema) string { // 零声母 if contains([]string{"a", "e", "o", "ai", "ei", "ou", "an", "en", "ang", "eng", "ao", "er"}, code) { return s.mapping["-"+code+"-"] } // 分割为声母和韵母 consonantRegexp := regexp.MustCompile(`^(b|p|m|f|d|t|n|l|g|k|h|j|q|x|zh|ch|sh|r|z|c|s|y|w)`) initial := consonantRegexp.FindString(code) final := consonantRegexp.ReplaceAllString(code, "") // 声母转换 if initial == "zh" || initial == "ch" || initial == "sh" { initial = s.mapping[initial] } // 韵母转换 if len(final) > 1 { final = s.mapping[final] } // 其余单个的声母和韵母不转换 return initial + final } func convertToDoublePinyinMulti(code string, s schema) []string { // 零声母 i := []string{"a", "e", "o", "ai", "ei", "ou", "an", "en", "ang", "eng", "ao", "er"} if contains(i, code) { return []string{s.mapping["-"+code+"-"]} } // 分割为声母和韵母 consonantRegexp := regexp.MustCompile(`^(b|p|m|f|d|t|n|l|g|k|h|j|q|x|zh|ch|sh|r|z|c|s|y|w)`) initial := consonantRegexp.FindString(code) final := consonantRegexp.ReplaceAllString(code, "") // 声母转换 isRetroflex := initial == "zh" || initial == "ch" || initial == "sh" if isRetroflex { initial = s.mapping[initial] } // 韵母转换 if len(final) > 1 { final = s.mapping[final] } var result []string if isRetroflex || len(final) > 1 { leadings := strings.Split(initial, ",") followings := strings.Split(final, ",") for _, leading := range leadings { for _, following := range followings { if exclusion, ok := s.excludingMapping[code]; ok { if exclusion == (leading + following) { continue } } result = append(result, leading+following) } } } else { // 其余单个的声母和韵母不转换 result = append(result, initial+final) } if addition, ok := s.additionalMapping[code]; ok { result = append(result, addition) } return result }