Coloured Mandelbrot set with generativepy

By Martin McBride, 2021-06-12
Tags: mandelbrot set
Categories: generativepy generative art

This article has been moved to my blog. Please refer to that article as it might be more up to date.

Following on from the previous article on the Mandelbrot set, we will now see how to create a colour image. it is worth reading the previous article first, as we will be building on the code in that article.

Adding some colour

The black and white Mandelbrot is a little stark, and while it might be mathematically very interesting it lacks a little visual appeal.

There is something we can do fairly easily to rectify this. The calc function doesn't just return a true or false value. It actually returns:

  • 0 if the point is inside the set.
  • Otherwise, it returns a i + 1, where i is the count of how many times the loop executes before the value of x*x + y*y exceeds 4 (the point of no return).

As you might expect, points that are a long way outside the boundary of the set tend to escape very quickly. Points that are very close to the boundary take longer to escape. So we can create a bit of extra interest in our image by setting the colour of the pixels outside the image according to the value returned by calc.

Colorising the values

In a different article we saw how to colorise a Tinkerbell fractal. We will use a similar technique here.

In summary:

  • We will use make_nparray_data to write the count values for each pixel to an integer NumPy array dimensions height x width x 1.
  • Use a colorise function to convert the counts array to a height x width x 3 array (an RGB vale for each pixel).
  • Use save_nparray_image to save the RGB data as an image.

Here is the result:

The final code

Here is the final code for the colour Mandelbrot:

from generativepy.bitmap import Scaler
from generativepy.nparray import make_nparray_data, make_npcolormap, save_nparray, load_nparray, save_nparray_image, apply_npcolormap
from generativepy.color import Color
import numpy as np


def calc(c1, c2):
    x = y = 0
    for i in range(MAX_COUNT):
        x, y = x*x - y*y + c1, 2*x*y + c2
        if x*x + y*y > 4:
            return i + 1
    return 0

def paint(image, pixel_width, pixel_height, frame_no, frame_count):
    scaler = Scaler(pixel_width, pixel_height, width=3, startx=-2, starty=-1.5)

    for px in range(pixel_width):
        for py in range(pixel_height):
            x, y = scaler.device_to_user(px, py)
            count = calc(x, y)
            image[py, px] = count

def colorise(counts):
    counts = np.reshape(counts, (counts.shape[0], counts.shape[1]))

    colormap = make_npcolormap(MAX_COUNT+1,
                               [Color('black'), Color('darkblue'), Color('green'), Color('cyan'), Color('white')],
                               [8, 8, 32, 128])

    outarray = np.zeros((counts.shape[0], counts.shape[1], 3), dtype=np.uint8)
    apply_npcolormap(outarray, counts, colormap)
    return outarray

data = make_nparray_data(paint, 600, 600, channels=1)

save_nparray("/tmp/temp.dat", data)
data = load_nparray("/tmp/temp.dat")

frame = colorise(data)

save_nparray_image('mandelbrot.png', frame)

Here is what the colorisefunction does:

  • Reshape our counts array from (height, width, 1) to (height, width).
  • Create a colormap with MAX_COUNT+1 elements.
  • Create an output array that is height by width by 3, to hold RGB image data. The array is of type uint8, that is unsigned byte value. We call apply_npcolormap to convert the normalised count array into an RGB image array.

The colormap goes from black to dark blue to green to cyan to white. However, we have also supplied a bands array [8, 8, 32, 128]. This specifies the relative size of each band. This means that black-darkblue and darkblue-green bands are very small, but the green-cyan band is bigger, and the cyan-white band is even bigger. This basically equalises the transitions over the image. so that the less interesting areas well away from the fractal boundary have subtle colour changes, whereas the exiting bit of the image is enhanced by a rapid green-cyan-white change.

You can easily experiment with other colour schemes.

Things to try

Here are a few things to try with the basic code.

Experiment with other colour schemes, you can use any scheme you like by modifying the colorise function. Remember that the outer areas of the image are mainly quite low values, whereas most of the higher values are concentrated closer to the edge of the set.

Create a super high resolution image by increasing the pixel dimensions you pass into make_bitmap. The scaling will automatically ensure that the same region of the image is visible. Be aware that every time you double the width and height the image will take 4 times longer to render.

Try zooming in on some other areas of the image by changing the width, startx, and starty you pass into the Scaler.

See also

If you found this article useful, you might be interested in the book NumPy Recipes or other books by the same author.

Join the PythonInformer Newsletter

Sign up using this form to receive an email when new content is added:

Popular tags

2d arrays abstract data type alignment and angle animation arc array arrays bar chart bar style behavioural pattern bezier curve built-in function callable object chain circle classes clipping close closure cmyk colour combinations comparison operator comprehension context context manager conversion count creational pattern data science data types decorator design pattern device space dictionary drawing duck typing efficiency ellipse else encryption enumerate fill filter font font style for loop formula function function composition function plot functools game development generativepy tutorial generator geometry gif global variable gradient greyscale higher order function hsl html image image processing imagesurface immutable object in operator index inner function input installing iter iterable iterator itertools join l system lambda function latex len lerp line line plot line style linear gradient linspace list list comprehension logical operator lru_cache magic method mandelbrot mandelbrot set map marker style matplotlib monad mutability named parameter numeric python numpy object open operator optimisation optional parameter or pandas partial application path pattern permutations pie chart pil pillow polygon pong positional parameter print product programming paradigms programming techniques pure function python standard library radial gradient range recipes rectangle recursion reduce regular polygon repeat rgb rotation roundrect scaling scatter plot scipy sector segment sequence setup shape singleton slice slicing sound spirograph sprite square str stream string stroke structural pattern subpath symmetric encryption template tex text text metrics tinkerbell fractal transform translation transparency triangle truthy value tuple turtle unpacking user space vectorisation webserver website while loop zip zip_longest