Fix reset

ipi_extract.py

@@ -444,7 +444,10 @@ class Callback(object):
         self.song = song
         self.song_resample = song_resample
         self._shit()
+        sr_update = False
+        if self.sr != sr:
             self.sr = sr
+            sr_update = True
         self.curr = 0  # current view selected
         self.offset = np.zeros((0, 2))
         self.scat.set_offsets(self.offset)
@@ -453,8 +456,12 @@ class Callback(object):
         self.curr_vert = 3 * [0]  # Current vertical line of sig/spec for each  plot
         for i in range(3):
             self.view_data[i][0][0].set_ydata(np.zeros(int(20e-3 * sr)))
-            self.view_data[i][2][0].set_ydata(np.zeros(int(20e-3 * sr)))
-            self.view_data[i][3][0].set_ydata(np.zeros(int(20e-3 * sr)))
+            self.view_data[i][2][0].set_ydata(np.zeros(int(10e-3 * sr)))
+            self.view_data[i][3][0].set_ydata(np.zeros(int(10e-3 * sr)))
+            if sr_update:
+                self.view_data[i][0][0].set_xdata(np.linspace(0, 20, int(20e-3*sr), False))
+                self.view_data[i][2][0].set_xdata(np.linspace(0, 10, int(10e-3*sr), False))
+                self.view_data[i][3][0].set_xdata(np.linspace(0, 10, int(10e-3*sr), False))
             self.view_data[i][1][0].set_clim(2000,2100)
         for i in range(3):
             self._set_label(i, EMLN)
@@ -536,8 +543,7 @@ def init(in_path, channel, low=2e3, high=20e3):
     callback.cursor = m_cursor
     for i in range(3):
         data_view[i][0][1] = (ax_view[i][0].axvline(10, c='k', linestyle='--'), ax_view[i][0].axvline(10, c='k'))
-        data_view[i][0][0] = ax_view[i][0].plot(np.linspace(0, 20, int(20e-3 * sr), False), np.zeros(int(20e-3 * sr)))[
-            0]
+        data_view[i][0][0] = ax_view[i][0].plot(np.linspace(0, 20, int(20e-3*sr), False), np.zeros(int(20e-3*sr)))[0]
         ax_view[i][0].set_xlim(0, 20)
         ax_view[i][0].set_xlabel('IPI man:None')
         ax_view[i][0].set_ylim(-1, 1)