automatyczny_kelner/agent.py
Vadzim Valchkovich 27050f5bfe some changes
2023-06-17 17:16:58 +02:00

26 lines
726 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
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)
params = layout.train_loop(12)
layout.create_and_subscribe(
params['radius'],
params['neighbors_count'],
params['block_probability']
)
engine.loop()