1

2026-03-02 09:19:06 +08:00
parent da4649101e
commit 781e557e80
34 changed files with 3926 additions and 87 deletions
--- a/CustomKeyboard/Manager/KBSuggestionEngine.m
+++ b/CustomKeyboard/Manager/KBSuggestionEngine.m
@@ -10,6 +10,8 @@
@property (nonatomic, copy) NSArray<NSString *> *words;
@property (nonatomic, strong) NSMutableDictionary<NSString *, NSNumber *> *selectionCounts;
@property (nonatomic, strong) NSSet<NSString *> *priorityWords;
+@property (nonatomic, copy) NSArray<NSString *> *traditionalChineseWords;
+@property (nonatomic, copy) NSArray<NSString *> *simplifiedChineseWords;
@end

@implementation KBSuggestionEngine
@@ -25,49 +27,32 @@

 - (instancetype)init {
    if (self = [super init]) {
+        _engineType = KBSuggestionEngineTypeLatin;
        _selectionCounts = [NSMutableDictionary dictionary];
        NSArray<NSString *> *defaults = [self.class kb_defaultWords];
        _priorityWords = [NSSet setWithArray:defaults];
        _words = [self kb_loadWords];
+        _traditionalChineseWords = [self kb_loadTraditionalChineseWords];
+        _simplifiedChineseWords = [self kb_loadSimplifiedChineseWords];
    }
    return self;
 }

 - (NSArray<NSString *> *)suggestionsForPrefix:(NSString *)prefix limit:(NSUInteger)limit {
    if (prefix.length == 0 || limit == 0) { return @[]; }
-    NSString *lower = prefix.lowercaseString;
-    NSMutableArray<NSString *> *matches = [NSMutableArray array];

-    for (NSString *word in self.words) {
-        if ([word hasPrefix:lower]) {
-            [matches addObject:word];
-            if (matches.count >= limit * 3) {
-                // Avoid scanning too many matches for long lists.
-                break;
-            }
-        }
+    // 根据引擎类型选择不同的联想逻辑
+    switch (self.engineType) {
+        case KBSuggestionEngineTypePinyinTraditional:
+            return [self kb_traditionalPinyinSuggestionsForPrefix:prefix limit:limit];
+        case KBSuggestionEngineTypePinyinSimplified:
+            return [self kb_simplifiedPinyinSuggestionsForPrefix:prefix limit:limit];
+        case KBSuggestionEngineTypeBopomofo:
+            return [self kb_bopomofoSuggestionsForPrefix:prefix limit:limit];
+        case KBSuggestionEngineTypeLatin:
+        default:
+            return [self kb_latinSuggestionsForPrefix:prefix limit:limit];
    }
-
-    if (matches.count == 0) { return @[]; }
-
-    [matches sortUsingComparator:^NSComparisonResult(NSString *a, NSString *b) {
-        NSInteger ca = self.selectionCounts[a].integerValue;
-        NSInteger cb = self.selectionCounts[b].integerValue;
-        if (ca != cb) {
-            return (cb > ca) ? NSOrderedAscending : NSOrderedDescending;
-        }
-        BOOL pa = [self.priorityWords containsObject:a];
-        BOOL pb = [self.priorityWords containsObject:b];
-        if (pa != pb) {
-            return pa ? NSOrderedAscending : NSOrderedDescending;
-        }
-        return [a compare:b];
-    }];
-
-    if (matches.count > limit) {
-        return [matches subarrayWithRange:NSMakeRange(0, limit)];
-    }
-    return matches.copy;
 }

 - (void)recordSelection:(NSString *)word {
@@ -164,4 +149,148 @@
    ];
 }

