Fix ant v1

ant_example/antHillExample.py

@@ -13,9 +13,6 @@ class AntHillExample(Amas):
         super().__init__(env)
         self.__observer = None
 
-    def on_initialization(self):
-        self._Amas__execution_policy = ExecutionPolicy.TWO_PHASES
-
     def get_Agents_Sorted(self):
         agents = self.get_agents()
         agents.sort(key=lambda x: x.get_id())

ant_example_2_phases/antExample.py

+"""
+class antExample
+"""
+from math import *
+from random import randint
+
+from pyAmakCore.classes.agent import Agent
+
+from color import Color
+
+
+class AntExample(Agent):
+
+    def __init__(self,
+                 amas: 'antHillExample',
+                 startX: float,
+                 startY: float
+                 ) -> None:
+        super().__init__(amas)
+        self._dx = startX
+        self._dy = startY
+        self._color = Color.BLACK
+        self._old_color = Color.BLACK
+
+    def get_color(self):
+        return self._color
+
+    def get_dx(self):
+        return self._dx
+
+    def get_dy(self):
+        return self._dy
+
+    def get_old_color(self):
+        return self._old_color
+
+    def on_perceive(self) -> None:
+        self.reset_neighbour()
+        for agent in self.get_amas().get_agents():
+            length = sqrt(pow(self._dx - agent.get_dx(), 2) + pow(self._dy - agent.get_dy(), 2))
+            if length < self.get_environment().field_of_view:
+                self.add_neighbour(agent)
+
+    def on_act(self) -> None:
+        # couleur
+        color_changed = False
+        for agent in self.get_neighbour():
+            if agent.get_color() != Color.BLACK:
+                self._color = agent.get_color()
+                color_changed = True
+
+        if not color_changed:
+            color = {
+                1: Color.BLUE,
+                2: Color.BLACK,
+                3: Color.RED,
+                4: Color.YELLOW,
+                5: Color.GREEN
+            }
+            if randint(1, 1000) <= 4:
+                    self._color = color.get(randint(1, 5))
+
+        # déplacement
+        self._dx += (randint(-1, 1) * self.get_environment().coef_deplacement)
+        self._dy += (randint(-1, 1) * self.get_environment().coef_deplacement)
+
+        if self._dx < self.get_environment().xmin:
+            self._dx = self.get_environment().xmin
+
+        if self._dx > self.get_environment().xmax:
+            self._dx = self.get_environment().xmax
+
+        if self._dy < self.get_environment().ymin:
+            self._dy = self.get_environment().ymin
+
+        if self._dy > self.get_environment().ymax:
+            self._dy = self.get_environment().ymax
+
+    def on_cycle_begin(self):
+        self._old_color = self._color

ant_example_2_phases/antHillExample.py

+"""
+class antHillExample
+"""
+from pyAmakCore.enumeration.executionPolicy import ExecutionPolicy
+
+from antExample import AntExample
+from pyAmakCore.classes.amas import Amas
+
+
+class AntHillExample(Amas):
+
+    def __init__(self, env):
+        super().__init__(env)
+        self.__observer = None
+
+    def on_initialization(self):
+        self.set_execution_policy(ExecutionPolicy.TWO_PHASES)
+
+    def get_Agents_Sorted(self):
+        agents = self.get_agents()
+        agents.sort(key=lambda x: x.get_id())
+        return agents
+
+    def on_initial_agents_creation(self) -> None:
+        for i in range(50):
+            self.add_agent(AntExample(self, self.get_environment().xmax/2, self.get_environment().ymax/2))
+
+    def on_cycle_end(self) -> None:
+        self.__observer.updateCycle(self.get_environment(), self)
+
+    def attach(self, observer: 'Controleur') -> None:
+        self.__observer = observer

ant_example_2_phases/antsLaunchExample.py

+"""
+Class antsLaunchExample
+"""
+from random import seed
+
+from pyAmakIHM.classes.fenetre import Fenetre
+from controleurAntsExample import ControleurAntsExample
+from worldExample import WorldExample
+from antHillExample import *
+from threading import Thread
+
+fenetre = Fenetre("Prototype Ants")
+
+seed()
+
+env = WorldExample(0, fenetre.get_canvas_width(), 0, fenetre.get_canvas_height(), 5, 7)
+amas = AntHillExample(env)
+
+controleur = ControleurAntsExample(fenetre, amas)
+
+
+def run():
+    controleur.get_amas().start()
+
+
+def main():
+    controleur.initialisation()
+    Thread(target=run).start()
+    controleur.get_fenetre().display()
+
+
+main()

ant_example_2_phases/color.py

+from enum import *
+class Color(Enum):
+    BLACK= auto()
+    RED=auto()
+    BLUE=auto()
+    GREEN=auto()
+    YELLOW=auto()
\ No newline at end of file

ant_example_2_phases/controleurAntsExample.py

+from color import Color
+from pyAmakIHM.classes.controleur import Controleur
+
+
+class ControleurAntsExample(Controleur):
+
+    def __init__(self, fenetre, amas):
+        super().__init__(fenetre, amas)
+        self.__ants = []
+        self.__numberAnts = 50
+
+    def initialisation(self):
+
+        widthCanvas = self.get_fenetre().get_canvas_width()
+        heightCanvas = self.get_fenetre().get_canvas_height()
+
+        for i in range(self.__numberAnts):
+            self.__ants.append(self.draw_image(widthCanvas / 2, heightCanvas / 2, 'images/blackAnt.png'))
+
+    def updateCycle(self, env, amas):
+        ants = amas.get_Agents_Sorted()
+
+        for i in range(len(ants)):
+            x = ants[i].get_dx()
+            y = ants[i].get_dy()
+            color = ants[i].get_color()
+
+            if color != ants[i].get_old_color():
+                self.remove_element(self.__ants[i])
+                if color == Color.BLUE:
+                    self.__ants[i] = self.draw_image(x, y, 'images/blueAnt.png')
+
+                elif color == Color.GREEN:
+                    self.__ants[i] = self.draw_image(x, y, 'images/greenAnt.png')
+
+                elif color == Color.RED:
+                    self.__ants[i] = self.draw_image(x, y, 'images/redAnt.png')
+
+                elif color == Color.YELLOW:
+                    self.__ants[i] = self.draw_image(x, y, 'images/yellowAnt.png')
+
+            self.move_image(self.__ants[i], x, y)

ant_example_2_phases/worldExample.py

+"""
+Class worldExample
+"""
+from pyAmakCore.classes.environment import Environment
+
+
+class WorldExample(Environment):
+
+    def __init__(self, xmin, xmax, ymin, ymax, field_of_view, coef_deplacement):
+        super().__init__()
+        self.xmin = xmin
+        self.xmax = xmax
+        self.ymin = ymin
+        self.ymax = ymax
+
+        self.field_of_view = field_of_view
+        self.coef_deplacement = coef_deplacement