X-Git-Url: https://plomlompom.com/repos/berlin_corona.txt?a=blobdiff_plain;f=game_common.py;h=b37e4475ffaf3fd420e53cd39eea6551c169bce2;hb=1c48dfdf85d5549202412f01eced7712adf2d468;hp=c6fb220ee04cfa7ba90ee512e4239e24fd45b69e;hpb=1d3411457a0ac41e86b44c635b6896ba7f3ab0f1;p=plomrogue2-experiments

diff --git a/game_common.py b/game_common.py
index c6fb220..b37e447 100644
--- a/game_common.py
+++ b/game_common.py
@@ -1,31 +1,37 @@
 from parser import ArgError
 
 
-class World:
+class Map:
 
-    def __init__(self):
-        self.turn = 0
-        self.map_size = (0, 0)
-        self.terrain_map = ''
-        self.things = []
-        self.Thing = Thing  # child classes may use an extended Thing class here
+    def __init__(self, size=(0, 0)):
+        self.size = size
+        self.terrain = '?'*self.size_i
 
-    def set_map_size(self, yx):
-        y, x = yx
-        self.map_size = (y, x)
-        self.terrain_map = ''
-        for y in range(self.map_size[0]):
-            self.terrain_map += '?' * self.map_size[1]
+    @property
+    def size_i(self):
+        return self.size[0] * self.size[1]
 
-    def set_map_line(self, y, line):
-        width_map = self.map_size[1]
-        if y >= self.map_size[0]:
+    def set_line(self, y, line):
+        height_map = self.size[0]
+        width_map = self.size[1]
+        if y >= height_map:
             raise ArgError('too large row number %s' % y)
         width_line = len(line)
         if width_line > width_map:
             raise ArgError('too large map line width %s' % width_line)
-        self.terrain_map = self.terrain_map[:y * width_map] + line + \
-                           self.terrain_map[(y + 1) * width_map:]
+        self.terrain = self.terrain[:y * width_map] + line +\
+                       self.terrain[(y + 1) * width_map:]
+
+    def get_position_index(self, yx):
+        return yx[0] * self.size[1] + yx[1]
+
+
+class World:
+
+    def __init__(self):
+        self.Thing = Thing  # child classes may use an extended Thing class here
+        self.turn = 0
+        self.things = []
 
     def get_thing(self, id_):
         for thing in self.things:
@@ -35,6 +41,10 @@ class World:
         self.things += [t]
         return t
 
+    def new_map(self, geometry, yx):
+        map_type = getattr(self, 'Map' + geometry)
+        self.map_ = map_type(yx)
+
 
 class Thing:
 
@@ -43,3 +53,25 @@ class Thing:
         self.id_ = id_
         self.type_ = '?'
         self.position = [0,0]
+
+
+class CommonCommandsMixin:
+
+    def cmd_MAP(self, geometry, yx):
+        """Create new map of grid geometry, size yx and only '?' cells."""
+        legal_grids = {'Hex', 'Square'}
+        if geometry not in legal_grids:
+            raise ArgError('First map argument must be one of: ' +
+                           ', '.join(legal_grids))
+        self.world.new_map(geometry, yx)
+    cmd_MAP.argtypes = 'string yx_tuple:nonneg'
+
+    def cmd_THING_TYPE(self, i, type_):
+        t = self.world.get_thing(i)
+        t.type_ = type_
+    cmd_THING_TYPE.argtypes = 'int:nonneg string'
+
+    def cmd_THING_POS(self, i, yx):
+        t = self.world.get_thing(i)
+        t.position = list(yx)
+    cmd_THING_POS.argtypes = 'int:nonneg yx_tuple:nonneg'