"""2025-08-10
Colorido geométrico IX
Hexágonos e Pentágonos distribuídos em uma grade
ericof.com
png
Sketch,py5,CreativeCoding
"""

from sketches.padroes.poligonos import gera_poligono_regular
from sketches.utils.draw import canvas
from sketches.utils.draw.grade import cria_grade_ex
from sketches.utils.helpers import sketches as helpers

import py5


sketch = helpers.info_for_sketch(__file__, __doc__)


LARGURA = 25
ALTURA = 25
LADOS = 6
forma1: py5.Py5Shape | None = None
forma2: py5.Py5Shape | None = None


def setup():
    global forma1, forma2
    py5.size(*helpers.DIMENSOES.external, py5.P3D)
    py5.color_mode(py5.HSB, 360, 100, 100)
    py5.shape_mode(py5.CORNER)
    forma1 = gera_poligono_regular(
        lados=6, largura=LARGURA, altura=ALTURA, rotacao=py5.PI / 6
    )
    forma2 = gera_poligono_regular(
        lados=5, largura=LARGURA, altura=ALTURA, rotacao=py5.PI / 5
    )


def cores_posicao(idx, idy) -> tuple[int, int | bool]:
    """Calcula as cores baseadas na posição da grade."""
    h = (40 * idx % 4) + 40
    s = 100
    b = 100
    preenchimento = py5.color(h, s, b)
    h -= 180
    s -= 100 if (idx + idy) % 2 == 0 else 50
    traco = py5.color(abs(h), abs(s), abs(b))
    return preenchimento, traco


def draw():
    cor_fundo = py5.color(0)
    py5.background(cor_fundo)
    grade = cria_grade_ex(
        largura=py5.width,
        altura=py5.height,
        margem_x=0,
        margem_y=0,
        celula_x=LARGURA,
        celula_y=ALTURA,
        alternada=True,
    )
    if forma1 is not None and forma2 is not None:
        for idx, x, idy, y in grade:
            preenchimento, traco = cores_posicao(idx, idy)
            forma = forma1 if idy % 2 == 0 else forma2
            with py5.push():
                z = -20
                forma.set_fill(preenchimento)
                forma.set_stroke(traco)
                forma.set_stroke_weight(3)
                py5.translate(x, y, z)
                # Desenho da forma
                py5.shape(forma, 0, 0)

    # Credits and go
    canvas.sketch_frame(
        sketch, cor_fundo, "large_transparent_white", "transparent_white"
    )


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


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


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