"""2023-12-29"""
from helpers import HEIGHT
from helpers import save_image
from helpers import WIDTH
from helpers import write_legend
from pathlib import Path

import py5


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

PASSO = 15
LIMITE = 500


def setup():
    py5.size(WIDTH, HEIGHT, py5.P3D)
    py5.background(0)
    py5.color_mode(py5.HSB, 360, 100, 100)
    quadrantes = [
        ((0, 0), (0, LIMITE + 1, PASSO), (LIMITE, -1, -PASSO)),
        ((WIDTH, 0), (0, -LIMITE - 1, -PASSO), (LIMITE, -1, -PASSO)),
        ((WIDTH, HEIGHT), (0, -LIMITE - 1, -PASSO), (-LIMITE, 1, PASSO)),
        ((0, HEIGHT), (0, LIMITE + 1, PASSO), (-LIMITE, 1, PASSO)),
    ]
    for center, qx, qy in quadrantes:
        with py5.push_matrix():
            py5.translate(*center, 0)
            x0_ = [x0 for x0 in range(*qx)]
            y1_ = [y1 for y1 in range(*qy)]
            pontos = list(zip(x0_, y1_))
            total = len(pontos)
            for idx, (x0, y1) in enumerate(pontos):
                x1 = 0
                y0 = 0
                py5.stroke_weight((total - idx) // 10 + 1)
                py5.stroke(160 + (idx * 3), 100, 100)
                py5.line(x0, y0, x1, y1)
    write_legend([py5.color(360, 0, 100)], IMG_NAME)


def key_pressed():
    key = py5.key
    if key == " ":
        save_and_close()


def save_and_close():
    py5.no_loop()
    save_image(IMG_NAME, "png")
    py5.exit_sketch()


py5.run_sketch()