import pygame, sys
from waiter import Waiter
from table import Table
from gridElement import GridElement


class Game(object):
    def __init__(self):
        pygame.init()

        self.screen = pygame.display.set_mode((1000, 900))
        self.fpsClock = pygame.time.Clock()
        self.idTable = 0
        self.idOrder = 0
        self.idItem = -1
        self.waiterNumberInGrid = 0
        pygame.display.set_caption('Automatic Waiter')
        self.background = pygame.image.load("./Images/tlo.jpg")

        # The most important list
        self.grid = []


        for b in range(18):
            for i in range(20):
                if ((i == 0) or (i == 8) or (i == 16)) and ((b == 2) or (b == 8)):
                    self.grid.append(Table(i * 50, b * 50, self))
                elif i == 18 and b == 14:
                    self.grid.append(Waiter(self, i * 50, b * 50))
                    self.waiterNumberInGrid = (b * 20) + i
                else:
                    self.grid.append(GridElement(i * 50, b * 50, self))

        while True:

            for event in pygame.event.get():
                if event.type == pygame.QUIT:
                    sys.exit(0)

            self.draw()
            self.grid[self.waiterNumberInGrid].move(self)
            pygame.display.flip()
            print(self.waiterNumberInGrid)
            self.fpsClock.tick(10)

    def draw(self):

        tempTable = []
        for gridElement in self.grid:
            if gridElement.type == "gridElement":
                gridElement.draw()
            else:
                tempTable.append(gridElement)
        for otherElement in tempTable:
            otherElement.draw()


if __name__ == "__main__":
    Game()