diff --git a/ipi_extract.py b/ipi_extract.py index f403b164ec532c01f7e83f9021c18402441389dc..87a61482c95cfa48a3063af82790e45384a626d0 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)