dodanie informacji o pozostalych smieciach

game.py

@@ -5,6 +5,8 @@ import random
 import os
 import shutil
 
+pygame.init()
+
 # kolory
 BLACK = (0, 0, 0)
 WHITE = (255, 255, 255)
@@ -25,7 +27,7 @@ ILOSC_WIERSZY = 15
 ILOSC_KOLUMN = 15
 
 # rozmiar okna
-WINDOW_SIZE = [980, 980]
+WINDOW_SIZE = [1300, 980]
 
 # Tworzenie planszy i kratek
 plansza = np.array([[modele.Kratka(i, j) for i in range(ILOSC_KOLUMN)] for j in range(ILOSC_WIERSZY)])
@@ -83,6 +85,37 @@ smieci_lista = [os.path.join(path, filename)
                 for path, dirs, files in os.walk("resources/smieci")
                 for filename in files]
 
+# informacje o ilosci smieci w domach
+font = pygame.font.SysFont("arial", 20, bold = True)
+text_pozostalo = font.render("Pozostało śmieci w domach:", True, WHITE)
+
+def liczSmieci(domy):
+    ile_metalu = 0
+    ile_szkla = 0
+    ile_papieru = 0
+    ile_plastiku = 0
+    ile_pozostalych = 0
+    for d in domy:
+        for s in d.smieci:
+            if "metal" in s:
+                ile_metalu += 1
+            elif "paper" in s:
+                ile_papieru += 1
+            elif "plastic" in s:
+                ile_plastiku += 1
+            elif "glass" in s:
+                ile_szkla += 1
+            elif "trash" in s:
+                ile_pozostalych += 1
+
+    text_metal = font.render("Metal: " + str(ile_metalu), True, WHITE)
+    text_papier = font.render("Papier: " + str(ile_papieru), True, WHITE)
+    text_plastik = font.render("Plastik: " + str(ile_plastiku), True, WHITE)
+    text_szklo = font.render("Szkło: " + str(ile_szkla), True, WHITE)
+    text_pozostale = font.render("Pozostałe: " + str(ile_pozostalych), True, WHITE)
+
+    return text_metal, text_papier, text_plastik, text_szklo, text_pozostale;
+
 for i in range(len(wspolrzedne_domow)):
     dom = modele.Dom(wspolrzedne_domow[i][0], wspolrzedne_domow[i][1])
     dom.setImage(pygame.image.load(random.choice(doms_array)))
@@ -91,15 +124,17 @@ for i in range(len(wspolrzedne_domow)):
     domy_lista.add(dom)
     all_sprites_list.add(dom)
 
-for d in domy_lista.sprites():
+lista_domow = domy_lista.sprites()
+for d in lista_domow:
     for j in range(5):
         smiec = random.choice(smieci_lista)
         d.dodajSmiec(smiec)
         smieci_lista.remove(smiec)
 
 ########## TEST DODAWANIA SMIECI Z POJEDYNCZEGO DOMU DO FOLDERU KONTENERA ZE SZKLEM
-# for u in domy_lista.sprites()[0].smieci:
+# for d in lista_domow:
-#     kontener_szklo.dodajSmiec(u)
+#     for s in d.smieci:
+#         kontener_szklo.dodajSmiec(s)
 
 # ustawienie wysypiska, rozmiar wysypiska 5x5
 for i in range(5):
@@ -111,8 +146,6 @@ all_sprites_list.add(kontener_plastik, kontener_metal, kontener_organiczne, kont
 
 
 def game():
-    pygame.init()
-
     obraz = pygame.display.set_mode(WINDOW_SIZE)
 
     # Tytul okienka
@@ -139,7 +172,6 @@ def game():
 
                 print("Click ", pozycja_myszki,
                       "Grid coordinates: ", wiersz, kolumna)
-                # plansza[wiersz, kolumna].setKolor(BLUE)
 
             elif event.type == pygame.KEYDOWN:
                 if event.key == pygame.K_LEFT:
@@ -179,5 +211,14 @@ def game():
         all_sprites_list.draw(obraz)
         smieciarka.rand_move()
 
+        text_metal, text_papier, text_plastik, text_szklo, text_pozostale = liczSmieci(lista_domow)
+        obraz.blit(text_pozostalo,
+                   (1020, 240))
+        obraz.blit(text_metal, (1020, 280))
+        obraz.blit(text_plastik, (1020, 320))
+        obraz.blit(text_szklo, (1020, 360))
+        obraz.blit(text_papier, (1020, 400))
+        obraz.blit(text_pozostale, (1020, 440))
+
         pygame.display.flip()
     pygame.quit()

modele.py

@@ -153,7 +153,7 @@ class Kratka(pygame.sprite.Sprite):
     def setImage(self, image):
         self.image = image
 
-    def setObiekt(self,obiekt):
+    def setObiekt(self, obiekt):
         self.obiekt = obiekt
 
     def setJestDomem(self, bool):
@@ -184,7 +184,7 @@ def generujWspolrzedneDomow(ilosc_domow):
         wspolrzedne_domu = []
         x = random.randrange(15)
         if x < 5:
-            y = random.randrange(5,15)
+            y = random.randrange(5, 15)
         else:
             y = random.randrange(15)
         wspolrzedne_domu.append(x)