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