Browse Source

add attack brain part to ant

Bastien Sevajol 9 years ago
parent
commit
1a78cefad3

+ 4 - 2
intelligine/simulation/object/brain/AntBrain.py View File

@@ -1,7 +1,8 @@
1 1
 from intelligine.simulation.object.brain.Brain import Brain
2
+from intelligine.simulation.object.brain.part.attack.AttackBrainPart import AttackBrainPart
2 3
 from intelligine.simulation.object.brain.part.move.AntMoveBrainPart import AntMoveBrainPart
3 4
 from intelligine.cst import MOVE_MODE, MOVE_MODE_EXPLO, MOVE_MODE_GOHOME, PHEROMON_DIR_HOME, PHEROMON_DIR_EXPLO, \
4
-    BRAIN_PART_TAKE, BRAIN_PART_PUT, MOVE_MODE_NURSE, PHEROMON_DIR_NONE
5
+    BRAIN_PART_TAKE, BRAIN_PART_PUT, MOVE_MODE_NURSE, PHEROMON_DIR_NONE, BRAIN_PART_ATTACK
5 6
 from intelligine.cst import PHEROMONE_SEARCHING
6 7
 from intelligine.cst import BRAIN_PART_MOVE
7 8
 from intelligine.simulation.object.brain.part.transport.AntPutBrainPart import AntPutBrainPart
@@ -14,7 +15,8 @@ class AntBrain(Brain):
14 15
     _brain_parts = {
15 16
         BRAIN_PART_MOVE: AntMoveBrainPart,
16 17
         BRAIN_PART_TAKE: AntTakeBrainPart,
17
-        BRAIN_PART_PUT: AntPutBrainPart
18
+        BRAIN_PART_PUT: AntPutBrainPart,
19
+        BRAIN_PART_ATTACK: AttackBrainPart
18 20
     }
19 21
 
20 22
     def __init__(self, context, host):

+ 1 - 0
intelligine/synergy/event/attack/NearAttackableEvent.py View File

@@ -14,6 +14,7 @@ class NearAttackableEvent(NearEvent):
14 14
 
15 15
     def _prepare(self, object_id, context, parameters={}):
16 16
         obj_colony_id = context.metas.value.get(COLONY, object_id)
17
+        # TODO: Le critère pas la même colonie est du ressort du brain.
17 18
         filter = lambda near_object_id, context: obj_colony_id != context.metas.value.get(COLONY, near_object_id)
18 19
 
19 20
         try: