41 lines
1.8 KiB
Python
41 lines
1.8 KiB
Python
|
from __future__ import annotations
|
|||
|
|
from dataclasses import dataclass
|
|||
|
|
import numpy as np
|
|||
|
|
|
|||
|
|
|
|||
|
|
@dataclass(frozen=True)
|
|||
|
|
class BlockGrid8:
|
|||
|
|
"""Разбиение и сборка блоков 8×8."""
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def split(component: np.ndarray) -> np.ndarray:
|
|||
|
|
"""Разбить 2D-компоненту на блоки (N,8,8)."""
|
|||
|
|
if component.ndim != 2:
|
|||
|
|
raise ValueError("Ожидается 2D-массив компоненты.")
|
|||
|
|
h, w = component.shape
|
|||
|
|
if h % 8 != 0 or w % 8 != 0:
|
|||
|
|
raise ValueError("Размеры должны быть кратны 8.")
|
|||
|
|
# Перестановка осей даёт (nH, nW, 8, 8) -> (N, 8, 8)
|
|||
|
|
arr = np.asarray(component)
|
|||
|
|
nH, nW = h // 8, w // 8
|
|||
|
|
blocks = arr.reshape(nH, 8, nW, 8).transpose(
|
|||
|
|
0, 2, 1, 3).reshape(nH * nW, 8, 8)
|
|||
|
|
return blocks
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def merge(blocks: np.ndarray, height: int, width: int) -> np.ndarray:
|
|||
|
|
"""Собрать 2D-компоненту из блоков (N,8,8)."""
|
|||
|
|
if blocks.ndim != 3 or blocks.shape[1:] != (8, 8):
|
|||
|
|
raise ValueError("Ожидается массив блоков формы (N,8,8).")
|
|||
|
|
if height % 8 != 0 or width % 8 != 0:
|
|||
|
|
raise ValueError("Размеры должны быть кратны 8.")
|
|||
|
|
nH, nW = height // 8, width // 8
|
|||
|
|
if blocks.shape[0] != nH * nW:
|
|||
|
|
raise ValueError(
|
|||
|
|
"Количество блоков не соответствует размеру выходной матрицы.")
|
|||
|
|
# Обратная перестановка из (N,8,8) -> (nH, nW, 8, 8) -> (H, W)
|
|||
|
|
arr = np.asarray(blocks)
|
|||
|
|
comp = arr.reshape(nH, nW, 8, 8).transpose(
|
|||
|
|
0, 2, 1, 3).reshape(height, width)
|
|||
|
|
return comp
|