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()