.gitignore

@@ -1,4 +1,3 @@
 __pycache__/
 .DS_Store
 .env
-src/field

.vscode/extensions.json

@@ -0,0 +1,3 @@
+{
+  "recommendations": ["sbsnippets.pytorch-snippets"]
+}

README.md

@@ -8,8 +8,6 @@ Wymagane biblioteki do pobrania:
 pip install pygame
 pip install python-dotenv
 pip install pytholog
-pip install torch
-pip install sklearn
 ```
 
 Stwórz plik `.env` w głównym folderze projektu o poniższej treści:
@@ -27,6 +25,10 @@ STARTING_DIRECTION = north
 START_X = 0
 START_Y = 0
 
+# coordinates of destination tile
+FINAL_X = 15
+FINAL_Y = 15
+
 # tiles without plants modifier
 FREE_TILES = 2
 ```
@@ -35,12 +37,9 @@ Uruchom środowisko używając komend:
 
 ```
 cd src
-python generate_field.py
 python main.py
 ```
 
-Skrypt `generate_field.py` musi zostać przy pierwszym uruchomieniu, aby wygenerować pole. Kolejne wykonania skryptu nie są wymagane.
-
 ## 🧑‍🌾 Członkowie grupy
 
 - Wojciech Kubicki (483780)

src/.~lock.tree.csv#

@@ -1 +0,0 @@
-,adam,adam-thinkpad,10.06.2024 15:05,file:///home/adam/.config/libreoffice/4;

src/field.py

@@ -1,16 +1,13 @@
 import pygame
 from tile import Tile
 from tractor import Tractor
-from ast import literal_eval
 
 class Field:
     def __init__(self):
         self.tiles = pygame.sprite.Group()
-        with open('./field', 'r', encoding='UTF-8') as file:
-            content = file.read()
-        tiles = literal_eval(content)
-        for x in range(len(tiles)):
-            self.tiles.add(Tile(x, self, tiles[x]))
+        # TODO: enable resizing field grid from 16x16 to any size
+        for x in range(256):
+            self.tiles.add(Tile(x, self))
         self.tractor = Tractor(self)
 
 

src/generate_field.py

@@ -1,7 +1,6 @@
 from random import randint, choices, random
 from kb import tractor_kb, multi_sasiedzi
 import pytholog as pl
-from numpy.random import choice as npchoice
 
 
 def score_field(field):
@@ -18,32 +17,33 @@ def score_field(field):
         if index % 16 != 0 and field[index-1] != 'water':
             neighbours.append(field[index-1])
 
-        mod = multi_sasiedzi(field[index], neighbours)[0]["Mul"]
-        if mod > 10:
-            print(mod, '= multi(', field[index], ', ', neighbours, ')')
-        score += mod
+        score += multi_sasiedzi(field[index], neighbours)[0]["Mul"]
 
-    score = score / 256
     return score
 
 
 def choose_parents(population):
-    total_weights = sum(entity[0] for entity in population)
+    weights = [x[0] for x in population]
+    
+    weights_sum = 0
+    for weight in weights:
+        weights_sum += weight
 
-    weights = [entity[0] / total_weights for entity in population]
+    weights = [weight/weights_sum for weight in weights]
 
-    selection = npchoice(len(population), size=2, replace=False, p=weights)
+    mom = choices(population, cum_weights=weights, k=1)[0]
+    remaining_elements = [el for el in population if el != mom]
+    remaining_weights = [weights[population.index(el)] for el in remaining_elements]
+    dad = choices(remaining_elements, weights=remaining_weights, k=1)[0]
 
-    parents = [population[i] for i in selection]
-
-    return parents[0], parents[1]
+    return mom, dad
 
 
 def breed_and_mutate(mom, dad):
     crossover_point = randint(1, len(mom[1]) - 2)
     offspring = mom[1][:crossover_point] + dad[1][crossover_point:]
     if len(offspring) != len(mom):
-        ValueError("offspring length is not equal to mom length")
+        ValueError("offspring lenght is not equal to mom length")
 
     if random() < 0.1:
         mutation_index = randint(0, len(offspring) - 1)
@@ -57,7 +57,6 @@ def breed_and_mutate(mom, dad):
         offspring[mutation_index] = mutation
 
     offspring_score = score_field(offspring)
-    # print('offspring score', offspring_score, 'for parents', mom[0], 'and', dad[0])
 
     return [offspring_score, offspring]
 
@@ -74,10 +73,9 @@ def genetic_algorithm(population, iterations):
     for entity in population:
         entity[0] = score_field(entity[1])
 
-    for iteration in range(iterations):
+    for _ in range(iterations):
         population.sort(key=lambda x: x[0], reverse=True)
