From 430ca8dc056f7faf429afc347de4635d6461f90c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=81ukasz?= <lukkoz2@st.amu.edu.pl>
Date: Tue, 13 Apr 2021 21:50:08 +0200
Subject: [PATCH] graphserach v0.6

---
 FindPath.py | 14 +++++++-------
 main.py     |  2 ++
 2 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/FindPath.py b/FindPath.py
index 44b9419..b0e8cbc 100644
--- a/FindPath.py
+++ b/FindPath.py
@@ -1,6 +1,7 @@
 from constants import *
 
 def whichStateLookingFor(tractor, TillageUnit):
+    searching_field = "toPlow"
     if tractor.header and tractor.hitch == "Crop Trailer":
         searching_field = "toCut"
     elif isinstance(tractor.hitch, TillageUnit) and tractor.hitch.load == "Nothing":
@@ -14,19 +15,18 @@ def whichStateLookingFor(tractor, TillageUnit):
     return searching_field
 
 def nearestLookingField(board, tractor, TillageUnit):
-    x = whichStateLookingFor(tractor, TillageUnit)
+    end_horizontal_index = 0
+    end_vertical_index = 0
+    searching_field = whichStateLookingFor(tractor, TillageUnit)
     for i in range(0, int(HORIZONTAL_TILES_NUMBER)):
         for j in range(0, int(VERTICAL_TILES_NUMBER)):
             field = board[i][j]
-            if x == field.state:
+            if searching_field == field.state:
                 end_horizontal_index = field.horizontal_index
                 end_vertical_index = field.vertical_index
                 break
-            else:
-                print('Wroc do bazy zmienic narzedzie')
-                end_horizontal_index = 0
-                end_vertical_index = 0
-    return(end_horizontal_index, end_vertical_index)
+
+    return end_horizontal_index, end_vertical_index
 
 def graphsearch(tractor, board, TillageUnit):
     start_horizontal_index = tractor.horizontal_index
diff --git a/main.py b/main.py
index 7e44402..45be0b5 100644
--- a/main.py
+++ b/main.py
@@ -58,5 +58,7 @@ while working:
 
     clock.tick(FPS)
 
+graphsearch(tractor, board, TillageUnit)
+
 pygame.quit()
 quit()