"""2025-07-12
Hexágonos e gradientes II
Exercício de criação de hexágonos com gradientes de cores.
ericof.com
png
Sketch,py5,CreativeCoding
"""

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

import numpy as np
import py5


sketch = helpers.info_for_sketch(__file__, __doc__)


def setup():
    py5.size(*helpers.DIMENSOES.external, py5.P3D)
    cor_fundo = py5.color(0, 98)
    py5.background(cor_fundo)
    py5.rect_mode(py5.CORNER)
    py5.shape_mode(py5.CENTER)
    py5.color_mode(py5.HSB, 360, 100, 100)
    passo = 60
    traco = 6
    forma_largura = passo * 2.5
    forma_altura = passo * 2.5
    lados = 6
    rotacao = np.pi / lados
    forma = gera_poligono_regular(
        lados=lados,
        largura=forma_largura,
        altura=forma_altura,
        rotacao=rotacao,
    )
    buffer = int(forma_largura * 2)
    largura = helpers.DIMENSOES.internal[0] + buffer
    altura = helpers.DIMENSOES.internal[1] + buffer
    x0 = helpers.DIMENSOES.pos_interno[0] - buffer // 2
    y0 = helpers.DIMENSOES.pos_interno[1] - buffer // 2
    xx, yy = np.meshgrid(
        np.arange(0, largura + 1, passo), np.arange(0, altura + 1, passo), indexing="ij"
    )
    pontos = np.column_stack((xx.ravel(), yy.ravel()))
    with py5.push_matrix():
        py5.translate(x0, y0, -5)
        py5.blend_mode(py5.BLEND)
        forma.set_fill(False)
        for x, y in pontos:
            h = py5.random_int(0, 360)
            s = py5.random_int(80, 100)
            b = py5.random_int(80, 100)
            cores = []
            for _ in range(lados):
                cor = py5.color(h, s, b)
                cores.append(cor)
                h = py5.random_int(0, 360)
                s *= 0.85
                b *= 0.9
            forma.set_stroke_weight(traco)
            forma.set_strokes(cores)
            py5.shape(forma, x, y, forma_largura, forma_altura)

    # 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()