podprojekt kacper

game.py

@@ -5,6 +5,7 @@ import random
 import os
 import shutil
 import astar
+import uczenie_kacper as kacper
 
 pygame.init()
 
@@ -42,8 +43,7 @@ def game():
     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_organiczne"].x, obiekty["kontener_organiczne"].y)]
+                               (obiekty["kontener_plastik"].x, obiekty["kontener_plastik"].y),]
     # Petla az uzytkownik zamknie program
     done = False
     clock = pygame.time.Clock()
@@ -66,9 +66,10 @@ def game():
                       "Grid coordinates: ", wiersz, kolumna)
             elif event.type == pygame.KEYDOWN:
                 if event.key == pygame.K_LEFT:
-                    obiekty["smieciarka"].w_lewo()
+                    kacper.przewidz()
+
                 if event.key == pygame.K_RIGHT:
-                    obiekty["smieciarka"].w_prawo()
+                    kacper.trainModel()
                 if event.key == pygame.K_UP:
                     obiekty["smieciarka"].w_gore()
                 if event.key == pygame.K_DOWN:
@@ -79,11 +80,45 @@ def game():
             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]
+            pozY = cel[1]
+            for dom in obiekty["domy"]:
+                if dom.x == pozX and dom.y == pozY:
+                    while dom.smieci:
+                        smiec = dom.smieci.pop(0)
+                        rodzaj = kacper.przewidz(smiec)
+                        if rodzaj == "paper":
+                            obiekty["smieciarka"].dodajPapier(smiec)
+                        elif rodzaj == "glass":
+                            obiekty["smieciarka"].dodajSzklo(smiec)
+                        elif rodzaj == "metal":
+                            obiekty["smieciarka"].dodajMetal(smiec)
+                        elif rodzaj == "plastic":
+                            obiekty["smieciarka"].dodajPlastik(smiec)
 
         while nieodwiedzone_kontenery:
             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]
+            pozY = cel[1]
+
+            if obiekty["plansza"][pozX, pozY].obiekt == obiekty["kontener_papier"]:
+                while obiekty["smieciarka"].papier:
+                    smiec = obiekty["smieciarka"].papier.pop(0)
+                    obiekty["kontener_papier"].dodajSmiec(smiec)
+            elif obiekty["plansza"][pozX, pozY].obiekt == obiekty["kontener_szklo"]:
+                while obiekty["smieciarka"].szklo:
+                    smiec = obiekty["smieciarka"].szklo.pop(0)
+                    obiekty["kontener_szklo"].dodajSmiec(smiec)
+            elif obiekty["plansza"][pozX, pozY].obiekt == obiekty["kontener_metal"]:
+                while obiekty["smieciarka"].metal:
+                    smiec = obiekty["smieciarka"].metal.pop(0)
+                    obiekty["kontener_metal"].dodajSmiec(smiec)
+            elif obiekty["plansza"][pozX, pozY].obiekt == obiekty["kontener_plastik"]:
+                while obiekty["smieciarka"].plastik:
+                    smiec = obiekty["smieciarka"].plastik.pop(0)
+                    obiekty["kontener_plastik"].dodajSmiec(smiec)
 
         clock.tick(7)
 
@@ -120,7 +155,7 @@ def rysowaniePlanszy(obiekty):
     obiekty["obraz"].blit(text_plastik, (1020, 320))
     obiekty["obraz"].blit(text_szklo, (1020, 360))
     obiekty["obraz"].blit(text_papier, (1020, 400))
-    obiekty["obraz"].blit(text_pozostale, (1020, 440))
+    # obiekty["obraz"].blit(text_pozostale, (1020, 440))
     obiekty["obraz"].blit(text_odwiedzone_domy, (1020, 480))
 
     pygame.display.update()
@@ -159,11 +194,11 @@ 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_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(
@@ -203,18 +238,25 @@ 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)
-            smieci_lista.remove(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):
         for j in range(5):
             plansza[i, j].setJestWysypiskiem(True)
 
-    all_sprites_list.add(kontener_plastik, kontener_metal, kontener_organiczne, kontener_papier, kontener_szklo,
+    all_sprites_list.add(kontener_plastik, kontener_metal, kontener_papier, kontener_szklo,
                          smieciarka)
 
     obiekty = {
@@ -224,7 +266,6 @@ def utworzObiekty():
         "kontener_plastik": kontener_plastik,
         "kontener_szklo": kontener_szklo,
         "kontener_metal": kontener_metal,
-        "kontener_organiczne": kontener_organiczne,
         "kontener_papier": kontener_papier,
         "sprajty": all_sprites_list,
         "domy": lista_domow,
@@ -254,8 +295,6 @@ def liczSmieci(domy, obiekty):
                 ile_plastiku += 1
             elif "glass" in s:
                 ile_szkla += 1
-            elif "trash" in s:
-                ile_pozostalych += 1
 
     text_metal = obiekty["font"].render("Metal: " + str(ile_metalu), True, WHITE)
     text_papier = obiekty["font"].render("Papier: " + str(ile_papieru), True, WHITE)

requirements.txt

@@ -1,2 +1,2 @@
 pygame==1.9.6
-numpy==1.18.2
+numpy==1.18