X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tabakrolletjie%2Fenemies.py;h=e4c868e89ae3dabd6d4ca6d3b9b08448ad93ce76;hb=24d25dfaffe99f2f18221668cb34383d6358634a;hp=d40af93750b64be2592558d14e13c45242b2ba2d;hpb=b1cbbc6a48f68e43b72293ef0d6d24450ce40c7e;p=tabakrolletjie.git

diff --git a/tabakrolletjie/enemies.py b/tabakrolletjie/enemies.py
index d40af93..e4c868e 100644
--- a/tabakrolletjie/enemies.py
+++ b/tabakrolletjie/enemies.py
@@ -20,21 +20,35 @@ MOULD_FILTER = pymunk.ShapeFilter(
 EAT_TURNIP_FILTER = pymunk.ShapeFilter(mask=TURNIP_CATEGORY)
 
 
+# Boyd parameters
+SPAWN_RATE = 10
+MAX_AGE = 60
+MAX_ELEMENTS = 400
+MAX_HEALTH = 100
+
+# Increase in health per day
+DAY_HEALTH = 10
+
+HEAL_FACTOR = 1
+
+MOULD_STAGES = [15, 25]
+
+
 class Mould(pymunk.Body):
     """A segment of Boyd"""
 
-    def __init__(self, gamestate, space, pos):
+    def __init__(self, gamestate, space, pos, resistances):
         super(Mould, self).__init__(0, 0, pymunk.Body.STATIC)
         self.position = pos
         self._shape = pymunk.Circle(self, 16)
         space.add(self, self._shape)
         self._shape.filter = MOULD_FILTER
-        self._resistances = {}
         self._age = 0
         self._img = None
         self._health = 500
         self.has_eyeball = False
         self._eyeball = None
+        self._resistances = resistances
 
     def pygame_pos(self, surface):
         """Convert to pygame coordinates and offset position so
@@ -47,7 +61,7 @@ class Mould(pymunk.Body):
         if not self._img:
             name = random.choice(
                 ('mouldA.png', 'mouldB.png', 'mouldC.png'))
-            size = "16" if self._age < 10 else "32" if self._age < 20 else "64"
+            size = "16" if self._age < MOULD_STAGES[0] else "32" if self._age < MOULD_STAGES[1] else "64"
             self._img = loader.load_image(size, name)
         return self._img
 
@@ -58,21 +72,24 @@ class Mould(pymunk.Body):
             self._eyeball = loader.load_image("32", name)
         return self._eyeball
 
+    def set_health(self, new_health):
+        self._health = new_health
+
     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
+        if self._health < MAX_HEALTH:
+            self._health += HEAL_FACTOR
 
         refresh = False
 
-        if (self._age % 15) == 0 and len(moulds) < 1000:
+        if (self._age % SPAWN_RATE) == 0 and len(moulds) < MAX_ELEMENTS:
             # Spawn a new child, if we can
             spawn = True
-            choice = random.randint(0, 4)
+            choice = random.randint(0, 3)
             if choice == 0:
                 pos = self.position + (0, 24)
             elif choice == 1:
@@ -90,28 +107,25 @@ class Mould(pymunk.Body):
             # We allow some overlap, hence not checking full radius
             query = space.point_query(pos, 8, MOULD_FILTER)
             if query:
-                # for x in query:
-                #     if not isinstance(x.shape.body, Mould):
-                #         print x.shape, x.shape.body
                 spawn = False
             if spawn:
-                child = Mould(gamestate, space, pos)
+                child = Mould(gamestate, space, pos, self._resistances)
                 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):
+        if self._age in MOULD_STAGES:
             # We grow in size
             refresh = True
             self._img = None  # invalidate cached image
 
-        if self._age > 20 and random.randint(0, 500) < 1:
+        if self._age > MOULD_STAGES[1] and random.randint(0, 500) < 1:
             # Maybe we grow an eyeball
             self.has_eyeball = True
 
-        if self._age > 120:
+        if self._age > MAX_AGE:
             # We die of old age
             space.remove(self, self._shape)
             moulds.remove(self)
@@ -126,8 +140,11 @@ class Mould(pymunk.Body):
 
     def damage(self, light, space, moulds):
         """Take damage for light, adjusted for resistances."""
-        self._health -= 3
-        if self._health <= 0 and self._age <= 120:
+        damage = light.base_damage()
+        colour = light.colour
+        damage = int(damage * (3 - self._resistances.get(colour, 0)) / 3.0)
+        self._health -= damage
+        if self._health <= 0 and self._age <= MAX_AGE:
             # We die of damage
             space.remove(self, self._shape)
             moulds.remove(self)
@@ -138,8 +155,13 @@ class Mould(pymunk.Body):
 class Boyd(object):
 
     def __init__(self, gamestate, space):
-        seed = Mould(gamestate, space, (275, 300))
-        self._moulds = [seed]
+        self._moulds = []
+        self._seen_colours = set()
+        for position in gamestate.get_spawn_positions():
+            seed = Mould(gamestate, space, position,
+                         gamestate.resistances)
+            seed.set_health(MAX_HEALTH + gamestate.days * DAY_HEALTH)
+            self._moulds.append(seed)
         self._image = pygame.surface.Surface(SCREEN_SIZE)
         self._image = self._image.convert_alpha(pygame.display.get_surface())
         self._draw_moulds()
@@ -165,7 +187,7 @@ class Boyd(object):
             # Check for damage
             lit_by = lights.light_query(mould._shape)
             for light in lit_by:
-                # Todo: extract colour and intensity from light
+                self._seen_colours.add(light.colour)
                 if mould.damage(light, space, self._moulds):
                     redraw = True
                     break  # we only die once
@@ -175,3 +197,17 @@ class Boyd(object):
     def render(self, surface):
         """Draw ourselves"""
         surface.blit(self._image, (0, 0), None, 0)
+
+    def alive(self):
+        return len(self._moulds) > 0
+
+    def update_resistances(self, gamestate):
+        for colour in self._seen_colours:
+            cur_reistance = gamestate.resistances.get(colour, 0)
+            gamestate.resistances[colour] = cur_reistance + 2
+        for colour in gamestate.resistances:
+            gamestate.resistances[colour] -= 1
+            if gamestate.resistances[colour] > 3:
+                gamestate.resistances[colour] = 3
+            if gamestate.resistances[colour] < 0:
+                gamestate.resistances[colour] = 0