refactorise brain parts populate

intelligine/simulation/object/brain/AntBrain.py

@@ -10,29 +10,19 @@ from intelligine.simulation.object.brain.part.transport.AntTakeBrainPart import
 
 class AntBrain(Brain):
 
-    _brain_part_move_class = AntMoveBrainPart
-    _brain_part_take_class = AntTakeBrainPart
-    _brain_part_put_class = AntPutBrainPart
+    # TODO: methode __init_ pour la classe ? Pour surcharger ici.
+    _brain_parts = {
+        BRAIN_PART_MOVE: AntMoveBrainPart,
+        BRAIN_PART_TAKE: AntTakeBrainPart,
+        BRAIN_PART_PUT: AntPutBrainPart
+    }
 
     def __init__(self, context, host):
         super().__init__(context, host)
-        # TODO: Gerer les BrainPart avec un dictionnaire ?
-        self._set_brain_part(BRAIN_PART_MOVE, self._get_move_brain_part_instance())
-        self._set_brain_part(BRAIN_PART_TAKE, self._get_take_brain_part_instance())
-        self._set_brain_part(BRAIN_PART_PUT, self._get_put_brain_part_instance())
         self._movement_mode = MOVE_MODE_EXPLO
         self._distance_from_objective = 0  # TODO rename: distance_since_objective
         self._pheromone_searching = PHEROMON_DIR_EXPLO
 
-    def _get_move_brain_part_instance(self):
-        return self._brain_part_move_class()
-
-    def _get_take_brain_part_instance(self):
-        return self._brain_part_take_class()
-
-    def _get_put_brain_part_instance(self):
-        return self._brain_part_put_class()
-
     def switch_to_mode(self, mode):
         self._movement_mode = mode
         self._update_pheromone_gland(mode)

intelligine/simulation/object/brain/Brain.py

@@ -4,13 +4,18 @@ from intelligine.cst import BRAIN_SCHEMA
 
 class Brain():
 
-    _brain_part_class = {}
+    _brain_parts = {}
 
     def __init__(self, context, host):
         self._context = context
         self._host = host
         self._schema = {}
         self._parts = {}
+        self._init_parts()
+
+    def _init_parts(self):
+        for brain_part_name in self._brain_parts:
+            self._set_brain_part(brain_part_name, self._brain_parts[brain_part_name](self))
 
     def _set_brain_part(self, name, brain_part, replace=False):
         if name in self._parts and not replace:

intelligine/simulation/object/brain/part/BrainPart.py

@@ -1,4 +1,7 @@
 class BrainPart():
 
+    def __init__(self, host_brain):
+        self._host_brain = host_brain
+
     def done(self, obj, context):
         pass

intelligine/simulation/object/brain/part/transport/AntTakeBrainPart.py

@@ -6,7 +6,7 @@ from intelligine.cst import MOVE_MODE_EXPLO, MOVE_MODE, TYPE_RESOURCE_TRANSFORMA
 
 class AntTakeBrainPart(TakeBrainPart):
 
-    # TODO: methode __nit_ pour la classe ? vt mieux surcharger !
+    # TODO: methode __init_ pour la classe ? Pour surcharger ici.
     _mode_matches = {
         MOVE_MODE_EXPLO: [TYPE_RESOURCE_TRANSFORMABLE],
         MOVE_MODE_NURSE: [TYPE_NURSERY]