28 lines
771 B
Python
28 lines
771 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))
|
|
# '''
|