-        print('\n=====\n\n💪 Best individual in iteration', iteration, 'has a score of', population[0][0])
-        population = population[:population_size//2]
+        population = population[:5]
         new_offspring = []
 
         while len(population) + len(new_offspring) < population_size:
@@ -90,27 +88,36 @@ def genetic_algorithm(population, iterations):
     return population[0]
 
 
-population = []
-# each field has unmutable locations of water and grass tiles
-water_tile_indexes = [1, 2, 3, 34, 37, 44, 45, 53, 60, 61, 69, 81, 82, 83, 84, 119, 120, 121, 136, 152, 187, 194, 202, 203, 204, 210, 219, 226, 227, 228]
-grass_tile_indexes = [0, 39, 40, 56, 71, 72, 73, 86, 88, 114, 115, 130, 146, 147, 163, 164, 166, 167, 180, 181, 182, 231, 232, 233]
 vegetables = [x['Nazwa_warzywa'] for x in tractor_kb.query(pl.Expr("warzywo(Nazwa_warzywa)"))]
+# water tiles locations
+# _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _
+# _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _
+# _ _ X X _ _ _ _ _ _ X X X _ _ _
+# _ X X X _ _ _ _ _ X X X _ _ _ _
+# _ _ X _ _ _ _ _ _ _ _ _ _ _ _ _
+# _ _ X _ _ _ _ X X _ X X _ _ _ _
+# _ _ _ _ _ _ X X X X X X _ _ _ _
+# _ _ _ _ _ _ X X _ _ _ _ _ _ _ _
+# _ _ _ _ _ _ _ X X _ _ _ _ _ _ _
+# _ _ _ _ X _ _ _ _ _ _ _ _ _ _ _
+# _ _ _ _ X X _ _ _ _ _ _ _ _ _ _
+# _ _ _ _ X _ _ _ _ _ X _ _ _ _ _
+# _ _ _ _ _ _ _ _ X X X X _ _ _ _
+# _ _ _ X _ _ _ _ _ X X X _ _ _ _
+# _ X _ _ _ _ _ _ _ _ _ _ _ _ _ _
+# _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _
+water_tile_indexes = [37, 38, 45, 46, 47, 53, 54, 55, 61, 62, 63, 71, 88, 93, 94, 96, 97, 109, 110, 111, 112, 113, 114, 126, 127, 144, 145, 158, 175, 176, 192, 198, 213, 214, 215, 216, 225, 231, 232, 233, 240]
+population = []
 
-for _ in range(100):
+for _ in range(10):
     field = [vegetables[randint(0, 24)] for _ in range(256)]
     for index in water_tile_indexes:
         field[index] = "water"
-    for index in grass_tile_indexes:
-        field[index] = "grass"
-
     # entities of the population are stored with two properties
     # the first being the average score of the field
     # and the second being the layout of the field 
     population.append([0, field])
 
-best = genetic_algorithm(population, 20)
-print('\n=====\n\nfinal field multiplier score is', best[0])
-with open('field', 'w', encoding='utf-8') as file:
-    file.write(str(best[1]))
-file.close
-print('final field layout saved to file "field" in the current working directory\n')
+best = genetic_algorithm(population, 10)
+print('final field layout', best[1])
+print('final field multiplier score', best[0])

src/kb.py

@@ -370,7 +370,7 @@ tractor_kb([
     "przeszkadza(burak, szpinak, 0.85)",
     "przeszkadza(burak, ziemniak, 0.85)",
     "przeszkadza(cebula, fasola, 0.80)",
-    "przeszkadza(cebula, groch, 0.85)",
+    "przeszkadza(cebula, groch, 85)",
     "przeszkadza(cebula, kalafior, 0.70)",
     "przeszkadza(cebula, kapusta, 0.75)",
     "przeszkadza(cebula, marchew, 0.85)",

src/tile.py

@@ -11,26 +11,25 @@ from PIL import Image
 
 class Tile(pygame.sprite.Sprite):
 
-    def __init__(self, id, field, tile_type):
+    def __init__(self, id, field):
         super().__init__()
         self.id = id
         x = id%16
         y = id//16
 
         self.field = field
-        self.set_type(tile_type)
-        print('tile type set as', tile_type)
-        if self.type == 'water':
-            self.stage = 'no_plant'
-            self.prediction = 'water'
-            self.water_level = 100
-        elif self.type == 'grass':
-            self.stage = 'no_plant'
-            self.prediction = 'grass'
+
+        # temporary solution to have vegetables act as obstacles
+        if random.randint(1, 10) % FREE_TILES == 0:
+            vegetables = tractor_kb.query(pl.Expr("warzywo(Nazwa_warzywa)"))
+            random_vegetable = vegetables[random.randint(0, len(vegetables)-1)]['Nazwa_warzywa']
+
+            if random_vegetable in {'cebula','pietruszka','bób', 'dynia','ziemniak'}:
+                random_vegetable = 'marchew'
+
+            self.set_type(random_vegetable)
             self.water_level = random.randint(1, 5) * 10
-        else:
             self.stage = 'planted' # wczesniej to była self.faza = 'posadzono' ale stwierdzilem ze lepiej po angielsku???
-            self.water_level = random.randint(1, 5) * 10
 
             classes = [
                     "bób", "brokuł", "brukselka", "burak", "cebula",
@@ -52,7 +51,20 @@ class Tile(pygame.sprite.Sprite):
             ])
 
             self.prediction = self.predict(model, image_transforms, self.image_path, classes)
+            
+        else:
+            if random.randint(1, 10) % 3 == 0:
+                self.set_type('water')
+                self.water_level = 100
+                self.stage = 'no_plant'
+                self.prediction = 'water'
+            else:
+                self.set_type('grass')
+                self.water_level = random.randint(1, 5) * 10
+                self.stage = 'no_plant'
+                self.prediction = 'grass'
         
+
         self.rect = self.image.get_rect()
         self.rect.topleft = (x * TILE_SIZE, y * TILE_SIZE)
     
@@ -97,6 +109,9 @@ class Tile(pygame.sprite.Sprite):
 
         result = classes[predicted.item()]
         
+
+        if result == "ziemniak":
+            result = 'marchew'
         return result
 
 

src/tractor.py

@@ -165,13 +165,8 @@ class Tractor(pygame.sprite.Sprite):
             self.move()
     
     def decision_tree(self):
-        match (self.get_current_tile().type):
-            case ('grass'):
-                action = self.make_decision()
-            case ('water'):
-                action = 'nothing'
-            case _:
-                action = 'water'
+        action = self.make_decision()
+        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)
@@ -185,7 +180,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('bób')
+                self.get_current_tile().set_type('marchew')
             case ('plant(brokuł)'):
                 self.get_current_tile().set_type('brokuł')
             case ('plant(brukselka)'):
@@ -193,11 +188,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('cebula')
+                self.get_current_tile().set_type('marchew')
             case ('plant(cukinia)'):
                 self.get_current_tile().set_type('cukinia')
             case ('plant(dynia)'):
-                self.get_current_tile().set_type('dynia')
+                self.get_current_tile().set_type('fasola')
             case ('plant(fasola)'):
                self.get_current_tile().set_type('fasola')
             case ('plant(groch)'):
@@ -217,7 +212,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('pietruszka')
+                self.get_current_tile().set_type('marchew')
             case ('plant(pomidor)'):
                 self.get_current_tile().set_type('pomidor')
             case ('plant(por)'):

src/tree.csv

@@ -217,41 +217,3 @@ cebula,40,growing,szpinak,grass,kapusta,szpinak,move
 cebula,50,growing,water,szpinak,kalarepa,grass,move
 cebula,10,growing,szpinak,grass,marchew,grass,move
 cebula,20,growing,grass,cebula,burak,grass,move
-grass,10,no_plant,szpinak,dynia,szpinak,water,plant(seler)
-grass,10,no_plant,marchew,dynia,grass,water,plant(seler)
-grass,10,no_plant,water,dynia,cebula,cebula,plant(seler)
-grass,10,no_plant,ogórek,dynia,cebula,grass,plant(seler)
-grass,10,no_plant,szpinak,dynia,fasola,grass,plant(seler)
-ziemniak,40,planted,water,grass,szpinak,grass,move
-ziemniak,50,grown,szpinak,szpinak,grass,grass,move
-ziemniak,10,planted,water,ziemniak,seler,water,move
-ziemniak,20,planted,szpinak,szpinak,grass,seler,move
-ziemniak,30,planted,grass,water,ziemniak,grass,move
-ziemniak,40,planted,growing,grass,grass,burak,move
-ziemniak,50,grown,growing,szpinak,water,grass,move
-ziemniak,10,planted,growing,water,szpinak,marchew,move
-ziemniak,20,planted,growing,szpinak,grass,kapusta,move
-cukinia,50,grown,brukselka,water,grass,jarmuż,move
-dynia,0,grown,burak,grass,pietruszka,kalafior,move
-grass,10,grown,cebula,grass,pietruszka,kalarepa,move
-grass,20,grown,cukinia,groch,pietruszka,kapusta,move
-jarmuż,30,grown,grass,water,pietruszka,brokuł,move
-kalafior,40,grown,fasola,pietruszka,water,brukselka,move
-grass,10,grown,cebula,pietruszka,grass,kalarepa,move
-grass,20,grown,cukinia,pietruszka,groch,kapusta,move
-jarmuż,30,grown,grass,pietruszka,water,brokuł,move
-kalafior,40,grown,fasola,kalafior,water,brukselka,move
-bób,40,planted,jarmuż,grass,water,bób,move
-brokuł,10,growing,kalafior,brokuł,bób,kalafior,move
-brukselka,20,growing,water,brukselka,grass,bób,move
-burak,30,planted,bób,burak,brukselka,bób,move
-cebula,40,growing,brokuł,bób,burak,bób,move
-cukinia,40,growing,grass,bób,water,grass,move
-dynia,10,planted,grass,grass,cukinia,kalafior,move
-grass,40,planted,jarmuż,grass,water,bób,move
-grass,10,growing,kalafior,brokuł,bób,kalafior,move
-grass,20,growing,water,bób,grass,bób,move
-grass,30,planted,bób,burak,brukselka,bób,move
-grass,40,growing,brokuł,bób,burak,bób,move
-grass,40,growing,grass,bób,water,grass,move
-grass,10,planted,bób,grass,cukinia,kalafior,move