add animation for coloring

Author: edabot

README.md

@@ -52,4 +52,4 @@ python3 coloring.py
 
 ![coloring]
 
-[coloring]: ./images/coloring.png
+[coloring]: ./images/colored_and_animated.gif

__pycache__/grid.cpython-36.pyc

@@ -0,0 +1,16 @@
+from colored_grid import ColoredGrid
+from sidewinder import Sidewinder
+from hunt_and_kill import HuntAndKill
+
+height, width = [30, 30]
+
+grid = ColoredGrid(height,width)
+HuntAndKill.mutate(grid, True, 10, 30)
+
+start = grid.grid[int(height/2)][int(width/2)]
+
+grid.distances = start.distances()
+
+grid.maximum = grid.distances.max()[1]
+
+grid.to_svg()

__pycache__/hunt_and_kill.cpython-36.pyc

@@ -116,18 +116,22 @@ def to_png(self, cell_size=20, index = "0"):
                 y1 = cell.row * cell_size + left_offset
                 x2 = (cell.column + 1) * cell_size + top_offset
                 y2 = (cell.row + 1) * cell_size + left_offset
+                color = "#fff"
                 if len(cell.get_links()) > 0:
-                    draw.rectangle((x1 + 1, y1+1 , x2-1, y2-1), fill="#fff")
+                    if self.background_color_for(cell) is not None:
+                        color=self.background_color_for(cell)
+
+                    draw.rectangle((x1 + 1, y1+1 , x2-1, y2-1), fill=color)
                 if not cell.north:
                     draw.line((x1, y1, x2, y1), (0,0,0))
                 if not cell.west:
                     draw.line((x1, y1, x1, y2), (0,0,0))
                 if not cell.linked(cell.east):
                     draw.line((x2, y1, x2, y2), (0,0,0))
                 elif cell.east:
-                    draw.line((x2, y1 + 1, x2, y2 -1), (255,255,255))
+                    draw.line((x2, y1 + 1, x2, y2 -1), fill=color)
                 if not cell.linked(cell.south):
                     draw.line((x1, y2, x2, y2), (0,0,0))
                 elif cell.south:
-                    draw.line((x1 + 1, y2, x2 - 1, y2), (255,255,255))
+                    draw.line((x1 + 1, y2, x2 - 1, y2), fill=color)
             im.save("./exports/maze"+index+".png", "PNG")

__pycache__/make_animation.cpython-36.pyc

@@ -3,8 +3,8 @@
 
 class HuntAndKill:
 
-    def mutate(grid, animation = False):
-        current = grid.random_cell()
+    def mutate(grid, animation = False, cell_size=20, fps=10):
+        start = current = grid.random_cell()
         index = 0
         filenames = []
 
@@ -17,7 +17,9 @@ def mutate(grid, animation = False):
                 current = neighbor
                 index += 1
                 if animation:
-                    grid.to_png(20, str(index))
+                    grid.distances = start.distances()
+                    grid.maximum = grid.distances.max()[1]
+                    grid.to_png(cell_size, str(index))
                     filenames.append("./exports/maze"+str(index)+".png")
             else:
                 current = None
@@ -31,8 +33,10 @@ def mutate(grid, animation = False):
                         current.link(neighbor)
                         index += 1
                         if animation:
-                            grid.to_png(20, str(index))
+                            grid.distances = start.distances()
+                            grid.maximum = grid.distances.max()[1]
+                            grid.to_png(cell_size, str(index))
                             filenames.append("./exports/maze"+str(index)+".png")
                         break
         if animation:
-            MakeAnimation(filenames, 'hunt_and_kill.gif')
+            MakeAnimation(filenames, 'hunt_and_kill.gif', fps)

coloring_animation.py

@@ -1,10 +1,10 @@
 import imageio
 import os
 
-def MakeAnimation(filenames, moviename):
+def MakeAnimation(filenames, moviename, fps=10):
     images = []
     for filename in filenames:
         images.append(imageio.imread(filename))
-    imageio.mimsave('./exports/' + moviename, images)
+    imageio.mimsave('./exports/' + moviename, images, fps=fps)
     for filename in filenames:
         os.remove(filename)