# tk_native_grayscale_ani.py

# tk_native_grayscale_ani.py

from tkinter import *
from PIL import Image, ImageTk, ImageFilter, ImageDraw
from math import sin, cos, pi
import random

ww = 384
hh = 384
cx, cy = ww//2, hh//2

root = Tk()
root.title("native_grayscale_ani")
root.geometry("%dx%d+0+0"%(ww,hh))

def rgb2hex(r,g,b):
    return '#%02X%02X%02X'%(r,g,b)

rgb = range(0, 256, 50)
colors = [rgb2hex(r, g, b) for r in rgb for g in rgb for b in rgb]

img = Image.new('RGB', (ww, hh), (0, 0, 0))
draw = {}
for k in (0, 1, 2, 3):
    draw[k] = ImageDraw.Draw(img)
blur_radius = 0.04 * min(img.size)

# set the minimum and maximum colors of the water
black = (0, 0, 0)
gray = (128, 128, 128)
white = (255, 255, 255)

# create a custom color palette for interpolation
palette = []
for i in range(0,256):
    j = i % 32
    if j > 20:
        color = white
    elif j > 18:
        color = gray
    else:
        color = black
    palette.append(color)
Lc = len(palette)

canvas = Canvas(root, width=ww, height=hh, bg='white')
canvas.pack(side=LEFT, fill=BOTH, expand=True)

def display():
    tkimg = ImageTk.PhotoImage(draw['source'])
    canvas.create_image((cx, cy), image=tkimg)
    canvas.update()

sz = 16
c = 0
xy = range(0, 512, sz)
for y in xy:
    c = (c + 5) % 11
    for x in xy:
        color = colors.pop(c)
        colors.append(color)
        draw[1].rectangle((x, y, x+sz, y+sz), fill=color, outline=color)
        c = (c + 5) % 11
draw['source'] = img.filter(ImageFilter.GaussianBlur(radius=blur_radius))
source = {}
target = {}
for y in range(hh):
    for x in range(ww):
        source[x,y] = draw['source'].getpixel((x, y))

def wave_transition(x, y, alpha):
    source_color = source[x, y]
    target_color = draw['target'].getpixel((x, y))

    r = int((1 - alpha) * source_color[0] + alpha * target_color[0])
    g = int((1 - alpha) * source_color[1] + alpha * target_color[1])
    b = int((1 - alpha) * source_color[2] + alpha * target_color[2])

    # apply the custom color palette
    color = int(r + g + b) % Lc
    draw['source'].putpixel((x, y), palette[color])


def waves():
    alpha = 0.1
    while alpha < 0.8:
        for y in range(hh):
            for x in range(ww):
                wave_transition(x, y, alpha)
        alpha += 0.1
        display()

    for y in range(hh):
        for x in range(ww):
            target_color = draw['target'].getpixel((x, y))
            wave_transition(x, y, alpha=0.9)
            source[x, y] = target_color # fastest solution I can think of
    display()

while 1:
    for y in xy:
        c = (c + 1) % 11
        for x in xy:
            color = colors.pop(c)
            colors.append(color)
            draw[0].rectangle((x, y, x+sz, y+sz), fill=color, outline=color)
            c = (c + 1) % 11
    draw['target'] = img.filter(ImageFilter.GaussianBlur(radius=blur_radius))

    waves()