SZI-Smieciarka/game.py

309 lines
12 KiB
Python
Raw Normal View History

2020-03-28 12:02:27 +01:00
import pygame
2020-03-29 16:12:46 +02:00
import modele
2020-03-29 17:27:02 +02:00
import numpy as np
2020-04-03 17:39:57 +02:00
import random
2020-04-05 18:22:28 +02:00
import os
import shutil
2020-04-25 22:30:39 +02:00
import astar
2020-05-07 16:55:31 +02:00
import uczenie_kacper as kacper
2020-04-05 18:22:28 +02:00
pygame.init()
2020-03-29 16:12:46 +02:00
# kolory
2020-03-28 12:02:27 +01:00
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
GREEN = (0, 255, 0)
RED = (255, 0, 0)
2020-04-17 15:16:00 +02:00
BLUE = (40, 50, 200)
2020-03-29 05:01:12 +02:00
GREY = (128, 128, 128)
2020-03-28 12:02:27 +01:00
2020-03-29 16:12:46 +02:00
# wysokosc i szerokosc kazdej kratki
2020-03-28 12:02:27 +01:00
WIDTH = 60
HEIGHT = 60
2020-03-29 16:12:46 +02:00
# margines pomiedzy kratkami
2020-03-28 12:02:27 +01:00
MARGIN = 5
2020-03-29 17:27:02 +02:00
# rozmiar kratki
2020-03-29 17:32:48 +02:00
ILOSC_WIERSZY = 15
ILOSC_KOLUMN = 15
2020-03-29 17:27:02 +02:00
2020-03-29 16:12:46 +02:00
# rozmiar okna
WINDOW_SIZE = [1300, 980]
2020-03-28 15:34:30 +01:00
2020-04-07 20:41:17 +02:00
# 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)
2020-04-03 17:39:57 +02:00
2020-03-29 17:27:02 +02:00
def game():
obiekty = utworzObiekty()
2020-04-28 17:33:36 +02:00
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),
2020-05-07 16:55:31 +02:00
(obiekty["kontener_plastik"].x, obiekty["kontener_plastik"].y),]
2020-04-06 17:13:01 +02:00
# Petla az uzytkownik zamknie program
2020-03-29 16:12:46 +02:00
done = False
2020-04-17 15:05:28 +02:00
clock = pygame.time.Clock()
2020-03-29 16:12:46 +02:00
2020-04-06 17:13:01 +02:00
# -------- Glowna petla programu -----------
2020-03-29 16:12:46 +02:00
while not done:
2020-03-29 21:02:00 +02:00
2020-03-29 19:59:24 +02:00
# obsluga zdarzen typu nacisniecie klawisza lub przycisku myszy
2020-03-29 16:12:46 +02:00
for event in pygame.event.get(): # User did something
if event.type == pygame.QUIT: # If user clicked close
done = True # Flag that we are done so we exit this loop
elif event.type == pygame.MOUSEBUTTONDOWN:
2020-04-06 17:13:01 +02:00
# Zapisywanie pozycji myszki po kliknieciu
2020-03-29 19:59:24 +02:00
pozycja_myszki = pygame.mouse.get_pos()
2020-04-06 17:13:01 +02:00
# Zamiana pozycji na konkretne koordy
2020-03-29 19:59:24 +02:00
kolumna = pozycja_myszki[0] // (WIDTH + MARGIN)
wiersz = pozycja_myszki[1] // (HEIGHT + MARGIN)
print("Click ", pozycja_myszki,
"Grid coordinates: ", wiersz, kolumna)
2020-03-29 16:12:46 +02:00
elif event.type == pygame.KEYDOWN:
if event.key == pygame.K_LEFT:
2020-05-07 17:05:22 +02:00
obiekty["smieciarka"].w_lewo()
2020-03-29 16:12:46 +02:00
if event.key == pygame.K_RIGHT:
2020-05-07 17:05:22 +02:00
#kacper.trainModel()
obiekty["smieciarka"].w_prawo()
2020-03-29 16:12:46 +02:00
if event.key == pygame.K_UP:
obiekty["smieciarka"].w_gore()
2020-03-29 16:12:46 +02:00
if event.key == pygame.K_DOWN:
obiekty["smieciarka"].w_dol()
2020-03-29 16:12:46 +02:00
rysowaniePlanszy(obiekty)
2020-04-26 16:13:54 +02:00
while nieodwiedzone_domy:
2020-04-28 17:31:16 +02:00
nieodwiedzone_domy.sort(key=lambda x: astar.heurystyka((obiekty["smieciarka"].x, obiekty["smieciarka"].y), x))
2020-04-26 16:13:54 +02:00
cel = nieodwiedzone_domy.pop(0)
obiekty["smieciarka"].astar_move(obiekty, (obiekty["smieciarka"].x, obiekty["smieciarka"].y), cel)
2020-05-07 16:55:31 +02:00
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)
2020-04-26 16:13:54 +02:00
while nieodwiedzone_kontenery:
2020-04-28 17:31:16 +02:00
nieodwiedzone_kontenery.sort(key=lambda x: astar.heurystyka((obiekty["smieciarka"].x, obiekty["smieciarka"].y), x))
2020-04-26 16:13:54 +02:00
cel = nieodwiedzone_kontenery.pop(0)
obiekty["smieciarka"].astar_move(obiekty, (obiekty["smieciarka"].x, obiekty["smieciarka"].y), cel)
2020-05-07 16:55:31 +02:00
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)
2020-04-26 16:13:54 +02:00
2020-04-17 15:05:28 +02:00
clock.tick(7)
2020-04-25 23:46:33 +02:00
2020-03-29 16:12:46 +02:00
pygame.quit()
2020-04-17 13:56:29 +02:00
def rysowaniePlanszy(obiekty):
obiekty["obraz"].fill(BLACK)
2020-04-17 13:56:29 +02:00
# rysowanie planszy
for i in range(ILOSC_WIERSZY):
for j in range(ILOSC_KOLUMN):
pygame.draw.rect(obiekty["obraz"],
obiekty["plansza"][i, j].kolor,
[(MARGIN + WIDTH) * obiekty["plansza"][i, j].pozY + MARGIN,
2020-04-17 13:56:29 +02:00
(MARGIN + HEIGHT) *
obiekty["plansza"][i, j].pozX + MARGIN,
2020-04-17 13:56:29 +02:00
WIDTH,
HEIGHT])
obiekty["obraz"].blit(pygame.image.load(
2020-04-17 13:56:29 +02:00
"resources/plansza/wysypisko.jpg"), (5, 5))
obiekty["obraz"].blit(pygame.image.load(
2020-04-17 13:56:29 +02:00
"resources/plansza/jezioro.png"), (395, 655))
obiekty["plansza"][6, 10].setJestPrzeszkoda(True)
obiekty["plansza"][6, 11].setJestPrzeszkoda(True)
obiekty["plansza"][7, 10].setJestPrzeszkoda(True)
obiekty["plansza"][7, 11].setJestPrzeszkoda(True)
obiekty["sprajty"].draw(obiekty["obraz"])
2020-04-17 13:56:29 +02:00
text_metal, text_papier, text_plastik, text_szklo, text_pozostale, text_odwiedzone_domy = liczSmieci(
2020-05-07 16:55:31 +02:00
obiekty["domy"], obiekty)
obiekty["obraz"].blit(obiekty["text_pozostalo"],
(1020, 240))
obiekty["obraz"].blit(text_metal, (1020, 280))
obiekty["obraz"].blit(text_plastik, (1020, 320))
obiekty["obraz"].blit(text_szklo, (1020, 360))
obiekty["obraz"].blit(text_papier, (1020, 400))
2020-05-07 16:55:31 +02:00
# obiekty["obraz"].blit(text_pozostale, (1020, 440))
obiekty["obraz"].blit(text_odwiedzone_domy, (1020, 480))
2020-04-17 13:56:29 +02:00
pygame.display.update()
def utworzObiekty():
# Tworzenie planszy i kratek
plansza = np.array([[modele.Kratka(i, j) for i in range(ILOSC_KOLUMN)]
for j in range(ILOSC_WIERSZY)])
all_sprites_list = pygame.sprite.Group()
# smieciarka
smieciarka = modele.Smieciarka(10, 10)
plansza[10, 10].setKolor(BLUE)
plansza[10, 10].setObiekt(smieciarka)
# tworzenie wyswietlanego okna
os.environ['SDL_VIDEO_WINDOW_POS'] = "%d,%d" % (0, 30)
obraz = pygame.display.set_mode(WINDOW_SIZE)
smieciarka.setObraz(obraz)
pygame.display.set_caption("Inteligentna śmieciarka")
# kontenery
if os.path.exists('resources/smieci w kontenerach'):
shutil.rmtree("resources/smieci w kontenerach")
os.makedirs('resources/smieci w kontenerach')
kontener_plastik = modele.Kontener(0, 0, "plastik")
kontener_plastik.setImage(pygame.image.load(
"resources/plansza/pojemnik_plastik.png"))
plansza[0, 0].setJestKontenerem(True)
plansza[0, 0].setObiekt(kontener_plastik)
kontener_metal = modele.Kontener(0, 4, "metal")
kontener_metal.setImage(pygame.image.load(
"resources/plansza/pojemnik_metal.png"))
plansza[0, 4].setJestKontenerem(True)
plansza[0, 4].setObiekt(kontener_metal)
2020-05-07 16:55:31 +02:00
# 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"))
plansza[4, 0].setJestKontenerem(True)
plansza[4, 0].setObiekt(kontener_papier)
kontener_szklo = modele.Kontener(4, 4, "szklo")
kontener_szklo.setImage(pygame.image.load(
"resources/plansza/pojemnik_szklo.png"))
plansza[4, 4].setJestKontenerem(True)
plansza[4, 4].setObiekt(kontener_szklo)
# domy
doms_array = ['resources/plansza/domy/dom1.png', 'resources/plansza/domy/dom2.png',
'resources/plansza/domy/dom3.png', 'resources/plansza/domy/dom4.png',
'resources/plansza/domy/dom5.png']
domy_lista = pygame.sprite.Group()
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)
wspolrzedne_domow = modele.generujWspolrzedneDomow(10)
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)))
plansza[wspolrzedne_domow[i][0],
wspolrzedne_domow[i][1]].setJestDomem(True)
plansza[wspolrzedne_domow[i][0], wspolrzedne_domow[i][1]].setObiekt(dom)
domy_lista.add(dom)
all_sprites_list.add(dom)
lista_domow = domy_lista.sprites()
2020-05-07 16:55:31 +02:00
acc = 0
for d in lista_domow:
for j in range(5):
smiec = random.choice(smieci_lista)
d.dodajSmiec(smiec)
2020-05-07 16:55:31 +02:00
# 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)
2020-05-07 16:55:31 +02:00
all_sprites_list.add(kontener_plastik, kontener_metal, kontener_papier, kontener_szklo,
smieciarka)
obiekty = {
"plansza": plansza,
"smieciarka": smieciarka,
"obraz": obraz,
"kontener_plastik": kontener_plastik,
"kontener_szklo": kontener_szklo,
"kontener_metal": kontener_metal,
"kontener_papier": kontener_papier,
"sprajty": all_sprites_list,
"domy": lista_domow,
"smieci": smieci_lista,
"font": font,
"text_pozostalo": text_pozostalo,
"wspolrzedne_domow": wspolrzedne_domow
}
smieciarka.setObiekty(obiekty)
smieciarka.setPlansza(plansza)
return obiekty
def liczSmieci(domy, obiekty):
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
text_metal = obiekty["font"].render("Metal: " + str(ile_metalu), True, WHITE)
text_papier = obiekty["font"].render("Papier: " + str(ile_papieru), True, WHITE)
text_plastik = obiekty["font"].render("Plastik: " + str(ile_plastiku), True, WHITE)
text_szklo = obiekty["font"].render("Szkło: " + str(ile_szkla), True, WHITE)
text_pozostale = obiekty["font"].render(
"Pozostałe: " + str(ile_pozostalych), True, WHITE)
text_odwiedzone_domy = obiekty["font"].render(
"Odwiedzone domy: " + str(obiekty["smieciarka"].getOdwiedzoneDomy()), True, WHITE)
return text_metal, text_papier, text_plastik, text_szklo, text_pozostale, text_odwiedzone_domy