dodany wybór osoby

game.py

@@ -7,6 +7,18 @@ import shutil
 import astar
 import uczenie_kacper as kacper
 
+smieci_path = ''
+
+# PODAJ OSOBE PRZED URUCHOMIENIEM (kacper/adamB/adamO)
+osoba = 'kacper'
+
+if osoba == 'kacper':
+    smieci_path = 'resources/smieci'
+elif osoba == 'adamB':
+    smieci_path = 'resources/smieci_stare'
+else:
+    smieci_path = 'resources/smieci_stare'
+
 pygame.init()
 
 # kolory
@@ -32,18 +44,13 @@ ILOSC_KOLUMN = 15
 WINDOW_SIZE = [1300, 980]
 
 
-# TEST DODAWANIA SMIECI Z POJEDYNCZEGO DOMU DO FOLDERU KONTENERA ZE SZKLEM
-# for d in lista_domow:
-#     for s in d.smieci:
-#         kontener_szklo.dodajSmiec(s)
-
 def game():
     obiekty = utworzObiekty()
     nieodwiedzone_domy = obiekty["wspolrzedne_domow"]
     nieodwiedzone_kontenery = [(obiekty["kontener_szklo"].x, obiekty["kontener_szklo"].y),
                                (obiekty["kontener_papier"].x, obiekty["kontener_papier"].y),
                                (obiekty["kontener_metal"].x, obiekty["kontener_metal"].y),
-                               (obiekty["kontener_plastik"].x, obiekty["kontener_plastik"].y),]
+                               (obiekty["kontener_plastik"].x, obiekty["kontener_plastik"].y), ]
     # Petla az uzytkownik zamknie program
     done = False
     clock = pygame.time.Clock()
@@ -52,9 +59,9 @@ def game():
     while not done:
 
         # obsluga zdarzen typu nacisniecie klawisza lub przycisku myszy
-        for event in pygame.event.get():  # User did something
+        for event in pygame.event.get():
-            if event.type == pygame.QUIT:  # If user clicked close
+            if event.type == pygame.QUIT:
-                done = True  # Flag that we are done so we exit this loop
+                done = True
             elif event.type == pygame.MOUSEBUTTONDOWN:
                 # Zapisywanie pozycji myszki po kliknieciu
                 pozycja_myszki = pygame.mouse.get_pos()
@@ -68,7 +75,7 @@ def game():
                 if event.key == pygame.K_LEFT:
                     obiekty["smieciarka"].w_lewo()
                 if event.key == pygame.K_RIGHT:
-                    #kacper.trainModel()
+                    # kacper.trainModel()
                     obiekty["smieciarka"].w_prawo()
                 if event.key == pygame.K_UP:
                     obiekty["smieciarka"].w_gore()
@@ -77,7 +84,8 @@ def game():
 
         rysowaniePlanszy(obiekty)
         while nieodwiedzone_domy:
-            nieodwiedzone_domy.sort(key=lambda x: astar.heurystyka((obiekty["smieciarka"].x, obiekty["smieciarka"].y), x))
+            nieodwiedzone_domy.sort(
+                key=lambda x: astar.heurystyka((obiekty["smieciarka"].x, obiekty["smieciarka"].y), x))
             cel = nieodwiedzone_domy.pop(0)
             obiekty["smieciarka"].astar_move(obiekty, (obiekty["smieciarka"].x, obiekty["smieciarka"].y), cel)
             pozX = cel[0]
@@ -86,7 +94,15 @@ def game():
                 if dom.x == pozX and dom.y == pozY:
                     while dom.smieci:
                         smiec = dom.smieci.pop(0)
+
+                        rodzaj = ""
+                        if osoba == 'kacper':
                             rodzaj = kacper.przewidz(smiec)
+                        elif osoba == 'adamB':
+                            pass
+                        else:
+                            pass
+
                         if rodzaj == "paper":
                             obiekty["smieciarka"].dodajPapier(smiec)
                         elif rodzaj == "glass":
@@ -97,7 +113,8 @@ def game():
                             obiekty["smieciarka"].dodajPlastik(smiec)
 
         while nieodwiedzone_kontenery:
-            nieodwiedzone_kontenery.sort(key=lambda x: astar.heurystyka((obiekty["smieciarka"].x, obiekty["smieciarka"].y), x))
+            nieodwiedzone_kontenery.sort(
+                key=lambda x: astar.heurystyka((obiekty["smieciarka"].x, obiekty["smieciarka"].y), x))
             cel = nieodwiedzone_kontenery.pop(0)
             obiekty["smieciarka"].astar_move(obiekty, (obiekty["smieciarka"].x, obiekty["smieciarka"].y), cel)
             pozX = cel[0]
@@ -194,12 +211,6 @@ def utworzObiekty():
     plansza[0, 4].setJestKontenerem(True)
     plansza[0, 4].setObiekt(kontener_metal)
 
-    # kontener_organiczne = modele.Kontener(2, 2, "pozostale")
-    # kontener_organiczne.setImage(pygame.image.load(
-    #     "resources/plansza/pojemnik_organiczne.png"))
-    # plansza[2, 2].setJestKontenerem(True)
-    # plansza[2, 2].setObiekt(kontener_organiczne)
-
     kontener_papier = modele.Kontener(4, 0, "papier")
     kontener_papier.setImage(pygame.image.load(
         "resources/plansza/pojemnik_papier.png"))
@@ -220,7 +231,7 @@ def utworzObiekty():
     domy_lista = pygame.sprite.Group()
 
     smieci_lista = [os.path.join(path, filename)
-                    for path, dirs, files in os.walk("resources/smieci")
+                    for path, dirs, files in os.walk(smieci_path)
                     for filename in files]
 
     # informacje o ilosci smieci w domach
@@ -238,18 +249,10 @@ def utworzObiekty():
         all_sprites_list.add(dom)
 
     lista_domow = domy_lista.sprites()
-    acc = 0
     for d in lista_domow:
         for j in range(5):
             smiec = random.choice(smieci_lista)
             d.dodajSmiec(smiec)
-    #         print('')
-    #         print(smiec)
-    #         print(kacper.przewidz(smiec))
-    #         if kacper.przewidz(smiec) in smiec:
-    #             acc += 1
-    #         smieci_lista.remove(smiec)
-    # print(acc)
 
     # ustawienie wysypiska, rozmiar wysypiska 5x5
     for i in range(5):

modele.py

@@ -58,7 +58,7 @@ class Smieciarka(pygame.sprite.Sprite):
     def astar_move(self, obiekty, start, cel):
 
         sciezka = astar.astar(obiekty, start, (cel[0], cel[1]))
-        print(sciezka)
+        # print(sciezka)
         if sciezka:
             for koord in sciezka:
                 if koord[0] == self.x - 1 and koord[1] == self.y: