from __future__ import print_function
 #  
 #  Copyright (c) 2012 Edward Langley
 #  All rights reserved.
 #  
 #  Redistribution and use in source and binary forms, with or without
 #  modification, are permitted provided that the following conditions
 #  are met:
 #  
 #  Redistributions of source code must retain the above copyright notice,
 #  this list of conditions and the following disclaimer.
 #  
 #  Redistributions in binary form must reproduce the above copyright
 #  notice, this list of conditions and the following disclaimer in the
 #  documentation and/or other materials provided with the distribution.
 #  
 #  Neither the name of the project's author nor the names of its
 #  contributors may be used to endorse or promote products derived from
 #  this software without specific prior written permission.
 #  
 #  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 #  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 #  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 #  FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 #  HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 #  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
 #  TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 #  PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 #  LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 #  NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 #  SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 #  
 #  
 
 import time
 import random
 import pygame
 import sys
 import collections
 import game
 
 import collections
 
 ############### Cell ################
 class Cell(game.Cell):
 	pass
 
 Cell.add_rule(3)(0)
 Cell.add_rule(2)(3)
 
 @Cell.add_rule(1)
 def hook_1(cell):
 	neighbors = [c and c.value for c in cell.neighbors]
 	neighbor_value = cell.neighbor_value()
 	rnd = random.random()
 	if 2 in neighbors or 3 in neighbors:
 		if rnd < .62 * neighbor_value/2: return 2
 
 	else:
 		if rnd < .01 * neighbor_value:
 			return 0
 
 @Cell.add_rule(0)
 def hook_0(cell):
 	neighbor_value = cell.neighbor_value()
 	if neighbor_value != 0:
 		nearby_fire = set([2,3]) & set(c and c.value for c in cell.neighbors)
 		if nearby_fire == set() and random.random() < .02 * neighbor_value:
 			return 1
 	return cell.value
 
 ########## End Cell #################
 
 class Presence(game.Presence):
 	def act(self):
 		value = int(round( (self.locale.neighbor_value() + self.locale.value) ))
 		self.locale.set_value(0)
 		for x in self.locale.neighbors:
 			if not x: continue
 			if x.value == 1:
 				x.set_value(2)
 
 import time
 
 if __name__ == '__main__':
 	pygame.init()
 	pygame.event.set_allowed([pygame.QUIT, pygame.KEYDOWN])
 	pygame.display.set_caption('Simulator')
 
 	board = game.Board(120, 120)
 	window = pygame.display.set_mode(board.rect.size, pygame.DOUBLEBUF)
 	window.set_alpha(None)
 	for sprite in board.board_group:
 		sprite.image.convert()
 
 	a = game.Controller(board)
 
 	p = a.add_observer(Presence, a.board.init.south)
 	np = a.board.init.southeast
 	for x in range(20):
 		np = np.east
 	np = a.add_observer(Presence, np)
 
 	window.fill( (64,64,64) )
 
 	running = True
 
 	board.board_group.draw(window)
 	pygame.display.flip()
 
 	clck = pygame.time.Clock()
 
 	paused = True
 	while running:
 		clck.tick(80)
 
 		for event in pygame.event.get():
 			if event.type == pygame.QUIT:
 				running=False
 				break
 			elif event.type == pygame.KEYDOWN and event.unicode == 'e':
 				paused = not paused
 			elif event.type == pygame.KEYDOWN and event.unicode == 'w':
 				new = p.locale.north
 				if new:
 					p.locale = new
 			elif event.type == pygame.KEYDOWN and event.unicode == 'a':
 				new = p.locale.west
 				if new:
 					p.locale = new
 			elif event.type == pygame.KEYDOWN and event.unicode == 's':
 				new = p.locale.south
 				if new:
 					p.locale = new
 			elif event.type == pygame.KEYDOWN and event.unicode == 'd':
 				new = p.locale.east
 				if new:
 					p.locale = new
 
 		if not paused:
 			a.step()
 			updates = a.board.board_group.draw(window)
 			pygame.display.update(updates)
 
 		print( clck.get_fps(), '\r', end='' )
 		sys.stdout.flush()