automatyczny_kelner/agent.py
2023-06-15 21:22:07 +02:00

28 lines
765 B
Python

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))
'''