#!/usr/bin/env python

# -*- coding: utf-8 -*-

# -----------------------------------------------------------------------------

import numpy as np

import matplotlib

import matplotlib.pyplot as plt

def iterate(Z):

# Count neighbours

N = (Z[0:-2,0:-2] + Z[0:-2,1:-1] + Z[0:-2,2:] +

Z[1:-1,0:-2] + Z[1:-1,2:] +

Z[2: ,0:-2] + Z[2: ,1:-1] + Z[2: ,2:])

# Apply rules

birth = (N==3) & (Z[1:-1,1:-1]==0)

survive = ((N==2) | (N==3)) & (Z[1:-1,1:-1]==1)

Z[...] = 0

Z[1:-1,1:-1][birth | survive] = 1

return Z

Z = np.random.uniform(0,1,(256,512))

Z = (Z < .2).astype(int)

size = np.array(Z.shape)

dpi = 72.0

figsize= size[1]/float(dpi),size[0]/float(dpi)

fig = plt.figure(figsize=figsize, dpi=dpi, facecolor="white")

fig.add_axes([0.0, 0.0, 1.0, 1.0], frameon=False)

plt.imshow(Z,interpolation='nearest', cmap=plt.cm.Purples)

plt.xticks([]), plt.yticks([])

plt.savefig('../figures/exercice-2.png', dpi=dpi)

plt.show()