File Coverage

pp-hmmexpect.c
Criterion Covered Total %
statement 85 132 64.3
branch 66 304 21.7
condition n/a
subroutine n/a
pod n/a
total 151 436 34.6


line stmt bran cond sub pod time code
1              
2             #line 453 "lib/PDL/PP.pm"
3             /*
4             * THIS FILE WAS GENERATED BY PDL::PP from HMM.pd! Do not modify!
5             */
6              
7             #define PDL_FREE_CODE(trans, destroy, comp_free_code, ntpriv_free_code) \
8             if (destroy) { \
9             comp_free_code \
10             } \
11             if ((trans)->dims_redone) { \
12             ntpriv_free_code \
13             }
14              
15             #include "EXTERN.h"
16             #include "perl.h"
17             #include "XSUB.h"
18             #include "pdl.h"
19             #include "pdlcore.h"
20             #define PDL PDL_HMM
21             extern Core* PDL; /* Structure hold core C functions */
22             #line 23 "pp-hmmexpect.c"
23              
24             #include
25              
26             /*#define DEBUG_ALPHA*/
27             /*#define DEBUG_BETA*/
28             /*#define DEBUG_VITERBI*/
29              
30              
31             /* logadd(x,y) = log(exp(x)+exp(y))
32             * + Code from Manning & Schütze (1997), Sec. 9.4, page 337
33             *
34             * LOG_BIG = log(1E31)
35             */
36             #define LOG_BIG 71.3801378828154
37             #define LOG_ZERO -1E+38
38             #define LOG_ONE 0
39             #define LOG_NONE 1
40 84           static inline double logadd1(double x, double y) {
41 84 100         if (y-x > LOG_BIG) return y;
42 45 100         else if (x-y > LOG_BIG) return x;
43             /*else return min(x,y) + log(exp(x-min(x,y)) + exp(y-min(x,y))); */
44 22 100         else if (x
45 14           else return y + log(exp(x-y) + 1);
46             }
47             static inline double logadd0(double x, double y) {
48             return log(exp(x)+exp(y));
49             }
50              
51             /* logdiff(x,y) = log(exp(x)-exp(y))
52             * + adapted from above
53             * + always returns positive (i.e. symmetric difference)
54             */
55             static inline double logdiff1(double x, double y) {
56             if (y-x > LOG_BIG) { return y; }
57             else if (x-y > LOG_BIG) { return x; }
58             /*else { return max(x,y) + log(exp(max(x,y)-max(x,y)) - exp(min(x,y)-max(x,y))); } */
59             /* = max(x,y) + log( 1 - exp(min(x,y)-max(x,y))); } */
60             else if (x>y) { return x + log( 1 - exp(y-x)); }
61             else { return y + log( 1 - exp(x-y)); }
62             }
63             static inline double logdiff0(double x, double y) {
64             return log(x>y ? (exp(x)-exp(y)) : (exp(y)-exp(x)));
65             }
66              
67             /*
68             #define logadd(x,y) logadd0(x,y)
69             #define logdiff(x,y) logdiff0(x,y)
70             */
71              
72             #define logadd(x,y) logadd1(x,y)
73             #define logdiff(x,y) logdiff1(x,y)
74              
75              
76             #line 1857 "lib/PDL/PP.pm"
77             pdl_error pdl_hmmexpect_readdata(pdl_trans *__privtrans) {
78             pdl_error PDL_err = {0, NULL, 0};
79             #line 80 "pp-hmmexpect.c"
80 6           register PDL_Indx __N_size = __privtrans->ind_sizes[1];
81 6 50         if (!__privtrans->broadcast.incs) return PDL->make_error(PDL_EUSERERROR, "Error in hmmexpect:" "broadcast.incs NULL");
82             /* broadcastloop declarations */
83             int __brcloopval;
84             register PDL_Indx __tind0,__tind1; /* counters along dim */
85 6           register PDL_Indx __tnpdls = __privtrans->broadcast.npdls;
86             /* dims here are how many steps along those dims */
87 6           register PDL_Indx __tinc0_a = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,0,0);
88 6           register PDL_Indx __tinc0_b = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,1,0);
89 6           register PDL_Indx __tinc0_pi = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,2,0);
90 6           register PDL_Indx __tinc0_omega = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,3,0);
91 6           register PDL_Indx __tinc0_o = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,4,0);
92 6           register PDL_Indx __tinc0_alpha = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,5,0);
93 6           register PDL_Indx __tinc0_beta = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,6,0);
94 6           register PDL_Indx __tinc0_ea = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,7,0);
95 6           register PDL_Indx __tinc0_eb = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,8,0);
96 6           register PDL_Indx __tinc0_epi = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,9,0);
97 6           register PDL_Indx __tinc0_eomega = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,10,0);
98 6           register PDL_Indx __tinc1_a = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,0,1);
99 6           register PDL_Indx __tinc1_b = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,1,1);
100 6           register PDL_Indx __tinc1_pi = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,2,1);
101 6           register PDL_Indx __tinc1_omega = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,3,1);
102 6           register PDL_Indx __tinc1_o = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,4,1);
103 6           register PDL_Indx __tinc1_alpha = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,5,1);
104 6           register PDL_Indx __tinc1_beta = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,6,1);
105 6           register PDL_Indx __tinc1_ea = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,7,1);
106 6           register PDL_Indx __tinc1_eb = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,8,1);
107 6           register PDL_Indx __tinc1_epi = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,9,1);
108 6           register PDL_Indx __tinc1_eomega = PDL_BRC_INC(__privtrans->broadcast.incs,__tnpdls,10,1);
109             #define PDL_BROADCASTLOOP_START_hmmexpect_readdata PDL_BROADCASTLOOP_START( \
110             readdata, \
111             __privtrans->broadcast, \
112             __privtrans->vtable, \
113             a_datap += __offsp[0]; \
114             b_datap += __offsp[1]; \
115             pi_datap += __offsp[2]; \
116             omega_datap += __offsp[3]; \
117             o_datap += __offsp[4]; \
118             alpha_datap += __offsp[5]; \
119             beta_datap += __offsp[6]; \
120             ea_datap += __offsp[7]; \
121             eb_datap += __offsp[8]; \
122             epi_datap += __offsp[9]; \
123             eomega_datap += __offsp[10]; \
124             , \
125             ( ,a_datap += __tinc1_a - __tinc0_a * __tdims0 \
126             ,b_datap += __tinc1_b - __tinc0_b * __tdims0 \
127             ,pi_datap += __tinc1_pi - __tinc0_pi * __tdims0 \
128             ,omega_datap += __tinc1_omega - __tinc0_omega * __tdims0 \
129             ,o_datap += __tinc1_o - __tinc0_o * __tdims0 \
130             ,alpha_datap += __tinc1_alpha - __tinc0_alpha * __tdims0 \
131             ,beta_datap += __tinc1_beta - __tinc0_beta * __tdims0 \
132             ,ea_datap += __tinc1_ea - __tinc0_ea * __tdims0 \
133             ,eb_datap += __tinc1_eb - __tinc0_eb * __tdims0 \
134             ,epi_datap += __tinc1_epi - __tinc0_epi * __tdims0 \
135             ,eomega_datap += __tinc1_eomega - __tinc0_eomega * __tdims0 \
136             ), \
137             ( ,a_datap += __tinc0_a \
138             ,b_datap += __tinc0_b \
139             ,pi_datap += __tinc0_pi \
140             ,omega_datap += __tinc0_omega \
141             ,o_datap += __tinc0_o \
142             ,alpha_datap += __tinc0_alpha \
143             ,beta_datap += __tinc0_beta \
144             ,ea_datap += __tinc0_ea \
145             ,eb_datap += __tinc0_eb \
146             ,epi_datap += __tinc0_epi \
147             ,eomega_datap += __tinc0_eomega \
148             ) \
149             )
150             #define PDL_BROADCASTLOOP_END_hmmexpect_readdata PDL_BROADCASTLOOP_END( \
151             __privtrans->broadcast, \
152             a_datap -= __tinc1_a * __tdims1 + __offsp[0]; \
153             b_datap -= __tinc1_b * __tdims1 + __offsp[1]; \
154             pi_datap -= __tinc1_pi * __tdims1 + __offsp[2]; \
155             omega_datap -= __tinc1_omega * __tdims1 + __offsp[3]; \
156             o_datap -= __tinc1_o * __tdims1 + __offsp[4]; \
157             alpha_datap -= __tinc1_alpha * __tdims1 + __offsp[5]; \
158             beta_datap -= __tinc1_beta * __tdims1 + __offsp[6]; \
159             ea_datap -= __tinc1_ea * __tdims1 + __offsp[7]; \
160             eb_datap -= __tinc1_eb * __tdims1 + __offsp[8]; \
161             epi_datap -= __tinc1_epi * __tdims1 + __offsp[9]; \
162             eomega_datap -= __tinc1_eomega * __tdims1 + __offsp[10]; \
163             )
164 6           register PDL_Indx __inc_a_N0 = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,0,0)]; (void)__inc_a_N0;register PDL_Indx __inc_a_N1 = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,0,1)]; (void)__inc_a_N1;
165 6           register PDL_Indx __inc_alpha_N = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,5,0)]; (void)__inc_alpha_N;register PDL_Indx __inc_alpha_T = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,5,1)]; (void)__inc_alpha_T;
166 6           register PDL_Indx __inc_b_N = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,1,0)]; (void)__inc_b_N;register PDL_Indx __inc_b_M = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,1,1)]; (void)__inc_b_M;
167 6           register PDL_Indx __inc_beta_N = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,6,0)]; (void)__inc_beta_N;register PDL_Indx __inc_beta_T = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,6,1)]; (void)__inc_beta_T;
168 6           register PDL_Indx __inc_ea_N0 = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,7,0)]; (void)__inc_ea_N0;register PDL_Indx __inc_ea_N1 = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,7,1)]; (void)__inc_ea_N1;
169 6           register PDL_Indx __inc_eb_N = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,8,0)]; (void)__inc_eb_N;register PDL_Indx __inc_eb_M = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,8,1)]; (void)__inc_eb_M;
170 6           register PDL_Indx __inc_eomega_N = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,10,0)]; (void)__inc_eomega_N;
171 6           register PDL_Indx __inc_epi_N = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,9,0)]; (void)__inc_epi_N;
172 6           register PDL_Indx __inc_o_T = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,4,0)]; (void)__inc_o_T;
173 6           register PDL_Indx __inc_omega_N = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,3,0)]; (void)__inc_omega_N;
174 6           register PDL_Indx __inc_pi_N = __privtrans->inc_sizes[PDL_INC_ID(__privtrans->vtable,2,0)]; (void)__inc_pi_N;
175             #ifndef PDL_DECLARE_PARAMS_hmmexpect_1
176             #define PDL_DECLARE_PARAMS_hmmexpect_1(PDL_TYPE_OP,PDL_PPSYM_OP) \
177             PDL_DECLARE_PARAMETER(PDL_TYPE_OP, a, (__privtrans->pdls[0]), 1, PDL_PPSYM_OP) \
178             PDL_DECLARE_PARAMETER(PDL_TYPE_OP, b, (__privtrans->pdls[1]), 1, PDL_PPSYM_OP) \
179             PDL_DECLARE_PARAMETER(PDL_TYPE_OP, pi, (__privtrans->pdls[2]), 1, PDL_PPSYM_OP) \
180             PDL_DECLARE_PARAMETER(PDL_TYPE_OP, omega, (__privtrans->pdls[3]), 1, PDL_PPSYM_OP) \
181             PDL_DECLARE_PARAMETER(PDL_TYPE_OP, o, (__privtrans->pdls[4]), 1, PDL_PPSYM_OP) \
182             PDL_DECLARE_PARAMETER(PDL_TYPE_OP, alpha, (__privtrans->pdls[5]), 1, PDL_PPSYM_OP) \
183             PDL_DECLARE_PARAMETER(PDL_TYPE_OP, beta, (__privtrans->pdls[6]), 1, PDL_PPSYM_OP) \
184             PDL_DECLARE_PARAMETER(PDL_TYPE_OP, ea, (__privtrans->pdls[7]), 1, PDL_PPSYM_OP) \
185             PDL_DECLARE_PARAMETER(PDL_TYPE_OP, eb, (__privtrans->pdls[8]), 1, PDL_PPSYM_OP) \
186             PDL_DECLARE_PARAMETER(PDL_TYPE_OP, epi, (__privtrans->pdls[9]), 1, PDL_PPSYM_OP) \
187             PDL_DECLARE_PARAMETER(PDL_TYPE_OP, eomega, (__privtrans->pdls[10]), 1, PDL_PPSYM_OP)
188             #endif
189             #define PDL_IF_BAD(t,f) f
190 6           switch (__privtrans->__datatype) { /* Start generic switch */
191 0           case PDL_F: {
192 0 0         PDL_DECLARE_PARAMS_hmmexpect_1(PDL_Float,F)
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
193 0 0         PDL_BROADCASTLOOP_START_hmmexpect_readdata {
    0          
    0          
    0          
    0          
    0          
    0          
194             int i,j,t;
195             int o_tp1, o_t;
196 0           double p_o = LOG_ZERO;
197             double gamma_it;
198             double xi_ijt;
199              
200             /*-- Initialize: t==(T-1): P(o|@theta) --*/
201 0           t = __privtrans->ind_sizes[2]-1;
202 0 0         {/* Open N */ PDL_EXPAND2(register PDL_Indx N=0, __N_stop=(__N_size)); for(; N<__N_stop; N+=1) { p_o = logadd(p_o, (omega_datap)[0+(__inc_omega_N*(N))] + (alpha_datap)[0+(__inc_alpha_N*(N))+(__inc_alpha_T*(t))]); }} /* Close N */
203              
204              
205             /*-- Initialize: t==(T-1): Iterate: state_t==i: get gamma(i,t) --*/
206 0           o_t = (o_datap)[0+(__inc_o_T*(t))];
207 0 0         for (i=0; i<__privtrans->ind_sizes[1]; i++) {
208 0           gamma_it = (alpha_datap)[0+(__inc_alpha_N*(i))+(__inc_alpha_T*(t))] + (beta_datap)[0+(__inc_beta_N*(i))+(__inc_beta_T*(t))] - p_o;
209 0           (eb_datap)[0+(__inc_eb_N*(i))+(__inc_eb_M*(o_t))] = logadd((eb_datap)[0+(__inc_eb_N*(i))+(__inc_eb_M*(o_t))], gamma_it);
210 0           (eomega_datap)[0+(__inc_eomega_N*(i))] = logadd((eomega_datap)[0+(__inc_eomega_N*(i))] , gamma_it);
211             }
212              
213             /*-- Main: Iterate: T-1 > t >= 0 --*/
214 0 0         for (t--; t>=0; t--) {
215 0           o_tp1 = o_t;
216 0           o_t = (o_datap)[0+(__inc_o_T*(t))];
217              
218             /*-- Main: Iterate: state_t == i --*/
219 0 0         for (i=0; i<__privtrans->ind_sizes[1]; i++) {
220 0           gamma_it = (alpha_datap)[0+(__inc_alpha_N*(i))+(__inc_alpha_T*(t))] + (beta_datap)[0+(__inc_beta_N*(i))+(__inc_beta_T*(t))] - p_o;
221              
222             /*-- Main: Iterate: state_(t+1) == j --*/
223 0 0         for (j=0; j<__privtrans->ind_sizes[1]; j++) {
224 0           xi_ijt = (alpha_datap)[0+(__inc_alpha_N*(i))+(__inc_alpha_T*(t))] + (a_datap)[0+(__inc_a_N0*(i))+(__inc_a_N1*(j))] + (b_datap)[0+(__inc_b_N*(j))+(__inc_b_M*(o_tp1))] + (beta_datap)[0+(__inc_beta_N*(j))+(__inc_beta_T*(t+1))] - p_o;
225              
226 0           (ea_datap)[0+(__inc_ea_N0*(i))+(__inc_ea_N1*(j))] = logadd(xi_ijt, (ea_datap)[0+(__inc_ea_N0*(i))+(__inc_ea_N1*(j))]);
227             }
228              
229             /*-- Main: Update: pi --*/
230 0 0         if (t==0) (epi_datap)[0+(__inc_epi_N*(i))] = logadd(gamma_it, (epi_datap)[0+(__inc_epi_N*(i))]);
231              
232             /*-- Main: Update: b --*/
233 0           (eb_datap)[0+(__inc_eb_N*(i))+(__inc_eb_M*(o_t))] = logadd(gamma_it, (eb_datap)[0+(__inc_eb_N*(i))+(__inc_eb_M*(o_t))]);
234             }
235             }
236 0 0         }PDL_BROADCASTLOOP_END_hmmexpect_readdata
    0          
237 0           } break;
238 6           case PDL_D: {
239 6 50         PDL_DECLARE_PARAMS_hmmexpect_1(PDL_Double,D)
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
    50          
240 24 50         PDL_BROADCASTLOOP_START_hmmexpect_readdata {
    50          
    50          
    50          
    50          
    100          
    100          
241             int i,j,t;
242             int o_tp1, o_t;
243 6           double p_o = LOG_ZERO;
244             double gamma_it;
245             double xi_ijt;
246              
247             /*-- Initialize: t==(T-1): P(o|@theta) --*/
248 6           t = __privtrans->ind_sizes[2]-1;
249 18 100         {/* Open N */ PDL_EXPAND2(register PDL_Indx N=0, __N_stop=(__N_size)); for(; N<__N_stop; N+=1) { p_o = logadd(p_o, (omega_datap)[0+(__inc_omega_N*(N))] + (alpha_datap)[0+(__inc_alpha_N*(N))+(__inc_alpha_T*(t))]); }} /* Close N */
250              
251              
252             /*-- Initialize: t==(T-1): Iterate: state_t==i: get gamma(i,t) --*/
253 6           o_t = (o_datap)[0+(__inc_o_T*(t))];
254 18 100         for (i=0; i<__privtrans->ind_sizes[1]; i++) {
255 12           gamma_it = (alpha_datap)[0+(__inc_alpha_N*(i))+(__inc_alpha_T*(t))] + (beta_datap)[0+(__inc_beta_N*(i))+(__inc_beta_T*(t))] - p_o;
256 12           (eb_datap)[0+(__inc_eb_N*(i))+(__inc_eb_M*(o_t))] = logadd((eb_datap)[0+(__inc_eb_N*(i))+(__inc_eb_M*(o_t))], gamma_it);
257 12           (eomega_datap)[0+(__inc_eomega_N*(i))] = logadd((eomega_datap)[0+(__inc_eomega_N*(i))] , gamma_it);
258             }
259              
260             /*-- Main: Iterate: T-1 > t >= 0 --*/
261 12 100         for (t--; t>=0; t--) {
262 6           o_tp1 = o_t;
263 6           o_t = (o_datap)[0+(__inc_o_T*(t))];
264              
265             /*-- Main: Iterate: state_t == i --*/
266 18 100         for (i=0; i<__privtrans->ind_sizes[1]; i++) {
267 12           gamma_it = (alpha_datap)[0+(__inc_alpha_N*(i))+(__inc_alpha_T*(t))] + (beta_datap)[0+(__inc_beta_N*(i))+(__inc_beta_T*(t))] - p_o;
268              
269             /*-- Main: Iterate: state_(t+1) == j --*/
270 36 100         for (j=0; j<__privtrans->ind_sizes[1]; j++) {
271 24           xi_ijt = (alpha_datap)[0+(__inc_alpha_N*(i))+(__inc_alpha_T*(t))] + (a_datap)[0+(__inc_a_N0*(i))+(__inc_a_N1*(j))] + (b_datap)[0+(__inc_b_N*(j))+(__inc_b_M*(o_tp1))] + (beta_datap)[0+(__inc_beta_N*(j))+(__inc_beta_T*(t+1))] - p_o;
272              
273 24           (ea_datap)[0+(__inc_ea_N0*(i))+(__inc_ea_N1*(j))] = logadd(xi_ijt, (ea_datap)[0+(__inc_ea_N0*(i))+(__inc_ea_N1*(j))]);
274             }
275              
276             /*-- Main: Update: pi --*/
277 12 50         if (t==0) (epi_datap)[0+(__inc_epi_N*(i))] = logadd(gamma_it, (epi_datap)[0+(__inc_epi_N*(i))]);
278              
279             /*-- Main: Update: b --*/
280 12           (eb_datap)[0+(__inc_eb_N*(i))+(__inc_eb_M*(o_t))] = logadd(gamma_it, (eb_datap)[0+(__inc_eb_N*(i))+(__inc_eb_M*(o_t))]);
281             }
282             }
283 6 50         }PDL_BROADCASTLOOP_END_hmmexpect_readdata
    50          
284 6           } break;
285 0           case PDL_LD: {
286 0 0         PDL_DECLARE_PARAMS_hmmexpect_1(PDL_LDouble,E)
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
    0          
287 0 0         PDL_BROADCASTLOOP_START_hmmexpect_readdata {
    0          
    0          
    0          
    0          
    0          
    0          
288             int i,j,t;
289             int o_tp1, o_t;
290 0           double p_o = LOG_ZERO;
291             double gamma_it;
292             double xi_ijt;
293              
294             /*-- Initialize: t==(T-1): P(o|@theta) --*/
295 0           t = __privtrans->ind_sizes[2]-1;
296 0 0         {/* Open N */ PDL_EXPAND2(register PDL_Indx N=0, __N_stop=(__N_size)); for(; N<__N_stop; N+=1) { p_o = logadd(p_o, (omega_datap)[0+(__inc_omega_N*(N))] + (alpha_datap)[0+(__inc_alpha_N*(N))+(__inc_alpha_T*(t))]); }} /* Close N */
297              
298              
299             /*-- Initialize: t==(T-1): Iterate: state_t==i: get gamma(i,t) --*/
300 0           o_t = (o_datap)[0+(__inc_o_T*(t))];
301 0 0         for (i=0; i<__privtrans->ind_sizes[1]; i++) {
302 0           gamma_it = (alpha_datap)[0+(__inc_alpha_N*(i))+(__inc_alpha_T*(t))] + (beta_datap)[0+(__inc_beta_N*(i))+(__inc_beta_T*(t))] - p_o;
303 0           (eb_datap)[0+(__inc_eb_N*(i))+(__inc_eb_M*(o_t))] = logadd((eb_datap)[0+(__inc_eb_N*(i))+(__inc_eb_M*(o_t))], gamma_it);
304 0           (eomega_datap)[0+(__inc_eomega_N*(i))] = logadd((eomega_datap)[0+(__inc_eomega_N*(i))] , gamma_it);
305             }
306              
307             /*-- Main: Iterate: T-1 > t >= 0 --*/
308 0 0         for (t--; t>=0; t--) {
309 0           o_tp1 = o_t;
310 0           o_t = (o_datap)[0+(__inc_o_T*(t))];
311              
312             /*-- Main: Iterate: state_t == i --*/
313 0 0         for (i=0; i<__privtrans->ind_sizes[1]; i++) {
314 0           gamma_it = (alpha_datap)[0+(__inc_alpha_N*(i))+(__inc_alpha_T*(t))] + (beta_datap)[0+(__inc_beta_N*(i))+(__inc_beta_T*(t))] - p_o;
315              
316             /*-- Main: Iterate: state_(t+1) == j --*/
317 0 0         for (j=0; j<__privtrans->ind_sizes[1]; j++) {
318 0           xi_ijt = (alpha_datap)[0+(__inc_alpha_N*(i))+(__inc_alpha_T*(t))] + (a_datap)[0+(__inc_a_N0*(i))+(__inc_a_N1*(j))] + (b_datap)[0+(__inc_b_N*(j))+(__inc_b_M*(o_tp1))] + (beta_datap)[0+(__inc_beta_N*(j))+(__inc_beta_T*(t+1))] - p_o;
319              
320 0           (ea_datap)[0+(__inc_ea_N0*(i))+(__inc_ea_N1*(j))] = logadd(xi_ijt, (ea_datap)[0+(__inc_ea_N0*(i))+(__inc_ea_N1*(j))]);
321             }
322              
323             /*-- Main: Update: pi --*/
324 0 0         if (t==0) (epi_datap)[0+(__inc_epi_N*(i))] = logadd(gamma_it, (epi_datap)[0+(__inc_epi_N*(i))]);
325              
326             /*-- Main: Update: b --*/
327 0           (eb_datap)[0+(__inc_eb_N*(i))+(__inc_eb_M*(o_t))] = logadd(gamma_it, (eb_datap)[0+(__inc_eb_N*(i))+(__inc_eb_M*(o_t))]);
328             }
329             }
330 0 0         }PDL_BROADCASTLOOP_END_hmmexpect_readdata
    0          
331 0           } break;
332 0           default: return PDL->make_error(PDL_EUSERERROR, "PP INTERNAL ERROR in hmmexpect: unhandled datatype(%d), only handles (FDE)! PLEASE MAKE A BUG REPORT\n", __privtrans->__datatype);
333             }
334             #undef PDL_IF_BAD
335 6           return PDL_err;
336             }
337              
338             static pdl_datatypes pdl_hmmexpect_vtable_gentypes[] = { PDL_F, PDL_D, PDL_LD, -1 };
339             static PDL_Indx pdl_hmmexpect_vtable_realdims[] = { 2, 2, 1, 1, 1, 2, 2, 2, 2, 1, 1 };
340             static char *pdl_hmmexpect_vtable_parnames[] = { "a","b","pi","omega","o","alpha","beta","ea","eb","epi","eomega" };
341             static short pdl_hmmexpect_vtable_parflags[] = {
342             0,
343             0,
344             0,
345             0,
346             0,
347             0,
348             0,
349             PDL_PARAM_ISCREAT|PDL_PARAM_ISOUT|PDL_PARAM_ISWRITE,
350             PDL_PARAM_ISCREAT|PDL_PARAM_ISOUT|PDL_PARAM_ISWRITE,
351             PDL_PARAM_ISCREAT|PDL_PARAM_ISOUT|PDL_PARAM_ISWRITE,
352             PDL_PARAM_ISCREAT|PDL_PARAM_ISOUT|PDL_PARAM_ISWRITE
353             };
354             static pdl_datatypes pdl_hmmexpect_vtable_partypes[] = { -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1 };
355             static PDL_Indx pdl_hmmexpect_vtable_realdims_starts[] = { 0, 2, 4, 5, 6, 7, 9, 11, 13, 15, 16 };
356             static PDL_Indx pdl_hmmexpect_vtable_realdims_ind_ids[] = { 1, 1, 1, 0, 1, 1, 2, 1, 2, 1, 2, 1, 1, 1, 0, 1, 1 };
357             static char *pdl_hmmexpect_vtable_indnames[] = { "M","N","T" };
358             pdl_transvtable pdl_hmmexpect_vtable = {
359             PDL_TRANS_DO_BROADCAST, 0, pdl_hmmexpect_vtable_gentypes, 7, 11, NULL /*CORE21*/,
360             pdl_hmmexpect_vtable_realdims, pdl_hmmexpect_vtable_parnames,
361             pdl_hmmexpect_vtable_parflags, pdl_hmmexpect_vtable_partypes,
362             pdl_hmmexpect_vtable_realdims_starts, pdl_hmmexpect_vtable_realdims_ind_ids, 17,
363             3, pdl_hmmexpect_vtable_indnames,
364             NULL, pdl_hmmexpect_readdata, NULL,
365             NULL,
366             0,"PDL::HMM::hmmexpect"
367             };
368              
369              
370 6           pdl_error pdl_run_hmmexpect(pdl *a,pdl *b,pdl *pi,pdl *omega,pdl *o,pdl *alpha,pdl *beta,pdl *ea,pdl *eb,pdl *epi,pdl *eomega) {
371 6           pdl_error PDL_err = {0, NULL, 0};
372 6 50         if (!PDL) return (pdl_error){PDL_EFATAL, "PDL core struct is NULL, can't continue",0};
373 6           pdl_trans *__privtrans = PDL->create_trans(&pdl_hmmexpect_vtable);
374 6 50         if (!__privtrans) return PDL->make_error_simple(PDL_EFATAL, "Couldn't create trans");
375 6           __privtrans->pdls[0] = a;
376 6           __privtrans->pdls[1] = b;
377 6           __privtrans->pdls[2] = pi;
378 6           __privtrans->pdls[3] = omega;
379 6           __privtrans->pdls[4] = o;
380 6           __privtrans->pdls[5] = alpha;
381 6           __privtrans->pdls[6] = beta;
382 6           __privtrans->pdls[7] = ea;
383 6           __privtrans->pdls[8] = eb;
384 6           __privtrans->pdls[9] = epi;
385 6           __privtrans->pdls[10] = eomega;
386 6 50         PDL_RETERROR(PDL_err, PDL->type_coerce(__privtrans));
387 6 50         PDL_RETERROR(PDL_err, PDL->make_trans_mutual(__privtrans));
388 6           return PDL_err;
389             }