Skip to content

Update channel visibility when unit is changed in unitview #259

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
alejoe91 merged 5 commits into from
Apr 1, 2026
+61 −40
Merged

Update channel visibility when unit is changed in unitview #259

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
7c734ce
update channel visibility on unit change from unitview
chrishalcrow Mar 30, 2026
0ebdd27
Revert "update channel visibility on unit change from unitview"
chrishalcrow Mar 30, 2026
c9d9033
update channel visibility on unitview change
chrishalcrow Mar 30, 2026
92c1958
Notify unit changes only when actual change
alejoe91 Apr 1, 2026
02d4a33
Fix panel bug in events
alejoe91 Apr 1, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 9 additions & 11 deletions spikeinterface_gui/traceview.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -394,13 +394,11 @@ def _panel_create_bottom_toolbar(self):

bottom_bar_items = [self.time_slider]
if self.controller.has_extension("events"):
self.event_line = None
if self.controller.has_extension("events"):
self.event_source = ColumnDataSource({"xs": [], "ys": []})
self.event_line = self.figure.multi_line(
source=self.event_source,
xs="xs", ys="ys", line_color="yellow", line_dash="dashed", line_width=2, line_alpha=0.8
)
self.event_source = ColumnDataSource({"xs": [], "ys": []})
self.event_line = self.figure.multi_line(
source=self.event_source,
xs="xs", ys="ys", line_color="yellow", line_dash="dashed", line_width=2, line_alpha=0.8
)
event_keys = list(self.controller.events.keys())
if len(event_keys) > 1:
self.event_selector = pn.widgets.Select(
Expand Down Expand Up @@ -582,9 +580,6 @@ def _panel_add_event_line(self):
yspan = [fig.y_range.start, fig.y_range.end]
self.event_source.data = {"xs": [[evt_time, evt_time]], "ys": [yspan]}

def _panel_remove_event_line(self):
self.event_source.data = {"xs": [], "ys": []}
Comment on lines -585 to -586
Copy link
Copy Markdown
Member

@alejoe91 alejoe91 Apr 1, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Already handled in the _panel_add_event_lines


# TODO: pan behavior like Qt?
# def _panel_on_pan_start(self, event):
# self.drag_state["x_start"] = event.x
Expand Down Expand Up @@ -828,6 +823,10 @@ def _panel_make_layout(self):
x="x", y="y", size=10, fill_color="color", fill_alpha=self.settings['alpha'], source=self.spike_source
)

# Placeholder for events
self.event_line = None
self.event_source = None

self.figure.on_event(DoubleTap, self._panel_on_double_tap)

self._panel_create_toolbars()
Expand All @@ -842,7 +841,6 @@ def _panel_make_layout(self):


def _panel_refresh(self):
self._panel_remove_event_line()
t, segment_index = self.controller.get_time()
xsize = self.xsize
t1, t2 = t - xsize / 3.0, t + xsize * 2 / 3.0
Expand Down
81 changes: 52 additions & 29 deletions spikeinterface_gui/unitlistview.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,13 @@ def update_manual_labels(self):
elif self.backend == 'panel':
self._panel_update_labels()

def notify_unit_visibility_changed(self):
selected_units = self.controller.get_visible_unit_ids()
visible_channel_inds = self.controller.get_common_sparse_channels(selected_units)
self.controller.set_channel_visibility(visible_channel_inds)
self.notify_channel_visibility_changed()
super().notify_unit_visibility_changed()

## Qt ##
def _qt_make_layout(self):

Expand Down Expand Up @@ -326,8 +333,11 @@ def _qt_on_item_changed(self, item):
is_visible = item.checkState() == QT.Qt.Checked
# visibility checkbox
unit_id = item.unit_id
current_visible_units = self.controller.get_visible_unit_ids()
self.controller.set_unit_visibility(unit_id, is_visible)
self.notify_unit_visibility_changed()
updated_visibile_units = self.controller.get_visible_unit_ids()
if set(current_visible_units) != set(updated_visibile_units):
self.notify_unit_visibility_changed()


elif col in self.label_columns:
Expand All @@ -342,12 +352,12 @@ def _qt_on_item_changed(self, item):

def _qt_on_double_clicked(self, row, col):
unit_id = self.table.item(row, 1).unit_id
current_visible_units = self.controller.get_visible_unit_ids()
self.controller.set_visible_unit_ids([unit_id])
# self.refresh()


