koperkapel/loaders/levelloader.py

   1 """Loader a level, using the pygame-zero ResourceLoader infrastructure"""
   2
   3 import json
   4
   5 from pgzero.loaders import ResourceLoader
   6
   7 from ..gamelib.tiles import Wall, Floor, Tunnel, Underground
   8 from ..gamelib.level import Level
   9
  10
  11
  12 TILES = {
  13     "cwall": Wall, # rename this everywhere
  14     "floor": Floor,
  15     "tunnel": Tunnel,
  16     "underground": Underground,
  17 }
  18
  19 class LevelLoader(ResourceLoader):
  20     """ Level loader. """
  21
  22     EXTNS = ['json']
  23     TYPE = 'level'
  24
  25     def _load(self, level_path):
  26         f = open(level_path, 'r')
  27         level_data = json.load(f)
  28         f.close()
  29         self._level = Level()
  30         self._level.height = len(level_data['tiles'])
  31         self._level.width = len(level_data['tiles'][0])
  32         self._level.tiles = level_data['tiles']
  33         self._level.tileset = level_data['tileset']
  34         # Consistency check, so we can assume things are correct
  35         # in the level renderer
  36         for row, row_data in enumerate(self._level.tiles):
  37             if len(row_data) != self._level.width:
  38                 raise RuntimeError("Incorrect len for row %d" % row)
  39         for tile in TILES.values():
  40             tile.TILESET = self._level.tileset
  41         self._load_tile_images()
  42         return self._level
  43
  44     def _load_tile_images(self):
  45         """Load all the tile images"""
  46         for y, row_data in enumerate(self._level.tiles):
  47             for x, tile in enumerate(row_data):
  48                 neighborhood = self._level.get_neighbors(x, y)
  49                 for layer in ['floor', 'tunnels']:
  50                     neighbors = [x[layer] if x else None for x in neighborhood]
  51                     tile['%s image' % layer] = \
  52                             TILES[tile[layer]['base']].image(neighbors)
  53
  54
  55 levels = LevelLoader('levels')