int
srch_FLAT_FWD_select_active_gmm(void *srch)
{
kbcore_t *kbc;
s3wid_t w;
whmm_t *h;
int32 st;
s3pid_t p;
s3senid_t *senp;
ascr_t *ascr;
srch_FLAT_FWD_graph_t *fwg;
srch_t *s;
mdef_t *mdef;
dict_t *dict;
s = (srch_t *) srch;
fwg = (srch_FLAT_FWD_graph_t *) s->grh->graph_struct;
ascr = s->ascr;
kbc = s->kbc;
mdef = kbcore_mdef(kbc);
dict = kbcore_dict(kbc);
ascr_clear_sen_active(ascr);
/* Flag active senones */
for (w = 0; w < dict->n_word; w++) {
for (h = fwg->whmm[w]; h; h = h->next) {
if (hmm_is_mpx(h)) {
for (st = 0; st < hmm_n_emit_state(h); ++st) {
p = hmm_mpx_ssid(h, st);
if (p == -1)
break; /* All subsequent ones are also inactive */
senp = mdef->sseq[p];
ascr->sen_active[senp[st]] = 1;
}
}
else {
p = hmm_nonmpx_ssid(h);
senp = mdef->sseq[p];
for (st = 0; st < hmm_n_emit_state(h); ++st)
ascr->sen_active[senp[st]] = 1;
}
}
}
return SRCH_SUCCESS;
}
static void
fwdflat_word_transition(ngram_search_t *ngs, int frame_idx)
{
int32 cf, nf, b, thresh, pip, i, w, newscore;
int32 best_silrc_score = 0, best_silrc_bp = 0; /* FIXME: good defaults? */
bptbl_t *bp;
int32 *rcss;
root_chan_t *rhmm;
int32 *awl;
float32 lwf;
dict_t *dict = ps_search_dict(ngs);
dict2pid_t *d2p = ps_search_dict2pid(ngs);
cf = frame_idx;
nf = cf + 1;
thresh = ngs->best_score + ngs->fwdflatbeam;
pip = ngs->pip;
best_silrc_score = WORST_SCORE;
lwf = ngs->fwdflat_fwdtree_lw_ratio;
/* Search for all words starting within a window of this frame.
* These are the successors for words exiting now. */
get_expand_wordlist(ngs, cf, ngs->max_sf_win);
/* Scan words exited in current frame */
for (b = ngs->bp_table_idx[cf]; b < ngs->bpidx; b++) {
xwdssid_t *rssid;
int32 silscore;
bp = ngs->bp_table + b;
ngs->word_lat_idx[bp->wid] = NO_BP;
if (bp->wid == ps_search_finish_wid(ngs))
continue;
/* DICT2PID location */
/* Get the mapping from right context phone ID to index in the
* right context table and the bscore_stack. */
rcss = ngs->bscore_stack + bp->s_idx;
if (bp->last2_phone == -1)
rssid = NULL;
else
rssid = dict2pid_rssid(d2p, bp->last_phone, bp->last2_phone);
/* Transition to all successor words. */
for (i = 0; ngs->expand_word_list[i] >= 0; i++) {
int32 n_used;
w = ngs->expand_word_list[i];
/* Get the exit score we recorded in save_bwd_ptr(), or
* something approximating it. */
if (rssid)
newscore = rcss[rssid->cimap[dict_first_phone(dict, w)]];
else
newscore = bp->score;
if (newscore == WORST_SCORE)
continue;
/* FIXME: Floating point... */
newscore += lwf
* (ngram_tg_score(ngs->lmset,
dict_basewid(dict, w),
bp->real_wid,
bp->prev_real_wid,
&n_used) >> SENSCR_SHIFT);
newscore += pip;
/* Enter the next word */
if (newscore BETTER_THAN thresh) {
rhmm = (root_chan_t *) ngs->word_chan[w];
if ((hmm_frame(&rhmm->hmm) < cf)
|| (newscore BETTER_THAN hmm_in_score(&rhmm->hmm))) {
hmm_enter(&rhmm->hmm, newscore, b, nf);
/* DICT2PID: This is where mpx ssids get introduced. */
/* Look up the ssid to use when entering this mpx triphone. */
hmm_mpx_ssid(&rhmm->hmm, 0) =
dict2pid_ldiph_lc(d2p, rhmm->ciphone, rhmm->ci2phone,
dict_last_phone(dict, bp->wid));
assert(IS_S3SSID(hmm_mpx_ssid(&rhmm->hmm, 0)));
E_DEBUG(6,("ssid %d(%d,%d) = %d\n",
rhmm->ciphone, dict_last_phone(dict, bp->wid), rhmm->ci2phone,
hmm_mpx_ssid(&rhmm->hmm, 0)));
bitvec_set(ngs->word_active, w);
}
}
}
/* Get the best exit into silence. */
if (rssid)
silscore = rcss[rssid->cimap[ps_search_acmod(ngs)->mdef->sil]];
else
silscore = bp->score;
if (silscore BETTER_THAN best_silrc_score) {
best_silrc_score = silscore;
best_silrc_bp = b;
}
}
/* Transition to <sil> */
newscore = best_silrc_score + ngs->silpen + pip;
if ((newscore BETTER_THAN thresh) && (newscore BETTER_THAN WORST_SCORE)) {
w = ps_search_silence_wid(ngs);
rhmm = (root_chan_t *) ngs->word_chan[w];
if ((hmm_frame(&rhmm->hmm) < cf)
|| (newscore BETTER_THAN hmm_in_score(&rhmm->hmm))) {
hmm_enter(&rhmm->hmm, newscore,
best_silrc_bp, nf);
bitvec_set(ngs->word_active, w);
}
}
/* Transition to noise words */
newscore = best_silrc_score + ngs->fillpen + pip;
if ((newscore BETTER_THAN thresh) && (newscore BETTER_THAN WORST_SCORE)) {
for (w = ps_search_silence_wid(ngs) + 1; w < ps_search_n_words(ngs); w++) {
rhmm = (root_chan_t *) ngs->word_chan[w];
/* Noise words that aren't a single phone will have NULL here. */
if (rhmm == NULL)
continue;
if ((hmm_frame(&rhmm->hmm) < cf)
|| (newscore BETTER_THAN hmm_in_score(&rhmm->hmm))) {
hmm_enter(&rhmm->hmm, newscore,
best_silrc_bp, nf);
bitvec_set(ngs->word_active, w);
}
}
}
/* Reset initial channels of words that have become inactive even after word trans. */
i = ngs->n_active_word[cf & 0x1];
awl = ngs->active_word_list[cf & 0x1];
for (w = *(awl++); i > 0; --i, w = *(awl++)) {
rhmm = (root_chan_t *) ngs->word_chan[w];
if (hmm_frame(&rhmm->hmm) == cf) {
hmm_clear_scores(&rhmm->hmm);
}
}
}
