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', pos=(50, SCREEN_SIZE[1] - 40)),
  26         ]
  27
  28     @debug_timer("day.render")
  29     def render(self, surface, gamestate):
  30         surface.fill((0, 0, 155))
  31         self._lights.render_light(surface)
  32         self._obstacles.render(surface)
  33         self._lights.render_fittings(surface)
  34         for tool in self._tools:
  35             tool.render(surface)
  36
  37     def event(self, ev, gamestate):
  38         if ev.type == pgl.KEYDOWN:
  39             if ev.key in (pgl.K_q, pgl.K_ESCAPE):
  40                 from .menu import MenuScene
  41                 SceneChangeEvent.post(scene=MenuScene())
  42         elif ev.type == pgl.MOUSEBUTTONDOWN:
  43             if ev.button == 1:
  44                 # Check tools
  45                 for tool in self._tools:
  46                     if tool.pressed(ev):
  47                         print 'tool'
  48                         return
  49                 # Not tool, so check lights
  50                 self._lights.toggle_nearest(ev.pos, surfpos=True)
  51                 print self._lights.lit_by(ev.pos, surfpos=True)
  52
  53     @debug_timer("day.tick")
  54     def tick(self, gamestate):
  55         self._lights.tick()