Merge branch 'main' into feat/hist-bin-params

Author: p-j-smith

src/napari_matplotlib/histogram.py

@@ -25,6 +25,16 @@
 _COLORS = {"r": "tab:red", "g": "tab:green", "b": "tab:blue"}
 
 
+def _get_bins(data: npt.NDArray[Any]) -> npt.NDArray[Any]:
+    if data.dtype.kind in {"i", "u"}:
+        # Make sure integer data types have integer sized bins
+        step = np.ceil(np.ptp(data) / 100)
+        return np.arange(np.min(data), np.max(data) + step, step)
+    else:
+        # For other data types, just have 128 evenly spaced bins
+        return np.linspace(np.min(data), np.max(data), 100)
+
+
 class HistogramWidget(SingleAxesWidget):
     """
     Display a histogram of the currently selected layer.
@@ -134,13 +144,7 @@ def _update_contrast_lims(self) -> None:
 
     def autoset_widget_bins(self, data: npt.NDArray[Any]) -> None:
         """Update widgets with bins determined from the image data"""
-        if data.dtype.kind in {"i", "u"}:
-            # Make sure integer data types have integer sized bins
-            step = abs(np.max(data) - np.min(data)) // 100
-            step = max(1, step)
-            bins = np.arange(np.min(data), np.max(data) + step, step)
-        else:
-            bins = np.linspace(np.min(data), np.max(data), 100)
+        bins = _get_bins(data)
 
         # Disable callbacks whilst setting widget values
         for widget in self._bin_widgets.values():
@@ -353,9 +357,9 @@ def draw(self) -> None:
         if data is None:
             return
 
-        _, bins, patches = self.axes.hist(
-            data, bins=50, edgecolor="white", linewidth=0.3
-        )
+        bins = _get_bins(data)
+
+        _, bins, patches = self.axes.hist(data, bins=bins.tolist())
         patches = cast(BarContainer, patches)
 
         # recolor the histogram plot