работающая стеганография LSB и DCT

я устал... DCT убивает

debug_dct_russian.py

@@ -0,0 +1,137 @@
+"""
+Отладка DCT: русский текст
+"""
+
+from core.dct import encode_dct, _apply_dct, _extract_bits_from_block, MID_FREQ_INDICES
+from core.utils import text_to_bits, bits_to_text
+from PIL import Image
+import numpy as np
+
+print("=" * 60)
+print("Отладка DCT: русский текст")
+print("=" * 60)
+
+# Создаем тестовое изображение
+img = Image.new('L', (400, 400), color=128)
+img.save("debug_russian.png")
+
+# Русское сообщение
+message = "Привет"
+print(f"Оригинальное сообщение: '{message}'")
+print(f"Длина в символах: {len(message)}")
+
+# Покажем биты оригинального сообщения
+bits_original = text_to_bits(message)
+print(f"\nОригинальные биты: {bits_original}")
+print(f"Длина битов: {len(bits_original)}")
+
+# Добавляем стоп-маркер
+bits_with_stop = bits_original + '00000000'
+print(f"Биты со стоп-маркером: {bits_with_stop}")
+print(f"Длина со стоп-маркером: {len(bits_with_stop)}")
+
+# Кодируем
+print("\n--- Кодирование ---")
+encode_dct("debug_russian.png", message, "debug_russian_encoded.png")
+
+# Загружаем закодированное изображение и извлекаем биты
+img_encoded = Image.open("debug_russian_encoded.png").convert('L')
+pixels = np.array(img_encoded, dtype=np.float64)
+height, width = pixels.shape
+
+print("\n--- Извлечение битов из блоков ---")
+all_bits = []
+block_count = 0
+
+for i in range(0, height - 8 + 1, 8):
+    for j in range(0, width - 8 + 1, 8):
+        block = pixels[i:i+8, j:j+8]
+        dct_block = _apply_dct(block)
+        bits_from_block = _extract_bits_from_block(dct_block, len(MID_FREQ_INDICES))
+        all_bits.extend(bits_from_block)
+        block_count += 1
+        
+        # Покажем первые 3 блока
+        if block_count <= 3:
+            print(f"Блок {block_count}: {bits_from_block[:30]}...")
+
+print(f"\nВсего извлечено битов: {len(all_bits)}")
+
+# Сравним оригинальные и извлеченные биты (первые 100)
+print("\n" + "=" * 60)
+print("Сравнение битов (первые 100):")
+print("=" * 60)
+print(f"Оригинал: {bits_with_stop[:100]}")
+print(f"Извлечено: {''.join(str(b) for b in all_bits[:100])}")
+
+# Найдем позиции, где биты отличаются
+print("\n" + "=" * 60)
+print("Различия в битах:")
+print("=" * 60)
+diff_count = 0
+for i in range(min(len(bits_with_stop), len(all_bits))):
+    orig_bit = int(bits_with_stop[i])
+    ext_bit = all_bits[i]
+    if orig_bit != ext_bit:
+        diff_count += 1
+        if diff_count <= 20:
+            print(f"Позиция {i}: было {orig_bit}, стало {ext_bit}")
+print(f"Всего различий: {diff_count}")
+
+# Попробуем декодировать извлеченные биты без стоп-маркера
+print("\n" + "=" * 60)
+print("Декодирование извлеченных битов:")
+print("=" * 60)
+
+# Преобразуем биты в байты и покажем их
+extracted_bytes = []
+for k in range(0, min(len(all_bits), 200), 8):
+    if k + 8 <= len(all_bits):
+        byte_bits = all_bits[k:k+8]
+        byte_val = 0
+        for b in byte_bits:
+            byte_val = (byte_val << 1) | b
+        extracted_bytes.append(byte_val)
+        # Покажем как число и как символ
+        if 32 <= byte_val < 127:
+            char = chr(byte_val)
+        else:
+            char = f'[{byte_val:02x}]'
+        print(f"Байт {k//8:2d}: {byte_val:3d} (0x{byte_val:02x}) -> {char}  биты={byte_bits}")
+
+# Найдем стоп-маркер
+print("\n" + "=" * 60)
+print("Поиск стоп-маркера:")
+print("=" * 60)
+
+stop_counter = 0
+stop_position = -1
+for i, bit in enumerate(all_bits):
+    if bit == 0:
+        stop_counter += 1
+        if stop_counter == 8:
+            stop_position = i - 7
+            print(f"Стоп-маркер найден на позиции бита {stop_position}")
+            break
+    else:
+        stop_counter = 0
+
+if stop_position == -1:
+    print("Стоп-маркер НЕ НАЙДЕН!")
+    # Используем все биты
+    bits_to_decode = all_bits
+else:
+    bits_to_decode = all_bits[:stop_position]
+    print(f"Битов до стоп-маркера: {len(bits_to_decode)}")
+
+# Декодируем
+bits_string = ''.join(str(b) for b in bits_to_decode)
+try:
+    decoded = bits_to_text(bits_string)
+    print(f"\nДекодированное сообщение: '{decoded}'")
+    if decoded == message:
+        print("✅ УСПЕХ!")
+    else:
+        print("❌ Сообщения не совпадают")
+except Exception as e:
+    print(f"\nОшибка декодирования: {e}")