# tk_rgb_spiral_zoom.py

# tk_rgb_spiral_zoom.py

# most colorful

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

ww = 640
hh = 640

root = tk.Tk()
root.title("tk_rgb_spiral_zoom")
root.geometry("%dx%d+0+0"%(ww,hh))
canvas = tk.Canvas(root, bg='white', width=ww, height=hh)
canvas.pack()

cx, cy = ww//2, hh//2
img = Image.new('RGB', (ww, hh), "white")

rgb = []
def create_rgb():
    t = list(range(0, 256, 15))
    rrr = t[:]
    ggg = t[:]
    bbb = t[:]
    r = g = b = 0
    while 1:
        for r in rrr:
            for g in ggg:
                for b in bbb:
                    if (r,g,b) not in rgb:
                        rgb.append((r,g,b))
                    else:
                        return
                if bbb.index(b) == len(bbb) - 1:
                    bbb = bbb[::-1]
            if ggg.index(g) == len(ggg) - 1:
                ggg = ggg[::-1]
        rrr = rrr[::-1]
create_rgb()
L = len(rgb)

cXY = []
t = ww//4
for y in range(t,t*3):
    for x in range(t,t*3):
        distance = ((cx-x)**2+(cy-y)**2)**0.5
        if distance < ww//4:
            xy2 = math.atan2(x-cx,y-cy)
            cXY.append(((int(distance), xy2),x,y))
cXY.sort()
cXY = [(x,y) for z,x,y in cXY]
LXY = len(cXY)

blur_radius = 0.0005 * min(img.size)
scale = 0
zoom = 0.01

def draw():
    x, y = cXY[i]
    r, g, b = rgb.pop(0)
    rgb.append((r, g, b))
    rrr, ggg, bbb = img.getpixel((x,y))
    r = rrr + max(-clamp, min(clamp, r - rrr))
    g = ggg + max(-clamp, min(clamp, g - ggg))
    b = bbb + max(-clamp, min(clamp, b - bbb))
    img.putpixel((x, y+1), (r, g, b))


clamp = 29
while True:
    if scale%20 < 20 and scale%20 + zoom > 20:
        for i in range(0, LXY, 5):
            draw()
    else:
        for i in range(0, LXY, 40):
            draw()
    cXY.append(cXY.pop(int(scale)%L))

    xt, yt = int(ww+scale), int(hh+scale)
    img = img.resize((xt, yt), resample=Image.LANCZOS)
    # img = img.filter(ImageFilter.GaussianBlur(radius=blur_radius))

    # crop parameters to zoom into the center
    img = img.crop((scale/2, scale/2, xt-scale/2, yt-scale/2))
    scale += zoom

    tkimg = ImageTk.PhotoImage(img)
    canvas.create_image((cx, cy), image=tkimg)
    canvas.update()