"""2026-10-23
Hope redux 07 (A very plural country)
A grid with Brasil maps.
ericof.com
png
Sketch,py5,CreativeCoding
"""

from random import shuffle
from sketches.utils.draw import canvas
from sketches.utils.draw.cores.paletas import gera_paleta
from sketches.utils.draw.grade import cria_grade_ex
from sketches.utils.helpers import recursos as rc
from sketches.utils.helpers import sketches as helpers

import py5


sketch = helpers.info_for_sketch(__file__, __doc__)

cor_fundo = py5.color(0)
nome_paleta = "Okazz-230905a"
celula_largura = 200
celula_altura = 80
meia_largura = celula_largura / 2
meia_altura = celula_altura / 2
proporcoes = (0.75, 0.5)


def inicializa():
    paleta = gera_paleta(nome_paleta, True)
    grade = []
    celulas = cria_grade_ex(
        largura=helpers.DIMENSOES.external[0],
        altura=helpers.DIMENSOES.external[1],
        margem_x=0,
        margem_y=0,
        celula_x=celula_largura,
        celula_y=celula_altura,
        alternada=True,
    )
    cores = []
    for _ in range(len(celulas)):
        shuffle(paleta)
        fundo, preenchimento = paleta[0], paleta[1]
        cores.append((paleta[0], paleta[1]))
    shuffle(cores)
    for cor, (idx, x, idy, y) in enumerate(celulas):
        proporcao_base = proporcoes[idy % len(proporcoes)]
        proporcao = min(0.95, max(0, py5.random_gaussian(proporcao_base, 0.2)))
        dimensao = min(celula_largura, celula_altura) * proporcao
        fundo, preenchimento = cores[cor]
        if idx % 4 == 0:
            fundo, preenchimento = preenchimento, fundo
        grade.append((x, y, dimensao, fundo, preenchimento))
    return grade


def setup():
    global grade, forma
    py5.size(*helpers.DIMENSOES.external, py5.P3D)
    py5.hint(py5.DISABLE_DEPTH_TEST)
    py5.color_mode(py5.HSB, 360, 100, 100)
    py5.rect_mode(py5.CENTER)
    py5.shape_mode(py5.CENTER)
    forma = rc.carrega_svg_forma("brasil.svg")
    grade = inicializa()


def draw():
    py5.background(cor_fundo)
    forma.set_stroke_join(py5.ROUND)
    with py5.push():
        py5.translate(*helpers.DIMENSOES.pos_interno, -1)
        for xb, yb, dimensao, fundo, preenchimento in grade:
            x, y = xb + meia_largura, yb + meia_altura
            py5.fill(fundo)
            py5.rect(x, y, celula_largura, celula_altura)
            forma.set_stroke(preenchimento)
            forma.set_stroke_weight(3)
            forma.set_fill(preenchimento)
            py5.shape(forma, x, y, dimensao, dimensao)
    # Credits and go
    canvas.sketch_frame(
        sketch,
        cor_fundo,
        "large_transparent_white",
        "transparent_white",
        version=2,
    )


def key_pressed():
    global grade
    key = py5.key
    if key == " ":
        save_and_close()
    elif key == "r":
        grade = inicializa()


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


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