import pygame, sys
from waiter import Waiter
from table import Table
from gridElement import GridElement
from pygame.math import Vector2
from orderTable import OrderTable
from pool import Pool
import copy

class Game(object):
    def __init__(self):
        pygame.init()
        self.x = 10
        self.y = 10
        self.screen = pygame.display.set_mode((self.x * 50, self.y * 50))
        self.fpsClock = pygame.time.Clock()
        self.idTable = 0
        self.idOrder = 0
        self.idItem = -1
        self.idOrderTable = 0
        self.waiterNumberInGrid = Vector2()
        pygame.display.set_caption('Automatic Waiter')
        self.background = pygame.image.load("./Images/tlo.jpg")

        # The most important lists
        self.grid = []
        self.row = []
        self.tableList = []

        for b in range(self.y):
            for i in range(self.x):
                if ((i == 0) or (i == 9)) and ((b == 0) or (b == 6)):
                    self.row.append(Table(i * 50, b * 50, self))
                elif (b == self.y-1 or b == self.y-2 or b == self.y-3) and (i == 5):
                    table = OrderTable(i * 50, b * 50, self)
                    self.row.append(table)
                    self.tableList.append(table)
                elif i == 4 and b == 9:
                    self.waiter = Waiter(self, i * 50, b * 50)
                    self.row.append(self.waiter)
                    self.waiterNumberInGrid.x = i
                    self.waiterNumberInGrid.y = b
                elif (((b > 0 and b < 6) and (i > 0 and i < 9)) and not((i==1 or i == 8) and (b==1 or b==5))) \
                        or (b>6 and (i<4 or i > 5) ):
                    self.row.append(Pool(i * 50, b * 50, self))
                else:
                    self.row.append(GridElement(i * 50, b * 50, self))
                if i == self.x-1:
                    self.grid.append(list(self.row))
                    self.row.clear()
        #copyGrid = copy.copy(self.grid)
        #copyGrid = []
        #kopia = copy.deepcopy(self.grid[9][4])
        kopia = [[num for num in line] for line in self.grid]
        self.waiter.dfsFind(kopia, [], 2)
        #print("Prawdziwa pozycja kelnera: {0}, {1} ", self.waiter.positionX, self.waiter.positionY)
        #print(self.waiter.findWaiter(self.grid[:]))

        while True:

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

            self.draw()
            self.waiter.move(self)
            #print(self.waiter.checkPoss(self.grid, self.waiter.lastStep))
            pygame.display.flip()
            #print(self.waiterNumberInGrid)
            self.fpsClock.tick(10)

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

    def convert(self, object):
        if object.type == "gridElement":
            return 'G'
        elif object.type == "table":
            return 'T'
        elif object.type == "waiter":
            return 'W'
        elif object.type == "orderTable":
            return 'O'
        elif object.type == "pool":
            return 'P'

    def showGrid(self, grid):
        charakterList = []
        row = []
        for y in range(self.y):
            row = list(map(self.convert, grid[y]))
            charakterList.append(row)
            print(row)



if __name__ == "__main__":
    Game()