plugin/calcmix.c

説明を見る。

/***************************************************************************/

/* we refer to Julius libsent header */
#include <sent/stddefs.h>
#include <sent/hmm_calc.h>

//#include "plugin_defs.h"

#define PLUGIN_TITLE "Gaussian calculation plugin for Julius"
#define GPRUNE_OPT "mycalc"

int
initialize()
{
  return 0;
}

int
get_plugin_info(int opcode, char *buf, int buflen)
{
  switch(opcode) {
  case 0:
    /* plugin description string */
    strncpy(buf, PLUGIN_TITLE, buflen);
    break;
  }

  return 0;
}

/************************************************************************/
/************************************************************************/

void
calcmix_get_optname(char *buf, int buflen)
{
  strncpy(buf, GPRUNE_OPT, buflen);
}

void
calcmix(HMMWork *wrk, HTK_HMM_Dens **g, int num, int *last_id, int lnum)
{
  int i;
  HTK_HMM_Dens *dens;
  LOGPROB *prob = wrk->OP_calced_score;
  int *id = wrk->OP_calced_id;
  VECT tmp, x;
  VECT *mean;
  VECT *var;
  VECT *vec;
  short veclen;

  for(i=0; i<num; i++) {
    /* store ID */
    *(id++) = i;
    /* get Gaussian to compute */
    dens = *(g++);
    if (dens == NULL) {
      /* no Gaussian, set LOG_ZERO as result */
      *(prob++) = LOG_ZERO;
      continue;
    }
    /* compute log outprob probability */
    mean = dens->mean;
    var = dens->var->vec;
    tmp = dens->gconst;
    vec = wrk->OP_vec;
    veclen = wrk->OP_veclen;
    for (; veclen > 0; veclen--) {
      x = *(vec++) - *(mean++);
      tmp += x * x * *(var++);
    }
    tmp *= -0.5;
    /* store it */
    *(prob++) = tmp;
  }
  wrk->OP_calced_num = num;
}

void
calcmix_free(HMMWork *wrk)
{
  free(wrk->OP_calced_score);
  free(wrk->OP_calced_id);
}
  
boolean
calcmix_init(HMMWork *wrk)
{
  /* maximum Gaussian set size = maximum mixture size * nstream */
  wrk->OP_calced_maxnum = wrk->OP_hmminfo->maxmixturenum * wrk->OP_nstream;
  wrk->OP_calced_score = (LOGPROB *)malloc(sizeof(LOGPROB) * wrk->OP_calced_maxnum);
  wrk->OP_calced_id = (int *)malloc(sizeof(int) * wrk->OP_calced_maxnum);
  /* force gprune_num to the max number */
  wrk->OP_gprune_num = wrk->OP_calced_maxnum;
  return TRUE;
}

/* end of file */