# Tk_rainbow_sq_spiral.py

# Tk_rainbow_sq_spiral.py

from Tkinter import *
from math import sqrt, atan2
from random import shuffle as rs

def oRGB(rgb): # pass
    r,g,b = rgb
    return "#%02x%02x%02x" % (r,g,b)

ww = 600
hh = 600
root = Tk()
root.title("# Tk_rainbow_sq_spiral")
root.geometry("%dx%d+0+0"%(ww,hh))
canvas = Canvas(root, width=ww, height=hh)
canvas.grid()

def switch(state):
    return {
        0 : (x + stepSize, y),
        1 : (x, y - stepSize),
        2 : (x - stepSize, y),
        3 : (x, y + stepSize),
    }[state]

# set up spiral
step = 0
state = 0
numSteps = 1
turnCounter = 1
stepSize = 5
cols = ww / stepSize
rows = hh / stepSize
totalSteps = cols * rows
x = ww / 2
y = hh / 2

xy = []
while 1:
    xy.append((x,y))

    # Change state
    if step % numSteps == 0:
        state = (state + 1) % 4
        turnCounter += 1
        if turnCounter % 2 == 0:
            numSteps += 1
    step += 1
    x, y = switch(state)

    # Are we done?
    if step > totalSteps:
        break

COLORS = '''
red
orange
yellow
green
blue
purple
'''.strip().splitlines()
t = stepSize
Lc = len(COLORS)

color = -1
loop = 0
for x,y in xy:
    if not loop:
        color += 1
        loop = color+1
    canvas.create_rectangle((x,y,x+t,y+t), fill=COLORS[int(color/3+(loop/3))%6], outline='')
    canvas.update()
    loop -= 1