# Tk_mandelbrot2.py

# Tk_mandelbrot2.py

import math
import os

#frame parameters
ww = 600
hh = 600

try:
    import tkinter as tk
except:
    import Tkinter as tk

root=tk.Tk()
root.title("Tk Mandelbrot")

xc=ww/2
yc=hh/2

canvas=tk.Canvas(root,width=ww,height=hh)
canvas.pack(fill="both",expand=True)

x = -0.65
y = 0
xRange = 3.4
aspectRatio = 4/3

precision = 500

yRange = xRange / aspectRatio
minX = x - xRange / 2
maxX = x + xRange / 2
minY = y - yRange / 2
maxY = y + yRange / 2

def rgb2hex(rgb):
    r,g,b = rgb
    return "#{:02x}{:02x}{:02x}".format(r,g,b)

rainbow='''
#ff0000
#ffa500
#ffff00
#008000
#0000ff
#4b0082
#ee82ee
'''.strip().split()
Lr = len(rainbow)

up = 9.08373867835
zz = 6.86105967195
zz = Lr / (up + zz)

for row in range(hh):
    for col in range(ww):
        x = minX + col * xRange / ww
        y = maxY - row * yRange / hh
        oldX = x
        oldY = y
        for i in range(precision + 1):
            a = x*x - y*y #real component of z^2
            b = 2 * x * y #imaginary component of z^2
            x = a + oldX #real component of new z
            y = b + oldY #imaginary component of new z
            if x*x + y*y > 4:
                break
        if i < precision:
            j = int(abs(oldX-x)+abs(oldY-y))

            try:
                rgb = rainbow[j]
            except:
                rgb = rainbow[-1]
            canvas.create_oval((col,row,col+1,row+1),outline=rgb)
        else:
            canvas.create_oval((col,row,col+1,row+1),outline='black')
        index = row * ww + col + 1
        # print("{} / {}, {}%".format(index, ww * hh, round(index / ww / hh * 100 * 10) / 10))
    canvas.update()