import kandinsky
WIDTH, HEIGHT = 320, 222
C = complex(-0.8, 0.156)  # paramètre de Julia (change pour d'autres formes)
MAX_ITER = 50             # nombre max d'itérations
ESCAPE_RADIUS = 4.0       # seuil de divergence

def julia(z):
    """Retourne le nombre d'itérations avant divergence."""
    i = 0
    while i < MAX_ITER and (z.real*z.real + z.imag*z.imag) <= ESCAPE_RADIUS:
        z = z*z + C
        i += 1
    return i

def iter_to_color(i):
    """Convertit une itération en couleur RGB."""
    if i >= MAX_ITER:
        return (0, 0, 0)  # noir dans l'ensemble
    t = i / MAX_ITER
    r = int(255 * t)
    g = int(128 * (1 - t))
    b = int(255 * (1 - t))
    return (r, g, b)

def pixel_to_complex(px, py):
    """Convertit un pixel écran en coordonnée complexe (ajustée au ratio)."""
    zx = 3.0 * (px / WIDTH) - 1.5      # intervalle [-1.5, +1.5]
    zy = 2.0 * (py / HEIGHT) - 1.0     # intervalle [-1.0, +1.0]
    return complex(zx, -zy)            # inversion verticale

def main():
    for y in range(HEIGHT):
        for x in range(WIDTH):
            z = pixel_to_complex(x, y)
            i = julia(z)
            kandinsky.set_pixel(x, y, iter_to_color(i))

main()