Initial commit

lab6-steganography/entities/channel_view.py

@@ -0,0 +1,39 @@
+from __future__ import annotations
+from dataclasses import dataclass
+import numpy as np
+
+
+@dataclass(frozen=True)
+class ChannelView:
+    """Доступ к компонентам Y, Cb, Cr в массиве YCbCr."""
+
+    _MAP = {"y": 0, "cb": 1, "cr": 2}
+
+    @staticmethod
+    def _idx(name: str) -> int:
+        """Вернуть индекс канала."""
+        k = name.strip().lower()
+        if k not in ChannelView._MAP:
+            raise ValueError("Канал должен быть Y, Cb или Cr.")
+        return ChannelView._MAP[k]
+
+    @staticmethod
+    def get(ycbcr: np.ndarray, channel: str) -> np.ndarray:
+        """Вернуть 2D-компоненту выбранного канала."""
+        if ycbcr.ndim != 3 or ycbcr.shape[-1] != 3:
+            raise ValueError("Ожидается массив HxWx3.")
+        i = ChannelView._idx(channel)
+        return ycbcr[..., i]
+
+    @staticmethod
+    def set(ycbcr: np.ndarray, channel: str, component: np.ndarray) -> np.ndarray:
+        """Вернуть копию YCbCr с заменённым каналом."""
+        if ycbcr.ndim != 3 or ycbcr.shape[-1] != 3:
+            raise ValueError("Ожидается массив HxWx3.")
+        if component.shape != ycbcr.shape[:2]:
+            raise ValueError("Размер компоненты не совпадает с HxW.")
+        out = ycbcr.copy()
+        i = ChannelView._idx(channel)
+        # Короткая отсечка и приведение типа
+        out[..., i] = np.clip(component, 0, 255).astype(np.uint8)
+        return out