cannam@16: #ifndef _HMM_H
cannam@16: #define _HMM_H
cannam@16: 
cannam@20: #ifdef __cplusplus
cannam@20: extern "C" {
cannam@20: #endif
cannam@20: 
cannam@16: /*
cannam@16:  *  hmm.h
cannam@16:  *  soundbite
cannam@16:  *
cannam@16:  *  Created by Mark Levy on 12/02/2006.
cannam@16:  *  Copyright 2006 Centre for Digital Music, Queen Mary, University of London. All rights reserved.
cannam@16:  *
cannam@16:  */
cannam@16: 
cannam@16: #ifndef PI    
cannam@16: #define PI 3.14159265358979323846264338327950288
cannam@16: #endif 
cannam@16: 
cannam@16: typedef struct _model_t {
cannam@16: 	int N;			/* number of states */
cannam@16: 	double* p0;		/* initial probs */
cannam@16: 	double** a;		/* transition probs */
cannam@16: 	int L;			/* dimensionality of data */
cannam@16: 	double** mu;	/* state means */
cannam@16: 	double** cov;	/* covariance, tied between all states */
cannam@16: } model_t;
cannam@16: 
cannam@16: void hmm_train(double** x, int T, model_t* model);							/* with scaling */
cannam@16: void forward_backwards(double*** xi, double** gamma, double* loglik, double* loglik1, double* loglik2, int iter, 
cannam@16: 					   int N, int T, double* p0, double** a, double** b);
cannam@16: void baum_welch(double* p0, double** a, double** mu, double** cov, int N, int T, int L, double** x, double*** xi, double** gamma);
cannam@16: void viterbi_decode(double** x, int T, model_t* model, int* q);				/* using logs */
cannam@16: model_t* hmm_init(double** x, int T, int L, int N);
cannam@16: void hmm_close(model_t* model);
cannam@16: void invert(double** cov, int L, double** icov, double* detcov);			/* uses LAPACK (included with Mac OSX) */
cannam@16: double gauss(double* x, int L, double* mu, double** icov, double detcov, double* y, double* z);
cannam@16: double loggauss(double* x, int L, double* mu, double** icov, double detcov, double* y, double* z);
cannam@16: void hmm_print(model_t* model);
cannam@16: 
cannam@20: #ifdef __cplusplus
cannam@20: }
cannam@16: #endif
cannam@16: 
cannam@20: #endif
cannam@20: