255 lines
8.7 KiB
Python
255 lines
8.7 KiB
Python
import ctypes
|
|
import os
|
|
import random
|
|
import threading
|
|
|
|
import pygame.transform
|
|
|
|
import szafka2
|
|
from agent import *
|
|
from neural_network import *
|
|
from okno import *
|
|
from ramy_czyli_wiedza_agenta import *
|
|
from rescue import *
|
|
|
|
# aby działalo w oknie + rozdzielczość ekranu
|
|
# ctypes.windll.shcore.SetProcessDpiAwareness(1)
|
|
|
|
okno_pygame = pygame.display.set_mode((SZEROKOSC_OKNA, WYSOKOSC_OKNA))
|
|
pygame.display.set_caption("Okno1")
|
|
krata_magazynu = Krata(okno_pygame)
|
|
Pomieszczenie = Pomieszczenie(WarunkiPowietrza(0, 0), 0)
|
|
|
|
|
|
def dodaj_agenta():
|
|
# pole_lewe_gorne = PoleKraty(krata_magazynu, random.randint(0, LICZBA_POL_W_PIONIE - BOK_AGENTA1_W_POLACH),
|
|
# random.randint(0, LICZBA_POL_W_POZIOMIE - BOK_AGENTA1_W_POLACH))
|
|
# pole_lewe_gorne = PoleKraty(krata_magazynu, LICZBA_POL_W_PIONIE - BOK_AGENTA1_W_POLACH, int(LICZBA_POL_W_POZIOMIE / 2))
|
|
pole_lewe_gorne = PoleKraty(krata_magazynu, 0, 0)
|
|
pom = 'wozek_widlowy.png'
|
|
ikona = pygame.transform.scale(pygame.image.load(os.path.join('Ikony', pom)),
|
|
(BOK_AGENTA1, BOK_AGENTA1))
|
|
Agent(krata_magazynu, pole_lewe_gorne, ikona)
|
|
# zawsze na poczatku polnoc
|
|
krata_magazynu.agent.kierunek = Kierunek.POLNOC
|
|
|
|
|
|
def dodaj_szafke(numerSzafki, iloscPolek, iloscMiejscNaPolce, dostepZeStrony, poczatek_wiersz, poczatek_kolumna):
|
|
wymiary_szafki = Wymiary(0, 0, 0)
|
|
szafka = Szafka(numerSzafki, wymiary_szafki, iloscPolek, iloscMiejscNaPolce, dostepZeStrony, poczatek_wiersz,
|
|
poczatek_kolumna, krata_magazynu)
|
|
Pomieszczenie.dodajSzafke(szafka)
|
|
|
|
|
|
def gdzie_paczka(numerSzafki):
|
|
global kolumna, wiersz
|
|
if numerSzafki == 1:
|
|
kolumna = random.randint(1, 2)
|
|
if kolumna == 1:
|
|
wiersz = random.randint(2, 13)
|
|
else:
|
|
wiersz = random.choice([1, 14])
|
|
if numerSzafki == 2:
|
|
kolumna = random.randint(3, 4)
|
|
if kolumna == 3:
|
|
wiersz = random.choice([1, 14])
|
|
else:
|
|
wiersz = random.randint(2, 13)
|
|
if numerSzafki == 3:
|
|
kolumna = random.randint(5, 8)
|
|
if kolumna == 5:
|
|
wiersz = random.randint(1, 5)
|
|
if kolumna == 6:
|
|
wiersz = random.choice([0, 6])
|
|
if kolumna == 7:
|
|
wiersz = random.choice([0, 7])
|
|
if kolumna == 8:
|
|
wiersz = random.randint(1, 6)
|
|
if numerSzafki == 4:
|
|
kolumna = random.randint(5, 8)
|
|
if kolumna == 5:
|
|
wiersz = random.randint(9, 12)
|
|
if kolumna == 6:
|
|
wiersz = random.choice([8, 13])
|
|
if kolumna == 7:
|
|
wiersz = random.choice([9, 13])
|
|
if kolumna == 8:
|
|
wiersz = random.randint(10, 12)
|
|
if numerSzafki == 5:
|
|
kolumna = random.randint(10, 12)
|
|
if kolumna == 11:
|
|
wiersz = random.choice([2, 11])
|
|
else:
|
|
wiersz = random.randint(3, 10)
|
|
if numerSzafki == 6:
|
|
kolumna = random.randint(12, 14)
|
|
if kolumna == 13:
|
|
wiersz = random.choice([1, 12])
|
|
else:
|
|
wiersz = random.randint(2, 11)
|
|
if numerSzafki == 7:
|
|
kolumna = random.randint(17, 19)
|
|
if kolumna == 18:
|
|
wiersz = random.choice([0, 13])
|
|
else:
|
|
wiersz = random.randint(1, 12)
|
|
if numerSzafki == 8:
|
|
kolumna = random.randint(22, 24)
|
|
if kolumna == 23:
|
|
wiersz = random.choice([4, 12])
|
|
else:
|
|
wiersz = random.randint(5, 11)
|
|
if numerSzafki == 9:
|
|
kolumna = random.randint(24, 26)
|
|
if kolumna == 25:
|
|
wiersz = random.choice([0, 13])
|
|
else:
|
|
wiersz = random.randint(1, 12)
|
|
if numerSzafki == 10:
|
|
kolumna = random.randint(27, 29)
|
|
if kolumna == 28:
|
|
wiersz = random.choice([4, 14])
|
|
else:
|
|
wiersz = random.randint(5, 13)
|
|
if numerSzafki == 11:
|
|
kolumna = 0
|
|
wiersz = 0
|
|
print(wiersz, kolumna)
|
|
return wiersz, kolumna
|
|
|
|
|
|
def ustawienie():
|
|
ostatnia = recognition()
|
|
print(ostatnia)
|
|
kierunek = Kierunek(random.randint(0, 3))
|
|
wiersz, kolumna = gdzie_paczka(ostatnia + 1)
|
|
return Stan(kierunek, PoleKraty(krata_magazynu, wiersz, kolumna))
|
|
|
|
|
|
def wroc():
|
|
kierunek = Kierunek(random.randint(0, 3))
|
|
wiersz, kolumna = gdzie_paczka(11)
|
|
return Stan(kierunek, PoleKraty(krata_magazynu, wiersz, kolumna))
|
|
|
|
|
|
def zaznacz_cel_na_mapie(cel: Stan):
|
|
wiersz = cel.poleStartoweGorne.wiersz
|
|
kolumna = cel.poleStartoweGorne.kolumna
|
|
if krata_magazynu.krata[wiersz][kolumna] == ZawartoscPola.PUSTE:
|
|
krata_magazynu.krata[wiersz][kolumna] = ZawartoscPola.CEL
|
|
|
|
|
|
def nadaj_cel_agentowi(agent: Agent, cel: Stan):
|
|
agent.cel = cel
|
|
zaznacz_cel_na_mapie(agent.cel)
|
|
print("CEL:", agent.cel.poleStartoweGorne.wiersz, agent.cel.poleStartoweGorne.kolumna)
|
|
|
|
|
|
def zdarzenie_osoba():
|
|
global flaga1
|
|
flaga1 = 1
|
|
|
|
|
|
def losowa_osoba():
|
|
wiersz = random.randint(0, krata_magazynu.liczbaPolPionowo - 1)
|
|
kolumna = random.randint(0, krata_magazynu.liczbaPolPoziomo - 1)
|
|
osoba = PoleKraty(krata_magazynu, wiersz, kolumna)
|
|
return osoba
|
|
|
|
|
|
def main():
|
|
# dla kraty 30 x 15
|
|
dodaj_szafke("A", 2, 12, "P", 2, 2)
|
|
dodaj_szafke("B", 2, 12, "L", 2, 3)
|
|
dodaj_szafke("C", 2, 5, "P", 1, 6)
|
|
dodaj_szafke("D", 2, 6, "L", 1, 7)
|
|
dodaj_szafke("C", 2, 4, "P", 9, 6)
|
|
dodaj_szafke("D", 2, 3, "L", 10, 7)
|
|
dodaj_szafke("E", 2, 8, "P", 3, 11)
|
|
dodaj_szafke("F", 2, 10, "L", 2, 13)
|
|
dodaj_szafke("H", 2, 12, "L", 1, 18)
|
|
dodaj_szafke("I", 2, 7, "P", 5, 23)
|
|
dodaj_szafke("J", 2, 12, "L", 1, 25)
|
|
dodaj_szafke("G", 2, 9, "P", 5, 28)
|
|
|
|
# # dla kraty 10 x 10
|
|
# dodaj_szafke("A", 1, 8, "P", 1, 1)
|
|
# dodaj_szafke("B", 1, 8, "L", 1, 4)
|
|
# dodaj_szafke("C", 1, 8, "P", 1, 6)
|
|
# dodaj_szafke("C", 1, 8, "P", 1, 8)
|
|
|
|
# # dla kraty 5 x 5
|
|
# dodaj_szafke("A", 1, 3, "P", 1, 1)
|
|
# dodaj_szafke("B", 1, 3, "L", 1, 3)
|
|
|
|
for i in (
|
|
(1, 10), (1, 3), (3, 23), (2, 23), (5, 15), (4, 15), (9, 12), (11, 20), (11, 27), (11, 26), (14, 19),
|
|
(14, 18),
|
|
(14, 20), (8, 29), (9, 29)):
|
|
krata_magazynu.krata[i[0]][i[1]] = ZawartoscPola.DYWAN
|
|
for i in (
|
|
(0, 10), (13, 20), (13, 6), (13, 14), (14, 13), (9, 26), (9, 16), (9, 15), (9, 27), (9, 16), (9, 26),
|
|
(5, 8),
|
|
(5, 9), (7, 9), (7, 10)):
|
|
krata_magazynu.krata[i[0]][i[1]] = ZawartoscPola.KALUZA
|
|
|
|
for i in range(LICZBA_POL_W_PIONIE):
|
|
krata_magazynu.krata[i][21] = ZawartoscPola.SCIANA2
|
|
krata_magazynu.krata[0][21] = ZawartoscPola.PUSTE
|
|
krata_magazynu.krata[7][21] = ZawartoscPola.PUSTE
|
|
krata_magazynu.krata[14][21] = ZawartoscPola.PUSTE
|
|
|
|
dodaj_agenta()
|
|
okno1 = Okno(krata_magazynu, krata_magazynu.agent)
|
|
okno1.wyswietlOkno()
|
|
|
|
t = threading.Timer(5.0, zdarzenie_osoba).start()
|
|
osoba = PoleKraty(krata_magazynu, 0, 0)
|
|
clf = drzewo_decyzyjne()
|
|
global flaga1
|
|
flaga1 = 0
|
|
|
|
while True:
|
|
# cel to Stan (pole kraty gdzie ma stać agent, aby położyć paczkę na półkę, w obiekcie klasy Miejsce jest to artybut dostęp + kierunek <-na razie niepotrzebny)
|
|
if krata_magazynu.agent.cel is None:
|
|
lista_paczek = []
|
|
#cyfra 5 oznacza ilosc paczek
|
|
for i in range(5):
|
|
cel = ustawienie()
|
|
wiersz_paczki = cel.poleStartoweGorne.wiersz
|
|
kolumna_paczki = cel.poleStartoweGorne.kolumna
|
|
lista_paczek.append((wiersz_paczki, kolumna_paczki))
|
|
print(lista_paczek)
|
|
|
|
# powrot
|
|
cel = wroc()
|
|
nadaj_cel_agentowi(krata_magazynu.agent, cel)
|
|
krata_magazynu.agent.idzDoCelu()
|
|
|
|
if flaga1 == 1:
|
|
osoba.krata.krata[osoba.wiersz][osoba.kolumna] = ZawartoscPola.PUSTE
|
|
okno1.wyswietlOkno()
|
|
osoba = losowa_osoba()
|
|
while osoba.krata.krata[osoba.wiersz][osoba.kolumna] != ZawartoscPola.PUSTE:
|
|
osoba = losowa_osoba()
|
|
osoba.krata.krata[osoba.wiersz][osoba.kolumna] = ZawartoscPola.OSOBA
|
|
okno1.wyswietlOkno()
|
|
pygame.time.wait(1000)
|
|
answer = decyzja_osoba(osoba, clf)
|
|
if answer == 1:
|
|
osoba.krata.krata[osoba.wiersz][osoba.kolumna] = ZawartoscPola.PUSTE
|
|
okno1.wyswietlOkno()
|
|
bieg = pygame.transform.scale(pygame.image.load(os.path.join('Ikony', 'bieg.png')),
|
|
(BOK_AGENTA1, BOK_AGENTA1))
|
|
okno_pygame.blit(bieg, (osoba.kolumna * (BOK_POLA + 1) + 1, osoba.wiersz * (BOK_POLA + 1) + 1))
|
|
pygame.display.flip()
|
|
pygame.time.wait(1500)
|
|
flaga1 = 0
|
|
t = threading.Timer(5.0, zdarzenie_osoba).start()
|
|
|
|
|
|
try:
|
|
main()
|
|
except pygame.error:
|
|
pygame.quit()
|