Save floors and visibility, fixes #61

squirrelbattle/interfaces.py

@@ -80,18 +80,18 @@ class Map:
     currentx: int
     currenty: int
 
-    def __init__(self, width: int, height: int, tiles: list,
-                 start_y: int, start_x: int):
+    def __init__(self, width: int = 0, height: int = 0, tiles: list = None,
+                 start_y: int = 0, start_x: int = 0):
         self.floor = 0
         self.width = width
         self.height = height
         self.start_y = start_y
         self.start_x = start_x
-        self.tiles = tiles
-        self.visibility = [[False for _ in range(len(tiles[0]))]
-                           for _ in range(len(tiles))]
+        self.tiles = tiles or []
+        self.visibility = [[False for _ in range(len(self.tiles[0]))]
+                           for _ in range(len(self.tiles))]
         self.seen_tiles = [[False for _ in range(len(tiles[0]))]
-                           for _ in range(len(tiles))]
+                           for _ in range(len(self.tiles))]
         self.entities = []
         self.logs = Logs()
 
@@ -338,9 +338,10 @@ class Map:
         for enti in self.entities:
             d["entities"].append(enti.save_state())
         d["map"] = self.draw_string(TexturePack.ASCII_PACK)
+        d["seen_tiles"] = self.seen_tiles
         return d
 
-    def load_state(self, d: dict) -> None:
+    def load_state(self, d: dict) -> "Map":
         """
         Loads the map's attributes from a dictionary.
         """
@@ -351,11 +352,16 @@ class Map:
         self.currentx = d["currentx"]
         self.currenty = d["currenty"]
         self.tiles = self.load_dungeon_from_string(d["map"])
+        self.seen_tiles = d["seen_tiles"]
+        self.visibility = [[False for _ in range(len(self.tiles[0]))]
+                           for _ in range(len(self.tiles))]
         self.entities = []
         dictclasses = Entity.get_all_entity_classes_in_a_dict()
         for entisave in d["entities"]:
             self.add_entity(dictclasses[entisave["type"]](**entisave))
 
+        return self
+
 
 class Tile(Enum):
     """