Initial commit

lab6-steganography/modes/embed_mode.py

@@ -0,0 +1,40 @@
+
+from __future__ import annotations
+from dataclasses import dataclass
+from typing import Dict
+
+from entities.rgb_image512 import RgbImage512
+from entities.label64 import Label64
+from entities.sequence8 import Sequence8
+from steg.embedder import Embedder
+from steg.extractor import Extractor
+from steg.component_pipeline import ComponentPipeline
+
+
+@dataclass
+class EmbedMode:
+    """CLI-режим встраивания метки в изображение."""
+
+    def run(
+        self,
+        input_path: str,
+        label_path: str,
+        output_path: str,
+        channel: str,
+        seq0: str,
+    ) -> Dict[str, str]:
+        """Выполнить встраивание и сохранить результат."""
+        rgb = RgbImage512.from_file(input_path)
+        label = Label64.from_image(label_path)
+        seq = Sequence8.from_string(seq0)
+
+        pipeline = ComponentPipeline(channel=channel, embedder=Embedder(), extractor=Extractor())
+        rgb_out = pipeline.embed_rgb(rgb, label, seq)
+        rgb_out.save(output_path)
+
+        return {
+            "status": "ok",
+            "output": output_path,
+            "channel": channel,
+            "seq0": seq0,
+        }