From b62a26ac20ffe86ba1836a1cea502d46fa83bd1a Mon Sep 17 00:00:00 2001
From: ferrari <maxence.ferrari@gmail.com>
Date: Fri, 5 Mar 2021 10:01:00 +0100
Subject: [PATCH] Fix reset
---
ipi_extract.py | 16 +++++++++++-----
1 file changed, 11 insertions(+), 5 deletions(-)
diff --git a/ipi_extract.py b/ipi_extract.py
index f403b16..87a6148 100644
--- a/ipi_extract.py
+++ b/ipi_extract.py
@@ -444,7 +444,10 @@ class Callback(object):
self.song = song
self.song_resample = song_resample
self._shit()
- self.sr = sr
+ 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)
--
GitLab