X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=koperkapel%2Fgamelib%2Fenemy_roach.py;h=c419af53282fca8967d8477376df6dd5ee5cf18a;hb=HEAD;hp=2a8438241f3bdc8ac3fa0380a57b28bc1b6e6834;hpb=1545515688e1bbc1ca4b7981bff6cc59636b0a98;p=koperkapel.git

diff --git a/koperkapel/gamelib/enemy_roach.py b/koperkapel/gamelib/enemy_roach.py
index 2a84382..c419af5 100644
--- a/koperkapel/gamelib/enemy_roach.py
+++ b/koperkapel/gamelib/enemy_roach.py
@@ -3,9 +3,10 @@
 import random
 
 from pgzero.clock import each_tick, unschedule
+from pgzero.loaders import sounds
 from functools import partial
 
-from ..roaches import t32_roaches, WorldRoach
+from ..roaches import t32_roaches, WorldRoach, default_rats, default_robots
 
 
 def get_enemy_roach(level):
@@ -13,15 +14,81 @@ def get_enemy_roach(level):
     roach.anchor = (-16, -16)  # this should center them on the tile
     roach.game_pos = (0, 0)
     roach.health = 5
+    roach.damage = 1
     roach.angle = 0
+    roach.hit_sound = sounds.load('enemy_hit')
     roach.level = level
+    roach.level_layer = 'floor'  # always the case for now
     roach.move = partial(move, roach)
     roach.last_moved = 0
+    roach.last_attacked= 0
     roach.start_pos = None
     each_tick(roach.move)
+    roach.attack = partial(attack, roach)
     return roach
 
 
+def get_rat(level):
+    roach = default_rats.assemble()
+    roach.anchor = (0, 0)
+    roach.game_pos = (0, 0)
+    roach.health = 10
+    roach.damage = 2
+    roach.angle = 0
+    roach.hit_sound = sounds.load('enemy_hit')
+    roach.level = level
+    roach.level_layer = 'floor'  # always the case for now
+    roach.move = partial(move, roach)
+    roach.last_moved = 0
+    roach.last_attacked= 0
+    roach.start_pos = None
+    each_tick(roach.move)
+    roach.attack = partial(attack, roach)
+    return roach
+
+
+def get_robot(level):
+    roach = default_robots.assemble()
+    roach.anchor = (0, 0)
+    roach.game_pos = (0, 0)
+    roach.health = 10
+    roach.damage = 5
+    roach.angle = 0
+    roach.level = level
+    roach.hit_sound = sounds.load('enemy_hit')
+    roach.level_layer = 'floor'  # always the case for now
+    roach.move = partial(move, roach)
+    roach.last_moved = 0
+    roach.last_attacked= 0
+    roach.start_pos = None
+    each_tick(roach.move)
+    roach.attack = partial(attack, roach)
+    return roach
+
+
+def attack(roach, player_pos, player_layer, dt):
+    """Attack the player if close enough"""
+    roach.last_attacked += dt
+    if roach.last_attacked > 0.6:
+        roach.last_attacked = 0
+        if player_layer != 'floor':
+            return None
+        if abs(player_pos[0] - roach.game_pos[0]) > 1:
+            return None
+        if abs(player_pos[1] - roach.game_pos[1]) > 1:
+            return None
+        # Attacking, so turn towards the player
+        if player_pos[0] - roach.game_pos[0] < 0:
+            roach.angle = 270
+        elif player_pos[1] - roach.game_pos[1] < 0:
+            roach.angle = 0
+        elif player_pos[0] - roach.game_pos[0] > 0:
+            roach.angle = 90
+        else:
+            roach.angle = 270
+        # Do 1 damage
+        return roach.damage
+
 def move(roach, dt):
     """Enemy roach move method"""
     roach.last_moved += dt