Add contrast limits to histogram plot

dstansby · dstansby · commit 6349077849a0 · 2023-10-23T11:22:51.000+01:00
diff --git a/src/napari_matplotlib/histogram.py b/src/napari_matplotlib/histogram.py
@@ -29,6 +29,24 @@ def __init__(
     ):
         super().__init__(napari_viewer, parent=parent)
         self._update_layers(None)
+        self.viewer.events.theme.connect(self._on_napari_theme_changed)
+
+    def on_update_layers(self) -> None:
+        """
+        Called when the selected layers are updated.
+        """
+        super().on_update_layers()
+        for layer in self.viewer.layers:
+            layer.events.contrast_limits.connect(self._update_contrast_lims)
+
+    def _update_contrast_lims(self) -> None:
+        [
+            line.set_xdata(lim)
+            for lim, line in zip(
+                self.layers[0].contrast_limits, self._contrast_lines
+            )
+        ]
+        self.figure.canvas.draw()
 
     def draw(self) -> None:
         """
@@ -62,6 +80,10 @@ def draw(self) -> None:
         else:
             self.axes.hist(data.ravel(), bins=bins, label=layer.name)
 
+        self._contrast_lines = [
+            self.axes.axvline(lim, color="white")
+            for lim in layer.contrast_limits
+        ]
         self.axes.legend()
 
 
