def __init__(self, width=50, height=50, block_size=10):
self.geometry = Vector(width, height)
self.block_size = block_size
self.score = 0
self.drawer = Drawer(width, height, block_size)
self.snake = Snake(self)
self.flora = Flora(self)
def add_flora(self):
"""Add places to hide."""
new_flora = Flora(variety=random.choice(range(0, 5)),
size=self.flora_size,
pos_x=self.max_x + self.flora_size,
pos_y=random.choice(range(0, self.min_y)))
self.flora += [new_flora]
# Let's draw from back to front...
self.flora.sort(key=lambda x: x.pos_y, reverse=False)
def add_flora(self):
"""Add places to hide."""
# TODO: Flora to spawn *not* all in a straight vertical column.
new_flora = Flora(variety=random.choice(range(0, 5)),
size=self.flora_size,
pos_x=self.max_x + self.flora_size,
pos_y=random.choice(range(0, self.min_y)))
# new_flora.draw(self.screen)
self.flora += [new_flora]
def add_castle(self):
"""Add victory castle."""
new_flora = Flora(variety=0,
is_home=True,
size=self.flora_size * 2,
pos_x=self.max_x + self.flora_size,
pos_y=self.flora_size)
self.flora += [new_flora]
# Let's draw from back to front...
self.flora.sort(key=lambda x: x.pos_y, reverse=False)
def initSimulation():
State['iter'] = 0
for i in range(INI_FLORA):
Flora(canvas)
for i in range(INI_BOIDS):
Boid(canvas)
for i in range(INI_PREDS):
Predator(canvas)
Boid.setPredatorGrid(Predator.grid)
class Game:
def __init__(self, width=50, height=50, block_size=10):
self.geometry = Vector(width, height)
self.block_size = block_size
self.score = 0
self.drawer = Drawer(width, height, block_size)
self.snake = Snake(self)
self.flora = Flora(self)
def handle_input(self):
for event in pygame.event.get():
if event.type == pygame.QUIT:
# handle window close
pygame.quit()
sys.exit()
elif event.type == pygame.KEYDOWN:
if event.key == pygame.K_ESCAPE:
# make it the same as "window close"
pygame.event.post(pygame.event.Event(pygame.QUIT))
elif event.key in Keyboard.DIRECTION_KEYS:
self.snake.turn(Keyboard.DIRECTION_KEYS[event.key])
def process_collisions(self):
if not self.within_limits(self.snake.head):
self.snake.die()
return
if self.snake.self_collide():
self.snake.die()
return
if self.flora.is_food_here(self.snake.head):
self.snake.eat()
self.flora.remove_food(self.snake.head)
def run(self):
# helps us set the pace
fps_controller = pygame.time.Clock()
while True:
self.handle_input()
self.snake.move()
self.process_collisions()
self.draw()
self.drawer.flip_display()
fps_controller.tick(10)
if not self.snake.is_alive:
break
def show_score(self, game_over=False):
font = pygame.font.SysFont('Arial', 24)
surface = font.render('Score: {}'.format(self.score), True,
Colors.WHITE)
rectangle = surface.get_rect()
rectangle.midtop = (80, 10)
self.surface.blit(surface, rectangle)
def game_over():
""" game over function
"""
font = pygame.font.SysFont('Arial', 72)
surface = font.render('Game over!', True, Colors.RED)
rectangle = surface.get_rect()
rectangle.midtop = (WIDTH_PX // 2, 15)
surface.blit(surface, rectangle)
self.show_score(surface)
pygame.display.flip()
time.sleep(2)
pygame.quit()
sys.exit()
def random_vector(self):
return Vector.random(self.geometry)
def within_limits(self, xyVector):
return xyVector.within(self.geometry)
def draw_cell(self, x, y, color):
self.drawer.draw_cell(x, y, color)
def draw(self):
self.drawer.draw_background()
self.flora.draw()
self.snake.draw()
def __init__(self):
Flora.__init__(self, "Mountain Apple Tree", 17)
IClay.__init__(self)
def __init__(self):
Flora.__init__(self, "Silversword", 22)
ISilt.__init__(self)
def __init__(self):
Flora.__init__(self, "Rainbow Eucalyptus Tree", 8)
ILoamy.__init__(self)
def __init__(self):
Flora.__init__(self, "Blue Jade Vine", 0)
ISilt.__init__(self)
IMarsh.__init__(self)