X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=koperkapel%2Fgamelib%2Flevel.py;h=34487af920d67e5ce59f97ddc5b8d0d1bb085fa2;hb=9e66c7b17935e886b9fbed0704727aacda3b0c38;hp=bcc319a2886d2f5e530d30ff70be6f7f50478a09;hpb=15f8404d39a46ec6243f2e75a0c03ba4e6294800;p=koperkapel.git

diff --git a/koperkapel/gamelib/level.py b/koperkapel/gamelib/level.py
index bcc319a..34487af 100644
--- a/koperkapel/gamelib/level.py
+++ b/koperkapel/gamelib/level.py
@@ -1,5 +1,7 @@
 """ Class holding the level info """
 
+import random
+
 
 class Level(object):
 
@@ -84,3 +86,34 @@ class Level(object):
 
     def get_exit_level(self):
         return self.exit["next level"]
+
+    def enemy_at(self, x, y):
+        for enemy in self.enemies:
+            if (x, y) == enemy.game_pos:
+                return True
+        return False
+
+    def get_enemy(self, x, y):
+        for enemy in self.enemies:
+            if (x, y) == enemy.game_pos:
+                return enemy
+        return None
+
+    def remove_enemy(self, enemy):
+        self.enemies.remove(enemy)
+        for generator in self.enemy_generators:
+            generator.killed(enemy)
+
+    def add_enemy(self, enemy, x, y):
+        """Add an enemy to an empty floor space near x, y"""
+        added = False
+        while not added:
+            if self.can_walk(x, y, 'floor'):
+                if not self.friend_at(x, y):
+                    if not self.enemy_at(x, y):
+                        added = True
+                        enemy.game_pos = (x, y)
+                        self.enemies.append(enemy)
+                        added = True
+            x += random.randint(-1, 1)
+            y += random.randint(-1, 1)