from PIL import Image
import numpy as np

# Размер холста
CANVAS_SIZE = 1000

# Размер блока динамита
BLOCK_SIZE = 5

# Цвет динамита в HEX и RGB
DYNAMITE_COLOR_HEX = '#171F2A'
DYNAMITE_COLOR_RGB = (23, 31, 42)  # Переводим HEX в RGB

# Цвет фона (белый)
BACKGROUND_COLOR = (255, 255, 255)

def process_image(image_path):
    # Загрузка изображения и преобразование в оттенки серого
    image = Image.open(image_path).convert('L')

    # Масштабирование изображения до размера холста
    image = image.resize((CANVAS_SIZE, CANVAS_SIZE), Image.LANCZOS)

    # Преобразование изображения в массив numpy
    pixel_data = np.array(image)

    # Преобразование в два цвета: BACKGROUND_COLOR и DYNAMITE_COLOR
    # Устанавливаем пороговое значение
    binary_mask = pixel_data < THRESHOLD  # True для пикселей, которые должны быть цветом динамита

    # Создаем цветное изображение с двумя цветами
    colored_image = np.zeros((CANVAS_SIZE, CANVAS_SIZE, 3), dtype=np.uint8)
    colored_image[binary_mask] = DYNAMITE_COLOR_RGB       # Пиксели цвета динамита
    colored_image[~binary_mask] = BACKGROUND_COLOR        # Остальные пиксели - белые

    # Инициализация списка координат блоков, которые нужно взорвать
    dynamite_coordinates = []

    # Разбиение на блоки и анализ
    for y in range(0, CANVAS_SIZE, BLOCK_SIZE):
        for x in range(0, CANVAS_SIZE, BLOCK_SIZE):
            # Извлекаем блок размером BLOCK_SIZE x BLOCK_SIZE
            block = colored_image[y:y + BLOCK_SIZE, x:x + BLOCK_SIZE]

            # Проверяем, есть ли в блоке пиксели цвета динамита
            if np.any(np.all(block == DYNAMITE_COLOR_RGB, axis=2)):
                # Вычисляем координаты центра блока
                center_x = x + BLOCK_SIZE // 2
                center_y = y + BLOCK_SIZE // 2

                # Добавляем координаты в список
                dynamite_coordinates.append((center_x, center_y))

    return dynamite_coordinates, colored_image

def save_coordinates(coordinates, output_file='dynamite_coordinates.txt'):
    with open(output_file, 'w') as f:
        for coord in coordinates:
            f.write(f"Color: {DYNAMITE_COLOR_HEX}, Coordinates: ({coord[0]}, {coord[1]})\n")
    print(f"Координаты сохранены в файл {output_file}")

def save_result_image(colored_image, output_file='result_image.png'):
    # Преобразуем массив numpy обратно в изображение PIL и сохраняем
    result_image = Image.fromarray(colored_image)
    result_image.save(output_file)
    print(f"Результирующее изображение сохранено в файл {output_file}")

if __name__ == "__main__":
    # Пороговое значение для определения цвета динамита
    THRESHOLD = 127  # Оттенки серого от 0 (чёрный) до 255 (белый)

    # Путь к изображению
    image_path = 'pngegg.png'  # Замените на путь к вашему изображению

    # Обработка изображения
    coords, colored_image = process_image(image_path)

    # Сохранение координат
    save_coordinates(coords)

    # Сохранение результирующего изображения
    save_result_image(colored_image)