Recompute retrieve time only if xsize increased

Author: alejoe91

spikeinterface_gui/tracemapview.py

@@ -40,6 +40,17 @@ def __init__(self, controller=None, parent=None, backend="qt"):
 
         self.make_color_lut()
 
+    def apply_gain_zoom(self, factor_ratio):
+        if self.color_limit is None:
+            return
+        self.color_limit = self.color_limit * factor_ratio
+        self.refresh()
+
+    def auto_scale(self):
+        if self.last_data_curves is not None:
+            self.color_limit = np.max(np.abs(self.last_data_curves))
+        self.refresh()
+
     def make_color_lut(self):
         N = 512
         cmap_name = self.settings['colormap']
@@ -55,17 +66,6 @@ def make_color_lut(self):
     def get_visible_channel_inds(self):
         return self.channel_order
 
-    def apply_gain_zoom(self, factor_ratio):
-        if self.color_limit is None:
-            return
-        self.color_limit = self.color_limit * factor_ratio
-        self.refresh()
-
-    def auto_scale(self):
-        if self.last_data_curves is not None:
-            self.color_limit = np.max(np.abs(self.last_data_curves))
-        self.refresh()
-
     ## Qt ##
     def _qt_make_layout(self, **kargs):
         from .myqt import QT

spikeinterface_gui/traceview.py

@@ -213,7 +213,8 @@ def _qt_on_combo_seg_changed(self):
     def _qt_on_xsize_changed(self):
         xsize = self.spinbox_xsize.value()
         # Reset trace retrieval check: might require more or less time now!
-        # self._retrieve_traces_time_checked = False
+        if xsize > self.xsize:
+            self._retrieve_traces_time_checked = False
         self.xsize = xsize
         if not self._block_auto_refresh_and_notify:
             self.refresh()
@@ -312,9 +313,10 @@ def _panel_change_segment(self, seg_index):
             self.notify_time_info_updated()
 
     def _panel_on_xsize_changed(self, event):
-        self.xsize = event.new
         # Reset trace retrieval check: might require more or less time now!
-        # self._retrieve_traces_time_checked = False
+        if event.new > self.xsize:
+            self._retrieve_traces_time_checked = False
+        self.xsize = event.new
         if not self._block_auto_refresh_and_notify:
             self.refresh()
             self.notify_time_info_updated()