intelligine/intelligine/display/Pygame.py

from intelligine.core.exceptions import NoPheromone
from xyzworld.display.Pygame import Pygame as XyzPygame
import pygame
from intelligine.cst import PHEROMON_INFOS, PHEROMON_DIRECTION, PHEROMON_DIR_HOME, PHEROMON_DIR_EXPLO, PHEROMON_POSITIONS
from intelligine.synergy.object.ant.PheromonExploration import PheromonExploration
from intelligine.synergy.object.ant.PheromonHome import PheromonHome


class Pygame(XyzPygame):

    def __init__(self, config, context):
        super().__init__(config, context)
        self._is_display_pheromones = False

    def receive(self, synergy_object_manager, context):
        super().receive(synergy_object_manager, context)
        if self._is_display_pheromones:
            self._display_pheromones(context.metas.list.get(PHEROMON_POSITIONS, PHEROMON_POSITIONS, allow_empty=True), context)

    def _display_pheromones(self, pheromones_positions, context):
        # TODO: Code de test bordelique !
        for point in pheromones_positions:
            flavour = context.pheromones().get_flavour(point)
            try:
                flavour.get_pheromone(category=PHEROMON_DIRECTION, type=PHEROMON_DIR_HOME)
                pheromon = PheromonHome(object(), context)
                pheromon.set_direction(11) # TODO: plus de direction avec ces nlles pheromones
                self._draw_objects_with_decal(point, [pheromon])
            except NoPheromone:
                pass
            try:
                flavour.get_pheromone(category=PHEROMON_DIRECTION, type=PHEROMON_DIR_EXPLO)
                pheromon = PheromonExploration(object(), context)
                pheromon.set_direction(11) # TODO: plus de direction avec ces nlles pheromones
                self._draw_objects_with_decal(point, [pheromon])
            except NoPheromone:
                pass

    def _key_pressed(self, key):
        if key == pygame.K_p:
            if self._is_display_pheromones:
                self._is_display_pheromones = False
            else:
                self._is_display_pheromones = True