Sandbox with blue and red colonys

config.py

 from xyzworld.Context import Context as XyzContext
 from socialintengine.synergy.Simulation import Simulation
-from socialintengine.synergy.Colony import Colony
-from socialintengine.synergy.ColonyConfiguration import ColonyConfiguration
-from socialintengine.synergy.Rocks import Rocks
-from socialintengine.synergy.RocksConfiguration import RocksConfiguration
 from socialintengine.display.Pygame import Pygame
 from socialintengine.display.pygame.visualisation import visualisation as pygame_visualisation
-
+from socialintengine.sandbox.redblue.red_blue_colonys import collections
 
 config = {
     'app': {
     'engine': {
         'fpsmax': 5,
         'debug': {
-            'mainprocess': False,
+            'mainprocess': True,
             'cycles': -1
         }
     },
-    'simulations' : [Simulation([Colony(ColonyConfiguration()), Rocks(RocksConfiguration())])],
+    'simulations' : [Simulation(collections)],
     'connections': [Pygame],
     'terminal': {
       'pygame': {

socialintengine/display/pygame/visualisation.py

 from xyworld.display.object.pygame.DirectionnedImage import DirectionnedImage
 from socialintengine.synergy.object.Bug import Bug
 from socialintengine.synergy.object.ant.Ant import Ant
+from socialintengine.sandbox.redblue.BlueAnt import BlueAnt
+from socialintengine.sandbox.redblue.RedAnt import RedAnt
 from socialintengine.synergy.object.Rock import Rock
 from os import getcwd
 from synergine.metas import metas
 from socialintengine.cst import PREVIOUS_DIRECTION
 
 ant = PygameImage.from_filepath(getcwd()+'/socialintengine/display/pygame/image/ant.png')
+red_ant = PygameImage.from_filepath(getcwd()+'/socialintengine/display/pygame/image/red_ant.png')
+blue_ant = PygameImage.from_filepath(getcwd()+'/socialintengine/display/pygame/image/blue_ant.png')
 bug = PygameImage.from_filepath(getcwd()+'/socialintengine/display/pygame/image/ant.png')
 rock = PygameImage.from_filepath(getcwd()+'/socialintengine/display/pygame/image/rock.png')
 
 directions_ant = DirectionnedImage(ant)
+directions_red_ant = DirectionnedImage(red_ant)
+directions_blue_ant = DirectionnedImage(blue_ant)
 
 def bug_direction(bug):
     try:
         previous_direction = 14
     return directions_ant.get_for_direction(previous_direction)
 
+def red_ant_direction(bug):
+    try:
+        previous_direction = metas.value.get(PREVIOUS_DIRECTION, bug.get_id())
+    except KeyError:
+        previous_direction = 14
+    return directions_red_ant.get_for_direction(previous_direction)
+
+def blue_ant_direction(bug):
+    try:
+        previous_direction = metas.value.get(PREVIOUS_DIRECTION, bug.get_id())
+    except KeyError:
+        previous_direction = 14
+    return directions_blue_ant.get_for_direction(previous_direction)
+
 visualisation = {
     'window': {},
     'objects': {
+        RedAnt: {
+            'default': red_ant,
+            'callbacks': [red_ant_direction]
+        },
+        BlueAnt: {
+            'default': blue_ant,
+            'callbacks': [blue_ant_direction]
+        },
         Ant: {
             'default': ant,
             'callbacks': [bug_direction]

socialintengine/sandbox/redblue/BlueAnt.py

+from socialintengine.synergy.object.ant.Ant import Ant
+
+class BlueAnt(Ant):
+    pass

socialintengine/sandbox/redblue/BlueColonyConfiguration.py

+from socialintengine.synergy.ColonyConfiguration import ColonyConfiguration
+from socialintengine.sandbox.redblue.BlueAnt import BlueAnt
+
+
+class BlueColonyConfiguration(ColonyConfiguration):
+
+    _start_position = (0, 20, 20)
+    _ant_class = BlueAnt

socialintengine/sandbox/redblue/RedAnt.py

+from socialintengine.synergy.object.ant.Ant import Ant
+
+class RedAnt(Ant):
+    pass

socialintengine/sandbox/redblue/RedColonyConfiguration.py

+from socialintengine.synergy.ColonyConfiguration import ColonyConfiguration
+from socialintengine.sandbox.redblue.RedAnt import RedAnt
+
+
+class RedColonyConfiguration(ColonyConfiguration):
+
+    _start_position = (0, 70, 20)
+    _ant_class = RedAnt

socialintengine/sandbox/redblue/red_blue_colonys.py

+from socialintengine.sandbox.redblue.RedColonyConfiguration import RedColonyConfiguration
+from socialintengine.sandbox.redblue.BlueColonyConfiguration import BlueColonyConfiguration
+from socialintengine.synergy.Colony import Colony
+from socialintengine.synergy.Rocks import Rocks
+from socialintengine.synergy.RocksConfiguration import RocksConfiguration
+
+collections = [Colony(BlueColonyConfiguration()), Colony(RedColonyConfiguration()), Rocks(RocksConfiguration())]

socialintengine/synergy/ColonyConfiguration.py

 
 class ColonyConfiguration(Configuration):
 
+    _start_position = (0, 20, 20)
+    _ant_class = Ant
+
     def get_start_objects(self):
 
       ants = []
-      for i in range(10):
-          ant = Ant()
-          ant.set_position((0, 20, 20))
+      for i in range(100):
+          ant = self._ant_class()
+          ant.set_position(self._start_position)
           metas.list.add(Simulation.STATE, ant.get_id(), ALIVE)
           ants.append(ant)
 

socialintengine/synergy/RocksConfiguration.py

 
       rocks = []
 
-      for i in range(50):
+      for i in range(100):
           rock = Rock()
           metas.list.add(Simulation.STATE, rock.get_id(), IMPENETRABLE)
           rock.set_position((0, 0+i, 0))
           rocks.append(rock)
 
-      for i in range(50):
+      for i in range(100):
           rock = Rock()
           metas.list.add(Simulation.STATE, rock.get_id(), IMPENETRABLE)
           rock.set_position((0, 0+i, 50))
           rocks.append(rock)
 
       for i in range(50):
+          if i is not 25:
+            rock = Rock()
+            rock.set_position((0, 50, 0+i))
+            metas.list.add(Simulation.STATE, rock.get_id(), IMPENETRABLE)
+            rocks.append(rock)
+
+      for i in range(50):
           rock = Rock()
-          rock.set_position((0, 50, 0+i))
+          rock.set_position((0, 100, 0+i))
           metas.list.add(Simulation.STATE, rock.get_id(), IMPENETRABLE)
           rocks.append(rock)
 
       metas.list.add(Simulation.STATE, rock.get_id(), IMPENETRABLE)
       rocks.append(rock)
 
+      rock = Rock()
+      rock.set_position((0, 100, 50))
+      metas.list.add(Simulation.STATE, rock.get_id(), IMPENETRABLE)
+      rocks.append(rock)
+
       return rocks

socialintengine/synergy/Simulation.py

 from synergine.synergy.Simulation import Simulation as BaseSimulation
+from xyzworld.cst import POSITION, POSITIONS
+
 
 class Simulation(BaseSimulation):
-  pass
+
+    def end_cycle(self, context):
+        if context.get_cycle() % 100 is 0:
+            context.metas.list.clean(POSITIONS)