fix unit presence time_range (#4571)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: tayheau

src/spikeinterface/widgets/unit_presence.py

@@ -73,6 +73,7 @@ def plot_matplotlib(self, data_plot, **backend_kwargs):
         self.figure, self.axes, self.ax = make_mpl_figure(**backend_kwargs)
 
         sorting = dp.sorting
+        unit_ids = sorting.unit_ids
 
         spikes = sorting.to_spike_vector(concatenated=False, use_cache=True)
         spikes = spikes[dp.segment_index]
@@ -85,18 +86,20 @@ def plot_matplotlib(self, data_plot, **backend_kwargs):
             ind1 = int(t1 * fs)
             mask = (spikes["sample_index"] >= ind0) & (spikes["sample_index"] <= ind1)
             spikes = spikes[mask]
+            duration = t1 - t0
+        else:
+            last = spikes["sample_index"][-1]
+            duration = last / fs
 
         if spikes.size == 0:
             return
 
-        last = spikes["sample_index"][-1]
-        max_time = last / fs
-
-        num_units = len(sorting.unit_ids)
-        num_time_bins = int(max_time / dp.bin_duration_s) + 1
+        num_units = len(unit_ids)
+        num_time_bins = int(duration / dp.bin_duration_s) + 1
         map = np.zeros((num_units, num_time_bins))
         ind0 = spikes["unit_index"]
-        ind1 = spikes["sample_index"] // int(dp.bin_duration_s * fs)
+        offset = int(t0 * fs) if dp.time_range is not None else 0
+        ind1 = (spikes["sample_index"] - offset) // int(dp.bin_duration_s * fs)
         map[ind0, ind1] += 1
 
         if dp.smooth_sigma is not None:
@@ -109,8 +112,13 @@ def plot_matplotlib(self, data_plot, **backend_kwargs):
             smooth_kernel = smooth_kernel[np.newaxis, :]
             map = scipy.signal.oaconvolve(map, smooth_kernel, mode="same", axes=1)
 
-        im = self.ax.matshow(map, cmap="inferno", aspect="auto")
+        extent = (dp.time_range[0], dp.time_range[1], len(unit_ids), 0) if dp.time_range is not None else None
+        im = self.ax.imshow(
+            map, cmap="inferno", aspect="auto", interpolation="nearest", extent=extent, vmax=1.0, vmin=0.0
+        )
         self.ax.set_xlabel("Time (s)")
         self.ax.set_ylabel("Units")
+        self.ax.set_yticks(np.arange(len(unit_ids)) + 0.5)
+        self.ax.set_yticklabels(unit_ids)
 
         self.figure.colorbar(im)