Add basic reality bubble mechanism.

[plomrogue2-experiments] / new / plomrogue / commands.py

diff --git a/new/plomrogue/commands.py b/new/plomrogue/commands.py
index 744d471fac100bc5e326813899d93466a353566e..c09bc4e177ff145bdcf17510d0aaf13c89bc40eb 100644 (file)
--- a/new/plomrogue/commands.py
+++ b/new/plomrogue/commands.py
@@ -18,10 +18,20 @@ def cmd_MAP_SIZE(game, size):
     game.map_size = size
 cmd_MAP_SIZE.argtypes = 'yx_tuple:pos'
 
-def cmd_MAP(game, map_pos):
-    """Ensure (possibly empty/'?'-filled) map at position map_pos."""
-    game.get_map(map_pos)
-cmd_MAP.argtypes = 'yx_tuple'
+def cmd_MAP(game, map_pos, awakeness):
+    """Ensure (possibly empty/'?'-filled) map at position map_pos.
+
+    Awakeness > 0 puts the map into the player's reality bubble.
+
+    """
+    m = game.get_map(map_pos)
+    m.awake = awakeness
+cmd_MAP.argtypes = 'yx_tuple int:nonneg'
+
+def cmd_MAP_STATS(game, map_pos, type_, population, health):
+    m = game.get_map(map_pos)
+    m.stats[type_] = {'population': population, 'health': health}
+cmd_MAP_STATS = 'yx_tuple string:thingtype int:nonneg int:nonneg'
 
 def cmd_THING_TYPE(game, i, type_):
     t_old = game.get_thing(i)
@@ -106,7 +116,12 @@ def cmd_SAVE(game):
         write(f, 'SEED %s' % game.rand.prngod_seed)
         write(f, 'MAP_SIZE %s' % (game.map_size,))
         for map_pos in game.maps:
-            write(f, 'MAP %s' % (map_pos,))
+            m = game.maps[map_pos]
+            write(f, 'MAP %s %s' % (map_pos, m.awake))
+            for t_type in m.stats:
+                write(f, 'MAP_STATS %s %s %s %s' %
+                      (map_pos, t_type, m.stats[t_type]['population'],
+                       m.stats[t_type]['health']))
         for map_pos in game.maps:
             for y, line in game.maps[map_pos].lines():
                  write(f, 'TERRAIN_LINE %s %5s %s' % (map_pos, y, quote(line)))