introduced FEATURES_LAYER_TYPES variable to be used across widgets

Author: jo-mueller

src/napari_matplotlib/features.py

@@ -0,0 +1,9 @@
+from napari.layers import Labels, Points, Shapes, Tracks, Vectors
+
+FEATURES_LAYER_TYPES = (
+    Labels,
+    Points,
+    Shapes,
+    Tracks,
+    Vectors,
+)

src/napari_matplotlib/histogram.py

@@ -7,6 +7,7 @@
 
 from .base import NapariMPLWidget
 from .util import Interval
+from .features import FEATURES_LAYER_TYPES
 
 __all__ = ["HistogramWidget", "FeaturesHistogramWidget"]
 
@@ -69,13 +70,7 @@ def draw(self) -> None:
 class FeaturesHistogramWidget(NapariMPLWidget):
     n_layers_input = Interval(1, 1)
     # All layers that have a .features attributes
-    input_layer_types = (
-        napari.layers.Labels,
-        napari.layers.Points,
-        napari.layers.Shapes,
-        napari.layers.Tracks,
-        napari.layers.Vectors,
-    )
+    input_layer_types = FEATURES_LAYER_TYPES
 
     def __init__(self, napari_viewer: napari.viewer.Viewer):
         super().__init__(napari_viewer)

src/napari_matplotlib/scatter.py

@@ -6,6 +6,7 @@
 
 from .base import NapariMPLWidget
 from .util import Interval
+from .features import FEATURES_LAYER_TYPES
 
 __all__ = ["ScatterBaseWidget", "ScatterWidget",
            "FeaturesScatterWidget"]
@@ -107,13 +108,7 @@ class FeaturesScatterWidget(ScatterBaseWidget):
 
     n_layers_input = Interval(1, 1)
     # All layers that have a .features attributes
-    input_layer_types = (
-        napari.layers.Labels,
-        napari.layers.Points,
-        napari.layers.Shapes,
-        napari.layers.Tracks,
-        napari.layers.Vectors,
-    )
+    input_layer_types = FEATURES_LAYER_TYPES
 
     def __init__(
         self,