"""2025-04-04
Maria's 12
A smile from the girl that puts a smile in my face every day
png
Sketch,py5,CreativeCoding
"""

from pathlib import Path

import numpy as np
import py5

from utils import helpers
from utils.draw import rgb_to_hsb

sketch = helpers.info_for_sketch(__file__, __doc__)


TAM_PIXEL = 6


def pixelar(img_array: np.array, tam_pixel: int = 24, func=np.median):
    pontos = []
    # py5.no_stroke()
    for idx, x in enumerate(range(0, helpers.LARGURA, tam_pixel)):
        for idy, y in enumerate(range(0, helpers.ALTURA, tam_pixel)):
            bloco = img_array[y : y + tam_pixel, x : x + tam_pixel]
            cor = func(bloco, axis=(0, 1))
            h, s, b = rgb_to_hsb(cor[0], cor[1], cor[2])
            pontos.append((idx, idy, x, y, (h, s, b)))
    return pontos


def calcula_raio(centro: int, idx: int, idy: int) -> float:
    banda = centro * 0.5
    base = 0.2
    mult = 1 / base
    dx = idx - centro
    dy = idy - centro
    distancia = dx**2 + dy**2
    valor = np.exp(-distancia / (mult * banda**2))
    return base + (1 - base) * valor


def setup():
    py5.size(helpers.LARGURA, helpers.ALTURA, py5.P3D)
    py5.background(0)
    py5.color_mode(py5.HSB, 360, 100, 100)
    largura = 800
    altura = 800
    path = Path(__file__).parent / "maria.jpg"
    img_array = helpers.image_as_array(path)
    pontos = pixelar(img_array, TAM_PIXEL, np.average)
    py5.no_stroke()
    centro = (py5.width / TAM_PIXEL) / 2
    with py5.push():
        py5.translate(py5.width // 2, py5.width // 2, -altura / 2)
        py5.lights()
        py5.light_specular(0, 150, 300)
        py5.specular(100, 100, 100)
        for idx, idy, xb, yb, cor in pontos:
            x = idx * TAM_PIXEL * 2 - largura
            y = idy * TAM_PIXEL * 2 - altura
            raio = calcula_raio(centro, idx, idy)
            z = -((1 - raio) * 220)
            with py5.push():
                py5.translate(x, y, z)
                py5.fill(*cor)
                py5.sphere(TAM_PIXEL * raio)
    helpers.write_legend(sketch=sketch, frame="#222", cor="#FFF")


def key_pressed():
    key = py5.key
    if key == " ":
        save_and_close()


def save_and_close():
    py5.no_loop()
    helpers.save_sketch_image(sketch)
    py5.exit_sketch()


if __name__ == "__main__":
    py5.run_sketch()