libsent/include/sent/htk_hmm.h

説明を見る。

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

#ifndef __SENT_HTK_HMM_2_H__
#define __SENT_HTK_HMM_2_H__

#include <sent/stddefs.h>
#include <sent/htk_defs.h>
#include <sent/ptree.h>
#include <sent/mfcc.h>

#define currentis(A)  (!strcasecmp(A, rdhmmdef_token))
#define NoTokErr(S)      if (!rdhmmdef_token) rderr(S)

#define HMMDEF_DELM " \t\n<>"




#define MAX_STATE_NUM 65535

#define HMM_RC_DLIM "+"         
#define HMM_LC_DLIM "-"         
#define HMM_RC_DLIM_C '+'       
#define HMM_LC_DLIM_C '-'       

#define SPMODEL_NAME_DEFAULT "sp"

#define MAX_HMMNAME_LEN 128

enum iwcd_type {
  IWCD_MAX,                     
  IWCD_AVG,                     
  IWCD_NBEST                    
};


/* options info */

typedef struct {
  short num;                    
  short vsize[50];              
} HTK_HMM_StreamInfo;

typedef struct {
  HTK_HMM_StreamInfo stream_info; 
  short vec_size;               
  short cov_type;               
  short dur_type;               
  short param_type;             
} HTK_HMM_Options;


typedef struct _HTK_HMM_trans {
  char *name;                   
  short statenum;               
  PROB **a;                     
  struct _HTK_HMM_trans *next;  
} HTK_HMM_Trans;

typedef struct _HTK_HMM_variance {
  char *name;                   
  VECT *vec;                    
  short len;                    
  struct _HTK_HMM_variance *next; 
} HTK_HMM_Var;

typedef struct _HTK_HMM_dens {
  char *name;                   
  VECT *mean;                   
  short meanlen;                
  HTK_HMM_Var *var;             

  LOGPROB gconst;
  struct _HTK_HMM_dens *next;   
} HTK_HMM_Dens;

typedef struct _HTK_HMM_state {
  char *name;                   
  short mix_num;                
  HTK_HMM_Dens **b;             
  PROB *bweight;                
  unsigned short id;            
  struct _HTK_HMM_state *next;  
} HTK_HMM_State;

typedef struct _HTK_HMM_data {
  char *name;                   
  short state_num;              
  HTK_HMM_State **s;            
  HTK_HMM_Trans *tr;            
  struct _HTK_HMM_data *next;   
} HTK_HMM_Data;

typedef struct {
  char *name;                   
  int num;                      
  HTK_HMM_Dens **d;             
  unsigned short id;            
} GCODEBOOK;

typedef struct {
  HTK_HMM_State *state;         
  /* GCODEBOOK *book;*/         /* pointer to the corresponding codebook in hmminfo */
} GS_SET;




typedef struct {
  HTK_HMM_State **s;            
  unsigned short num;           
  unsigned short maxnum;        
} CD_State_Set;
typedef struct _cd_set{
  char *name;                   
  CD_State_Set *stateset;       
  unsigned short state_num;     
  HTK_HMM_Trans *tr;            
  struct _cd_set *next;         
} CD_Set;
typedef struct {
  APATNODE *cdtree;             
} HMM_CDSET_INFO;


typedef struct _HMM_logical {
  char *name;                   
  boolean is_pseudo;            

  union {
    HTK_HMM_Data *defined;      
    CD_Set *pseudo;             
  } body;
  struct _HMM_logical *next;   
} HMM_Logical;

typedef struct {
  char *name;                   
  boolean bgnflag;              
  boolean endflag;              
} BASEPHONE;
typedef struct {
  int num;                      
  int bgnnum;                   
  int endnum;                   
  APATNODE *root;               
} HMM_basephone;

typedef struct {
  HTK_HMM_Options opt;          
  HTK_HMM_Trans *trstart;       
  HTK_HMM_Var *vrstart;         
  HTK_HMM_Dens *dnstart;        
  HTK_HMM_State *ststart;       
  HTK_HMM_Data *start;          


  HMM_Logical *lgstart;         

  
  APATNODE *tr_root;            
  APATNODE *vr_root;            
  APATNODE *dn_root;            
  APATNODE *st_root;            
  APATNODE *physical_root;      
  APATNODE *logical_root;       
  APATNODE *codebook_root;      


  HMM_basephone basephone;      
  HMM_CDSET_INFO cdset_info;    

  
  boolean is_triphone;          
  boolean is_tied_mixture;      
  short cdset_method;           
  short cdmax_num;              
  HMM_Logical *sp;              
#ifdef MULTIPATH_VERSION
  LOGPROB iwsp_penalty;         
#endif
  boolean variance_inversed;    
  
  int totalmixnum;              
  int totalstatenum;            
  int totalhmmnum;              
  int totallogicalnum;          
  int totalpseudonum;           
  int codebooknum;              
  int maxcodebooksize;          
  int maxmixturenum;            
  int maxstatenum;              

  BMALLOC_BASE *mroot;          
  BMALLOC_BASE *lroot;          

} HTK_HMM_INFO;



