From 30878adeddb3be19bd6edf67983df89528caa734 Mon Sep 17 00:00:00 2001
From: Adam Mikolajczak <adam.mikolajczak111@gmail.com>
Date: Sun, 9 Jun 2024 11:26:14 +0200
Subject: [PATCH] feat: After arriving at selected by user tile, Tractor makes
 decision by using decision tree. The decision is printed in terminal.

---
 src/tractor.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/tractor.py b/src/tractor.py
index b76c7058..ed2704b0 100644
--- a/src/tractor.py
+++ b/src/tractor.py
@@ -97,13 +97,11 @@ class Tractor(pygame.sprite.Sprite):
     def draw(self, surface):
         surface.blit(self.image, self.rect)
 
-
     def get_coordinates(self):
         x = self.rect.x // TILE_SIZE
         y = self.rect.y // TILE_SIZE
         return (x,y)
 
-
     def move(self):
         if self.direction == "north" and self.rect.y > 0:
             self.rect.y -= TILE_SIZE
@@ -168,12 +166,13 @@ class Tractor(pygame.sprite.Sprite):
     
     def decision_tree(self):
         action = self.make_decision()
-        if (self.get_current_tile().type != 'grass' or self.get_current_tile().type == 'water'): action = 'move'
+        if (self.get_current_tile().type != 'grass' or self.get_current_tile().type == 'water'): action = 'nothing'
         self.prev_action = action
 
+        print("Decyzja podjęta przez drzewo decyzyjne: ", action)
 
         match (action):
-            case ('move'):
+            case ('nothing'):
                 pass
                 #self.move_rotating()
             case ('harvest'):
@@ -255,6 +254,9 @@ class Tractor(pygame.sprite.Sprite):
             print("Rozpoznano: ", self.get_current_tile().prediction)
             print("Co jest faktycznie: ", self.get_current_tile().type)
         print("\n")
+
+        if self.get_coordinates() == self.final:
+            self.decision_tree()
         
         return