self.notify_unit_visibility_changed()
self._qt_refresh_visibility_items()
updated_visibile_units = self.controller.get_visible_unit_ids()
if set(current_visible_units) != set(updated_visibile_units):
self.notify_unit_visibility_changed()
self._qt_refresh_visibility_items()

def _qt_on_open_context_menu(self):
self.menu.popup(self.qt_widget.cursor().pos())
Expand All @@ -369,38 +379,45 @@ def _qt_get_selected_unit_ids(self):
def _qt_on_visible_shortcut(self):
rows = self._qt_get_selected_rows()

current_visible_units = self.controller.get_visible_unit_ids()
self.controller.set_visible_unit_ids(self.get_selected_unit_ids())
# self.refresh()
self.notify_unit_visibility_changed()
self._qt_refresh_visibility_items()

for row in rows:
self.table.selectRow(row)
updated_visibile_units = self.controller.get_visible_unit_ids()
if set(current_visible_units) != set(updated_visibile_units):
self.notify_unit_visibility_changed()
self._qt_refresh_visibility_items()

for row in rows:
self.table.selectRow(row)

def _qt_on_only_previous_shortcut(self):
sel_rows = self._qt_get_selected_rows()
if len(sel_rows) == 0:
sel_rows = [self.table.rowCount()]
new_row = max(sel_rows[0] - 1, 0)
unit_id = self.table.item(new_row, 1).unit_id
current_visible_units = self.controller.get_visible_unit_ids()
self.controller.set_visible_unit_ids([unit_id])
self.notify_unit_visibility_changed()
self._qt_refresh_visibility_items()

self.table.clearSelection()
self.table.selectRow(new_row)
updated_visibile_units = self.controller.get_visible_unit_ids()
if set(current_visible_units) != set(updated_visibile_units):
self.notify_unit_visibility_changed()
self._qt_refresh_visibility_items()
self.table.clearSelection()
self.table.selectRow(new_row)

def _qt_on_only_next_shortcut(self):
sel_rows = self._qt_get_selected_rows()
if len(sel_rows) == 0:
sel_rows = [-1]
new_row = min(sel_rows[-1] + 1, self.table.rowCount() - 1)
unit_id = self.table.item(new_row, 1).unit_id
current_visible_units = self.controller.get_visible_unit_ids()
self.controller.set_visible_unit_ids([unit_id])
self.notify_unit_visibility_changed()
self._qt_refresh_visibility_items()
self.table.clearSelection()
self.table.selectRow(new_row)
updated_visibile_units = self.controller.get_visible_unit_ids()
if set(current_visible_units) != set(updated_visibile_units):
self.notify_unit_visibility_changed()
self._qt_refresh_visibility_items()
self.table.clearSelection()
self.table.selectRow(new_row)

def _qt_on_delete_shortcut(self):
sel_rows = self._qt_get_selected_rows()
Expand Down Expand Up @@ -728,13 +745,16 @@ def _panel_update_labels(self):
def _panel_on_only_selection(self):
selected_unit = self.table.selection[0]
unit_id = self.table.value.index.values[selected_unit]
current_visible_units = self.controller.get_visible_unit_ids()
self.controller.set_visible_unit_ids([unit_id])
self._panel_refresh_colors()
# update the visible column
df = self.table.value
df.loc[self.controller.unit_ids, "visible"] = self.controller.get_units_visibility_mask()
self.table.value = df
self.notify_unit_visibility_changed()
updated_visibile_units = self.controller.get_visible_unit_ids()
if set(current_visible_units) != set(updated_visibile_units):
self._panel_refresh_colors()
# update the visible column
df = self.table.value
df.loc[self.controller.unit_ids, "visible"] = self.controller.get_units_visibility_mask()
self.table.value = df
self.notify_unit_visibility_changed()

def _panel_get_selected_unit_ids(self):
unit_ids = self.table.value.index.values
Expand Down Expand Up @@ -785,9 +805,12 @@ def _panel_handle_shortcut(self, event):
if self.controller.curation:
self._panel_merge_units()
elif event.data == "visible":
current_visibile_units = self.controller.get_visible_unit_ids()
self.controller.set_visible_unit_ids(selected_unit_ids)
self.notify_unit_visibility_changed()
self.refresh()
updated_visibile_units = self.controller.get_visible_unit_ids()
if set(current_visibile_units) != set(updated_visibile_units):
self.notify_unit_visibility_changed()
self.refresh()
elif event.data == "clear":
for unit_id in selected_unit_ids:
self.controller.set_label_to_unit(unit_id, "quality", None)
Expand Down
Loading