tabakrolletjie/scenes/day.py

   1 """ Be prepared. """
   2
   3 import pygame.locals as pgl
   4
   5 import pymunk
   6 import pymunk.pygame_util
   7
   8 from .base import BaseScene
   9 from ..lights import LightManager
  10 from ..obstacles import ObstacleManager
  11 from ..events import SceneChangeEvent
  12 from ..utils import debug_timer
  13
  14 from ..constants import SCREEN_SIZE
  15 from ..widgets import ImageButton
  16
  17
  18 class DayScene(BaseScene):
  19     def enter(self, gamestate):
  20         self._space = pymunk.Space()
  21         self._obstacles = ObstacleManager(self._space, gamestate)
  22         self._lights = LightManager(self._space, gamestate)
  23         # Toolbar
  24         self._tools = [
  25             ImageButton('32', 'seed.png', name='seed',
  26                         pos=(50, SCREEN_SIZE[1] - 40)),
  27         ]
  28
  29     def exit(self, gamestate):
  30         self._unset_cursor()
  31
  32     @debug_timer("day.render")
  33     def render(self, surface, gamestate):
  34         surface.fill((0, 0, 155))
  35         self._lights.render_light(surface)
  36         self._obstacles.render(surface)
  37         self._lights.render_fittings(surface)
  38         for tool in self._tools:
  39             tool.render(surface)
  40         self._draw_cursor(surface)
  41
  42     def event(self, ev, gamestate):
  43         if ev.type == pgl.KEYDOWN:
  44             if ev.key in (pgl.K_q, pgl.K_ESCAPE):
  45                 from .menu import MenuScene
  46                 SceneChangeEvent.post(scene=MenuScene())
  47             if ev.key == pgl.K_e:
  48                 from .night import NightScene
  49                 SceneChangeEvent.post(scene=NightScene())
  50         elif ev.type == pgl.MOUSEBUTTONDOWN:
  51             if ev.button == 1:
  52                 # Check tools
  53                 for tool in self._tools:
  54                     if tool.pressed(ev):
  55                         print 'tool', tool.name
  56                         self._set_cursor(tool.name)
  57                         return
  58                 # Not tool, so check lights
  59                 self._lights.toggle_nearest(ev.pos, surfpos=True)
  60                 print self._lights.lit_by(ev.pos, surfpos=True)
  61
  62     @debug_timer("day.tick")
  63     def tick(self, gamestate):
  64         self._lights.tick()