Sandbox with blue and red colonys

config.py

@@ -1,12 +1,8 @@
 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': {
@@ -18,11 +14,11 @@ config = {
     '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

@@ -2,16 +2,22 @@ from xyworld.display.object.pygame.PygameImage import PygameImage
 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:
@@ -20,9 +26,31 @@ def bug_direction(bug):
         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

@@ -0,0 +1,4 @@
+from socialintengine.synergy.object.ant.Ant import Ant
+
+class BlueAnt(Ant):
+    pass

socialintengine/sandbox/redblue/BlueColonyConfiguration.py

@@ -0,0 +1,8 @@
+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

@@ -0,0 +1,4 @@
+from socialintengine.synergy.object.ant.Ant import Ant
+
+class RedAnt(Ant):
+    pass

socialintengine/sandbox/redblue/RedColonyConfiguration.py

@@ -0,0 +1,8 @@
+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

@@ -0,0 +1,7 @@
+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

@@ -7,12 +7,15 @@ from synergine.synergy.Simulation import Simulation
 
 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

@@ -11,13 +11,13 @@ class RocksConfiguration(Configuration):
 
       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))
@@ -30,8 +30,15 @@ class RocksConfiguration(Configuration):
           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)
 
@@ -40,4 +47,9 @@ class RocksConfiguration(Configuration):
       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

@@ -1,4 +1,9 @@
 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)