Initial commit

2026-07-12 14:06:45 +04:00
commit a75028d68f
148 changed files with 3275 additions and 0 deletions
@@ -0,0 +1,3 @@
+__all__ = ["parse_args"]
+
+from .parse_args import parse_args
@@ -0,0 +1,147 @@
+
+from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
+
+
+def parse_args(argv=None):
+    ap = ArgumentParser(
+        prog="app.py",
+        description="CLI для лабораторной №6: Стеганография в изображениях",
+        formatter_class=ArgumentDefaultsHelpFormatter,
+    )
+    sub = ap.add_subparsers(dest="cmd", required=True)
+
+    # -------- Parent parsers --------
+    parent_channel_seq = ArgumentParser(add_help=False)
+    parent_channel_seq.add_argument(
+        "-c", "--channel",
+        choices=["Y", "Cb", "Cr", "y", "cb", "cr"],
+        default="Cb",
+        help="Канал для встраивания/извлечения: Y, Cb или Cr",
+    )
+    parent_channel_seq.add_argument(
+        "--seq0",
+        type=str,
+        default="01010101",
+        help="8-битная последовательность для бита 0 (seq1 считается инверсией seq0)",
+    )
+
+    parent_gen_common = ArgumentParser(add_help=False)
+    parent_gen_common.add_argument(
+        "--size",
+        type=int,
+        default=512,
+        help="Размер стороны квадратного изображения",
+    )
+    parent_gen_common.add_argument(
+        "--channels",
+        type=int,
+        choices=[1, 3],
+        default=3,
+        help="Количество каналов: 1 (grayscale) или 3 (RGB)",
+    )
+    parent_gen_common.add_argument(
+        "-o", "--output",
+        required=True,
+        help="Куда сохранить изображение",
+    )
+
+    # ---------------- EMBED ----------------
+    sp_embed = sub.add_parser(
+        "embed",
+        parents=[parent_channel_seq],
+        help="Встраивание метки в изображение-контейнер",
+        formatter_class=ArgumentDefaultsHelpFormatter,
+    )
+    sp_embed.add_argument(
+        "-i", "--input",
+        required=True,
+        help="Путь к исходному изображению-контейнеру (512x512, RGB)",
+    )
+    sp_embed.add_argument(
+        "-l", "--label",
+        required=True,
+        help="Путь к бинарной метке (64x64)",
+    )
+    sp_embed.add_argument(
+        "-o", "--output",
+        required=True,
+        help="Куда сохранить стего-изображение",
+    )
+
+    # ---------------- EXTRACT ----------------
+    sp_extract = sub.add_parser(
+        "extract",
+        parents=[parent_channel_seq],
+        help="Извлечение метки из стего-изображения",
+        formatter_class=ArgumentDefaultsHelpFormatter,
+    )
+    sp_extract.add_argument(
+        "-i", "--input",
+        required=True,
+        help="Путь к стего-изображению",
+    )
+    sp_extract.add_argument(
+        "-o", "--output",
+        required=True,
+        help="Куда сохранить восстановленную бинарную метку (64x64)",
+    )
+
+    # ---------------- ANALYZE ----------------
+    sp_analyze = sub.add_parser(
+        "analyze",
+        help="Сравнение изображений и метрик качества",
+        formatter_class=ArgumentDefaultsHelpFormatter,
+    )
+    sp_analyze.add_argument(
+        "-a", "--original",
+        required=True,
+        help="Путь к исходному изображению-контейнеру",
+    )
+    sp_analyze.add_argument(
+        "-b", "--stego",
+        required=True,
+        help="Путь к стего-изображению",
+    )
+    sp_analyze.add_argument(
+        "--space",
+        choices=["rgb", "ycbcr"],
+        default="ycbcr",
+        help="Цветовое пространство для метрик",
+    )
+    sp_analyze.add_argument(
+        "--metrics",
+        nargs="+",
+        choices=["psnr"],
+        default=["psnr"],
+        help="Набор метрик",
+    )
+
+    # ---------------- GENERATE ----------------
+    sp_gen = sub.add_parser(
+        "generate",
+        help="Генерация изображений",
+        formatter_class=ArgumentDefaultsHelpFormatter,
+    )
+    gen_sub = sp_gen.add_subparsers(dest="gen", required=True)
+
+    gen_sub.add_parser(
+        "gradient",
+        parents=[parent_gen_common],
+        help="Сгенерировать градиент",
+        formatter_class=ArgumentDefaultsHelpFormatter,
+    )
+
+    sp_chess = gen_sub.add_parser(
+        "chess",
+        parents=[parent_gen_common],
+        help="Сгенерировать шахматную доску",
+        formatter_class=ArgumentDefaultsHelpFormatter,
+    )
+    sp_chess.add_argument(
+        "--tile",
+        type=int,
+        default=32,
+        help="Размер квадрата (должен делить --size)",
+    )
+
+    return ap.parse_args(argv)