intelligine/intelligine/display/pygame/visualisation.py

from xyworld.display.object.pygame.PygameImage import PygameImage
from xyworld.display.object.pygame.DirectionnedImage import DirectionnedImage
from intelligine.synergy.object.Bug import Bug
from intelligine.synergy.object.Food import Food
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 intelligine.cst import PREVIOUS_DIRECTION

# 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')
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')


from intelligine.synergy.object.ant.PheromonExploration import PheromonExploration
from intelligine.synergy.object.ant.PheromonHome import PheromonHome
phee = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/phee.png')
pheh = PygameImage.from_filepath(getcwd()+'/intelligine/display/pygame/image/pheh.png')
dir_phee = DirectionnedImage(phee)
dir_pheh = DirectionnedImage(pheh)


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

def phee_direction(phee, context):
    return dir_phee.get_for_direction(phee.get_direction())

def pheh_direction(pheh, context):
    return dir_pheh.get_for_direction(pheh.get_direction())

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
    },
    '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
        },
        PheromonExploration: {
            'default': phee,
            'callbacks': [phee_direction]
        },
        PheromonHome: {
            'default': pheh,
            'callbacks': [pheh_direction]
        },
    }
}