juliaset.py

import math

class JuliaSet(object):
    def __init__(self, c, n = 100):
        self.c = c
        self.n = n
        self._d = 0.001
        self._complexplane = []
        self.set = []

    def juliamap(self, z):
        return (z ** 2) + self.c

    def iterate(self, z):
        m = 0
        while True:
            z = self.juliamap(z)
            m += 1
            if  abs(z) > 2:
                return m
            if m >= self.n:
                return 0

    def makeplane(self):
        plane = [i * self._d for i in range(int(-2 / self._d), int(2 / self._d), 1)]
        self._complexplane=[complex(x,y) for x in plane for y in plane]

    def set_spacing(self, d):
        self._d = d
        self.makeplane()

    def generate(self):
        self.set = [self.iterate(i) for i in self._complexplane]
        return self.set