Add image label to stretch and cuts

Author: mwcraig

src/astro_image_display_api/dummy_viewer.py

@@ -27,6 +27,8 @@ class ViewportInfo:
     fov: float | Quantity | None = None
     wcs: WCS | None = None
     largest_dimension: int | None = None
+    stretch: BaseStretch | None = None
+    cuts: BaseInterval | tuple[numbers.Real, numbers.Real] | None = None
 
 @dataclass
 class ImageViewer:
@@ -42,8 +44,6 @@ class ImageViewer:
     zoom_level: float = 1
     _cursor: str = ImageViewerInterface.ALLOWED_CURSOR_LOCATIONS[0]
     marker: Any = "marker"
-    _cuts: BaseInterval | tuple[numbers.Real, numbers.Real] = AsymmetricPercentileInterval(upper_percentile=95)
-    _stretch: BaseStretch = LinearStretch
     # viewer: Any
 
     # Allowed locations for cursor display
@@ -70,24 +70,37 @@ def __post_init__(self):
         self._images[None].fov = None
         self._images[None].wcs = None
 
-    def get_stretch(self) -> BaseStretch:
-        return self._stretch
+    def get_stretch(self, image_label: str | None = None) -> BaseStretch:
+        image_label = self._resolve_image_label(image_label)
+        if image_label not in self._images:
+            raise ValueError(f"Image label '{image_label}' not found. Please load an image first.")
+        return self._images[image_label].stretch
 
-    def set_stretch(self, value: BaseStretch) -> None:
+    def set_stretch(self, value: BaseStretch, image_label: str | None = None) -> None:
         if not isinstance(value, BaseStretch):
             raise ValueError(f"Stretch option {value} is not valid. Must be an Astropy.visualization Stretch object.")
-        self._stretch = value
+        image_label = self._resolve_image_label(image_label)
+        if image_label not in self._images:
+            raise ValueError(f"Image label '{image_label}' not found. Please load an image first.")
+        self._images[image_label].stretch = value
 
-    def get_cuts(self) -> tuple:
-        return self._cuts
+    def get_cuts(self, image_label: str | None = None) -> tuple:
+        image_label = self._resolve_image_label(image_label)
+        if image_label not in self._images:
+            raise ValueError(f"Image label '{image_label}' not found. Please load an image first.")
+        return self._images[image_label].cuts
 
-    def set_cuts(self, value: tuple[numbers.Real, numbers.Real] | BaseInterval) -> None:
+    def set_cuts(self, value: tuple[numbers.Real, numbers.Real] | BaseInterval, image_label: str | None = None) -> None:
         if isinstance(value, tuple) and len(value) == 2:
             self._cuts = ManualInterval(value[0], value[1])
         elif isinstance(value, BaseInterval):
             self._cuts = value
         else:
             raise ValueError("Cuts must be an Astropy.visualization Interval object or a tuple of two values.")
+        image_label = self._resolve_image_label(image_label)
+        if image_label not in self._images:
+            raise ValueError(f"Image label '{image_label}' not found. Please load an image first.")
+        self._images[image_label].cuts = self._cuts
 
     @property
     def cursor(self) -> str: