import pygame
from math import *
step=5
itr=50
data=[]

colours=[]
for n in range(itr):
	blue=int(sin(((2*pi)/itr)*n)*255)
	green=int(sin(((2*pi)/itr)*n-(pi/2))*255)
	red=int(sin(((2*pi)/itr)*n-pi)*255)
	if red<0:
		red=0
	if blue<0:
		blue=0
	if green<0:
		green=0
	colours.append((red,green,blue))

while len(colours)<50:
	colours.append((0,0,0))


for x in range(-1700,700+step,step):
   row_data=[]
   for y in range(-1200,1200+step,step):
     c=(float(x)/float(1000))+1j*(float(y)/float(1000))
     z=c
     for l in range(1,itr):
       z=(z**2)+c;
       if abs(z)>2:
         break
     row_data.append(l)
   data.append(row_data)

pygame.init()
screen=pygame.display.set_mode((480,480),pygame.DOUBLEBUF)

step2=step/5
for x in range(0,480,step2):
	for y in range(0,480,step2):
		screen.set_at((x,y),colours[data[x][y]])

pygame.display.flip()

running=True

while running:
    for event in pygame.event.get():
	if event.type==pygame.QUIT:
            running=False

pygame.display.quit()