sawine@2: import copy
sawine@3: from caugrid import Grid
sawine@2: 
sawine@2: class Rule(object):
sawine@2: 
sawine@2: 	def iterate(self, oldgrid):
sawine@2: 		grid = copy.deepcopy(oldgrid)
sawine@2: 		for x in xrange(oldgrid.minx - 1, oldgrid.maxx + 2):
sawine@2: 			for y in xrange(oldgrid.miny - 1, oldgrid.maxy + 2):
sawine@2: 				#print x, y, self.neighbours(oldgrid, x, y)
sawine@2: 				n = self.neighbours(oldgrid, (x, y))
sawine@2: 				if n > 2:
sawine@2: 					grid.set((x, y))					
sawine@2: 		return grid
sawine@2: 
sawine@2: 	def neighbours(self, grid, (testx, testy)):
sawine@2: 		n = 0
sawine@2: 		#print "testing ", testx, testy
sawine@2: 		for x in range(testx - 1, testx + 2):
sawine@2: 			for y in range(testy - 1, testy + 2):
sawine@2: 				if (x, y) in grid.cells:
sawine@2: 					n += 1
sawine@2: 				#print x, y, n
sawine@2: 		return n 
sawine@2: 	
sawine@2: 
sawine@2: class Rule2(object):
sawine@2: 
sawine@2: 	def iterate(self, oldgrid):
sawine@2: 		grid = copy.deepcopy(oldgrid)
sawine@2: 		for x in xrange(oldgrid.minx - 1, oldgrid.maxx + 2):
sawine@2: 			for y in xrange(oldgrid.miny, oldgrid.maxy + 2):
sawine@2: 				#print "testing ", x, y,
sawine@2: 				if (x+1, y) in oldgrid.cells or (x, y-1) in oldgrid.cells:
sawine@2: 					grid.set((x, y), 1)								
sawine@2: 		return grid
sawine@2: 
sawine@2: import random
sawine@3: import caumarshal as marshal
sawine@2: 
sawine@2: class PotentialGrowth(object):
sawine@2: 		
sawine@2: 	def __init__(self):
sawine@2: 		random.seed()
sawine@2: 
sawine@2: 	def iterate(self, oldgrid):
sawine@2: 		grid = copy.deepcopy(oldgrid)
sawine@2: 		for cell in oldgrid.cells.iteritems():
sawine@2: 			pos = cell[0]
sawine@2: 			value = cell[1]
sawine@2: 			if value > 1:
sawine@2: 				new_pos, new_value = self.grow(grid, pos)
sawine@2: 				grid.set(new_pos, new_value)
sawine@2: 				grid.set(pos, value - 1)
sawine@2: 		return grid
sawine@2: 
sawine@2: 	def grow(self, grid, (x, y)):
sawine@2: 		neighbours = [(x-1, y), (x, y+1), (x+1, y), (x, y-1)]
sawine@2: 		neighbours = [n for n in neighbours if n not in grid.cells]
sawine@2: 		try:
sawine@2: 			pos = random.choice(neighbours)	
sawine@2: 			value = 4
sawine@2: 		except IndexError:
sawine@2: 			pos = (x, y)
sawine@2: 			value = grid.cells[pos]
sawine@2: 		return pos, value
sawine@2: 
sawine@2: import sys
sawine@2: import math
sawine@2: 
sawine@2: def ruleTest():
sawine@2: 	rule = Rule()
sawine@2: 	grid = Grid()
sawine@2: 	grid.set((0, 0))
sawine@2: 	grid.set((0, 1)).set((1, 0)).set((0, -1)).set((-1, 0))
sawine@2: 	
sawine@2: 	
sawine@2: 	print "iteration radius(diff) area(diff) pi"
sawine@2: 
sawine@2: 	iterations = int(sys.argv[1])
sawine@2: 
sawine@2: 	for i in range(iterations):
sawine@2: 		A = len(grid)
sawine@2: 		r = grid.width() / 2.0
sawine@2: 		r_ideal = math.sqrt(A / math.pi)
sawine@2: 		A_ideal = r**2 * math.pi
sawine@2: 		pi = A / r**2
sawine@2: 		print "%i %f(%f) %i(%i) %f" % (i, r, r - r_ideal, 
sawine@2: 										A, A - A_ideal, pi)
sawine@2: 
sawine@2: 		grid = rule.iterate(grid)
sawine@3: 		marshal.dumpGrid(grid, "grid.cfg")
sawine@2: 		#print
sawine@2: 
sawine@2: def rule2Test():
sawine@2: 	rule = Rule2()
sawine@2: 	grid = Grid()
sawine@2: 	grid.set((0, 0)).set((-1, 0)).set((0, 1))
sawine@2: 	olda = 1
sawine@2: 
sawine@2: 	iterations = int(sys.argv[1])
sawine@2: 
sawine@2: 	for i in range(iterations):
sawine@2: 		a =  (float(len(grid)) - grid.width()) * 4.0
sawine@2: 		pi = 1.0 / ((grid.width()-1)**2 / ((a + olda) / 2.0))
sawine@2: 		print i, grid.width(), len(grid), pi
sawine@2: 		grid = rule.iterate(grid)
sawine@2: 		olda = a
sawine@3: 		marshal.dumpGrid(grid, "grid.cfg")
sawine@2: 
sawine@2: def potentialTest():
sawine@2: 	rule = PotentialGrowth()
sawine@2: 	grid = Grid()
sawine@2: 	grid.set((0, 0), 4)
sawine@2: 
sawine@2: 	iterations = int(sys.argv[1])
sawine@2: 
sawine@2: 	for i in range(iterations):
sawine@2: 		A = len(grid)
sawine@2: 		r = grid.width() / 4.0 + grid.height() / 4.0
sawine@2: 		r_ideal = math.sqrt(A / math.pi)
sawine@2: 		A_ideal = r**2 * math.pi
sawine@2: 		pi = A / r**2
sawine@2: 		print "%i %f(%f) %i(%i) %f" % (i, r, r - r_ideal, 
sawine@2: 										A, A - A_ideal, pi)
sawine@2: 
sawine@2: 		grid = rule.iterate(grid)
sawine@3: 		marshal.dumpGrid(grid, "grid.cfg")
sawine@2: 
sawine@2: def main():
sawine@3: 	#ruleTest()
sawine@2: 	#rule2Test()
sawine@3: 	potentialTest()
sawine@2: 
sawine@2: if __name__ == "__main__":
sawine@2: 	main()