+#pragma mark - Engine Type Management
+
+- (void)setEngineTypeFromString:(NSString *)engineTypeString {
+    if ([engineTypeString isEqualToString:@"latin"]) {
+        self.engineType = KBSuggestionEngineTypeLatin;
+    } else if ([engineTypeString isEqualToString:@"pinyin_traditional"]) {
+        self.engineType = KBSuggestionEngineTypePinyinTraditional;
+    } else if ([engineTypeString isEqualToString:@"pinyin_simplified"]) {
+        self.engineType = KBSuggestionEngineTypePinyinSimplified;
+    } else if ([engineTypeString isEqualToString:@"bopomofo"]) {
+        self.engineType = KBSuggestionEngineTypeBopomofo;
+    } else {
+        self.engineType = KBSuggestionEngineTypeLatin;
+    }
+    NSLog(@"[KBSuggestionEngine] Engine type set to: %@", engineTypeString);
+}
+
+#pragma mark - Latin Suggestions
+
+- (NSArray<NSString *> *)kb_latinSuggestionsForPrefix:(NSString *)prefix limit:(NSUInteger)limit {
+    NSString *lower = prefix.lowercaseString;
+    NSMutableArray<NSString *> *matches = [NSMutableArray array];
+
+    for (NSString *word in self.words) {
+        if ([word hasPrefix:lower]) {
+            [matches addObject:word];
+            if (matches.count >= limit * 3) {
+                break;
+            }
+        }
+    }
+
+    if (matches.count == 0) { return @[]; }
+
+    [matches sortUsingComparator:^NSComparisonResult(NSString *a, NSString *b) {
+        NSInteger ca = self.selectionCounts[a].integerValue;
+        NSInteger cb = self.selectionCounts[b].integerValue;
+        if (ca != cb) {
+            return (cb > ca) ? NSOrderedAscending : NSOrderedDescending;
+        }
+        BOOL pa = [self.priorityWords containsObject:a];
+        BOOL pb = [self.priorityWords containsObject:b];
+        if (pa != pb) {
+            return pa ? NSOrderedAscending : NSOrderedDescending;
+        }
+        return [a compare:b];
+    }];
+
+    if (matches.count > limit) {
+        return [matches subarrayWithRange:NSMakeRange(0, limit)];
+    }
+    return matches.copy;
+}
+
+#pragma mark - Traditional Chinese Pinyin Suggestions
+
+- (NSArray<NSString *> *)kb_traditionalPinyinSuggestionsForPrefix:(NSString *)prefix limit:(NSUInteger)limit {
+    // 繁体拼音联想：输入拼音，返回繁体中文候选词
+    NSString *lower = prefix.lowercaseString;
+    NSMutableArray<NSString *> *matches = [NSMutableArray array];
+
+    // 这里应该使用拼音到繁体字的映射表
+    // 目前先返回一些常用繁体词作为示例
+    for (NSString *word in self.traditionalChineseWords) {
+        // TODO: 实现拼音匹配逻辑
+        // 这里需要一个拼音库来将输入的拼音转换为繁体字
+        [matches addObject:word];
+        if (matches.count >= limit) {
+            break;
+        }
+    }
+
+    return matches.copy;
+}
+
+#pragma mark - Simplified Chinese Pinyin Suggestions
+
+- (NSArray<NSString *> *)kb_simplifiedPinyinSuggestionsForPrefix:(NSString *)prefix limit:(NSUInteger)limit {
+    // 简体拼音联想：输入拼音，返回简体中文候选词
+    NSString *lower = prefix.lowercaseString;
+    NSMutableArray<NSString *> *matches = [NSMutableArray array];
+
+    // 这里应该使用拼音到简体字的映射表
+    for (NSString *word in self.simplifiedChineseWords) {
+        // TODO: 实现拼音匹配逻辑
+        [matches addObject:word];
+        if (matches.count >= limit) {
+            break;
+        }
+    }
+
+    return matches.copy;
+}
+
+#pragma mark - Bopomofo (Zhuyin) Suggestions
+
+- (NSArray<NSString *> *)kb_bopomofoSuggestionsForPrefix:(NSString *)prefix limit:(NSUInteger)limit {
+    // 注音联想：输入注音符号，返回繁体中文候选词
+    NSMutableArray<NSString *> *matches = [NSMutableArray array];
+
+    // 这里应该使用注音到繁体字的映射表
+    // 注音符号：ㄅㄆㄇㄈㄉㄊㄋㄌㄍㄎㄏㄐㄑㄒㄓㄔㄕㄖㄗㄘㄙ
+    // 韵母：ㄚㄛㄜㄝㄞㄟㄠㄡㄢㄣㄤㄥㄦㄧㄨㄩ
+    // 声调：ˊˇˋ˙
+    for (NSString *word in self.traditionalChineseWords) {
+        // TODO: 实现注音匹配逻辑
+        [matches addObject:word];
+        if (matches.count >= limit) {
+            break;
+        }
+    }
+
+    return matches.copy;
+}
+
+#pragma mark - Chinese Word Loading
+
+- (NSArray<NSString *> *)kb_loadTraditionalChineseWords {
+    // 加载繁体中文常用词
+    // 这里先返回一些示例词，实际应该从文件或数据库加载
+    return @[
+        @"你好", @"謝謝", @"對不起", @"再見", @"早安",
+        @"晚安", @"請問", @"不好意思", @"沒關係", @"加油",
+        @"台灣", @"台北", @"高雄", @"台中", @"台南",
+        @"朋友", @"家人", @"工作", @"學習", @"生活",
+        @"時間", @"地點", @"方法", @"問題", @"答案",
+        @"喜歡", @"愛", @"想念", @"開心", @"快樂",
+        @"美麗", @"漂亮", @"帥氣", @"可愛", @"溫柔"
+    ];
+}
+
+- (NSArray<NSString *> *)kb_loadSimplifiedChineseWords {
+    // 加载简体中文常用词
+    return @[
+        @"你好", @"谢谢", @"对不起", @"再见", @"早安",
+        @"晚安", @"请问", @"不好意思", @"没关系", @"加油",
+        @"中国", @"北京", @"上海", @"广州", @"深圳",
+        @"朋友", @"家人", @"工作", @"学习", @"生活",
+        @"时间", @"地点", @"方法", @"问题", @"答案",
+        @"喜欢", @"爱", @"想念", @"开心", @"快乐",
+        @"美丽", @"漂亮", @"帅气", @"可爱", @"温柔"
+    ];
+}
+
@end