Add total power and damage.
[tabakrolletjie.git] / tabakrolletjie / enemies.py
index 830ef731bfd102fecaf82443f81cd6f3001750a0..eef2596e5300455ea7437d29642b27b9bdaabe02 100644 (file)
@@ -1,19 +1,24 @@
 # Boyd, the friendly, misunderstood turnip loving, light hating space mould
 
+import random
+
 import pymunk
 import pymunk.pygame_util
 import pygame.draw
 import pygame.surface
 import pygame.display
 
-import pygame.locals as pgl
-
-from .constants import SCREEN_SIZE, MOULD_CATEGORY, OBSTACLE_CATEGORY
+from .constants import (SCREEN_SIZE, MOULD_CATEGORY, OBSTACLE_CATEGORY,
+                        TURNIP_CATEGORY)
+from .loader import loader
+from .sound import sound
 
 MOULD_FILTER = pymunk.ShapeFilter(
     mask=MOULD_CATEGORY | OBSTACLE_CATEGORY,
     categories=MOULD_CATEGORY)
 
+EAT_TURNIP_FILTER = pymunk.ShapeFilter(mask=TURNIP_CATEGORY)
+
 
 class Mould(pymunk.Body):
     """A segment of Boyd"""
@@ -27,26 +32,46 @@ class Mould(pymunk.Body):
         self._resistances = {}
         self._age = 0
         self._img = None
+        self._health = 500
+        self.has_eyeball = False
+        self._eyeball = None
+
+    def pygame_pos(self, surface):
+        """Convert to pygame coordinates and offset position so
+           our position is the centre of the image."""
+        # The odd sign combination is because of the pymunk / pygame
+        # transform, but we do it this way to exploit Vec2d math magic
+        return pymunk.pygame_util.to_pygame(self.position + (-16, 16), surface)
 
     def get_image(self):
         if not self._img:
-            img = pygame.surface.Surface((32, 32))
-            img.convert_alpha(pygame.display.get_surface())
-            img.fill((0, 0, 0, 0))
-            pygame.draw.circle(img, (255, 255, 255, 255), (16, 16), 16)
-            self._img = img
+            name = random.choice(
+                ('mouldA.png', 'mouldB.png', 'mouldC.png'))
+            size = "16" if self._age < 10 else "32" if self._age < 20 else "64"
+            self._img = loader.load_image(size, name)
         return self._img
 
+    def get_eyeball(self):
+        if not self._eyeball:
+            name = random.choice(
+                ('eyeballA.png', 'eyeballB.png', 'eyeballC.png'))
+            self._eyeball = loader.load_image("32", name)
+        return self._eyeball
+
     def tick(self, gamestate, space, moulds):
         """Grow and / or Die"""
+
         self._age += 1
 
+        # we regain a health every tick, so we heal in the dark
+        if self._health < 100:
+            self._health += 1
+
         refresh = False
 
         if (self._age % 15) == 0 and len(moulds) < 1000:
             # Spawn a new child, if we can
             spawn = True
-            import random
             choice = random.randint(0, 4)
             if choice == 0:
                 pos = self.position + (0, 24)
@@ -71,41 +96,82 @@ class Mould(pymunk.Body):
                 spawn = False
             if spawn:
                 child = Mould(gamestate, space, pos)
+                child._health = self._health
                 moulds.append(child)
                 refresh = True
+                if random.randint(0, 10) < 2:
+                    sound.play_sound("mouth_pop_2a.ogg")
+
+        if self._age in (10, 20):
+            # We grow in size
+            refresh = True
+            self._img = None  # invalidate cached image
+
+        if self._age > 20 and random.randint(0, 500) < 1:
+            # Maybe we grow an eyeball
+            self.has_eyeball = True
 
         if self._age > 120:
             # We die of old age
             space.remove(self, self._shape)
             moulds.remove(self)
             refresh = True
+        else:
+            # Check for turnips we can eat
+            # Note that we can only eat a tick after we spawn
+            query = space.point_query(self.position, 16, EAT_TURNIP_FILTER)
+            if query:
+                query[0].shape.body.turnip.eaten = True
         return refresh
 
-    def damage(self, light_color, intensity):
+    def damage(self, light, space, moulds):
         """Take damage for light, adjusted for resistances."""
+        self._health -= light.base_damage()
+        if self._health <= 0 and self._age <= 120:
+            # We die of damage
+            space.remove(self, self._shape)
+            moulds.remove(self)
+            return True
+        return False
 
 
 class Boyd(object):
 
     def __init__(self, gamestate, space):
-        seed = Mould(gamestate, space, (400, 400))
+        seed = Mould(gamestate, space, (275, 300))
         self._moulds = [seed]
         self._image = pygame.surface.Surface(SCREEN_SIZE)
-        self._image.convert_alpha(pygame.display.get_surface())
-        self._image.fill((0, 0, 0, 0))
+        self._image = self._image.convert_alpha(pygame.display.get_surface())
+        self._draw_moulds()
 
-    def tick(self, gamestate, space):
+    def _draw_moulds(self):
+        self._image.fill((0, 0, 0, 0))
+        for m in self._moulds:
+            self._image.blit(m.get_image(),
+                             m.pygame_pos(self._image), None,
+                             0)
+        for m in self._moulds:
+            if m.has_eyeball:
+                self._image.blit(m.get_eyeball(), m.pygame_pos(self._image),
+                                 None, 0)
+
+    def tick(self, gamestate, space, lights):
         redraw = False
+        # Handle spawn events
         for mould in self._moulds[:]:
+            # Handle updates
             if mould.tick(gamestate, space, self._moulds):
                 redraw = True
+            # Check for damage
+            lit_by = lights.light_query(mould._shape)
+            for light in lit_by:
+                # Todo: extract colour and intensity from light
+                if mould.damage(light, space, self._moulds):
+                    redraw = True
+                    break  # we only die once
         if redraw:
-            self._image.fill((0, 0, 0, 0))
-            for mould in self._moulds:
-                pos = pymunk.pygame_util.to_pygame(mould.position, self._image)
-                self._image.blit(mould.get_image(), pos, None,
-                                 pgl.BLEND_RGBA_ADD)
+            self._draw_moulds()
 
     def render(self, surface):
         """Draw ourselves"""
-        surface.blit(self._image, (0, 0), None, pgl.BLEND_RGBA_ADD)
+        surface.blit(self._image, (0, 0), None, 0)