from synergine_xyz.display.object.pygame.PygameImage import PygameImage
from synergine_xyz.display.object.pygame.DirectionnedImage import DirectionnedImage
from intelligine.synergy.object.Bug import Bug
from intelligine.synergy.object.Food import Food
from intelligine.synergy.object.Hole import Hole
from intelligine.synergy.object.ant.Ant import Ant
from intelligine.sandbox.colored.BlueAnt import BlueAnt
from intelligine.sandbox.colored.RedAnt import RedAnt
from intelligine.sandbox.colored.GreenAnt import GreenAnt
from intelligine.synergy.object.Rock import Rock
from intelligine.synergy.object.ant.Egg import Egg
from os import getcwd
from synergine_xyz.cst import PREVIOUS_DIRECTION

SURFACE_PHEROMONE_HOME = 'pheromone_home'
SURFACE_PHEROMONE_EXPLORATION = 'pheromone_exploration'

# TODO: Analyser les procedes ici pour proposer des outils dans le framework

ant = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/ant.png')
food = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/food.png')
hole = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/hole.png')
dead_ant = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/dead_ant.png')
red_ant = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/red_ant.png')
green_ant = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/green_ant.png')
blue_ant = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/blue_ant.png')
dead_red_ant = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/dead_red_ant.png')
dead_green_ant = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/dead_green_ant.png')
dead_blue_ant = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/dead_blue_ant.png')
bug = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/ant.png')
rock = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/rock.png')
egg = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/egg.png')
eggc2 = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/egg_c2.png')
eggc3 = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/egg_c3.png')
eggc4 = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/egg_c4.png')
eggc5 = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/egg_c5.png')
eggc7 = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/egg_c7.png')
phee = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/phee.png')
pheh = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/pheh.png')

directions_ant = DirectionnedImage(ant)
directions_red_ant = DirectionnedImage(red_ant)
directions_blue_ant = DirectionnedImage(blue_ant)
directions_green_ant = DirectionnedImage(green_ant)

def bug_direction(bug, context):
    if bug.get_life_points() <= 0:
        return dead_ant
    try:
        previous_direction = context.metas.value.get(PREVIOUS_DIRECTION, bug.get_id())
    except KeyError:
        previous_direction = 14
    return directions_ant.get_for_direction(previous_direction)

def red_ant_direction(bug, context):
    if bug.get_life_points() <= 0:
        return dead_red_ant
    try:
        previous_direction = context.metas.value.get(PREVIOUS_DIRECTION, bug.get_id())
    except KeyError:
        previous_direction = 14
    return directions_red_ant.get_for_direction(previous_direction)

def blue_ant_direction(bug, context):
    if bug.get_life_points() <= 0:
        return dead_blue_ant
    try:
        previous_direction = context.metas.value.get(PREVIOUS_DIRECTION, bug.get_id())
    except KeyError:
        previous_direction = 14
    return directions_blue_ant.get_for_direction(previous_direction)

def green_ant_direction(bug, context):
    if bug.get_life_points() <= 0:
        return dead_green_ant
    try:
        previous_direction = context.metas.value.get(PREVIOUS_DIRECTION, bug.get_id())
    except KeyError:
        previous_direction = 14
    return directions_green_ant.get_for_direction(previous_direction)


def for_position(position, objects, context):
    # TODO: DEV TMP: refact, etc
    eggs = []
    for obj in objects:
        if isinstance(obj, Egg):
            eggs.append(obj)
    if len(eggs) == 2:
        return (eggc2, eggs)
    if len(eggs) == 3:
        return (eggc3, eggs)
    if len(eggs) == 4:
        return (eggc4, eggs)
    if len(eggs) > 4:
        return (eggc7, eggs)
    return (None, [])

visualisation = {
    'window': {},
    'callbacks': {
        'position': for_position
    },
    'surfaces': {
        SURFACE_PHEROMONE_EXPLORATION: {
            'default': phee,
            'callbacks': []
        },
        SURFACE_PHEROMONE_HOME: {
            'default': pheh,
            'callbacks': []
        },
    },
    'objects': {
        RedAnt: {
            'default': red_ant,
            'callbacks': [red_ant_direction]
        },
        BlueAnt: {
            'default': blue_ant,
            'callbacks': [blue_ant_direction]
        },
        GreenAnt: {
            'default': green_ant,
            'callbacks': [green_ant_direction]
        },
        Ant: {
            'default': ant,
            'callbacks': [bug_direction]
        },
        Bug: {
            'default': bug,
            'callbacks': [bug_direction]
        },
        Egg: {
            'default': egg
        },
        Rock: {
            'default': rock
        },
        Food: {
            'default': food
        },
        Hole: {
            'default': hole
        }
    }
}