add widget event handler

Author: JaerongA

element_array_ephys/plotting/unit_level.py

@@ -65,6 +65,7 @@ def plot_correlogram(
         template="simple_white",
         width=350,
         height=350,
+        yaxis_range=[0, None]
     )
     return fig
 

element_array_ephys/plotting/widget.py

@@ -1,30 +1,70 @@
-import ipywidgets
+import ipywidgets as widgets
 from .. import ephys_report
 import matplotlib.image as mpimg
 import matplotlib.pyplot as plt
 import pathlib
 import typing as T
 from IPython.display import display
+from .. import ephys_no_curation as ephys
+from skimage import io
 
 
-def _drop_down_wg(
-    dj_table, *, description="", width=100, description_width=100
-) -> ipywidgets.Dropdown:
-    return ipywidgets.Dropdown(
-        options=dj_table.fetch("KEY", as_dict=True),
-        description=description,
-        disabled=False,
-        layout=ipywidgets.Layout(
-            width=f"{width}%",
-            display="flex",
-            flex_flow="row",
-            justify_content="space-between",
-            grid_area="processed_dropdown",
-        ),
-        style={"description_width": f"{description_width}px"},
+# Widgets
+probe_dropdown_wg = widgets.Dropdown(
+    options=ephys.CuratedClustering & ephys_report.ProbeLevelReport.fetch("KEY"),
+    description="Select Probe Insertion : ",
+    disabled=False,
+    layout=widgets.Layout(
+        width="80%",
+    ),
+    style={"description_width": "150px"},
+)
+
+shank_dropdown_wg = widgets.Dropdown(
+    options=(ephys_report.ProbeLevelReport & probe_dropdown_wg.value).fetch("shank"),
+    description="Select Shank : ",
+    disabled=False,
+    layout=widgets.Layout(
+        width="15%",
+    ),
+    style={"description_width": "100px"},
+)
+
+unit_dropdown_wg = widgets.Dropdown(
+    options=(
+        (ephys_report.UnitLevelReport & probe_dropdown_wg.value)
+        & "cluster_quality_label='good'"
+    ).fetch("unit"),
+    description="Select Units : ",
+    disabled=False,
+    layout=widgets.Layout(
+        width="15%",
+    ),
+    style={"description_width": "100px"},
+)
+
+# Update shank & unit dropdown according to probe change
+def update_shank_options(change):
+    probe_key = change.new
+    shank_dropdown_wg.options = (ephys_report.ProbeLevelReport & probe_key.value).fetch(
+        "shank"
     )
 
 
+def update_unit_dropdown(change):
+    probe_key = change.new
+    unit_dropdown_wg.options = (
+        (
+            (ephys_report.UnitLevelReport & probe_key.value)
+            & "cluster_quality_label='good'"
+        ).fetch("unit"),
+    )
+
+
+probe_dropdown_wg.observe(update_shank_options, "value")
+probe_dropdown_wg.observe(update_unit_dropdown, "value")
+
+
 def plot_probe_level_report(probe_key: T.Dict[str, T.Any]) -> None:
     fig_name = (ephys_report.ProbeLevelReport & probe_key).fetch1("drift_map_plot")
     fig, ax = plt.subplots(1, 1, figsize=(10, 5))
@@ -35,16 +75,6 @@ def plot_probe_level_report(probe_key: T.Dict[str, T.Any]) -> None:
     pathlib.Path(fig_name).unlink()
 
 
-def probe_widget() -> ipywidgets.interactive:
-    drop_down_wg = _drop_down_wg(
-        ephys_report.ProbeLevelReport,
-        description="Select Probe Insertion : ",
-        width=85,
-        description_width=150,
-    )
-    return ipywidgets.interactive(plot_probe_level_report, probe_key=drop_down_wg)
-
-
 def plot_unit_level_report(unit_key: T.Dict[str, T.Any]) -> None:
     waveform_fig, autocorrelogram_fig, depth_waveform_fig = (
         ephys_report.UnitLevelReport & unit_key
@@ -54,17 +84,7 @@ def plot_unit_level_report(unit_key: T.Dict[str, T.Any]) -> None:
     depth_waveform_fig = go.FigureWidget(depth_waveform_fig)
 
     display(
-        ipywidgets.HBox(
-            [ipywidgets.VBox([waveform_fig, autocorrelogram_fig]), depth_waveform_fig]
+        widgets.HBox(
+            [widgets.VBox([waveform_fig, autocorrelogram_fig]), depth_waveform_fig]
         )
     )
-
-
-def unit_widget() -> ipywidgets.interactive:
-    drop_down_wg = _drop_down_wg(
-        ephys_report.UnitLevelReport & "cluster_quality_label='good'",
-        description="Select Units : ",
-        width=85,
-        description_width=150,
-    )
-    return ipywidgets.interactive(plot_unit_level_report, unit_key=drop_down_wg)