X-Git-Url: https://plomlompom.com/repos/feed.xml?a=blobdiff_plain;f=game_common.py;h=599e54bcda4c51b936b9dac9ac1d90b03debd8af;hb=2b91efc835f8e68c97830d13d1e3070fdbcf1cc5;hp=b37e4475ffaf3fd420e53cd39eea6551c169bce2;hpb=1c48dfdf85d5549202412f01eced7712adf2d468;p=plomrogue2-experiments

diff --git a/game_common.py b/game_common.py
index b37e447..599e54b 100644
--- a/game_common.py
+++ b/game_common.py
@@ -1,6 +1,27 @@
 from parser import ArgError
 
 
+class MapManager:
+
+    def __init__(self, globs):
+        """With globs a globals() call, collect caller's Map classes."""
+        self.map_classes = []
+        for name in globs:
+            if name[:3] == 'Map':
+                self.map_classes += [globs[name]]
+
+    def get_map_geometries(self):
+        geometries = []
+        for map_class in self.map_classes:
+            geometries += [map_class.__name__[3:]]
+        return geometries
+
+    def get_map_class(self, geometry):
+        for map_class in self.map_classes:
+            if map_class.__name__[3:] == geometry:
+                return map_class
+
+
 class Map:
 
     def __init__(self, size=(0, 0)):
@@ -42,7 +63,7 @@ class World:
         return t
 
     def new_map(self, geometry, yx):
-        map_type = getattr(self, 'Map' + geometry)
+        map_type = self.game.map_manager.get_map_class(geometry)
         self.map_ = map_type(yx)
 
 
@@ -59,7 +80,7 @@ class CommonCommandsMixin:
 
     def cmd_MAP(self, geometry, yx):
         """Create new map of grid geometry, size yx and only '?' cells."""
-        legal_grids = {'Hex', 'Square'}
+        legal_grids = self.map_manager.get_map_geometries()
         if geometry not in legal_grids:
             raise ArgError('First map argument must be one of: ' +
                            ', '.join(legal_grids))