check_hmm_restriction.c

来自「julius version 4.12.about sound recognit」· C语言 代码 · 共 239 行

/**
 * @file   check_hmm_restriction.c
 * 
 * <JA>
 * @brief  涂えられた %HMM の莲败が蝗脱材墙な妨及かどうかチェックする
 * </JA>
 * 
 * <EN>
 * @brief  Check if the given %HMM definition file can be used
 * </EN>
 *
 * TRANSITION RESTRICTIONS:
 *
 *   - for HTK and Julius:
 *      - no arc to initial state
 *      - no arc from final state
 * 
 *   - Normal version of Julius:
 *      - should have at least one output state
 *      - allow only one arc from initial state
 *      - allow only one arc to final state
 *        (internal skip/loop is allowed)
 *
 *   - Multipath version of Julius:
 *      - should have at least one output state
 *
 * In multipath version, all the transitions including model-skipping
 * transition is allowed.  However, in normal version, their transition
 * is restricted as above.
 * 
 * If such transition is found, Julius output warning and
 * proceed by modifying transition to suite for the restriction.
 *
 * @author Akinobu LEE
 * @date   Tue Feb 15 19:00:58 2005
 *
 * $Revision: 1.2 $
 * 
 */
/*
 * Copyright (c) 1991-2007 Kawahara Lab., Kyoto University
 * Copyright (c) 2000-2005 Shikano Lab., Nara Institute of Science and Technology
 * Copyright (c) 2005-2007 Julius project team, Nagoya Institute of Technology
 * All rights reserved
 */

#include <sent/stddefs.h>
#include <sent/htk_hmm.h>
#include <sent/htk_param.h>
#include <sent/hmm.h>

/** 
 * Return TRUE if it has more than one arc from initial state, or
 * to the final state.  In such case, Julius should be run in
 * multi-path version.
 * 
 * @param hmminfo [in] HMM definition
 * 
 * @return TRUE if has, or FALSE if not exist.
 * </EN>
 */
boolean
htk_hmm_has_several_arc_on_edge(HTK_HMM_INFO *hmminfo)
{
  HTK_HMM_Data *dt;
  HTK_HMM_Trans *t;
  boolean flag;
  int i;

  for (dt = hmminfo->start; dt; dt = dt->next) {
    t = dt->tr;
    flag = FALSE;
    for (i=0;i<t->statenum;i++) {
      if (t->a[0][i] != LOG_ZERO) {
	if (flag == FALSE) {
	  flag = TRUE;
	} else {
	  jlog("Stat: check_hmm_restriction: an HMM with several arcs from initial state found: \"%s\"\n", dt->name);
	  return (TRUE);
	}
      }
    }
    flag = FALSE;
    for (i=0;i<t->statenum;i++) {
      if (t->a[i][t->statenum-1] != LOG_ZERO) {
	if (flag == FALSE) {
	  flag = TRUE;
	} else {
	  jlog("Stat: check_hmm_restriction: an HMM with several arcs to final state found: \"%s\"\n", dt->name);
	  return (TRUE);
	}
      }
    }
  }

  return FALSE;
}

/** 
 * Scan the transition matrix to test the ristrictions.
 * 
 * @param t [in] a transition matrix to be tested
 * 
 * @return 0 if it conforms, 1 if unacceptable transition was found and
 * modification forced, 3 if totally unsupported transition as included and
 * cannot by handled.
 */
static boolean
trans_ok_p(HTK_HMM_Trans *t)
{
  int i;
  int tflag;
  int retflag = TRUE;

  /* check arc to initial state */
  tflag = FALSE;
  for (i=0;i<t->statenum;i++) {
    if (t->a[i][0] != LOG_ZERO) {
      tflag = TRUE;
      break;
    }
  }
  if (tflag) {
    jlog("Error: check_hmm_restriction: transition to initial state is not allowed\n");
    retflag = FALSE;
  }
  /* check arc from final state */
  tflag = FALSE;
  for (i=0;i<t->statenum;i++) {
    if (t->a[t->statenum-1][i] != LOG_ZERO) {
      tflag = TRUE;
      break;
    }
  }
  if (tflag) {
    jlog("Error: check_hmm_restriction: transition from final state is not allowed\n");
    retflag = FALSE;
  }
  /* check if arc from/to initial/final state exist */
  tflag = FALSE;
  for (i=0;i<t->statenum;i++) {
    if (t->a[0][i] != LOG_ZERO) {
      tflag = TRUE;
      break;
    }
  }
  if (tflag == FALSE) {
    jlog("Error: check_hmm_restriction: no transition from initial state\n");
    retflag = FALSE;
  }
  tflag = FALSE;
  for (i=0;i<t->statenum;i++) {
    if (t->a[i][t->statenum-1] != LOG_ZERO) {
      tflag = TRUE;
      break;
    }
  }
  if (tflag == FALSE) {
    jlog("Error: check_hmm_restriction: no transition to final state\n");
    retflag = FALSE;
  }
    
  return(retflag);
}

/** 
 * Check if the transition matrix conforms the ristrictions of Julius.
 * 
 * @param dt [in] HTK %HMM model to check.
 * 
 * @return TRUE on success, FALSE if the check failed.
 */
boolean
check_hmm_limit(HTK_HMM_Data *dt)
{
  boolean return_flag = TRUE;

  if (trans_ok_p(dt->tr) == FALSE) {
    return_flag = FALSE;
    jlog("Error: check_hmm_restriction: HMM \"%s\" has unsupported arc.\n", dt->name);
    put_htk_trans(jlog_get_fp(), dt->tr);
  }
  if (dt->tr->statenum < 3) {
    return_flag = FALSE;
    jlog("Error: HMM \"%s\" has no output state (statenum=%d)\n", dt->name, dt->tr->statenum);
  }
  return(return_flag);
}

/** 
 * Check all the %HMM definitions in a HTK %HMM definition data.
 * 
 * @param hmminfo [in] HTK %HMM data to check.
 * 
 * @return TRUE if there was no bad models, FALSE if at least one model is bad.
 */
boolean
check_all_hmm_limit(HTK_HMM_INFO *hmminfo)
{
  HTK_HMM_Data *dt;
  boolean return_flag = TRUE;

  for (dt = hmminfo->start; dt; dt = dt->next) {
    if (check_hmm_limit(dt) == FALSE) {
      return_flag = FALSE;
    }
  }
  return(return_flag);
}


/** 
 * <JA>
 * モデルが·叫蜗觉轮を沸统せずに掐蜗觉轮から叫蜗觉轮へ木儡莲败するような
 * 莲败を积つかどうかをチェックするˉ
 * 
 * @param d [in] 侠妄HMM
 * 
 * @return 掐蜗から叫蜗への木儡莲败を积つ眷圭 TRUE, 积たない眷圭 FALSE を手すˉ
 * </JA>
 * <EN>
 * Check if the model has direct transition from initial state to final state,
 * skipping all the output state.
 * 
 * @param d [in] logical HMM
 * 
 * @return TRUE if it has direct transition from initial state to final state,
 * that is, this is a "skippable" model.  Otherwise, return FALSE.
 * </EN>
 */
boolean
is_skippable_model(HTK_HMM_Data *d)
{
  if (d->tr->a[0][d->tr->statenum-1] != LOG_ZERO) {
    return TRUE;
  }
  return FALSE;
}