from PIL import Image
import sys
import os

def split_and_convert(img_path, tile_size=(64, 256)):
    # Открываем изображение
    img = Image.open(img_path)
    width, height = img.size
    
    # Получаем имя файла без расширения
    base_name = os.path.splitext(os.path.basename(img_path))[0]
    
    counter = 0
    
    for i in range(0, width, tile_size[0]):
        for j in range(0, height, tile_size[1]):
            box = (i, j, i+tile_size[0], j+tile_size[1])
            tile = img.crop(box)
            
            # Конвертируем в 256 цветов
            tile = tile.convert('P', palette=Image.Palette.ADAPTIVE, colors=256)
            
            # Сохраняем тайл с чётным индексом
            tile.save(f'{base_name}_{counter:02d}_00_00.png', optimize=True)
            
            # Извлекаем палитру
            palette = tile.getpalette()
            if palette:
                # Создаём изображение палитры 256x1
                # Каждый пиксель - один цвет из палитры
                palette_rgb = []
                for k in range(0, len(palette), 3):
                    r, g, b = palette[k], palette[k+1], palette[k+2]
                    palette_rgb.append((r, g, b))
                
                # Обрезаем до 256 цветов (на всякий случай)
                palette_rgb = palette_rgb[:256]
                
                # Создаём изображение палитры 256x10 (можно сделать 256x1, но так нагляднее)
                palette_img = Image.new('RGB', (256, 1))
                
                # Заполняем каждый столбец своим цветом
                for x in range(256):
                    color = palette_rgb[x] if x < len(palette_rgb) else (0, 0, 0)
                    for y in range(1):
                        palette_img.putpixel((x, y), color)
                
                # Сохраняем палитру с нечётным индексом как 16-bit PNG
                palette_img.save(
                    f'{base_name}_{counter+1:02d}_00_00.png', 
                    optimize=True,
                    bits=16  # 16-битное сохранение
                )
            
            counter += 2

def main():
    # Проверяем аргументы командной строки
    if len(sys.argv) < 2:
        print("Использование: python script.py <input_image>")
        print("Пример: python script.py TEST.png")
        sys.exit(1)
    
    input_image = sys.argv[1]
    
    # Проверяем существование файла
    if not os.path.exists(input_image):
        print(f"Ошибка: Файл '{input_image}' не найден")
        sys.exit(1)
    
    # Запускаем обработку
    split_and_convert(input_image)

if __name__ == "__main__":
    main()