Deduce length of sequence from observation array in PyViterbi.

Alexandre Marquet · Alexandre Marquet · commit d1fb0f526f5b · 2019-07-10T12:49:41.000+02:00
diff --git a/PyTurbo.pyx b/PyTurbo.pyx
@@ -46,7 +46,8 @@ cdef class PyViterbi:
     def __dealloc__(self):
         del self.cpp_viterbi
 
-    def viterbi_algorithm(self, K, S0, SK, float[::1] _in):
+    def viterbi_algorithm(self, S0, SK, float[::1] _in):
+        cdef int K = _in.shape[0]/self.O
         cdef unsigned char[::1] _out = numpy.zeros(K, dtype=numpy.uint8)
 
         self.cpp_viterbi.viterbi_algorithm(K, S0, SK, &_in[0], &_out[0])
diff --git a/examples/75_cc.py b/examples/75_cc.py
@@ -67,7 +67,7 @@ def branch_metrics(bits_rcvd):
     bm[O*i:O*(i+1)] = branch_metrics([r[2*i], r[2*i+1]])
 
 #decode message
-m_hat = dec.viterbi_algorithm(K_m, -1, -1, bm);
+m_hat = dec.viterbi_algorithm(-1, -1, bm);
 
 print(numpy.array(m, dtype=numpy.int))
 print('...')
