"""2024-05-29
Grade pastel 4
Exercício de animação de grade com formas geométricas concêntricas
png
Sketch,py5,CreativeCoding
"""

from dataclasses import dataclass

import py5

from utils import helpers

sketch = helpers.info_for_sketch(__file__, __doc__)


@dataclass
class Celula:
    x: int
    y: int
    pg: py5.Py5Graphics
    largura: int
    altura: int
    formas: list[
        tuple[int, int, int],
        py5.RECT | py5.ELLIPSE,
        tuple[float, float],
        tuple[int, int],
    ]


GRADE: list[Celula] = []
PASSO: int = 100
QUADRADO: int = 100
DIFF_PASSO_QUADRADO: int = PASSO - QUADRADO
MAX_INTERNO: int = 10
PASSO_INTERNO: float = QUADRADO / MAX_INTERNO


def setup():
    py5.size(helpers.LARGURA, helpers.ALTURA, py5.P3D)
    py5.background(0)
    formas = [py5.ELLIPSE]
    pontos_base = [0, QUADRADO / 4, QUADRADO / 3, QUADRADO / 2, QUADRADO]
    for x in range(100, 700, PASSO):
        for y in range(100, 700, PASSO):
            pg = py5.create_graphics(QUADRADO, QUADRADO)
            formas_ = []
            for _ in range(py5.random_int(2, MAX_INTERNO)):
                cor = [
                    py5.random_int(100, 240),
                    py5.random_int(80, 220),
                    py5.random_int(60, 220),
                ]
                base = (py5.random_choice(pontos_base), py5.random_choice(pontos_base))
                forma = py5.random_choice(formas)
                forma_mult = [
                    1,
                    1,
                ]
                formas_.append((cor, forma, forma_mult, base))
            celula = Celula(x, y, pg, QUADRADO, QUADRADO, formas_)
            GRADE.append(celula)


def draw():
    py5.background(0)
    f = py5.frame_count
    for idx, celula in enumerate(GRADE):
        pg = celula.pg
        pg.begin_draw()
        pg.shape_mode(py5.CENTER)
        pg.background(228, 221, 199)
        for cor, forma, (b_mult_x, b_mult_y), (xb, yb) in celula.formas:
            tamanho = idx * PASSO_INTERNO
            mult_x = py5.cos(py5.radians(f * 2) + idx * 2)
            mult_y = py5.sin(py5.radians(f * 2) + idx * 2)
            tamanho_x = tamanho * b_mult_x * mult_x
            tamanho_y = tamanho * b_mult_y * mult_y
            forma_ = pg.create_shape(forma, xb, yb, xb + tamanho_x, yb + tamanho_y)
            forma_.set_fill(py5.color(*cor))
            forma_.set_stroke(py5.color(*cor))
            forma_.rotate_x(py5.radians(f * 15))
            pg.shape(forma_, xb, yb)
        pg.end_draw()
        x = celula.x + (DIFF_PASSO_QUADRADO / 2)
        y = celula.y + (DIFF_PASSO_QUADRADO / 2)
        with py5.push_style():
            pg.fill(188, 181, 119)
            pg.stroke("#000")
            py5.square(x - 4, y - 4, QUADRADO + 8)
        py5.image(pg, x, y)
    helpers.write_legend(sketch=sketch, frame="#000")


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