b_life.py

[sourcecode language=”python” wraplines=”false” collapse=”false”]
import argparse
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation

def randomGrid(N):
return np.random.choice([1,0], N*N, p=[0.2, 0.8]).reshape(N, N)

def addGlider(i, j, grid):
glider = np.array([[0,0,1],
[1,0,1],
[0,1,1]])
grid[i:i+3, j:j+3] = glider

def addGosperGliderGun(i, j, grid):
gun = np.zeros((11,38))

gun[5][1] = gun[5][2] = 1
gun[6][1] = gun[6][2] = 1

gun[3][13] = gun[3][14] = 1
gun[4][12] = gun[4][16] = 1
gun[5][11] = gun[5][17] = 1
gun[6][11] = gun[6][15] = gun[6][17] = gun[6][18] = 1
gun[7][11] = gun[7][17] = 1
gun[8][12] = gun[8][16] = 1
gun[9][13] = gun[9][14] = 1

gun[1][25] = 1
gun[2][23] = gun[2][25] = 1
gun[3][21] = gun[3][22] = 1
gun[4][21] = gun[4][22] = 1
gun[5][21] = gun[5][22] = 1
gun[6][23] = gun[6][25] = 1
gun[7][25] = 1

gun[3][35] = gun[3][36] = 1
gun[4][35] = gun[4][36] = 1

grid[i:i+11, j:j+38] = gun

def update(frameNum, img, grid, N):
newGrid = grid.copy()
for i in xrange(N):
for j in xrange(N):
neigh = int(grid[i, (j-1)%N] + grid[i, (j+1)%N] +
grid[(i-1)%N, j] + grid[(i+1)%N, j] +
grid[(i-1)%N, (j-1)%N] + grid[(i-1)%N, (j+1)%N] +
grid[(i+1)%N, (j-1)%N] + grid[(i+1)%N, (j+1)%N])

if grid[i,j]==1:
if neigh3:
newGrid[i,j] = 0
elif neigh == 3:
newGrid[i,j] = 1

img.set_data(newGrid)
grid[:] = newGrid[:]
return img,

def main():
parser = argparse.ArgumentParser(description=”Runs Conway’s Game of Life simulation.”)
parser.add_argument(‘–grid-size’, dest=’N’, required=False)
parser.add_argument(‘–mov-file’, dest=’movfile’, required=False)
parser.add_argument(‘–interval’, dest=’interval’, required=False)
parser.add_argument(‘–glider’, action=’store_true’, required=False)
parser.add_argument(‘–gosper’, action=’store_true’, required=False)
args = parser.parse_args()

N = 100
if args.N and int(args.N) > 8:
N = int(args.N)

updateInterval = 50
if args.interval:
updateInterval = int(args.interval)

grid = np.array([])

if args.glider:
grid = np.zeros(N*N).reshape(N, N)
addGlider(1, 1, grid)
elif args.gosper:
grid = np.zeros(N*N).reshape(N, N)
addGosperGliderGun(10, 10, grid)

else:
grid = randomGrid(N)

fig, ax = plt.subplots()
cmap = plt.cm.get_cmap(‘Greys’)
img = ax.imshow(grid, interpolation=’nearest’, cmap=cmap)
ani = animation.FuncAnimation(fig, update, fargs=(img, grid, N, ),
frames = 10,
interval = updateInterval,
save_count = 50)

if args.movfile:
ani.save(args.movfile, fps=30, extra_args=[‘-vcodec’, ‘libx264’])

plt.show()

if __name__ == ‘__main__’:
main()
[/sourcecode]