X-Git-Url: https://git.ctpug.org.za/?p=erdslangetjie.git;a=blobdiff_plain;f=erdslangetjie%2Fplayer.py;h=6f3689e39cae0a3e919396c6950ad52f4032b9cb;hp=5bb6472b1b88fca7b942d28a60e97e8eb35690ff;hb=16f38ab01c4dc966c5be8c81361380739d36e480;hpb=ae83360a244732dabac64083f6915fc301d583f7

diff --git a/erdslangetjie/player.py b/erdslangetjie/player.py
index 5bb6472..6f3689e 100644
--- a/erdslangetjie/player.py
+++ b/erdslangetjie/player.py
@@ -43,19 +43,62 @@ class Nemesis(FigureSprite):
         super(Nemesis, self).__init__()
         self.sprite = load_image('sprites/nemesis.png')
         self.reset_pos()
+        self._deadends = set([])
+        self._been = {}
 
-    def move(self, level):
+    def move(self, level, player_pos):
         if not self.on_board():
             # Advance towards the map a step at a time
             self.pos = (self.pos[0] + 1, self.pos[1])
             if self.on_board():
                 self.pos = level.enter_pos
-            return False
+            return
         # AI goes here
-        return True
+        # First, if we're standing next to a gate, we attack it
+        neighbours = [(self.pos[0] + 1, self.pos[1]),
+                (self.pos[0] - 1, self.pos[1]),
+                (self.pos[0], self.pos[1] + 1),
+                (self.pos[0], self.pos[1] - 1)]
+        for cand in neighbours:
+            if level.is_gate(cand) and level.blocked(cand):
+                level.damage_gate(cand)
+                return
+        # check for dead-ends
+        if self._in_dead_end(level):
+            self._deadends.add(self.pos)
+        self._been.setdefault(self.pos, 0)
+        mindist = 999999
+        best_pos = None
+        for direction in [(1, 0), (-1, 0), (0, -1), (0, 1)]:
+            new_pos = (self.pos[0] + direction[0], self.pos[1] + direction[1])
+            if new_pos in self._deadends:
+                continue
+            if self.can_move(direction, level):
+                # We weigh stuff by the number of times we've stepped there, to
+                # avoid certain types of loop
+                dist = (level.calc_dist(new_pos, player_pos)
+                        + self._been.get(new_pos, 0))
+                if dist < mindist:
+                    mindist = dist
+                    best_pos = new_pos
+        if best_pos:
+            self._been[self.pos] += 1
+            self.pos = best_pos
 
     def reset_pos(self):
-        self.pos = (-10, 0)
+        self.pos = (-7, 0)
+        self._deadends = set([])
+        self._been = {}
 
     def on_board(self):
         return self.pos[0] >= 0
+
+    def _in_dead_end(self, level):
+        # Check if this is a dead end
+        blocked = 0
+        for direction in [(1, 0), (-1, 0), (0, -1), (0, 1)]:
+            new_pos = (self.pos[0] + direction[0], self.pos[1] + direction[1])
+            if level.is_wall(new_pos) or new_pos in self._deadends:
+                blocked += 1
+        # A dead end has only 1 exit, and recurse back from there
+        return blocked >= 3