# tk_flippant_cells_spiral_2.py

# tk_flippant_cells_spiral_2.py

import tkinter as tk
from PIL import Image, ImageTk, ImageDraw, ImageFilter
import math

cell_size = 10
SQ = 600 // cell_size * cell_size
GRID = SQ // cell_size
CENTER = (GRID - 1) / 2

rr = tk.Tk()
rr.geometry('+0+0')

cv = tk.Canvas(rr, width=SQ, height=SQ, bg='black')
cv.pack(side='left')

canvas_image_id = cv.create_image(0, 0, anchor='nw')

MIN_SPEED = 0.00001
MAX_SPEED = 5.0

c = []
loop = range(0, 256, int(MAX_SPEED))
for r in loop:
    for g in loop:
        for b in loop:
            c.append(f"#{r:02X}{g:02X}{b:02X}")

def hex_to_hue(hexcol):
    r = int(hexcol[1:3], 16) / 255.0
    g = int(hexcol[3:5], 16) / 255.0
    b = int(hexcol[5:7], 16) / 255.0
    return sum((r, g, b))

c = sorted(c, key=hex_to_hue)[::-1]
lc = len(c) - 1

img = Image.new('RGB', (SQ, SQ), (0, 0, 0))
draw = ImageDraw.Draw(img)

max_dist = math.hypot(CENTER, CENTER)
order = sorted(
    [(x, y) for y in range(GRID) for x in range(GRID)],
    key=lambda xy: (
        math.hypot(xy[0] - CENTER, xy[1] - CENTER) / max_dist,
        math.atan2(xy[1] - CENTER, xy[0] - CENTER)
    )
)


speeds = [0.0] * (GRID * GRID)
for i, (x, y) in enumerate(order):
    speeds[y * GRID + x] = MIN_SPEED + (MAX_SPEED - MIN_SPEED) * (1.0 - i / ((GRID * GRID) - 1))

cells = [0.0] * (GRID * GRID)

while 1:
    i = 0
    for y in range(GRID):
        for x in range(GRID):
            cells[i] += speeds[i]

            if cells[i] >= lc:
                cells[i] = lc
                speeds[i] *= -1
            elif cells[i] <= 0:
                cells[i] = 0
                speeds[i] *= -1

            col = c[int(cells[i])]
            x0 = x * cell_size
            y0 = y * cell_size

            draw.rectangle(
                (x0, y0, x0 + cell_size, y0 + cell_size),
                fill=col
            )

            i += 1

    img_out = img.filter(ImageFilter.GaussianBlur(radius=50 * 0.2))

    tk_img = ImageTk.PhotoImage(img_out)
    cv.itemconfig(canvas_image_id, image=tk_img)
    cv.tk_img = tk_img

    rr.update_idletasks()
    rr.update()