ProjektAI/kelner/main.py

import pygame
from kelner.src.components.GridBoard import GridBoard
from kelner.src.components.Waiter import Waiter
from kelner.src.components.Table import Table
from kelner.src.managers.DrawableCollection import DrawableCollection
from kelner.src.managers.MenuManager import MenuManager
from kelner.src.managers.TableManager import TableManager
from kelner.src.managers.WaiterManager import WaiterManager

# create screen consts
Scale        = 2                  # scale for all images used within project
CellSize     = round(50 * Scale)  # pixel size of 1 square cell in the grid
PaintOffset  = CellSize           # pixel size of paint offset for all drawables
GridCountX   = 15                 # number of columns in grid
GridCountY   = 9                  # number of rows in grid
ScreenWidth  = CellSize * GridCountX + 2 * PaintOffset  # screen width in pixels
ScreenHeight = CellSize * GridCountY + 2 * PaintOffset  # screen height in pixels

# initialize background
gridBoard = GridBoard(ScreenWidth, ScreenHeight)

# initialize drawable objects manager
drawableManager = DrawableCollection()

# initialize menu manager
menuManager = MenuManager()

# initialize waiter component
waiter1 = Waiter(0, 0, 0, GridCountX - 1, 0, GridCountY - 1, CellSize, PaintOffset)
waiter2 = Waiter(0, 1, 0, GridCountX - 1, 0, GridCountY - 1, CellSize, PaintOffset)
waiter3 = Waiter(0, 2, 0, GridCountX - 1, 0, GridCountY - 1, CellSize, PaintOffset)
waiter4 = Waiter(0, 3, 0, GridCountX - 1, 0, GridCountY - 1, CellSize, PaintOffset)

# adds waiter to drawable collection
drawableManager.add(waiter1)
drawableManager.add(waiter2)
drawableManager.add(waiter3)
drawableManager.add(waiter4)

# initialize a number of tables given in range
for i in range(1, 40):
    table = Table(0, GridCountX - 1, 0, GridCountY - 1, CellSize, PaintOffset)
    if drawableManager.generatePosition(table):
        drawableManager.add(table)

# new thread controlling tables
tableTask = TableManager(drawableManager, menuManager)
tableTask.start()

# new thread controlling waiter
waiter1Task = WaiterManager(drawableManager, [waiter1])
waiter1Task.start()

waiter2Task = WaiterManager(drawableManager, [waiter2])
waiter2Task.start()

waiter3Task = WaiterManager(drawableManager, [waiter3])
waiter3Task.start()

waiter4Task = WaiterManager(drawableManager, [waiter4])
waiter4Task.start()

# main loop
running = True
while running:

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            tableTask.stop()
            waiter1Task.stop()
            waiter2Task.stop()
            waiter3Task.stop()
            waiter4Task.stop()
            running = False

        # handles keyboard events
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_LEFT:
                # checks if new waiter's position to the left is not occupied by other object
                if drawableManager.isPositionAvailable(waiter1.getX() - 1, waiter1.getY()):
                    waiter1.moveLeft()
            if event.key == pygame.K_RIGHT:
                # checks if new waiter's position to the right is not occupied by other object
                if drawableManager.isPositionAvailable(waiter1.getX() + 1, waiter1.getY()):
                    waiter1.moveRight()
            if event.key == pygame.K_UP:
                # checks if new waiter's position up is not occupied by other object
                if drawableManager.isPositionAvailable(waiter1.getX(), waiter1.getY() - 1):
                    waiter1.moveUp()
            if event.key == pygame.K_DOWN:
                # checks if new waiter's position down is not occupied by other object
                if drawableManager.isPositionAvailable(waiter1.getX(), waiter1.getY() + 1):
                    waiter1.moveDown()

            if event.key == pygame.K_w:
                # checks if new waiter's position to the left is not occupied by other object
                if drawableManager.isPositionAvailable(waiter1.getX() - 1, waiter1.getY() - 1):
                    waiter1.moveLeft()
                    waiter1.moveUp()
            if event.key == pygame.K_d:
                # checks if new waiter's position to the right is not occupied by other object
                if drawableManager.isPositionAvailable(waiter1.getX() + 1, waiter1.getY() - 1):
                    waiter1.moveRight()
                    waiter1.moveUp()
            if event.key == pygame.K_s:
                # checks if new waiter's position up is not occupied by other object
                if drawableManager.isPositionAvailable(waiter1.getX() + 1, waiter1.getY() + 1):
                    waiter1.moveDown()
                    waiter1.moveRight()
            if event.key == pygame.K_a:
                # checks if new waiter's position down is not occupied by other object
                if drawableManager.isPositionAvailable(waiter1.getX() - 1, waiter1.getY() + 1):
                    waiter1.moveDown()
                    waiter1.moveLeft()
            drawableManager.forceRepaint()

    # repaints all objects to the screen
    # is set only on initial paint or after keyboard event or call to forceRepaint()
    if drawableManager.mustRepaint():
        gridBoard.reinitialize()
        gridBoard.draw(drawableManager)
        gridBoard.udpdate()