Fixes resample_ann freezing MIT-LCP#116

Lucas-Mc · Lucas-Mc · commit 510676bfda43 · 2020-05-07T11:43:34.000-04:00
Fixes the freezing seen in resample_ann when the fs_target is set too high. This causes the algorithm to develop an infinite loop which must be exited upon discovery. This pull request breaks the loop if this situation is detected. Fixes MIT-LCP#162.
diff --git a/wfdb/processing/basic.py b/wfdb/processing/basic.py
@@ -1,5 +1,6 @@
 import numpy as np
 from scipy import signal
+import pdb
 
 from ..io.annotation import Annotation
 
@@ -23,8 +24,10 @@ def resample_ann(resampled_t, ann_sample):
     """
     tmp = np.zeros(len(resampled_t), dtype='int16')
     j = 0
+    break_loop = 0
     tprec = resampled_t[j]
     for i, v in enumerate(ann_sample):
+        break_loop = 0
         while True:
             d = False
             if v < tprec:
@@ -44,6 +47,10 @@ def resample_ann(resampled_t, ann_sample):
                 d = True
             j += 1
             tprec = tnow
+            break_loop += 1
+            if (break_loop > 1000):
+                tmp[j] += 1
+                break
             if d:
                 break
 
