libsent/src/hmminfo/rdhmmdef_streamweight.c

説明を見る。

/*
 * Copyright (c) 1991-2011 Kawahara Lab., Kyoto University
 * Copyright (c) 2000-2005 Shikano Lab., Nara Institute of Science and Technology
 * Copyright (c) 2005-2011 Julius project team, Nagoya Institute of Technology
 * All rights reserved
 */

/* currenty cannot treat other sub macros (~u,~i,~x) */

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

extern char *rdhmmdef_token;    

static HTK_HMM_StreamWeight *
sw_new(HTK_HMM_INFO *hmm)
{
  HTK_HMM_StreamWeight *new;

  new = (HTK_HMM_StreamWeight *)mybmalloc2(sizeof(HTK_HMM_StreamWeight), &(hmm->mroot));

  new->name = NULL;
  new->weight = NULL;
  new->len = 0;
  new->next = NULL;

  return(new);
}

void
sw_add(HTK_HMM_INFO *hmm, HTK_HMM_StreamWeight *new)
{
  HTK_HMM_StreamWeight *match;

  /* link data structure */
  new->next = hmm->swstart;
  hmm->swstart = new;
  
  if (new->name != NULL) {
    /* add index to search index tree */
    if (hmm->sw_root == NULL) {
      hmm->sw_root = aptree_make_root_node(new, &(hmm->mroot));
    } else {
      match = aptree_search_data(new->name, hmm->sw_root);
      if (match != NULL && strmatch(match->name, new->name)) {
        jlog("Error: rdhmmdef_streamweight: ~w \"%s\" is already defined\n", new->name);
        rderr(NULL);
      } else {
        aptree_add_entry(new->name, new, match->name, &(hmm->sw_root), &(hmm->mroot));
      }
    }
  }
  
}

static HTK_HMM_StreamWeight *
sw_lookup(HTK_HMM_INFO *hmm, char *keyname)
{
  HTK_HMM_StreamWeight *sw;

  sw = aptree_search_data(keyname, hmm->sw_root);
  if (sw != NULL && strmatch(sw->name, keyname)) {
    return sw;
  } else {
    return NULL;
  }
}

static HTK_HMM_StreamWeight *
sw_read(FILE *fp, HTK_HMM_INFO *hmm)
{
  HTK_HMM_StreamWeight *new;
  int i;

  new = sw_new(hmm);
  /* read stream weight */
  if (!currentis("SWEIGHTS")) {
    jlog("Error: rdhmmdef_streamweight: failed to read stream weight: \"%s\"\n", rdhmmdef_token);
    rderr(NULL);
  } else {
    read_token(fp);
    NoTokErr("missing SWEIGHTS vector length");
    new->len = atoi(rdhmmdef_token);
    read_token(fp);
    new->weight = (VECT *)mybmalloc2(sizeof(VECT) * new->len, &(hmm->mroot));
    /* needs conversion if integerized */
    for (i=0;i<new->len;i++) {
      NoTokErr("missing some SWEIGHTS element");
      new->weight[i] = (VECT)atof(rdhmmdef_token);
      read_token(fp);
    }
  }

  return (new);
}

HTK_HMM_StreamWeight *
get_streamweight_data(FILE *fp, HTK_HMM_INFO *hmm)
{
  HTK_HMM_StreamWeight *tmp;

  if (currentis("~w")) {
    /* macro reference: lookup and return the pointer */
    read_token(fp);
    NoTokErr("missing SWEIGHTS macro name");
    tmp = sw_lookup(hmm, rdhmmdef_token);
    if (tmp == NULL) {
      jlog("Error: rdhmmdef_streamweight: ~w \"%s\" not defined\n", rdhmmdef_token);
      rderr(NULL);
    }
    read_token(fp);
    return tmp;
  } else if (currentis("SWEIGHTS")){
    /* definition: define stream weight data, and return the pointer */
    tmp = sw_read(fp, hmm);
    tmp->name = NULL; /* no name */
    sw_add(hmm, tmp);
    return tmp;
  } else {
    rderr("no stream weights data");
    return NULL;
  }
}

void
def_streamweight_macro(char *name, FILE *fp, HTK_HMM_INFO *hmm)
{
  HTK_HMM_StreamWeight *new;

  
  /* read in data and return newly malloced data */
  new = sw_read(fp, hmm);

  /* register it to the grobal HMM structure */
  new->name = name;
  sw_add(hmm, new);
}

/* end of file */