Fix gaps

Author: alejoe91

spikeinterface_gui/tests/test_mainwindow_panel.py

@@ -61,6 +61,18 @@ def test_mainwindow(start_app=False, verbose=True, curation=False, only_some_ext
     )
     win = None
 
+    for segment_index in range(analyzer.get_num_segments()):
+        shift = (segment_index + 1) * 100
+        # add a gap to times
+        gap = 5
+        times = analyzer.recording.get_times(segment_index)
+        times = times + shift
+        times[len(times)//2:] += gap  # add a gap in the middle
+        analyzer.recording.set_times(
+            times,
+            segment_index=segment_index
+        )
+
     win = run_mainwindow(
         analyzer,
         mode="web",

spikeinterface_gui/tests/test_mainwindow_qt.py

@@ -70,8 +70,13 @@ def test_mainwindow(start_app=False, verbose=True, curation=False, only_some_ext
 
     for segment_index in range(analyzer.get_num_segments()):
         shift = (segment_index + 1) * 100
+        # add a gap to times
+        gap = 1
+        times = analyzer.recording.get_times(segment_index)
+        times = times + shift
+        times[len(times)//2:] += gap  # add a gap in the middle
         analyzer.recording.set_times(
-            analyzer.recording.get_times(segment_index) + shift,
+            times,
             segment_index=segment_index
         )
 

spikeinterface_gui/tracemapview.py

@@ -80,7 +80,11 @@ def get_data_in_chunk(self, t1, t2, segment_index):
             data_curves = data_curves.astype("float32")
 
         t_start, _ = self.controller.get_t_start_t_stop()
-        times_chunk = np.arange(traces_chunk.shape[0], dtype='float64') / self.controller.sampling_frequency + max(t1, t_start)
+        if self.controller.main_settings["use_times"]:
+            recording = self.controller.analyzer.recording
+            times_chunk = recording.get_times(segment_index=segment_index)[ind1:ind2]
+        else:
+            times_chunk = np.arange(traces_chunk.shape[0], dtype='float64') / self.controller.sampling_frequency + max(t1, t_start)
 
         scatter_x = []
         scatter_y = []
@@ -197,6 +201,11 @@ def _qt_seek(self, t):
         segment_index = self.controller.get_time()[1]
         times_chunk, data_curves, scatter_x, scatter_y, scatter_colors, scatter_unit_ids = \
             self.get_data_in_chunk(t1, t2, segment_index)
+
+        if times_chunk.size == 0:
+            self.image.hide()
+            self.scatter.clear()
+            return
 
         if self.color_limit is None:
             self.color_limit = np.max(np.abs(data_curves))

spikeinterface_gui/traceview.py

@@ -321,6 +321,9 @@ def apply_gain_zoom(self, factor_ratio):
 
     def get_data_in_chunk(self, t1, t2, segment_index):
         ind1, ind2 = self.get_chunk_indices(t1, t2, segment_index)
+        # handle blank spots
+        if ind1 == ind2:
+            return np.array([]), np.array([[]]), [], [], [], []
 
         traces_chunk = self.controller.get_traces(segment_index=segment_index, start_frame=ind1, end_frame=ind2)
 
@@ -345,7 +348,11 @@ def get_data_in_chunk(self, t1, t2, segment_index):
         data_curves += offsets[:, None]
 
         t_start, _ = self.controller.get_t_start_t_stop()
-        times_chunk = np.arange(traces_chunk.shape[0], dtype='float64') / self.controller.sampling_frequency + max(t1, t_start)
+        if self.controller.main_settings["use_times"]:
+            recording = self.controller.analyzer.recording
+            times_chunk = recording.get_times(segment_index=segment_index)[ind1:ind2]
+        else:
+            times_chunk = np.arange(traces_chunk.shape[0], dtype='float64') / self.controller.sampling_frequency + max(t1, t_start)
 
         scatter_x = []
         scatter_y = []
@@ -355,7 +362,6 @@ def get_data_in_chunk(self, t1, t2, segment_index):
         global_to_local_chan_inds = np.zeros(self.controller.channel_ids.size, dtype='int64')
         global_to_local_chan_inds[visible_channel_inds] = np.arange(visible_channel_inds.size, dtype='int64')
 
-
         for unit_index, unit_id in self.controller.iter_visible_units():
 
             inds = np.flatnonzero(spikes_chunk["unit_index"] == unit_index)
@@ -482,6 +488,12 @@ def _qt_seek(self, t):
 
         times_chunk, data_curves, scatter_x, scatter_y, scatter_colors, scatter_unit_ids = \
             self.get_data_in_chunk(t1, t2,  self.controller.get_time()[1])
+
+        if times_chunk.size == 0:
+            self.signals_curve.setData([], [])
+            self.scatter.setData(x=[], y=[], brush=[])
+            return
+
         connect = np.ones(data_curves.shape, dtype='bool')
         connect[:, -1] = 0