fastplotlib
diff --git a/‎fastplotlib/widgets/image.py
Copy file name to clipboardExpand all lines: fastplotlib/widgets/image.py
+81-10Lines changed: 81 additions & 10 deletions b/‎fastplotlib/widgets/image.py
Copy file name to clipboardExpand all lines: fastplotlib/widgets/image.py
+81-10Lines changed: 81 additions & 10 deletions
@@ -1,7 +1,7 @@
 from ..plot import Plot
 from ..layouts import GridPlot
 from ..graphics import Image
-from ipywidgets.widgets import IntSlider, VBox, HBox
+from ipywidgets.widgets import IntSlider, VBox, HBox, Layout
 import numpy as np
 from typing import *
 from warnings import warn
@@ -24,6 +24,12 @@ def calc_gridshape(n):
     )
 
 
+def get_indexer(ndim: int, dim_index: int, slice_index: int) -> slice:
+    dim_index = [slice(None)] * ndim
+    dim_index[dim_index] = slice_index
+    return tuple(dim_index)
+
+
 class ImageWidget:
     def __init__(
             self,
@@ -33,6 +39,7 @@ def __init__(
             slider_axes: Union[int, str, dict] = None,
             frame_apply: Union[callable, dict] = None,
             grid_shape: Tuple[int, int] = None,
+            **kwargs
     ):
         # single image
         if isinstance(data, np.ndarray):
@@ -71,21 +78,85 @@ def __init__(
         if axes_order is None:
             self.axes_order: List[str] = [DEFAULT_AXES_ORDER[ndim] for i in range(len(data))]
 
-        if isinstance(slider_axes, (int)):
-            self._slider_axes: List[int] = [slider_axes for i in range(len(data))]
+        # if a single one is provided
+        if isinstance(slider_axes, (int, str)):
+            if isinstance(slider_axes, (int)):
+                self._slider_axes = slider_axes
 
-        elif isinstance(slider_axes, str):
-            self._slider_axes: List[int] = [self.axes_order.index(slider_axes)]
+            # also if a single one is provided, get the integer dimension index from the axes_oder string
+            elif isinstance(slider_axes, str):
+                self._slider_axes = self.axes_order.index(slider_axes)
 
-        self.sliders: List[IntSlider] = [
-            IntSlider(
+            self.slider: IntSlider = IntSlider(
                 min=0,
-                max=data.shape[slider_axes] - 1,
+                max=data.shape[self._slider_axes] - 1,
                 value=0,
                 step=1,
-                description=f"slider axis: {slider_axes}"
+                description=f"slider axis: {self._slider_axes}"
+            )
+
+        # individual slider for each data array
+        elif isinstance(slider_axes, dict):
+            if not len(slider_axes.keys()) == len(self.data):
+                raise ValueError(
+                    f"Must provide slider_axes entry for every input `data` array"
+                )
+
+            if not isinstance(axes_order, dict):
+                raise ValueError("Must pass `axes_order` dict if passing a dict of `slider_axes`")
+
+            if not len(axes_order.keys()) == len(self.data):
+                raise ValueError(
+                    f"Must provide `axes_order` entry for every input `data` array"
+                )
+
+            # convert str type desired slider axes to dimension index integers
+            # matchup to the given axes_order dict
+            _axes = [
+                self.axes_order[array].index(slider_axes[array])
+                    if isinstance(dim_index, str)
+                    else dim_index
+                        for
+                            array, dim_index in slider_axes.items()
+            ]
+
+            self.sliders: Dict[IntSlider] = {
+                array: IntSlider(
+                    min=0,
+                    max=array.shape[dim] -1,
+                    step=1,
+                    value=0,
+                )
+                for array, dim in self.axes_order.items()
+            }
+
+        if self.plot_type == Plot:
+            self.plot = Plot()
+
+            slice_index = get_indexer(ndim, self._slider_axes, slice_index=0)
+
+            self.image_graphics: List[Image] = [self.plot.image(
+                data=data[0][slice_index],
+                **kwargs
+            )]
+
+            self.slider.observe(
+                lambda x: self.image_graphics[0].update_data(
+                    data[0][
+                        get_indexer(ndim, self._slider_axes, slice_index=x["new"])
+                    ]
+                ),
+                names="value"
             )
-        ]
+
+            self.widget = VBox([self.plot, self.slider])
+
+        elif self.plot_type == GridPlot:
+            pass
+
+    def set_frame_slider_width(self):
+        w, h = self.plot.renderer.logical_size
+        self.slider.layout = Layout(width=f"{w}px")
 
     def slider_changed(self):
         pass