| | #include <pocketsphinx.h> |
| | #include <stdio.h> |
| | #include <string.h> |
| | #include <time.h> |
| |
|
| | #include "pocketsphinx_internal.h" |
| | #include "ngram_search_fwdtree.h" |
| | #include "test_macros.h" |
| |
|
| | int |
| | main(int argc, char *argv[]) |
| | { |
| | ps_decoder_t *ps; |
| | cmd_ln_t *config; |
| | acmod_t *acmod; |
| | ngram_search_t *ngs; |
| |
|
| | (void)argc; |
| | (void)argv; |
| | TEST_ASSERT(config = |
| | ps_config_parse_json( |
| | NULL, |
| | "hmm: \"" MODELDIR "/en-us/en-us\"," |
| | "lm: \"" MODELDIR "/en-us/en-us.lm.bin\"," |
| | "dict: \"" MODELDIR "/en-us/cmudict-en-us.dict\"," |
| | "fwdtree: true," |
| | "fwdflat: false," |
| | "bestpath: false," |
| | "samprate: 16000")); |
| | TEST_ASSERT(ps = ps_init(config)); |
| |
|
| | ngs = (ngram_search_t *)ps->search; |
| | acmod = ps->acmod; |
| |
|
| | setbuf(stdout, NULL); |
| | { |
| | FILE *rawfh, *senfh; |
| | int16 buf[2048]; |
| | size_t nread; |
| | int16 const *bptr; |
| | int nfr; |
| |
|
| | TEST_ASSERT(rawfh = fopen(DATADIR "/goforward.raw", "rb")); |
| | TEST_EQUAL(0, acmod_start_utt(acmod)); |
| | TEST_ASSERT(senfh = fopen("goforward.sen", "wb")); |
| | TEST_EQUAL(0, acmod_set_senfh(acmod, senfh)); |
| | ngram_fwdtree_start(ngs); |
| | while (!feof(rawfh)) { |
| | nread = fread(buf, sizeof(*buf), 2048, rawfh); |
| | bptr = buf; |
| | while ((nfr = acmod_process_raw(acmod, &bptr, &nread, FALSE)) > 0) { |
| | while (acmod->n_feat_frame > 0) { |
| | ngram_fwdtree_search(ngs, acmod->output_frame); |
| | acmod_advance(acmod); |
| | } |
| | } |
| | } |
| | ngram_fwdtree_finish(ngs); |
| | printf("%s\n", |
| | ngram_search_bp_hyp(ngs, ngram_search_find_exit(ngs, -1, NULL))); |
| |
|
| | TEST_ASSERT(acmod_end_utt(acmod) >= 0); |
| | fclose(rawfh); |
| | TEST_EQUAL(0, strcmp("go forward ten meters", |
| | ngram_search_bp_hyp(ngs, |
| | ngram_search_find_exit(ngs, -1, NULL)))); |
| |
|
| | TEST_EQUAL(0, acmod_set_senfh(acmod, NULL)); |
| | TEST_EQUAL(0, acmod_start_utt(acmod)); |
| | TEST_ASSERT(senfh = fopen("goforward.sen", "rb")); |
| | TEST_EQUAL(0, acmod_set_insenfh(acmod, senfh)); |
| | ngram_fwdtree_start(ngs); |
| | while ((nfr = acmod_read_scores(acmod)) > 0) { |
| | while (acmod->n_feat_frame > 0) { |
| | ngram_fwdtree_search(ngs, acmod->output_frame); |
| | acmod_advance(acmod); |
| | } |
| | } |
| | ngram_fwdtree_finish(ngs); |
| | printf("%s\n", |
| | ngram_search_bp_hyp(ngs, ngram_search_find_exit(ngs, -1, NULL))); |
| |
|
| | TEST_ASSERT(acmod_end_utt(acmod) >= 0); |
| | TEST_EQUAL(0, strcmp("go forward ten meters", |
| | ngram_search_bp_hyp(ngs, |
| | ngram_search_find_exit(ngs, -1, NULL)))); |
| | fclose(senfh); |
| | } |
| | ps_free(ps); |
| | ps_config_free(config); |
| |
|
| | return 0; |
| | } |
| |
|
