forked from s444399/AI
41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
|
import pygame
|
||
|
|
from os import sys
|
||
|
|
from generate import Generate
|
||
|
|
from floor import Floor
|
||
|
|
from wall import Wall
|
||
|
|
from shelf import Shelf
|
||
|
|
from wheelchair import Wheelchair
|
||
|
|
|
||
|
|
class MainWindow:
|
||
|
|
def __init__(self, szerokosc, wysokosc, kruche, latwopalne, radioaktywne, niebezpieczne):
|
||
|
|
#config
|
||
|
|
self.cell = 50
|
||
|
|
#init
|
||
|
|
pygame.init()
|
||
|
|
pygame.display.set_caption('Inteligentny wózek widłowy')
|
||
|
|
self.map = Generate.generate(szerokosc+2, wysokosc+2, kruche, latwopalne, radioaktywne, niebezpieczne)
|
||
|
|
self.screen = pygame.display.set_mode((((szerokosc+2)*self.cell), ((wysokosc+2)*self.cell)))
|
||
|
|
#create
|
||
|
|
for i in range(len(self.map)):
|
||
|
|
for j in range(len(self.map[i])):
|
||
|
|
if (self.map[i][j]==1):
|
||
|
|
self.map[i][j] = Wall(self.screen, self.cell, i, j)
|
||
|
|
elif (self.map[i][j]==2):
|
||
|
|
self.map[i][j] = Floor(self.screen, self.cell, i, j)
|
||
|
|
else:
|
||
|
|
self.map[i][j] = Shelf(self.screen, self.cell, i, j, (self.map[i][j]-3)%4, (self.map[i][j]-3)//4)
|
||
|
|
#loop
|
||
|
|
while True:
|
||
|
|
self.events()
|
||
|
|
self.draw()
|
||
|
|
|
||
|
|
def events(self):
|
||
|
|
for event in pygame.event.get():
|
||
|
|
if(event.type==pygame.QUIT):
|
||
|
|
sys.exit()
|
||
|
|
def draw(self):
|
||
|
|
self.screen.fill((255,255,255))
|
||
|
|
for i in range(len(self.map)):
|
||
|
|
for j in range(len(self.map[i])):
|
||
|
|
self.map[i][j].draw()
|
||
|
|
pygame.display.flip()
|