From b09561705ab9c67c93a384248f7c3bc9ad5bdd32 Mon Sep 17 00:00:00 2001
From: Mario Mulansky <mario.mulansky@gmx.net>
Date: Wed, 3 Feb 2016 13:07:26 +0100
Subject: fixed spike-sync bug

fixed ugly bugs in code for computing multi-variate spike sync profile and
multi-variate spike sync value.
---
 pyspike/cython/cython_add.pyx       | 33 +++++++++++++++++----------------
 pyspike/cython/cython_distances.pyx |  5 -----
 pyspike/cython/python_backend.py    | 33 ++++++++++++++++++---------------
 3 files changed, 35 insertions(+), 36 deletions(-)

(limited to 'pyspike/cython')

diff --git a/pyspike/cython/cython_add.pyx b/pyspike/cython/cython_add.pyx
index 8da1e53..25f1181 100644
--- a/pyspike/cython/cython_add.pyx
+++ b/pyspike/cython/cython_add.pyx
@@ -182,8 +182,8 @@ def add_discrete_function_cython(double[:] x1, double[:] y1, double[:] mp1,
     cdef int index1 = 0
     cdef int index2 = 0
     cdef int index = 0
-    cdef int N1 = len(y1)
-    cdef int N2 = len(y2)
+    cdef int N1 = len(y1)-1
+    cdef int N2 = len(y2)-1
     x_new[0] = x1[0]
     while (index1+1 < N1) and (index2+1 < N2):
         if x1[index1+1] < x2[index2+1]:
@@ -206,20 +206,21 @@ def add_discrete_function_cython(double[:] x1, double[:] y1, double[:] mp1,
             y_new[index] = y1[index1] + y2[index2]
             mp_new[index] = mp1[index1] + mp2[index2]
     # one array reached the end -> copy the contents of the other to the end
-    if index1+1 < len(y1):
-        x_new[index+1:index+1+len(x1)-index1-1] = x1[index1+1:]
-        y_new[index+1:index+1+len(x1)-index1-1] = y1[index1+1:]
-        mp_new[index+1:index+1+len(x1)-index1-1] = mp1[index1+1:]
-        index += len(x1)-index1-1
-    elif index2+1 < len(y2):
-        x_new[index+1:index+1+len(x2)-index2-1] = x2[index2+1:]
-        y_new[index+1:index+1+len(x2)-index2-1] = y2[index2+1:]
-        mp_new[index+1:index+1+len(x2)-index2-1] = mp2[index2+1:]
-        index += len(x2)-index2-1
-    # else:  # both arrays reached the end simultaneously
-    #     x_new[index+1] = x1[-1]
-    #     y_new[index+1] = y1[-1] + y2[-1]
-    #     mp_new[index+1] = mp1[-1] + mp2[-1]
+    if index1+1 < N1:
+        x_new[index+1:index+1+N1-index1] = x1[index1+1:]
+        y_new[index+1:index+1+N1-index1] = y1[index1+1:]
+        mp_new[index+1:index+1+N1-index1] = mp1[index1+1:]
+        index += N1-index1
+    elif index2+1 < N2:
+        x_new[index+1:index+1+N2-index2] = x2[index2+1:]
+        y_new[index+1:index+1+N2-index2] = y2[index2+1:]
+        mp_new[index+1:index+1+N2-index2] = mp2[index2+1:]
+        index += N2-index2
+    else:  # both arrays reached the end simultaneously
+        x_new[index+1] = x1[index1+1]
+        y_new[index+1] = y1[index1+1] + y2[index2+1]
+        mp_new[index+1] = mp1[index1+1] + mp2[index2+1]
+        index += 1
 
     y_new[0] = y_new[1]
     mp_new[0] = mp_new[1]
diff --git a/pyspike/cython/cython_distances.pyx b/pyspike/cython/cython_distances.pyx
index 7dc9cb9..ac5f226 100644
--- a/pyspike/cython/cython_distances.pyx
+++ b/pyspike/cython/cython_distances.pyx
@@ -428,9 +428,4 @@ def coincidence_value_cython(double[:] spikes1, double[:] spikes2,
             mp += 2
             coinc += 2
 
-    if coinc == 0 and mp == 0:
-        # empty spike trains -> spike sync = 1 by definition
-        coinc = 1
-        mp = 1
-
     return coinc, mp
diff --git a/pyspike/cython/python_backend.py b/pyspike/cython/python_backend.py
index a1e3a65..6b7209a 100644
--- a/pyspike/cython/python_backend.py
+++ b/pyspike/cython/python_backend.py
@@ -560,7 +560,9 @@ def add_discrete_function_python(x1, y1, mp1, x2, y2, mp2):
     index1 = 0
     index2 = 0
     index = 0
-    while (index1+1 < len(y1)) and (index2+1 < len(y2)):
+    N1 = len(x1)-1
+    N2 = len(x2)-1
+    while (index1+1 < N1) and (index2+1 < N2):
         if x1[index1+1] < x2[index2+1]:
             index1 += 1
             index += 1
@@ -581,20 +583,21 @@ def add_discrete_function_python(x1, y1, mp1, x2, y2, mp2):
             y_new[index] = y1[index1] + y2[index2]
             mp_new[index] = mp1[index1] + mp2[index2]
     # one array reached the end -> copy the contents of the other to the end
-    if index1+1 < len(y1):
-        x_new[index+1:index+1+len(x1)-index1-1] = x1[index1+1:]
-        y_new[index+1:index+1+len(x1)-index1-1] = y1[index1+1:]
-        mp_new[index+1:index+1+len(x1)-index1-1] = mp1[index1+1:]
-        index += len(x1)-index1-1
-    elif index2+1 < len(y2):
-        x_new[index+1:index+1+len(x2)-index2-1] = x2[index2+1:]
-        y_new[index+1:index+1+len(x2)-index2-1] = y2[index2+1:]
-        mp_new[index+1:index+1+len(x2)-index2-1] = mp2[index2+1:]
-        index += len(x2)-index2-1
-    # else:  # both arrays reached the end simultaneously
-    #     x_new[index+1] = x1[-1]
-    #     y_new[index+1] = y1[-1] + y2[-1]
-    #     mp_new[index+1] = mp1[-1] + mp2[-1]
+    if index1+1 < N1:
+        x_new[index+1:index+1+N1-index1] = x1[index1+1:]
+        y_new[index+1:index+1+N1-index1] = y1[index1+1:]
+        mp_new[index+1:index+1+N1-index1] = mp1[index1+1:]
+        index += N1-index1
+    elif index2+1 < N2:
+        x_new[index+1:index+1+N2-index2] = x2[index2+1:]
+        y_new[index+1:index+1+N2-index2] = y2[index2+1:]
+        mp_new[index+1:index+1+N2-index2] = mp2[index2+1:]
+        index += N2-index2
+    else:  # both arrays reached the end simultaneously
+        x_new[index+1] = x1[-1]
+        y_new[index+1] = y1[-1] + y2[-1]
+        mp_new[index+1] = mp1[-1] + mp2[-1]
+        index += 1
 
     y_new[0] = y_new[1]
     mp_new[0] = mp_new[1]
-- 
cgit v1.2.3