automatyczny_kelner/agent.py

from src.Engine import Engine
from src.obj.Waiter import Waiter
from src.obj.Kitchen import Kitchen
from src.controller.LayoutController import LayoutController
from src.controller.ImageController import ImageController

SCREEN_SIZE = [800, 800]
SQUARE_SIZE = 80
ACTION_DURATION = 0.1
COUNT_OF_OBJECTS = 25

store = ImageController(SQUARE_SIZE)
waiter = Waiter([0, 0], 0, SQUARE_SIZE, SCREEN_SIZE, store)
kitchen = Kitchen([0, 0], 0, SQUARE_SIZE, SCREEN_SIZE, store)
engine = Engine(SCREEN_SIZE, SQUARE_SIZE, kitchen, waiter, ACTION_DURATION)
layout = LayoutController(engine, store).create_and_subscribe(COUNT_OF_OBJECTS)


engine.loop()

'''
def example_stop(action_clock: int) -> bool:
    return action_clock < 1000


print(engine.train_loop(example_stop))
'''