mandelbrot.py

# mandelbrot.py
# posdouv mandelbrot calculator
import Image

x = 0.0
y = 0.0
zoom = .01
width = 200
height = 200
dwell =  20 
base_x = x - (width / 2) * zoom
base_y = y - (height / 2) * zoom

def job_init(args):
    global x, y, zoom, width, height, dwell
    global base_x, base_y, img
    if len(args) == 6:
        x, y, zoom, width, height, dwell = args
    base_x = x - (width / 2) * zoom
    base_y = y - (height / 2) * zoom
    img = Image.new('RGB', (width, height))
    return 0

def job_get_globals():
    global x, y, zoom, width, height, dwell
    job_globals = (x, y, zoom, width, height, dwell, base_x, base_y)
    return job_globals
    
def job_get_arg(task_num): 
    global width, height, base_x, base_y, zoom, dwell   
    if (task_num >= width * height): return None
    
    return task_num
    
def job_add_result(task_num, result):
    global width, height, img
    if (task_num >= width * height): return
    
    x = task_num % width
    y = task_num / width
    
    img.putpixel((x, y), result)

def job_notify_failure(task_num): pass

def job_finish():
    img.save('fractal.png')

import cmath

def job_worker(arg):
    global job_globals
    x, y, zoom, width, height, dwell, base_x, base_y = job_globals
    task_num = arg

    x = base_x + (task_num % width) * zoom
    y = base_y + (task_num / width) * zoom
    
    c = complex(x, y)
    z = complex(0, 0)
    
    iter = 0
    while iter < dwell:
        z = z * z + c
        if (z.conjugate() * z).real > 4.0:
            break
        iter = iter + 1
        
    if iter == dwell:
        result = (0, 0, 0)
    else: 
        r = 4*(iter % 64)
        g = 4*((iter >> 5) % 64)
        b = 4*((iter >> 10) % 64)
        result = (r, g, b)
        #result = (255 - iter % 256, ((iter >> 4) % 256) * 15, 0)
    
    return result