"""2025-03-05
Rotatio XVII
Rotações de retângulos com paletas diversas
png
Sketch,py5,CreativeCoding
"""

from collections import deque

import py5

from utils import helpers
from utils.draw import cria_grade_ex, gera_paleta

sketch = helpers.info_for_sketch(__file__, __doc__)


def rotatio(
    paleta: list | deque,
    rot: int,
    rot_passo: int,
    diametro: float,
    larg: float,
    alt: float,
    xc: float,
    yc: float,
    razao_larg_alt: float,
    voltas: float,
    passo: float,
) -> py5.Py5Graphics:
    pg = py5.create_graphics(larg, alt, py5.P3D)
    pg.rect_mode(py5.CENTER)
    pg.ellipse_mode(py5.CENTER)
    direcao = 1
    tamanho_paleta = len(paleta)
    angulo_final = voltas * 360
    with pg.begin_draw():
        pg.no_stroke()
        with pg.push_matrix():
            pg.translate(xc, yc, -30)
            idx = 0
            while diametro > 5:
                if idx and idx % tamanho_paleta == 0:
                    direcao *= -1
                    rot += rot_passo
                angulo = 0
                cor = paleta[tamanho_paleta // 2]
                paleta.rotate(direcao)
                pg.fill(cor)
                while angulo < angulo_final:
                    pg.rotate_z(py5.radians(rot))
                    altura = diametro * razao_larg_alt
                    pg.rect(0, 0, diametro, altura)
                    angulo += abs(rot)
                idx += 1
                diametro *= passo
    return pg


def setup():
    py5.size(helpers.LARGURA, helpers.ALTURA, py5.P3D)
    py5.background(0)
    py5.color_mode(py5.HSB, 360, 100, 100)
    py5.rect_mode(py5.CORNER)
    py5.blend_mode(py5.BLEND)
    alt = 100
    larg = 100
    grade = cria_grade_ex(py5.width, py5.height, 0, 0, alt, larg, False)
    paletas = deque(
        [
            gera_paleta("sunset-ocean", True),
            gera_paleta("laranja", True),
        ]
    )
    paletas.rotate(py5.random_int(len(paletas)))
    paleta = paletas[0]
    rot = py5.random_gaussian(-30, 5)
    rot_passo = py5.random_gaussian(0, 5)
    razao_larg_alt = py5.random(0.4, 0.9)
    voltas = py5.random_int(3, 8)
    passo = py5.random(0.75, 0.90)
    diametro = 800
    idt = 0
    paleta = paletas[0]
    for idx, x, idy, y in grade:
        if idx == 0:
            paletas.rotate()
        paleta = paletas[idx % 2]
        xf = x + larg
        yf = y + alt
        params = [
            paleta,
            rot,
            rot_passo,
            diametro,
            py5.width,
            py5.height,
            400,
            400,
            razao_larg_alt,
            voltas,
            passo,
        ]
        with py5.push_style():
            img_array = rotatio(*params).get_np_pixels()[y:yf, x:xf]
            image = py5.create_image_from_numpy(img_array)
            py5.image(image, x, y, larg, alt)
        with py5.push_style():
            py5.stroke("#777")
            py5.stroke_weight(3)
            py5.no_fill()
            py5.rect(x, y, larg, alt)
        idt += 1
    helpers.write_legend(sketch=sketch, frame="#000", cor="#FFF")


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