user
32 lines
1.5 KiB
Python
32 lines
1.5 KiB
Python
from __future__ import annotations
|
||
from dataclasses import dataclass
|
||
import numpy as np
|
||
|
||
from entities.block_grid8 import BlockGrid8
|
||
from entities.lsb_row8 import LsbRow8
|
||
from entities.label64 import Label64
|
||
from entities.sequence8 import Sequence8
|
||
|
||
|
||
@dataclass(frozen=True)
|
||
class Embedder:
|
||
"""Встраивание метки в 2D-компоненту Y/Cb/Cr через LSB верхней строки блоков 8×8."""
|
||
|
||
def embed_component(self, component: np.ndarray, label: Label64, seq: Sequence8) -> np.ndarray:
|
||
"""Вернуть новую компоненту с вшитой меткой."""
|
||
if component.ndim != 2:
|
||
raise ValueError("Ожидается 2D-компонента.")
|
||
h, w = component.shape
|
||
if h % 8 != 0 or w % 8 != 0:
|
||
raise ValueError("Размеры компоненты должны быть кратны 8.")
|
||
blocks = BlockGrid8.split(component)
|
||
bits = label.to_vector() # 4096
|
||
if blocks.shape[0] != bits.size:
|
||
raise ValueError("Количество блоков должно равняться 4096.")
|
||
out_blocks = blocks.copy()
|
||
# Для каждого бита берём 8-битную сигнатуру и встраиваем в верхнюю строку блока
|
||
for i in range(bits.size):
|
||
pattern = seq.for_bit(int(bits[i]))
|
||
out_blocks[i] = LsbRow8.embed(out_blocks[i], pattern)
|
||
return BlockGrid8.merge(out_blocks, h, w)
|