pocketsphinx/test/unit/test_align_fsg_beam.c at 33df974bc997d2b789d38e0b7353ce6797b772b2 · cmusphinx/pocketsphinx · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
/* -*- c-basic-offset: 4 -*- */
#include <pocketsphinx.h>

#include "pocketsphinx_internal.h"
#include "fsg_search_internal.h"
#include "util/hash_table.h"
#include "test_macros.h"

int
main(int argc, char *argv[])
{
    ps_decoder_t *ps;
    ps_config_t *config;
    void *search_p;
    fsg_search_t *fsgs;

    (void)argc;
    (void)argv;
    err_set_loglevel(ERR_INFO);
    /* Stock asymmetry: wbeam differs from beam; forced-align FSG must use align_* */
    TEST_ASSERT(config =
                ps_config_parse_json(
                    NULL,
                    "loglevel: INFO, bestpath: false,"
                    "hmm: \"" MODELDIR "/en-us/en-us\","
                    "dict: \"" MODELDIR "/en-us/cmudict-en-us.dict\","
                    "samprate: 16000,"
                    "wbeam: 7e-29"));
    TEST_ASSERT(ps = ps_init(config));
    TEST_EQUAL(0, ps_set_align_text(ps, "go forward ten meters"));
    TEST_EQUAL(0, hash_table_lookup(ps->searches, PS_DEFAULT_ALIGN_SEARCH,
                                    &search_p));
    fsgs = (fsg_search_t *)search_p;
    TEST_EQUAL(fsgs->wbeam_orig, fsgs->beam_orig);
    TEST_EQUAL(fsgs->pbeam_orig, fsgs->beam_orig);

    ps_free(ps);
    ps_config_free(config);

    /* With align_use_main_beams, FSG uses global wbeam (asymmetric from beam). */
    TEST_ASSERT(config =
                ps_config_parse_json(
                    NULL,
                    "loglevel: ERROR, bestpath: false,"
                    "hmm: \"" MODELDIR "/en-us/en-us\","
                    "dict: \"" MODELDIR "/en-us/cmudict-en-us.dict\","
                    "samprate: 16000,"
                    "wbeam: 7e-29,"
                    "align_use_main_beams: yes"));
    TEST_ASSERT(ps = ps_init(config));
    TEST_EQUAL(0, ps_set_align_text(ps, "go forward ten meters"));
    TEST_EQUAL(0, hash_table_lookup(ps->searches, PS_DEFAULT_ALIGN_SEARCH,
                                    &search_p));
    fsgs = (fsg_search_t *)search_p;
    TEST_ASSERT(fsgs->wbeam_orig != fsgs->beam_orig);

    ps_free(ps);
    ps_config_free(config);
    return 0;
}