python gifmaze_GIFMaze：制作各种迷宫生成额迷宫解决算法的 GIF 动画

GIFMaze

What's cool

The following image illustrates two graph algorithms: it firstly used Wilson's algorithm to generate an uniform spanning tree of the 2-D grid (the result is a perfectly random sampled maze) and then used breadth first search to solve this maze. It contains roughly 1520 frames, but the file size is only 670KB!

Installation

You can install either via pip:

pip install gifmaze

or via git

git clone https://github.com/neozhaoliang/gifmaze gifmaze && cd gifmaze && python setup.py install

Why you need this lib

Q: I'm a fan of Python and also a fan of maze generation and maze solving algorithms. I have always been jealous of other people's awesome animations of this kind (like here, here and here), how could I make my own animations with python and share them with other people? (I know tkinter, pyglet and pyqt are nice GUIs but they cannot be published directly to the web ...)

A: Now you have this lib gifmaze which can help you make even more awesome GIF animations! It has some very nice features:

It's written in pure Python, no third-party libs/softwares are required, only built-in modules! (If you want to embed the animation into another image, then PIL is required, which is not built-in but comes with all Python distributions, that's all!)

update: I have added the tqdm module to show the process bar, which can be installed via pip install tqdm.

It runs very fast and generates optimized GIF files in a few seconds. Usually the output file contains more than one thousand frames but the file size is only around a few hundreds of KBs.

You can make GIF animations of all kinds of maze generation and maze solving algorithms on the 2-D grid.

It's fully commented, fully exampled and fully documented!

A tutorial on this lib

Q: Cool! Is it easy to use?

A: Yes! Let me show you with an example:

The first thing is to declare a GIFSurface object which specifies the size of the image and on which the animation is drawn:

import gifmaze as gm

surface = gm.GIFSurface(600, 400, bg_color=0)

Here bg_color=0 means the 0-th color in the global color table is used as the background color. You may define the global color table at any time except must before you save the image and it must contain at least one (r,g,b) triple. Let's say it's

surface.set_palette([0, 0, 0, 255, 255, 255])

So the colors available in the image are black and white. This surface is the "bottom layer" object that handles the raw information of the resulting GIF image, i.e. size, palette, loop, and background color.

The second task is to declare a Maze object which specifies the size and position of the maze and on which the algorithm runs:

maze = gm.Maze(149, 99, mask=None).scale(4).translate((2, 2))

Here the size of the maze is 149x99 but is scaled by 4 and translated right and downward by (2, 2), so the top-left pixel of the maze is at (2, 2) and the maze occupies 596x396 pixels. This maze is the "top layer" object on which the algorithm runs, it does not know any information about the GIF image.

The last thing is to define an animation environment to run the algorithm:

from gifmaze.algorithms import random_dfs

anim = gm.Animation(surface)

anim.pause(100)

anim.run(random_dfs, maze, speed=15, delay=5, trans_index=None, cmap={0: 0, 1: 1}, mcl=2, start=(0, 0))

anim.pause(300)

So anim is the "middle layer" object which calls the renderer to draw the maze onto the surface. The parameters are:

speed: controls the speed of the animation.

delay: controls the delay between successive frames.

trans_index: the transparent color index.

cmap: controls how the cells are mapped to colors. Here cmap={0: 0, 1: 1} means the cells of state 0 (the walls) are colored with the 0-th color (black), cells of state 1 (the tree) are colored with the 1-th color (white).

mcl: the minimum code length for initializing the LZW compression.

Finally we save the image and finish the animation by

surface.save('random_dfs.gif')

surface.close()

The result is shown below (~470 frames, ~65KB):

For more usage please see the examples/ folder in the github repository. To implement your own maze generation/solving algorithms you may refer to the examples in algorithms.py.

How it works

The most tricky part when implementing this program is that, even for a 2-D grid of a morderate size, there are usually more than one thousand frames in the animation (it's almost always this case when animating Wilson's uniform spanning tree algorithm) and packing such many frames into a GIF image is definitely a formiddable task. So it's quite surprising that our program costs only a few seconds and can produce highly optimized images. The key points are:

Find a way to reduce the file size. This is accomplished by maintaining a rectangular region that holds the size and position of current frame and allowing variable mimimum code length for the LZW compression.

Write the frames to the file as quickly as possible. This is accomplished by writing them to a in-memory io file first and then flush the data to disk all at once.

References

What's in a gif. The most helpful and may be the only resource you will need for learning the GIF89a specification.

Maze generation algorithms. A useful webpage for learning various maze generation algorithms.