"""2023-05-13"""
from helpers import HEIGHT
from helpers import save_image
from helpers import WIDTH
from helpers import write_legend
from pathlib import Path
from random import shuffle

import numpy as np
import py5


IMG_NAME = Path(__file__).name.replace(".py", "")


def calcula_retangulo(lado_x: int, lado_y: int):
    pontos = []
    step = lado_x * 0.04
    idx = 0
    while idx < lado_x:
        pontos.append((idx, 0))
        pontos.append((0, idx))
        step = step * 1.04
        idx += step
    while idx < 1.5 * lado_x:
        pontos.append((lado_x, lado_x - idx))
        pontos.append((lado_x - idx, lado_x))
        step = step * 1.2
        idx += step
    return pontos


def desenha_poligono(pontos, stroke_weight, stroke) -> py5.SHAPE:
    s = py5.create_shape()
    with s.begin_shape():
        py5.stroke(stroke)
        py5.stroke_weight(stroke_weight)
        for x, y in pontos:
            s.vertex(x, y)
    return s


def settings():
    py5.size(WIDTH, HEIGHT)


def setup():
    py5.background(py5.color(0))
    py5.color_mode(py5.HSB, 360, 100, 100, 100)
    py5.no_fill()
    py5.blend_mode(py5.LIGHTEST)
    y0 = np.arange(-200, HEIGHT + 180, 80)
    x0 = np.arange(-200, WIDTH + 180, 80)
    grade = [(x, y) for x in x0 for y in y0]
    shuffle(grade)
    with py5.push_matrix():
        py5.translate(WIDTH / 2, HEIGHT / 2)
        py5.rotate(py5.radians(30))
        py5.translate(-WIDTH / 2, -HEIGHT / 2)
        for x, y in grade:
            lado_x = py5.random_int(20, 40)
            lado_y = py5.random_int(20, 40)
            pontos = calcula_retangulo(lado_x, lado_y)
            cor = py5.color(py5.random_int(0, 360), 100, 100)
            poligono = desenha_poligono(pontos, 2, cor)
            poligono.rotate(py5.radians(py5.random(40, 50)))
            py5.shape(poligono, x, y, lado_x, lado_y)

    write_legend([py5.color(100)], IMG_NAME)


def key_pressed():
    key = py5.key
    if key == " ":
        py5.no_loop()
        save_image(IMG_NAME, "png")
        py5.exit_sketch()


py5.run_sketch()