body parts

intelligine/core/exceptions.py

@@ -54,6 +54,14 @@ class BrainPartAlreadyExist(BrainException):
     pass
 
 
+class BodyException(Exception):
+    pass
+
+
+class BodyPartAlreadyExist(BodyException):
+    pass
+
+
 class DirectionException(Exception):
     pass
 

intelligine/cst.py

@@ -43,4 +43,7 @@ BRAIN_SCHEMA = IncrementedNamedInt.get('intelligine.brain_schema')
 BRAIN_PART_MOVE = IncrementedNamedInt.get('intelligine.brain.part.move')
 BRAIN_PART_TAKE = IncrementedNamedInt.get('intelligine.brain.part.take')
 BRAIN_PART_PUT = IncrementedNamedInt.get('intelligine.brain.part.put')
-BRAIN_PART_ATTACK = IncrementedNamedInt.get('intelligine.brain.part.attack')
+BRAIN_PART_ATTACK = IncrementedNamedInt.get('intelligine.brain.part.attack')
+
+BODY_SCHEMA = IncrementedNamedInt.get('intelligine.body_schema')
+BODY_PART_PHEROMONE_GLAND = IncrementedNamedInt.get('intelligine.body.part.pheromone_gland')

intelligine/simulation/object/brain/Brain.py

@@ -30,4 +30,6 @@ class Brain():
         self._schema = {}
         for part_name in self._parts:
             self._schema[part_name] = self._parts[part_name].__class__
+        # TODO: N'est-ce pas un schema appartenant a la classe ? Ne suffirai t-il pas de stocker ce schema par classe
+        # plutôt que par objet ?
         self._context.metas.value.set(BRAIN_SCHEMA, self._host.get_id(), self._schema)

intelligine/synergy/object/BaseBug.py

@@ -1,3 +1,4 @@
+from intelligine.core.exceptions import BodyPartAlreadyExist
 from intelligine.synergy.object.Transportable import Transportable
 from intelligine.cst import ALIVE, ATTACKABLE, COL_ALIVE
 from intelligine.simulation.object.brain.Brain import Brain
@@ -5,6 +6,8 @@ from intelligine.simulation.object.brain.Brain import Brain
 
 class BaseBug(Transportable):
 
+    _body_parts = {}
+
     def __init__(self, collection, context):
         super().__init__(collection, context)
         context.metas.states.add_list(self.get_id(), [ALIVE, ATTACKABLE])
@@ -12,6 +15,20 @@ class BaseBug(Transportable):
         self._life_points = 10
         self._movements_count = -1
         self._brain = self._get_brain_instance()
+        self._parts = {}
+        self._init_parts()
+
+    def _init_parts(self):
+        for body_part_name in self._body_parts:
+            self._set_body_part(body_part_name, self._body_parts[body_part_name](self, self._context))
+
+    def _set_body_part(self, name, body_part, replace=False):
+        if name in self._parts and not replace:
+            raise BodyPartAlreadyExist()
+        self._parts[name] = body_part
+
+    def get_body_part(self, name):
+        return self._parts[name]
 
     def hurted(self, points):
         self._life_points -= points

intelligine/synergy/object/ant/Ant.py

@@ -1,9 +1,7 @@
 from intelligine.core.exceptions import PheromoneException
 from intelligine.synergy.object.Bug import Bug
-from intelligine.cst import CARRYING, TRANSPORTER, ATTACKER, \
-                            COL_TRANSPORTER, COL_TRANSPORTER_NOT_CARRYING, \
-                            COL_FIGHTER, MOVE_MODE_EXPLO, MOVE_MODE_GOHOME, \
-                            PHEROMON_DIR_EXPLO, CARRIED
+from intelligine.cst import CARRYING, TRANSPORTER, ATTACKER, COL_TRANSPORTER, COL_TRANSPORTER_NOT_CARRYING, \
+    COL_FIGHTER, MOVE_MODE_EXPLO, MOVE_MODE_GOHOME, CARRIED, BODY_PART_PHEROMONE_GLAND
 from intelligine.synergy.object.Food import Food
 from intelligine.simulation.object.pheromone.MovementPheromoneGland import MovementPheromoneGland
 from intelligine.simulation.object.brain.AntBrain import AntBrain
@@ -11,6 +9,10 @@ from intelligine.simulation.object.brain.AntBrain import AntBrain
 
 class Ant(Bug):
 
+    _body_parts = {
+        BODY_PART_PHEROMONE_GLAND: MovementPheromoneGland
+    }
+
     def __init__(self, collection, context):
         super().__init__(collection, context)
         context.metas.states.add_list(self.get_id(), [TRANSPORTER, ATTACKER])
@@ -18,15 +20,13 @@ class Ant(Bug):
                                                            COL_TRANSPORTER_NOT_CARRYING,
                                                            COL_FIGHTER])
         self._carried = None
-        # TODO: Faire un body_part schema pour ces trucs la
-        self._movement_pheromone_gland = MovementPheromoneGland(self, self._context)
         self._brain.switch_to_mode(MOVE_MODE_EXPLO)
 
     def _get_brain_instance(self):
         return AntBrain(self._context, self)
 
     def get_movement_pheromone_gland(self):
-        return self._movement_pheromone_gland
+        return self.get_body_part(BODY_PART_PHEROMONE_GLAND)
 
     def put_carry(self, obj, position=None):
         if position is None: