from src.agent.map.gameMap import GameMap
from mesa.visualization.modules import CanvasGrid
from mesa.visualization.ModularVisualization import ModularServer


def player_representation(agent):
    portrayal = {"Shape": "sprites/hero.png",
                 "Layer": 1}
    if agent.isBox:
        portrayal["Shape"] = "sprites/box.png"
        portrayal["Layer"] = 0
    elif agent.isCreature:
        portrayal["Shape"] = 'sprites/goblin.png'
    return portrayal


grid = CanvasGrid(player_representation, 10, 10, 500, 500)
server = ModularServer(GameMap,
                       [grid],
                       "Map",
                       {"x": 10, "y": 10})
server.port = 8081
server.launch()