From 790438a9a29a1b1af2ed8ef3d8ac0e63b3c811e3 Mon Sep 17 00:00:00 2001
From: Wojciech Kubicki <wojt.kubicki@gmail.com>
Date: Mon, 10 Jun 2024 15:36:47 +0200
Subject: [PATCH] fix: apply decision tree for choosing new crops

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

diff --git a/src/tractor.py b/src/tractor.py
index ed2704b0..0cb57331 100644
--- a/src/tractor.py
+++ b/src/tractor.py
@@ -165,8 +165,13 @@ class Tractor(pygame.sprite.Sprite):
             self.move()
     
     def decision_tree(self):
-        action = self.make_decision()
-        if (self.get_current_tile().type != 'grass' or self.get_current_tile().type == 'water'): action = 'nothing'
+        match (self.get_current_tile().type):
+            case ('grass'):
+                action = self.make_decision()
+            case ('water'):
+                action = 'nothing'
+            case _:
+                action = 'water'
         self.prev_action = action
 
         print("Decyzja podjęta przez drzewo decyzyjne: ", action)
@@ -180,7 +185,7 @@ class Tractor(pygame.sprite.Sprite):
             case ('water'):
                 self.get_current_tile().water_level += 10
             case ('plant(bób)'):
-                self.get_current_tile().set_type('marchew')
+                self.get_current_tile().set_type('bób')
             case ('plant(brokuł)'):
                 self.get_current_tile().set_type('brokuł')
             case ('plant(brukselka)'):
@@ -188,11 +193,11 @@ class Tractor(pygame.sprite.Sprite):
             case ('plant(burak)'):
                 self.get_current_tile().set_type('burak')
             case ('plant(cebula)'):
-                self.get_current_tile().set_type('marchew')
+                self.get_current_tile().set_type('cebula')
             case ('plant(cukinia)'):
                 self.get_current_tile().set_type('cukinia')
             case ('plant(dynia)'):
-                self.get_current_tile().set_type('fasola')
+                self.get_current_tile().set_type('dynia')
             case ('plant(fasola)'):
                self.get_current_tile().set_type('fasola')
             case ('plant(groch)'):
@@ -212,7 +217,7 @@ class Tractor(pygame.sprite.Sprite):
             case ('plant(papryka)'):
                 self.get_current_tile().set_type('papryka')
             case ('plant(pietruszka)'):
-                self.get_current_tile().set_type('marchew')
+                self.get_current_tile().set_type('pietruszka')
             case ('plant(pomidor)'):
                 self.get_current_tile().set_type('pomidor')
             case ('plant(por)'):