libsent/include/sent/hmm_calc.h

説明を見る。

/*
 * 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
 */

#ifndef __SENT_HMM_CALC_H__
#define __SENT_HMM_CALC_H__

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

enum{GPRUNE_SEL_UNDEF, GPRUNE_SEL_NONE, GPRUNE_SEL_SAFE, GPRUNE_SEL_HEURISTIC, GPRUNE_SEL_BEAM, GPRUNE_SEL_USER};

#define TMBEAMWIDTH 5.0

typedef struct {
  LOGPROB score;                
  int id;               
} MIXCACHE;

typedef struct __hmmwork__{

  /* pointer to functions, selected by model configuration */

  LOGPROB (*calc_outprob)(struct __hmmwork__ *);

  LOGPROB (*calc_outprob_state)(struct __hmmwork__ *);

  void (*compute_gaussset)(struct __hmmwork__ *, HTK_HMM_Dens **g, int num, int *last_id, int lnum);

  boolean (*compute_gaussset_init)(struct __hmmwork__ *);

  void (*compute_gaussset_free)(struct __hmmwork__ *);

  /* local storage of pointers to the HMM */
  HTK_HMM_INFO *OP_hmminfo; 
  HTK_HMM_INFO *OP_gshmm; 

  /* local storage of input parameters */
  HTK_Param *OP_param;  
  int OP_gprune_num; 
  int OP_time;          
  int OP_last_time;     

  /* current computing state */
  HTK_HMM_State *OP_state;      
  int OP_state_id;              

  /* for multi-stream input */
  short OP_nstream;             
  VECT *OP_vec_stream[MAXSTREAMNUM]; 
  short OP_veclen_stream[MAXSTREAMNUM]; 

  /* temporal buffers to hold result of mixture computation at each stream */
  VECT *OP_vec;         
  short OP_veclen;              
  int OP_calced_maxnum; 
  LOGPROB *OP_calced_score; 
  int *OP_calced_id; 
  int OP_calced_num; 

  /* state level cache */
  int statenum;         
  LOGPROB **outprob_cache; 
  int outprob_allocframenum;    
  BMALLOC_BASE *croot;  
  LOGPROB *last_cache;  

  /* mixture level cache for tied-mixture model */
  MIXCACHE ***mixture_cache; 
  short **mixture_cache_num; 
  BMALLOC_BASE *mroot;  

  /* work area for tied-mixture computation */
  int *tmix_last_id;            
  int tmix_allocframenum;       

  /* work area for gaussian pruning (common) */
  boolean *mixcalced;   
  /* for beam gaussian pruning */
  LOGPROB *dimthres;    
  int dimthres_num;     
  /* for heuristic gaussian pruning */
  LOGPROB *backmax;     
  int backmax_num;              

  /* work area for outprob_cd_nbest */
  LOGPROB *cd_nbest_maxprobs;   
  int cd_nbest_maxn;            

  /* work area for GMS */
  /* GMS variables */
  int my_nbest;         
  int gms_allocframenum;        
  /* GMS info */
  GS_SET *gsset;                
  int gsset_num;                
  int *state2gs; 
  /* GMS results */
  boolean *gms_is_selected;     
  LOGPROB **fallback_score; 
  /* GMS calculation */
  int *gsindex;         
  LOGPROB *t_fs;                
  /* GMS gprune local cache */
  int **gms_last_max_id_list;   

} HMMWork;  


#ifdef __cplusplus
extern "C" {
#endif

/* addlog.c */
void make_log_tbl();
LOGPROB addlog(LOGPROB x, LOGPROB y);
LOGPROB addlog_array(LOGPROB *x, int n);

/* outprob_init.c */
boolean
outprob_init(HMMWork *wrk, HTK_HMM_INFO *hmminfo,
             HTK_HMM_INFO *gshmm, int gms_num,
             int gprune_method, int gprune_mixnum
             );
boolean outprob_prepare(HMMWork *wrk, int framenum);
void outprob_free(HMMWork *wrk);
/* outprob.c */
boolean outprob_cache_init(HMMWork *wrk);
boolean outprob_cache_prepare(HMMWork *wrk);
void outprob_cache_free(HMMWork *wrk);
LOGPROB outprob_state(HMMWork *wrk, int t, HTK_HMM_State *stateinfo, HTK_Param *param);
void outprob_cd_nbest_init(HMMWork *wrk, int num);
void outprob_cd_nbest_free(HMMWork *wrk);
LOGPROB outprob_cd(HMMWork *wrk, int t, CD_State_Set *lset, HTK_Param *param);
/* gms.c */
boolean gms_init(HMMWork *wrk);
boolean gms_prepare(HMMWork *wrk, int framelen);
void gms_free(HMMWork *wrk);
LOGPROB gms_state(HMMWork *wrk);
/* gms_gprune.c */
void gms_gprune_init(HMMWork *wrk);
void gms_gprune_prepare(HMMWork *wrk);
void gms_gprune_free(HMMWork *wrk);
void compute_gs_scores(HMMWork *wrk);

/* calc_mix.c */
LOGPROB calc_mix(HMMWork *wrk);
/* calc_tied_mix.c */
boolean calc_tied_mix_init(HMMWork *wrk);
boolean calc_tied_mix_prepare(HMMWork *wrk, int framenum);
void calc_tied_mix_free(HMMWork *wrk);
LOGPROB calc_tied_mix(HMMWork *wrk);
LOGPROB calc_compound_mix(HMMWork *wrk);

/* gprune_common.c */
int cache_push(HMMWork *wrk, int id, LOGPROB score, int len);
/* gprune_none.c */
LOGPROB compute_g_base(HMMWork *wrk, HTK_HMM_Dens *binfo);
boolean gprune_none_init(HMMWork *wrk);
void gprune_none_free(HMMWork *wrk);
void gprune_none(HMMWork *wrk, HTK_HMM_Dens **g, int num, int *last_id, int lnum);
/* gprune_safe.c */
LOGPROB compute_g_safe(HMMWork *wrk, HTK_HMM_Dens *binfo, LOGPROB thres);
boolean gprune_safe_init(HMMWork *wrk);
void gprune_safe_free(HMMWork *wrk);
void gprune_safe(HMMWork *wrk, HTK_HMM_Dens **g, int gnum, int *last_id, int lnum);
/* gprune_heu.c */
boolean gprune_heu_init(HMMWork *wrk);
void gprune_heu_free(HMMWork *wrk);
void gprune_heu(HMMWork *wrk, HTK_HMM_Dens **g, int gnum, int *last_id, int lnum);
/* gprune_beam.c */
boolean gprune_beam_init(HMMWork *wrk);
void gprune_beam_free(HMMWork *wrk);
void gprune_beam(HMMWork *wrk, HTK_HMM_Dens **g, int gnum, int *last_id, int lnum);


#ifdef __cplusplus
}
#endif

#endif /* __SENT_HMM_CALC_H__ */