/* init_phmm.c */
void htk_hmm_set_pause_model(HTK_HMM_INFO *hmminfo, char *spmodel_name);
/* rdhmmdef.c */
void rderr(char *str);
char *read_token(FILE *fp);
boolean rdhmmdef(FILE *, HTK_HMM_INFO *);
void htk_hmm_inverse_variances(HTK_HMM_INFO *hmm);
/* rdhmmdef_options.c */
void set_global_opt(FILE *fp, HTK_HMM_INFO *hmm);
char *get_cov_str(short covtype);
char *get_dur_str(short durtype);
/* rdhmmdef_trans.c */
void trans_add(HTK_HMM_INFO *hmm, HTK_HMM_Trans *new);
HTK_HMM_Trans *get_trans_data(FILE *, HTK_HMM_INFO *);
void def_trans_macro(char *, FILE *, HTK_HMM_INFO *);
/* rdhmmdef_state.c */
HTK_HMM_State *get_state_data(FILE *, HTK_HMM_INFO *);
void def_state_macro(char *, FILE *, HTK_HMM_INFO *);
HTK_HMM_State *state_lookup(HTK_HMM_INFO *hmm, char *keyname);
void state_add(HTK_HMM_INFO *hmm, HTK_HMM_State *new);
/* rdhmmdef_dens.c */
HTK_HMM_Dens *get_dens_data(FILE *, HTK_HMM_INFO *);
void def_dens_macro(char *, FILE *, HTK_HMM_INFO *);
HTK_HMM_Dens *dens_lookup(HTK_HMM_INFO *hmm, char *keyname);
void dens_add(HTK_HMM_INFO *hmm, HTK_HMM_Dens *new);
/* rdhmmdef_var.c */
HTK_HMM_Var *get_var_data(FILE *, HTK_HMM_INFO *);
void def_var_macro(char *, FILE *, HTK_HMM_INFO *);
void var_add(HTK_HMM_INFO *hmm, HTK_HMM_Var *new);
/* rdhmmdef_data.c */
void def_HMM(char *, FILE *, HTK_HMM_INFO *);
HTK_HMM_Data *htk_hmmdata_new(HTK_HMM_INFO *);
void htk_hmmdata_add(HTK_HMM_INFO *hmm, HTK_HMM_Data *new);
/* rdhmmdef_tiedmix.c */
void tmix_read(FILE *fp, HTK_HMM_State *state, HTK_HMM_INFO *hmm);
void codebook_add(HTK_HMM_INFO *hmm, GCODEBOOK *new);
/* rdhmmdef_regtree.c */
void def_regtree_macro(char *name, FILE *fp, HTK_HMM_INFO *hmm);
/* rdhmmdef_hmmlist.c */
boolean rdhmmlist(FILE *fp, HTK_HMM_INFO *hmminfo);
/* put_htkdata_info.c */
void put_htk_trans(HTK_HMM_Trans *t);
void put_htk_var(HTK_HMM_Var *v);
void put_htk_dens(HTK_HMM_Dens *d);
void put_htk_state(HTK_HMM_State *s);
void put_htk_hmm(HTK_HMM_Data *h);
void put_logical_hmm(HMM_Logical *l);
void print_hmmdef_info(HTK_HMM_INFO *);
HTK_HMM_INFO *hmminfo_new();
boolean hmminfo_free(HTK_HMM_INFO *);
void init_hmminfo(HTK_HMM_INFO *hmminfo, char *filename, char *mapfile, Value *para);
HTK_HMM_Data *htk_hmmdata_lookup_physical(HTK_HMM_INFO *, char *);
HMM_Logical *htk_hmmdata_lookup_logical(HTK_HMM_INFO *, char *);
void hmm_add_physical_to_logical(HTK_HMM_INFO *);
void hmm_add_pseudo_phones(HTK_HMM_INFO *hmminfo);
/* chkhmmlist.c */
void make_hmm_basephone_list(HTK_HMM_INFO *hmminfo);

/* HMM type check functions */
boolean check_hmm_limit(HTK_HMM_Data *dt);
boolean check_all_hmm_limit(HTK_HMM_INFO *hmm);
boolean check_hmm_options(HTK_HMM_INFO *hmm);
#ifdef MULTIPATH_VERSION
boolean is_skippable_model(HTK_HMM_Data *d);
#endif

/* CCD related */
boolean guess_if_cd_hmm(HTK_HMM_INFO *hmm);
HMM_Logical *get_right_context_HMM(HMM_Logical *base, char *rc_name, HTK_HMM_INFO *hmminfo);
HMM_Logical *get_left_context_HMM(HMM_Logical *base, char *lc_name, HTK_HMM_INFO *hmminfo);
void add_right_context(char name[], char *rc);
void add_left_context(char name[], char *lc);
char *center_name(char *hmmname, char *buf);
char *leftcenter_name(char *hmmname, char *buf);
char *rightcenter_name(char *hmmname, char *buf);

/* CD_SET related */
boolean regist_cdset(APATNODE **root, HTK_HMM_Data *d, char *cdname);
boolean make_cdset(HTK_HMM_INFO *hmminfo);
void put_all_cdinfo(HTK_HMM_INFO *hmminfo);
void free_cdset(APATNODE **root);
CD_Set *cdset_lookup(HTK_HMM_INFO *hmminfo, char *cdstr);
CD_Set *lcdset_lookup_by_hmmname(HTK_HMM_INFO *hmminfo, char *hmmname);
CD_Set *rcdset_lookup_by_hmmname(HTK_HMM_INFO *hmminfo, char *hmmname);
int hmm_logical_state_num(HMM_Logical *lg);
HTK_HMM_Trans *hmm_logical_trans(HMM_Logical *lg);

#include <sent/htk_param.h>
boolean check_param_coherence(HTK_HMM_INFO *hmm, HTK_Param *pinfo);
boolean check_param_basetype(HTK_HMM_INFO *hmm, HTK_Param *pinfo);
HTK_Param *new_param_check_and_adjust(HTK_HMM_INFO *hmm, HTK_Param *pinfo, boolean vflag);


/* binary format */
boolean write_binhmm(FILE *fp, HTK_HMM_INFO *hmm, Value *para);
boolean read_binhmm(FILE *fp, HTK_HMM_INFO *hmm, boolean gzfile_p, Value *para);

#endif /* __SENT_HTK_HMM_2_H__ */

---