ekran.py

@@ -50,11 +50,10 @@ def zwroc_regaly_kategoria(kategoria):
             lista_reg.append(reg)
     return lista_reg
 
-def zwroc_regaly_wspolrzedne(x_reg, y_reg):
+# def zwroc_regaly_wspolrzedne(x_reg, y_reg):
-    for regal in lista_regalow:
+#     for regal in lista_regalow:
-        if regal.wiersz == obliczPixeleNaPodstawieKratek(x_reg) and regal.kolumna == obliczPixeleNaPodstawieKratek(y_reg):
+#         if regal.wiersz == obliczPixeleNaPodstawieKratek(x_reg) and regal.kolumna == obliczPixeleNaPodstawieKratek(y_reg):
-            return regal 
+#             return regal 
-
 
 def narysuj_siatke():
     blockSize = 70  # Set the size of the grid block

main.py

@@ -15,7 +15,7 @@ pygame.init()
 def main():
     wozek = Wozek()
 
-    paczka = Paczka('duzy', 10, 'ogród', False, True, False, any, any, any, any, any)
+    paczka = Paczka('duzy', 10, 'narzedzia', False, True, False, any, any, any, any, any)
     ekran.lista_paczek.add(paczka)
     paczka.update_position(a_pix, b_pix)
 
@@ -43,12 +43,11 @@ def main():
                         if wozek.ln == 0:
                             wozek.load_package()
                             wozek.dynamic_wozek_picture
-                            print(wozek.storage)
 
                         ## wozek jedzie odlozyc paczke na regal
-                            paczka1 = wozek.storage.pop()
+                            paczka1 = wozek.storage[0]
 
-                            array, reg = paczka1.tablica_do_drzewa('ogród')
+                            array, reg = paczka1.tablica_do_drzewa(paczka1.kategoria)
 
                             plikZPrzecinkami = open("DecisionTree/training_data.txt", 'w')
 
@@ -83,9 +82,11 @@ def main():
                             wozek.przemiesc_wozek_po_sciezce(sciezka)
 
                             # print(wozek.storage)
-                            # if wozek.ln != 0:   # drops package
+                            if wozek.ln != 0:   # drops package
-                            #     wozek.drop_package()
+                                wozek.drop_package()
-                            #     wozek.dynamic_wozek_picture()
+                                reg.put_package_on_the_regal(paczka1, predictions)
+                                wozek.dynamic_wozek_picture()
+
 
             if event.type == pygame.MOUSEBUTTONDOWN:
                 # lewy przycisk myszy

paczka.py

@@ -72,7 +72,6 @@ class Paczka(pygame.sprite.Sprite):
     
     def tablica_do_drzewa(self, kategoria):
         tablica = []
-
         # rozmiar
         if self.rozmiar == 'maly':
             tablica.append(0)
@@ -103,7 +102,6 @@ class Paczka(pygame.sprite.Sprite):
         else: tablica.append(1)
 
         reg = ekran.zwroc_regaly_kategoria(kategoria)
-
         # czy dolna wolna
         if reg[0].is_dolna_free() is True:
             tablica.append(0)

regal.py

@@ -85,5 +85,11 @@ class Regal(pygame.sprite.Sprite):
             return True
         return False
     
+    def put_package_on_the_regal(self, package, where):
+        if(where == 0):
+            self.dolna.append(package)
+        else:
+            self.gorna.append(package)
+