forked from s444417/ProjektAI
80 lines
2.9 KiB
Python
80 lines
2.9 KiB
Python
import pygame
|
|
from src.components.GridBoard import GridBoard
|
|
from src.managers.DrawableCollection import DrawableCollection
|
|
from src.managers.MenuManager import MenuManager
|
|
from src.components.Waiter import Waiter
|
|
from src.components.Table import Table
|
|
from src.managers.TaskManager import TaskManager
|
|
|
|
#create screen consts
|
|
CellSize = 50 #pixel size of 1 square cell in the grid
|
|
GridCountX = 15 #number of columns in grid
|
|
GridCountY = 11 #number of rows in grid
|
|
ScreenWidth = CellSize * GridCountX #screen width in pixels
|
|
ScreenHeight = CellSize * GridCountY #screen height in pixels
|
|
|
|
#initialize background
|
|
gridBoard = GridBoard(ScreenWidth, ScreenHeight, CellSize)
|
|
|
|
#initialize drawable objects manager
|
|
drawableManager = DrawableCollection()
|
|
|
|
#initialize menu manager
|
|
menuManager = MenuManager()
|
|
|
|
#initialize waiter component
|
|
waiter = Waiter(0, 0, 0, GridCountX - 1, 0, GridCountY - 1, CellSize)
|
|
|
|
#adds waiter to drawable collection
|
|
drawableManager.add(waiter)
|
|
|
|
#initialize a number of tables given in range
|
|
for i in range(1, 20):
|
|
table = Table(0, GridCountX - 1, 0, GridCountY - 1, CellSize)
|
|
drawableManager.generatePosition(table)
|
|
drawableManager.add(table)
|
|
|
|
#main loop
|
|
|
|
#object that controlls repainting of changed objects
|
|
doRepaint = [True]
|
|
|
|
#new thread
|
|
task = TaskManager(drawableManager, menuManager, doRepaint)
|
|
task.start()
|
|
|
|
running = True
|
|
while running:
|
|
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
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(waiter.getX() - 1, waiter.getY()):
|
|
waiter.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(waiter.getX() + 1, waiter.getY()):
|
|
waiter.moveRight()
|
|
if event.key == pygame.K_UP:
|
|
# checks if new waiter's position up is not occupied by other object
|
|
if drawableManager.isPositionAvailable(waiter.getX(), waiter.getY() - 1):
|
|
waiter.moveUp()
|
|
if event.key == pygame.K_DOWN:
|
|
# checks if new waiter's position down is not occupied by other object
|
|
if drawableManager.isPositionAvailable(waiter.getX(), waiter.getY() + 1):
|
|
waiter.moveDown()
|
|
doRepaint[0] = True
|
|
|
|
# repaints all objects to the screen
|
|
# is set only on initial paint or after keyboard event
|
|
if doRepaint[0]:
|
|
gridBoard.reinitialize()
|
|
gridBoard.draw(drawableManager)
|
|
gridBoard.udpdate()
|
|
doRepaint[0] = False
|
|
drawableManager.collectOrders() |