diff --git a/ipi_extract.py b/ipi_extract.py index 27d5cb41092949d0bce290810e85ec0ed2f03bc3..e5f30b2f533e741880b22d54850d57a8809bd626 100644 --- a/ipi_extract.py +++ b/ipi_extract.py @@ -310,7 +310,7 @@ class Callback(object): self.curr_ind[self.curr] = len(self.offset) - 1 self.curr_vert[self.curr] = 0 self.cursor[self.curr].linev[0].set_linestyle('--') - self.cursor[self.curr].linev[1].set_linestyle('--') + self.cursor[self.curr].linev[self.cursor[self.curr].num_cur].set_linestyle('--') for i in range(4): if i < 2: ax_group[i].set_visible(False) @@ -470,15 +470,15 @@ class Callback(object): plt.draw() def increase_freq(self, event): - lim = self.view_data[0].specgram.get_ylim()[1] + 1e3 + lim = self.view_data[0].spectrogram.axes.get_ylim()[1] + 1e3 for ax_group in self.view_data: - ax_group.specgram.set_ylim(0, lim) + ax_group.spectrogram.axes.set_ylim(0, lim) plt.draw() def decrease_freq(self, event): - lim = max(self.view_data[0].specgram.get_ylim()[1] + 1e3, 1e3) + lim = max(self.view_data[0].spectrogram.axes.get_ylim()[1] - 1e3, 1e3) for ax_group in self.view_data: - ax_group.specgram.set_ylim(0, lim) + ax_group.spectrogram.axes.set_ylim(0, lim) plt.draw() def _update_spectrogram(self, ax_group): @@ -612,6 +612,7 @@ class AxesGroup: tuple(ax_spec.axvline(0.01, c='k') for _ in range(num_cur))) self.spectrogram.im.set_clim(2000, 2100) ax_spec.set_ylim(0, min(20e3, sr/2)) + ax_spec.set_yticks(ax_spec.get_yticks()) # Needed, otherwise updating ylim doesn't update ticks properly ax_spec.set_yticklabels((ax_spec.get_yticks() / 1e3).astype(int)) self.correlation = AxesWithCursor(ax_corr, ax_corr.plot(np.linspace(0, 10, int(10e-3 * sr), False),