54 lines
2.1 KiB
Python
54 lines
2.1 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 wheel import Wheel
|
||
|
|
from boxOnTheFloor import BoxOnTheFloor
|
||
|
|
from box import Box
|
||
|
|
from unboxOnTheFloor import UnboxOnTheFloor
|
||
|
|
|
||
|
|
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
|
||
|
|
self.wheel = Wheel(self.screen, self.cell);
|
||
|
|
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)
|
||
|
|
elif (self.map[i][j]==23):
|
||
|
|
self.map[i][j] = UnboxOnTheFloor(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()
|
||
|
|
elif(event.type==pygame.KEYDOWN):
|
||
|
|
self.wheel.move(event, self.map)
|
||
|
|
elif(event.type==pygame.MOUSEBUTTONDOWN):
|
||
|
|
if (type(self.map[0][2]) == Floor):
|
||
|
|
self.map[0][2] = BoxOnTheFloor(self.screen, self.cell, 0, 2, Box())
|
||
|
|
def draw(self):
|
||
|
|
self.screen.fill((33,69,108))
|
||
|
|
for i in range(len(self.map)):
|
||
|
|
for j in range(len(self.map[i])):
|
||
|
|
self.map[i][j].draw()
|
||
|
|
self.wheel.draw()
|
||
|
|
pygame.display.flip()
|