From e068fb861436b834556b234884c07fac7986fb1b Mon Sep 17 00:00:00 2001
From: Daniel Dolezal <d.dolezal97@protonmail.com>
Date: Wed, 8 May 2024 23:45:07 +0200
Subject: [PATCH] add function when not eat food to remove snake tail

---
 snakes/BetterMasterSnake.py |  1 +
 snakes/TemplateSnake.py     | 25 +++++++++++++++++++++++--
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/snakes/BetterMasterSnake.py b/snakes/BetterMasterSnake.py
index 6f2d925..a5f185e 100644
--- a/snakes/BetterMasterSnake.py
+++ b/snakes/BetterMasterSnake.py
@@ -72,6 +72,7 @@ class BetterMasterSnake(TemplateSnake):
         if len(self.food_positions) > 0:
             # Choose the closest food source based on the heuristic
             closest_food = min(self.food_positions, key=lambda food: abs(food['x'] - self.game_board.get_my_snake_head()['x']) + abs(food['y'] - self.game_board.get_my_snake_head()['y']))
+            self.set_target_food(closest_food)
 
             # Use A* to search for a safe path
             return self.a_star_search(self.game_board.get_my_snake_head(), closest_food, obstacles)
diff --git a/snakes/TemplateSnake.py b/snakes/TemplateSnake.py
index ecde468..77eebe2 100644
--- a/snakes/TemplateSnake.py
+++ b/snakes/TemplateSnake.py
@@ -4,6 +4,7 @@ import random
 class TemplateSnake:
   def __init__(self):
     self.history = []
+    self.target_food = None
 
   def clear_history(self):
     self.history = []
@@ -60,7 +61,7 @@ class TemplateSnake:
     snake.pop()
     return snake
 
-  def avoid_my_body(self, my_body:list[dict], safe_positions:dict[str, dict], add_to_calculations:bool=False) -> list:
+  def avoid_my_body(self, my_body:list[dict], my_head:dict, safe_positions:dict[str, dict], add_to_calculations:bool=False) -> list:
     """
     my_body: List of dictionaries of x/y coordinates for every segment of a Battlesnake.
             e.g. [ {"x": 0, "y": 0}, {"x": 1, "y": 0}, {"x": 2, "y": 0} ]
@@ -70,6 +71,7 @@ class TemplateSnake:
     return: The list of remaining possible_moves, with the 'neck' direction removed
     """
     remove = []
+    my_body = self.did_snake_eat_food(my_body, my_head, add_to_calculations)
     for direction, location in safe_positions.items():
       if location in my_body:
         remove.append(direction)
@@ -154,7 +156,7 @@ class TemplateSnake:
     if add_to_calculations:
       self.add_calculations({"function": "get_possible_moves", "safe_positions": safe_positions})
 
-    safe_positions = self.avoid_my_body(self.game_board.get_my_snake_body(), safe_positions, add_to_calculations)
+    safe_positions = self.avoid_my_body(self.game_board.get_my_snake_body(), self.game_board.get_my_snake_head(), safe_positions, add_to_calculations)
     safe_positions = self.avoid_walls(safe_positions, add_to_calculations)
     safe_positions = self.avoid_snakes(self.game_board.get_other_snakes(), safe_positions, add_to_calculations)
     safe_positions = self.avoid_get_eaten_by_other_snakes(self.game_board.get_other_snakes(), safe_positions, add_to_calculations)
@@ -173,3 +175,22 @@ class TemplateSnake:
 
       return body
     return move
+
+  def did_snake_eat_food(self, my_body:list[dict], my_head:dict, add_to_calculations:bool=False):
+    if self.target_food is None:
+      if add_to_calculations:
+        self.add_calculations({"function": "did_snake_eat_food", "my_body": my_body, "my_head": my_head, "target_food": self.target_food, "action": "No Target Food"})
+      return my_body
+
+    if self.target_food["x"] == my_head["x"] and self.target_food["y"] == my_head["y"]:
+      if add_to_calculations:
+        self.add_calculations({"function": "did_snake_eat_food", "my_body": my_body, "my_head": my_head, "target_food": self.target_food, "action": "Snake Eat no food"})
+      return my_body
+
+    if add_to_calculations:
+      self.add_calculations({"function": "did_snake_eat_food", "my_body": my_body[:-1], "my_head": my_head, "target_food": self.target_food, "action": "Remove Tail from Body"})
+    return my_body[:-1]
+
+  def set_target_food(self, target_food:dict):
+    self.target_food = target_food
+    return True