"""2025-08-12
Colorido geométrico XI
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
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)
    largura, altura = LARGURA * 2.2, ALTURA * 2.2
    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 = ((idx * 15) + (idy * 10)) % 360
    s = (idy**2) % 50 + 20
    b = 100
    preenchimento = py5.color(h, s, b)
    h -= 180
    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:
            _, traco = cores_posicao(idx, idy)
            forma = forma1 if idy % 2 == 0 else forma2
            stroke = 4 if idy % 2 == 0 else 5
            with py5.push():
                z = -20
                forma.set_fill(False)
                forma.set_stroke(traco)
                forma.set_stroke_weight(stroke)
                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()