Refactor giant on_key_down.

[koperkapel.git] / koperkapel / scenes / level.py

diff --git a/koperkapel/scenes/level.py b/koperkapel/scenes/level.py
index 54449dcc8513f7d2d91b5768d7549307f54e1e94..a17de55d70ece5aa8a37e2bf88891580eadf728c 100644 (file)
--- a/koperkapel/scenes/level.py
+++ b/koperkapel/scenes/level.py
@@ -6,6 +6,7 @@ import pygame.locals as pgl
 from ..loaders.levelloader import levels
 from .base import Scene, ChangeSceneEvent, MoveViewportEvent, defer_to_update
 from ..constants import TILE_SIZE, WIDTH, HEIGHT
+from ..roaches import build_roach
 from ..vehicles.base import Vehicle
 
 
@@ -87,6 +88,7 @@ class GameLevelScene(BaseLevelScene):
             return
         super().enter(world)
         self._roaches = self.actors.add_layer("roaches", level=10)
+        self._friends = self.actors.add_layer("friendly roaches", level=9)
         self._vehicle = Vehicle.current(world)
         self._mode = 'walk'
         self._angle = 0
@@ -99,6 +101,8 @@ class GameLevelScene(BaseLevelScene):
         self._set_pos(x, y)
         self._avatar.pos = (WIDTH // 2, HEIGHT // 2)
         self._roaches.add(self._avatar)
+        for friend in self._level.friends:
+            self._friends.add(friend)
         # Fix viewport offset
         return [
             MoveViewportEvent((
@@ -107,6 +111,7 @@ class GameLevelScene(BaseLevelScene):
 
     def _set_pos(self, x, y):
         self._player_pos = (x, y)
+        # print('At ', (x, y))
 
     def _can_move(self, x, y):
         if self._mode == 'walk':
@@ -129,6 +134,11 @@ class GameLevelScene(BaseLevelScene):
         self._roaches.add(self._avatar)
         self._set_angle(self._angle)
 
+    @defer_to_update
+    def _add_roach(self, world):
+        world.roaches.append(build_roach(world))
+        self._vehicle_changed()
+
     @defer_to_update
     def _change_vehicle(self, world):
         vehicle = Vehicle.random()
@@ -138,55 +148,63 @@ class GameLevelScene(BaseLevelScene):
     def update(self, world, engine, dt):
         super().update(world, engine, dt)
         events = world.pop_events()
+        for friend in self._friends:
+            friend.pos = self.calc_offset(
+                friend.game_pos[0] * TILE_SIZE, friend.game_pos[1] * TILE_SIZE)
         return events
 
+    def _movement_key(self, key):
+        x, y = self._player_pos
+        for k, dp, angle in (
+            (keys.DOWN, (0, 1), 180),
+            (keys.UP, (0, -1), 0),
+            (keys.LEFT, (-1, 0), 90),
+            (keys.RIGHT, (1, 0), -90),
+        ):
+            if key == k:
+                nx, ny = x + dp[0], y + dp[1]
+                if self._can_move(nx, ny):
+                    self._set_pos(nx, ny)
+                    offset = (TILE_SIZE * dp[0], TILE_SIZE * dp[1])
+                    self._set_angle(angle)
+                    return [MoveViewportEvent(offset)]
+
+    def _activate_key(self):
+        x, y = self._player_pos
+        if self._level.is_grate(x, y):
+            if (self._level_layer == 'floor' and
+                    self._level.can_crawl(x, y, 'floor')):
+                if self._level.can_crawl(x, y, 'tunnels'):
+                    self._level_layer = 'tunnels'
+                    self._mode = 'crawl'
+            elif self._level.can_crawl(x, y, 'floor'):
+                # Must be in the tunnels already
+                self._level_layer = 'floor'
+                self._mode = 'walk'
+        elif self._level.is_keypad(x, y):
+            self._level.press_keypad(x, y, self._roaches)
+        elif self._level.is_on_friend(x, y):
+            friend = self._level.remove_friend(x, y)
+            self._friends.remove(friend)
+            self._add_roach()
+
+    def _fire_key(self):
+        print('Boom')
+
+    def _vehicle_management_key(self):
+        from .roach_management import RoachesScene
+        return [ChangeSceneEvent(RoachesScene(level_scene=self))]
+
     def on_key_down(self, key, mod, unicode):
-        offset = None
         x, y = self._player_pos
-        if key == keys.DOWN:
-            if self._can_move(x, y + 1):
-                self._set_pos(x, y + 1)
-                offset = (0, TILE_SIZE)
-                self._set_angle(180)
-        elif key == keys.UP:
-            if self._can_move(x, y - 1):
-                self._set_pos(x, y - 1)
-                offset = (0, -TILE_SIZE)
-                self._set_angle(0)
-        elif key == keys.LEFT:
-            if self._can_move(x - 1, y):
-                self._set_pos(x - 1, y)
-                offset = (-TILE_SIZE, 0)
-                self._set_angle(90)
-        elif key == keys.RIGHT:
-            if self._can_move(x + 1, y):
-                self._set_pos(x + 1, y)
-                offset = (TILE_SIZE, 0)
-                self._set_angle(270)
+        if key in (keys.DOWN, keys.UP, keys.LEFT, keys.RIGHT):
+            return self._movement_key(key)
         elif key == keys.C:
-            # Activate
-            if self._level.is_grate(x, y):
-                if self._level_layer == 'floor' and self._level.can_crawl(x, y, 'floor'):
-                    if self._level.can_crawl(x, y, 'tunnels'):
-                        self._level_layer = 'tunnels'
-                        self._mode = 'crawl'
-                elif self._level.can_crawl(x, y, 'floor'):
-                    # Must be in the tunnels already
-                    self._level_layer = 'floor'
-                    self._mode = 'walk'
-            elif self._level.is_keypad(x, y):
-                self._level.press_keypad(x, y, self._roaches)
-            return
+            return self._activate_key()
         elif key == keys.X:
-            # Fire
-            print('Boom')
+            return self._fire_key()
         elif key == keys.V:
-            # Leave vehicle
-            self._change_vehicle()
+            return self._change_vehicle()
         elif key == keys.Z:
-            # Vehicle management
-            from .roach_management import RoachesScene
-            return[ChangeSceneEvent(RoachesScene(level_scene=self))]
-        if offset:
-            return [MoveViewportEvent(offset)]
+            return self._vehicle_management_key()
         return super(GameLevelScene, self).on_key_down(key, mod, unicode)