diff options
Diffstat (limited to 'tests/test_pinyin.cpp')
-rw-r--r-- | tests/test_pinyin.cpp | 97 |
1 files changed, 97 insertions, 0 deletions
diff --git a/tests/test_pinyin.cpp b/tests/test_pinyin.cpp new file mode 100644 index 0000000..f94263b --- /dev/null +++ b/tests/test_pinyin.cpp @@ -0,0 +1,97 @@ +/* + * libpinyin + * Library to deal with pinyin. + * + * Copyright (C) 2011 Peng Wu <alexepico@gmail.com> + * + * 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 2 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 + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + */ + + +#include "pinyin.h" +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +int main(int argc, char * argv[]){ + pinyin_context_t * context = + pinyin_init("../data", "../data"); + + pinyin_option_t options = + PINYIN_CORRECT_ALL | USE_DIVIDED_TABLE | USE_RESPLIT_TABLE | + DYNAMIC_ADJUST; + pinyin_set_options(context, options); + + pinyin_instance_t * instance = pinyin_alloc_instance(context); + + char * prefixbuf = NULL; size_t prefixsize = 0; + char * linebuf = NULL; size_t linesize = 0; + ssize_t read; + + while( TRUE ){ + fprintf(stdout, "prefix:"); + fflush(stdout); + + if ((read = getline(&prefixbuf, &prefixsize, stdin)) == -1) + break; + + if ( '\n' == prefixbuf[strlen(prefixbuf) - 1] ) { + prefixbuf[strlen(prefixbuf) - 1] = '\0'; + } + + fprintf(stdout, "pinyin:"); + fflush(stdout); + + if ((read = getline(&linebuf, &linesize, stdin)) == -1) + break; + + if ( '\n' == linebuf[strlen(linebuf) - 1] ) { + linebuf[strlen(linebuf) - 1] = '\0'; + } + + if ( strcmp ( linebuf, "quit" ) == 0) + break; + + pinyin_parse_more_full_pinyins(instance, linebuf); + pinyin_guess_sentence_with_prefix(instance, prefixbuf); + pinyin_guess_full_pinyin_candidates(instance, 0); + + guint len = 0; + pinyin_get_n_candidate(instance, &len); + for (size_t i = 0; i < len; ++i) { + lookup_candidate_t * candidate = NULL; + pinyin_get_candidate(instance, i, &candidate); + + const char * word = NULL; + pinyin_get_candidate_string(instance, candidate, &word); + + printf("%s\t", word); + } + printf("\n"); + + pinyin_train(instance); + pinyin_reset(instance); + pinyin_save(context); + } + + pinyin_free_instance(instance); + + pinyin_mask_out(context, 0x0, 0x0); + pinyin_save(context); + pinyin_fini(context); + + free(prefixbuf); free(linebuf); + return 0; +} |