clean up parse_one_key in full pinyin parser2
[platform/upstream/libpinyin.git] / src / storage / pinyin_parser2.cpp
1 /* 
2  *  libpinyin
3  *  Library to deal with pinyin.
4  *  
5  *  Copyright (C) 2011 Peng Wu <alexepico@gmail.com>
6  *  
7  *  This program is free software; you can redistribute it and/or modify
8  *  it under the terms of the GNU General Public License as published by
9  *  the Free Software Foundation; either version 2 of the License, or
10  *  (at your option) any later version.
11  * 
12  *  This program is distributed in the hope that it will be useful,
13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15  *  GNU General Public License for more details.
16  *  
17  *  You should have received a copy of the GNU General Public License
18  *  along with this program; if not, write to the Free Software
19  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
20  */
21
22
23 #include "pinyin_parser2.h"
24 #include <ctype.h>
25 #include <assert.h>
26 #include <stdio.h>
27 #include <string.h>
28 #include "stl_lite.h"
29 #include "pinyin_phrase2.h"
30 #include "pinyin_custom2.h"
31 #include "chewing_key.h"
32 #include "pinyin_parser_table.h"
33 #include "double_pinyin_table.h"
34 #include "chewing_table.h"
35
36
37 using namespace pinyin;
38
39 static bool check_pinyin_options(pinyin_option_t options, const pinyin_index_item_t * item) {
40     guint32 flags = item->m_flags;
41     assert (flags & IS_PINYIN);
42
43     /* handle incomplete pinyin. */
44     if (flags & PINYIN_INCOMPLETE) {
45         if (!(options & PINYIN_INCOMPLETE))
46             return false;
47     }
48
49     /* handle correct pinyin, currently only one flag per item. */
50     flags &= PINYIN_CORRECT_ALL;
51     options &= PINYIN_CORRECT_ALL;
52
53     if (flags) {
54         if ((flags & options) != flags)
55             return false;
56     }
57
58     return true;
59 }
60
61 static bool check_chewing_options(pinyin_option_t options, const chewing_index_item_t * item) {
62     guint32 flags = item->m_flags;
63     assert (flags & IS_CHEWING);
64
65     /* handle incomplete chewing. */
66     if (flags & CHEWING_INCOMPLETE) {
67         if (!(options & CHEWING_INCOMPLETE))
68             return false;
69     }
70
71     return true;
72 }
73
74
75 /* methods for Chewing Keys to access pinyin parser table. */
76 const char * ChewingKeyRest::get_pinyin_string(){
77     if (m_table_index == 0)
78         return NULL;
79
80     /* check end boundary. */
81     assert(m_table_index < G_N_ELEMENTS(content_table));
82     return content_table[m_table_index].m_pinyin_str;
83 }
84
85 const char * ChewingKeyRest::get_chewing_string(){
86     if (m_table_index == 0)
87         return NULL;
88
89     /* check end boundary. */
90     assert(m_table_index < G_N_ELEMENTS(content_table));
91     return content_table[m_table_index].m_chewing_str;
92 }
93
94
95 /* Pinyin Parsers */
96
97 /* internal information for pinyin parsers. */
98 struct parse_value_t{
99     ChewingKey m_key;
100     ChewingKeyRest m_key_rest;
101     gint16 m_num_keys;
102     gint16 m_parsed_len;
103     gint16 m_last_step;
104
105     /* constructor */
106 public:
107     parse_value_t(){
108         m_num_keys = 0;
109         m_parsed_len = 0;
110         m_last_step = -1;
111     }
112 };
113
114 const guint16 max_full_pinyin_length   = 7;  /* include tone. */
115
116 const guint16 max_double_pinyin_length = 3;  /* include tone. */
117
118 const guint16 max_chewing_length       = 4;  /* include tone. */
119
120 static bool compare_pinyin_less_than(const pinyin_index_item_t & lhs,
121                                      const pinyin_index_item_t & rhs){
122     return 0 > strcmp(lhs.m_pinyin_input, rhs.m_pinyin_input);
123 }
124
125 static inline bool search_pinyin_index(pinyin_option_t options, const char * pinyin,
126                                        ChewingKey & key,
127                                        ChewingKeyRest & key_rest){
128     pinyin_index_item_t item;
129     memset(&item, 0, sizeof(item));
130     item.m_pinyin_input = pinyin;
131
132     std_lite::pair<const pinyin_index_item_t *,
133                    const pinyin_index_item_t *> range;
134     range = std_lite::equal_range
135         (pinyin_index, pinyin_index + G_N_ELEMENTS(pinyin_index),
136          item, compare_pinyin_less_than);
137
138     guint16 range_len = range.second - range.first;
139     assert(range_len <= 1);
140     if (range_len == 1) {
141         const pinyin_index_item_t * index = range.first;
142
143         if (!check_pinyin_options(options, index))
144             return false;
145
146         key_rest.m_table_index = index->m_table_index;
147         key = content_table[key_rest.m_table_index].m_chewing_key;
148         return true;
149     }
150
151     return false;
152 }
153
154 static bool compare_chewing_less_than(const chewing_index_item_t & lhs,
155                                       const chewing_index_item_t & rhs){
156     return 0 > strcmp(lhs.m_chewing_input, rhs.m_chewing_input);
157 }
158
159 static inline bool search_chewing_index(pinyin_option_t options, const char * chewing,
160                                         ChewingKey & key,
161                                         ChewingKeyRest & key_rest){
162     chewing_index_item_t item;
163     memset(&item, 0, sizeof(item));
164     item.m_chewing_input = chewing;
165
166     std_lite::pair<const chewing_index_item_t *,
167                    const chewing_index_item_t *> range;
168     range = std_lite::equal_range
169         (chewing_index, chewing_index + G_N_ELEMENTS(chewing_index),
170          item, compare_chewing_less_than);
171
172     guint16 range_len = range.second - range.first;
173     assert (range_len <= 1);
174
175     if (range_len == 1) {
176         const chewing_index_item_t * index = range.first;
177
178         if (!check_chewing_options(options, index))
179             return false;
180
181         key_rest.m_table_index = index->m_table_index;
182         key = content_table[key_rest.m_table_index].m_chewing_key;
183         return true;
184     }
185
186     return false;
187 }
188
189 /* Full Pinyin Parser */
190 FullPinyinParser2::FullPinyinParser2 (){
191     m_parse_steps = g_array_new(TRUE, FALSE, sizeof(parse_value_t));
192 }
193
194
195 bool FullPinyinParser2::parse_one_key (pinyin_option_t options, ChewingKey & key,
196                                        ChewingKeyRest & key_rest,
197                                        const char * pinyin, int len) const {
198     /* "'" are not accepted in parse_one_key. */
199     gchar * input = g_strndup(pinyin, len);
200     assert(NULL == strchr(input, '\''));
201
202     guint16 tone = CHEWING_ZERO_TONE; guint16 tone_pos = 0;
203     guint16 parsed_len = len;
204     key = ChewingKey(); key_rest = ChewingKeyRest();
205
206     if (options & USE_TONE) {
207         /* find the tone in the last character. */
208         char chr = input[parsed_len - 1];
209         if ( '0' < chr && chr <= '5' ) {
210             tone = chr - '0';
211             parsed_len --;
212             tone_pos = parsed_len;
213         }
214     }
215
216     /* parse pinyin core staff here. */
217
218     /* Note: optimize here? */
219     input[parsed_len] = '\0';
220     if (!search_pinyin_index(options, input, key, key_rest)) {
221         g_free(input);
222         return false;
223     }
224
225     if (options & USE_TONE) {
226         /* post processing tone. */
227         if ( parsed_len == tone_pos ) {
228             if (tone != CHEWING_ZERO_TONE) {
229                 key.m_tone = tone;
230                 parsed_len ++;
231             }
232         }
233     }
234
235     key_rest.m_raw_begin = 0; key_rest.m_raw_end = parsed_len;
236     g_free(input);
237     return parsed_len == len;
238 }
239
240
241 int FullPinyinParser2::parse (pinyin_option_t options, ChewingKeyVector & keys,
242                               ChewingKeyRestVector & key_rests,
243                               const char *str, int len) const {
244     int i;
245     /* clear arrays. */
246     g_array_set_size(keys, 0);
247     g_array_set_size(key_rests, 0);
248
249     /* init m_parse_steps, and prepare dynamic programming. */
250     int step_len = len + 1;
251     g_array_set_size(m_parse_steps, 0);
252     parse_value_t value;
253     for (i = 0; i < step_len; ++i) {
254         g_array_append_val(m_parse_steps, value);
255     }
256
257     size_t next_sep = 0;
258     gchar * input = g_strndup(str, len);
259     parse_value_t * curstep = NULL, * nextstep = NULL;
260
261     for (i = 0; i < len; ++i) {
262         if (input[i] == '\'') {
263             curstep = &g_array_index(m_parse_steps, parse_value_t, i);
264             nextstep = &g_array_index(m_parse_steps, parse_value_t, i + 1);
265
266             /* propagate current step into next step. */
267             nextstep->m_key = ChewingKey();
268             nextstep->m_key_rest = ChewingKeyRest();
269             nextstep->m_num_keys = curstep->m_num_keys;
270             nextstep->m_parsed_len = curstep->m_parsed_len + 1;
271             nextstep->m_last_step = i;
272             next_sep = 0;
273             continue;
274         }
275
276         /* forward to next "'" */
277         if ( 0 == next_sep ) {
278             int k;
279             for (k = i;  k < len; ++k) {
280                 if (input[k] == '\'')
281                     break;
282             }
283             next_sep = k;
284         }
285
286         /* dynamic programming here. */
287         for (size_t m = i; m < next_sep; ++m) {
288             curstep = &g_array_index(m_parse_steps, parse_value_t, m);
289             size_t try_len = std_lite::min
290                 (m + max_full_pinyin_length, next_sep);
291             for (size_t n = m + 1; n < try_len + 1; ++n) {
292                 nextstep = &g_array_index(m_parse_steps, parse_value_t, n);
293
294                 /* gen next step */
295                 const char * onepinyin = input + m;
296                 gint16 onepinyinlen = n - m;
297                 value = parse_value_t();
298
299                 ChewingKey key; ChewingKeyRest rest;
300                 bool parsed = parse_one_key
301                     (options, key, rest, onepinyin, onepinyinlen);
302                 rest.m_raw_begin = m; rest.m_raw_end = n;
303                 if (!parsed)
304                     continue;
305                 value.m_key = key; value.m_key_rest = rest;
306                 value.m_num_keys = curstep->m_num_keys + 1;
307                 value.m_parsed_len = curstep->m_parsed_len + onepinyinlen;
308                 value.m_last_step = m;
309
310                 /* save next step */
311                 if (-1 == nextstep->m_last_step)
312                     *nextstep = value;
313                 if (value.m_parsed_len > nextstep->m_parsed_len)
314                     *nextstep = value;
315                 if (value.m_parsed_len == nextstep->m_parsed_len &&
316                     value.m_num_keys < nextstep->m_num_keys)
317                     *nextstep = value;
318             }
319         }
320     }
321
322     /* final step for back tracing. */
323     gint16 parsed_len = final_step(step_len, keys, key_rests);
324
325     /* post processing for re-split table. */
326     if (options & USE_RESPLIT_TABLE) {
327         post_process(options, keys, key_rests);
328     }
329
330     g_free(input);
331     return parsed_len;
332 }
333
334 int FullPinyinParser2::final_step(size_t step_len, ChewingKeyVector & keys,
335                                   ChewingKeyRestVector & key_rests) const{
336     int i;
337     gint16 parsed_len = 0;
338     parse_value_t * curstep = NULL;
339
340     /* find longest match, which starts from the beginning of input. */
341     for (i = step_len - 1; i >= 0; --i) {
342         curstep = &g_array_index(m_parse_steps, parse_value_t, i);
343         if (i == curstep->m_parsed_len)
344             break;
345     }
346     /* prepare saving. */
347     parsed_len = curstep->m_parsed_len;
348     gint16 num_keys = curstep->m_num_keys;
349     g_array_set_size(keys, num_keys);
350     g_array_set_size(key_rests, num_keys);
351
352     /* save the match. */
353     while (curstep->m_last_step != -1) {
354         gint16 pos = curstep->m_num_keys - 1;
355
356         /* skip "'" */
357         if (0 != curstep->m_key_rest.m_table_index) {
358             ChewingKey * key = &g_array_index(keys, ChewingKey, pos);
359             ChewingKeyRest * rest = &g_array_index
360                 (key_rests, ChewingKeyRest, pos);
361             *key = curstep->m_key; *rest = curstep->m_key_rest;
362         }
363
364         /* back ward */
365         curstep = &g_array_index(m_parse_steps, parse_value_t,
366                                  curstep->m_last_step);
367     }
368     return parsed_len;
369 }
370
371
372 bool FullPinyinParser2::post_process(pinyin_option_t options,
373                                      ChewingKeyVector & keys,
374                                      ChewingKeyRestVector & key_rests) const {
375     int i;
376     assert(keys->len == key_rests->len);
377     gint16 num_keys = keys->len;
378
379     ChewingKey * cur_key = NULL, * next_key = NULL;
380     ChewingKeyRest * cur_rest = NULL, * next_rest = NULL;
381     guint16 cur_tone = CHEWING_ZERO_TONE, next_tone = CHEWING_ZERO_TONE;
382
383     for (i = 0; i < num_keys - 1; ++i) {
384         cur_rest = &g_array_index(key_rests, ChewingKeyRest, i);
385         next_rest = &g_array_index(key_rests, ChewingKeyRest, i + 1);
386
387         /* some "'" here */
388         if (cur_rest->m_raw_end != next_rest->m_raw_begin)
389             continue;
390
391         cur_key = &g_array_index(keys, ChewingKey, i);
392         next_key = &g_array_index(keys, ChewingKey, i + 1);
393
394         if (options & USE_TONE) {
395             cur_tone = cur_key->m_tone;
396             next_tone = next_key->m_tone;
397             cur_key->m_tone = next_key->m_tone = CHEWING_ZERO_TONE;
398         }
399
400         /* lookup re-split table */
401         size_t k;
402         const resplit_table_item_t * item = NULL;
403         for (k = 0; k < G_N_ELEMENTS(resplit_table); ++k) {
404             item = resplit_table + k;
405             /* no ops */
406             if (item->m_orig_freq >= item->m_new_freq)
407                 continue;
408
409             /* use pinyin_exact_compare2 here. */
410             if (0 == pinyin_exact_compare2(item->m_orig_keys,
411                                            cur_key, 2))
412                 break;
413
414         }
415
416         /* find the match */
417         if (k < G_N_ELEMENTS(resplit_table)) {
418             /* do re-split */
419             item = resplit_table + k;
420             *cur_key = item->m_new_keys[0];
421             *next_key = item->m_new_keys[1];
422             /* assumes only moved one char in gen_all_resplit script. */
423             cur_rest->m_raw_end --;
424             next_rest->m_raw_begin --;
425         }
426
427         /* save back tones */
428         if (options & USE_TONE) {
429             cur_key->m_tone = cur_tone;
430             next_key->m_tone = next_tone;
431         }
432     }
433
434     return true;
435 }
436
437 #define IS_KEY(x)   (('a' <= x && x <= 'z') || x == ';')
438
439 bool DoublePinyinParser2::parse_one_key(pinyin_option_t options, ChewingKey & key,
440                                         ChewingKeyRest & key_rest,
441                                         const char *str, int len) const {
442
443     if (1 == len) {
444         if (!(options & PINYIN_INCOMPLETE))
445             return false;
446
447         char ch = str[0];
448         if (!IS_KEY(ch))
449             return false;
450
451         int charid = ch == ';' ? 26 : ch - 'a';
452         const char * sheng = m_shengmu_table[charid].m_shengmu;
453         if (NULL == sheng || strcmp(sheng, "'") == 0)
454             return false;
455
456         if (search_pinyin_index(options, sheng, key, key_rest)) {
457             key_rest.m_raw_begin = 0; key_rest.m_raw_end = len;
458             return true;
459         } else {
460             return false;
461         }
462     }
463
464     ChewingTone tone = CHEWING_ZERO_TONE;
465     options &= ~(PINYIN_CORRECT_ALL|PINYIN_AMB_ALL);
466
467     /* parse tone */
468     if (3 == len) {
469         if (!(options & USE_TONE))
470             return false;
471         char ch = str[2];
472         if (!('0' < ch && ch <= '5'))
473             return false;
474         tone = (ChewingTone) (ch - '0');
475     }
476
477     if (2 == len || 3 == len) {
478         /* parse shengmu here. */
479         char ch = str[0];
480         if (!IS_KEY(ch))
481             return false;
482
483         int charid = ch == ';' ? 26 : ch - 'a';
484         const char * sheng = m_shengmu_table[charid].m_shengmu;
485         if (NULL == sheng)
486             return false;
487         if (strcmp(sheng, "'") == 0)
488             sheng = "";
489
490         /* parse yunmu here. */
491         ch = str[1];
492         if (!IS_KEY(ch))
493             return false;
494
495         charid = ch == ';' ? 26 : ch - 'a';
496         /* first yunmu */
497         const char * yun = m_yunmu_table[charid].m_yunmus[0];
498         gchar * pinyin = g_strdup_printf("%s%s", sheng, yun);
499         if (search_pinyin_index(options, pinyin, key, key_rest)) {
500             key_rest.m_raw_begin = 0; key_rest.m_raw_end = len;
501             key.m_tone = tone;
502             g_free(pinyin);
503             return true;
504         }
505         g_free(pinyin);
506
507         /* second yunmu */
508         yun = m_yunmu_table[charid].m_yunmus[1];
509         pinyin = g_strdup_printf("%s%s", sheng, yun);
510         if (search_pinyin_index(options, pinyin, key, key_rest)) {
511             key_rest.m_raw_begin = 0; key_rest.m_raw_end = len;
512             key.m_tone = tone;
513             g_free(pinyin);
514             return true;
515         }
516         g_free(pinyin);
517
518     }
519
520     return false;
521 }
522
523
524 /* only 'a'-'z' and ';' are accepted here. */
525 int DoublePinyinParser2::parse(pinyin_option_t options, ChewingKeyVector & keys,
526                                ChewingKeyRestVector & key_rests,
527                                const char *str, int len) const {
528     g_array_set_size(keys, 0);
529     g_array_set_size(key_rests, 0);
530
531     int maximum_len = 0; int i;
532     /* probe the longest possible double pinyin string. */
533     for (i = 0; i < len; ++i) {
534         const char ch = str[i];
535         if (!(IS_KEY(ch) || ('0' < ch && ch <= '5')))
536             break;
537     }
538     maximum_len = i;
539
540     /* maximum forward match for double pinyin. */
541     int parsed_len = 0;
542     while (parsed_len < maximum_len) {
543         const char * cur_str = str + parsed_len;
544         i = std_lite::min(maximum_len - parsed_len,
545                           (int)max_double_pinyin_length);
546
547         ChewingKey key; ChewingKeyRest key_rest;
548         for (; i > 0; --i) {
549             bool success = parse_one_key(options, key, key_rest, cur_str, i);
550             if (success)
551                 break;
552         }
553
554         if (0 == i)        /* no more possible double pinyins. */
555             break;
556
557         key_rest.m_raw_begin = parsed_len; key_rest.m_raw_end = parsed_len + i;
558         parsed_len += i;
559
560         /* save the pinyin */
561         g_array_append_val(keys, key);
562         g_array_append_val(key_rests, key_rest);
563     }
564
565     return parsed_len;
566 }
567
568 #undef IS_KEY
569
570 bool DoublePinyinParser2::set_scheme(DoublePinyinScheme scheme) {
571
572     switch (scheme) {
573     case DOUBLE_PINYIN_ZRM:
574         m_shengmu_table = double_pinyin_zrm_sheng;
575         m_yunmu_table   = double_pinyin_zrm_yun;
576         return true;
577     case DOUBLE_PINYIN_MS:
578         m_shengmu_table = double_pinyin_mspy_sheng;
579         m_yunmu_table   = double_pinyin_mspy_yun;
580         return true;
581     case DOUBLE_PINYIN_ZIGUANG:
582         m_shengmu_table = double_pinyin_zgpy_sheng;
583         m_yunmu_table   = double_pinyin_zgpy_yun;
584         return true;
585     case DOUBLE_PINYIN_ABC:
586         m_shengmu_table = double_pinyin_abc_sheng;
587         m_yunmu_table   = double_pinyin_abc_yun;
588         return true;
589     case DOUBLE_PINYIN_PYJJ:
590         m_shengmu_table = double_pinyin_pyjj_sheng;
591         m_yunmu_table   = double_pinyin_pyjj_yun;
592         return true;
593     case DOUBLE_PINYIN_XHE:
594         m_shengmu_table = double_pinyin_xhe_sheng;
595         m_yunmu_table   = double_pinyin_xhe_yun;
596         return true;
597     case DOUBLE_PINYIN_CUSTOMIZED:
598         assert(FALSE);
599     };
600
601     return false; /* no such scheme. */
602 }
603
604 /* the chewing string must be freed with g_free. */
605 static bool search_chewing_symbols(const chewing_symbol_item_t * symbol_table,
606                                    const char key, char ** chewing) {
607     *chewing = NULL;
608     /* just iterate the table, as we only have < 50 items. */
609     while (symbol_table->m_input != '\0') {
610         if (symbol_table->m_input == key) {
611             *chewing = g_strdup(symbol_table->m_chewing);
612             return true;
613         }
614         symbol_table ++;
615     }
616     return false;
617 }
618
619 static bool search_chewing_tones(const chewing_tone_item_t * tone_table,
620                                  const char key, char * tone) {
621     *tone = CHEWING_ZERO_TONE;
622     /* just iterate the table, as we only have < 10 items. */
623     while (tone_table->m_input != '\0') {
624         if (tone_table->m_input == key) {
625             *tone = tone_table->m_tone;
626             return true;
627         }
628         tone_table ++;
629     }
630     return false;
631 }
632
633
634 bool ChewingParser2::parse_one_key(pinyin_option_t options, ChewingKey & key, ChewingKeyRest & key_rest, const char *str, int len) const {
635     char tone = CHEWING_ZERO_TONE;
636
637     int symbols_len = len;
638     /* probe whether the last key is tone key in str. */
639     if (options & USE_TONE) {
640         char ch = str[len - 1];
641         /* remove tone from input */
642         if (search_chewing_tones(m_tone_table, ch, &tone))
643             symbols_len --;
644     }
645
646     int i;
647     gchar * chewing = NULL, * onechar = NULL;
648
649     /* probe the possible chewing map in the rest of str. */
650     for (i = 0; i < symbols_len; ++i) {
651         if (!search_chewing_symbols(m_symbol_table, str[i], &onechar)) {
652             g_free(onechar);
653             g_free(chewing);
654             return false;
655         }
656
657         if (!chewing) {
658             chewing = g_strdup(onechar);
659         } else {
660             gchar * tmp = chewing;
661             chewing = g_strconcat(chewing, onechar, NULL);
662             g_free(tmp);
663         }
664         g_free(onechar);
665     }
666
667     /* search the chewing in the chewing index table. */
668     if (search_chewing_index(options, chewing, key, key_rest)) {
669         key_rest.m_raw_begin = 0; key_rest.m_raw_end = len;
670         /* save back tone if available. */
671         key.m_tone = tone;
672         g_free(chewing);
673         return true;
674     }
675
676     g_free(chewing);
677     return false;
678 }
679
680
681 /* only characters in chewing keyboard scheme are accepted here. */
682 int ChewingParser2::parse(pinyin_option_t options, ChewingKeyVector & keys,
683                           ChewingKeyRestVector & key_rests,
684                           const char *str, int len) const {
685     g_array_set_size(keys, 0);
686     g_array_set_size(key_rests, 0);
687
688     int maximum_len = 0; int i;
689     /* probe the longest possible chewing string. */
690     for (i = 0; i < len; ++i) {
691         if (!in_chewing_scheme(str[i]))
692             break;
693     }
694     maximum_len = i;
695
696     /* maximum forward match for chewing. */
697     int parsed_len = 0;
698     while (parsed_len < maximum_len) {
699         const char * cur_str = str + parsed_len;
700         i = std_lite::min(maximum_len - parsed_len,
701                           (int)max_chewing_length);
702
703         ChewingKey key; ChewingKeyRest key_rest;
704         for (; i > 0; --i) {
705             bool success = parse_one_key(options, key, key_rest, cur_str, i);
706             if (success)
707                 break;
708         }
709
710         if (0 == i)        /* no more possible chewings. */
711             break;
712
713         key_rest.m_raw_begin = parsed_len; key_rest.m_raw_end = parsed_len + i;
714         parsed_len += i;
715
716         /* save the pinyin. */
717         g_array_append_val(keys, key);
718         g_array_append_val(key_rests, key_rest);
719     }
720
721     return parsed_len;
722 }
723
724
725 bool ChewingParser2::set_scheme(ChewingScheme scheme) {
726     switch(scheme) {
727     case CHEWING_STANDARD:
728         m_symbol_table = chewing_standard_symbols;
729         m_tone_table   = chewing_standard_tones;
730         return true;
731     case CHEWING_IBM:
732         m_symbol_table = chewing_ibm_symbols;
733         m_tone_table   = chewing_ibm_tones;
734         return true;
735     case CHEWING_GINYIEH:
736         m_symbol_table = chewing_ginyieh_symbols;
737         m_tone_table   = chewing_ginyieh_tones;
738         return true;
739     case CHEWING_ETEN:
740         m_symbol_table = chewing_eten_symbols;
741         m_tone_table   = chewing_eten_tones;
742         return true;
743     }
744
745     return false;
746 }
747
748
749 bool ChewingParser2::in_chewing_scheme(const char key) const {
750     gchar * chewing = NULL;
751     char tone = CHEWING_ZERO_TONE;
752
753     bool retval = search_chewing_symbols(m_symbol_table, key, &chewing) ||
754         search_chewing_tones(m_tone_table, key, &tone);
755     g_free(chewing);
756
757     return retval;
758 }