rhubarb-lip-sync/lib/sphinxbase-rev13216/test/unit/test_feat/test_feat_fe.c

154 lines
4.1 KiB
C
Raw Normal View History

2015-10-19 19:45:08 +00:00
#include <stdio.h>
#include <errno.h>
#include "fe.h"
#include "feat.h"
#include "cmd_ln.h"
#include "ckd_alloc.h"
#include "test_macros.h"
int
main(int argc, char *argv[])
{
static const arg_t fe_args[] = {
waveform_to_cepstral_command_line_macro(),
{ NULL, 0, NULL, NULL }
};
FILE *raw;
cmd_ln_t *config;
fe_t *fe;
feat_t *fcb;
int16 buf[2048];
mfcc_t **cepbuf, **cptr;
mfcc_t ***featbuf1, ***featbuf2, ***fptr;
size_t nsamp;
int32 total_frames, ncep, nfr, i;
if ((raw = fopen(TESTDATADIR "/chan3.raw", "rb")) == NULL) {
perror(TESTDATADIR "/chan3.raw");
return 1;
}
config = cmd_ln_parse_r(NULL, fe_args, argc, argv, FALSE);
fe = fe_init_auto_r(config);
fcb = feat_init("1s_c_d_dd", CMN_NONE, FALSE, AGC_NONE,
TRUE, fe_get_output_size(fe));
/* Determine how much data and how many MFCC frames we need. */
fseek(raw, 0, SEEK_END);
nsamp = ftell(raw) / sizeof(int16);
fe_process_frames(fe, NULL, &nsamp, NULL, &total_frames, NULL);
printf("%d samples, %d + 1 frames\n", nsamp, total_frames);
total_frames++; /* For the possible fe_end_utt() frame */
cepbuf = ckd_calloc_2d(total_frames + 1, fe_get_output_size(fe), sizeof(**cepbuf));
fseek(raw, 0, SEEK_SET);
/* Pay close attention, kids. This is how you use fe_process_frames(). */
fe_start_utt(fe);
cptr = cepbuf;
nfr = total_frames;
while ((nsamp = fread(buf, sizeof(int16), 2048, raw)) > 0) {
int16 const *bptr = buf;
while (nsamp) {
int32 ncep = nfr;
fe_process_frames(fe, &bptr, &nsamp, cptr, &ncep, NULL);
cptr += ncep;
nfr -= ncep;
}
}
fe_end_utt(fe, *cptr, &nfr);
/* Now test some feature extraction problems. */
featbuf1 = feat_array_alloc(fcb, total_frames);
featbuf2 = feat_array_alloc(fcb, total_frames);
/* Whole utterance: canonical, assumed to be correct. */
ncep = total_frames;
TEST_EQUAL(total_frames,
feat_s2mfc2feat_live(fcb, cepbuf,
&ncep, TRUE, TRUE,
featbuf1));
TEST_EQUAL(ncep, total_frames);
/* Process one frame at a time. */
cptr = cepbuf;
fptr = featbuf2;
ncep = 1;
nfr = feat_s2mfc2feat_live(fcb, cptr, &ncep, TRUE, FALSE, fptr);
TEST_EQUAL(nfr, 0); /* Not possible to make any frames yet. */
TEST_EQUAL(ncep, 1); /* But we shold have consumed one. */
cptr += ncep;
for (i = 1; i < total_frames - 1; ++i) {
ncep = 1;
nfr = feat_s2mfc2feat_live(fcb, cptr, &ncep, FALSE, FALSE, fptr);
cptr += ncep;
fptr += nfr;
}
nfr = feat_s2mfc2feat_live(fcb, cptr, &ncep, FALSE, TRUE, fptr);
TEST_EQUAL(nfr, 4); /* This should have dumped the trailing window. */
TEST_EQUAL(ncep, 1); /* And only consumed one frame of MFCCs. */
cptr += ncep;
fptr += nfr;
/* Verify that we actually got the correct number of frames. */
TEST_EQUAL(cptr - cepbuf, total_frames);
TEST_EQUAL(fptr - featbuf2, total_frames);
/* Now verify that the results are equal. */
for (i = 0; i < total_frames; ++i) {
int32 j;
printf("%-4d ", i);
for (j = 0; j < feat_dimension(fcb); ++j) {
TEST_EQUAL_FLOAT(featbuf1[i][0][j], featbuf2[i][0][j]);
}
if (i % 10 == 9)
printf("\n");
}
printf("\n");
/* Process large chunks of frames at once, so as to exceed the
* internal ringbuffer size in feat_s2mfc2feat_live(). */
cptr = cepbuf;
fptr = featbuf2;
ncep = total_frames;
nfr = feat_s2mfc2feat_live(fcb, cptr, &ncep, TRUE, FALSE, fptr);
TEST_ASSERT(ncep != nfr);
cptr += ncep;
fptr += nfr;
ncep = total_frames - ncep;
while (ncep) {
int32 tmp_ncep;
tmp_ncep = ncep;
nfr = feat_s2mfc2feat_live(fcb, cptr, &tmp_ncep, FALSE, FALSE, fptr);
cptr += tmp_ncep;
fptr += nfr;
ncep -= tmp_ncep;
}
nfr = feat_s2mfc2feat_live(fcb, cptr, &ncep, FALSE, TRUE, fptr);
cptr += ncep;
fptr += nfr;
TEST_EQUAL(cptr - cepbuf, total_frames);
TEST_EQUAL(fptr - featbuf2, total_frames);
/* Now verify that the results are equal. */
for (i = 0; i < total_frames; ++i) {
int32 j;
printf("%-4d ", i);
for (j = 0; j < feat_dimension(fcb); ++j)
TEST_EQUAL_FLOAT(featbuf1[i][0][j], featbuf2[i][0][j]);
if (i % 10 == 9)
printf("\n");
}
printf("\n");
fclose(raw);
fe_free(fe);
feat_array_free(featbuf1);
feat_array_free(featbuf2);
feat_free(fcb);
ckd_free_2d(cepbuf);
cmd_ln_free_r(config);
return 0;
}