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, Grate
   8 from ..gamelib.level import Level
   9 from ..gamelib.door import Door
  10 from ..gamelib.keypad import Keypad
  11 from ..gamelib.friendlyroach import get_friendly_roach
  12
  13
  14
  15 TILES = {
  16     "cwall": Wall, # rename this everywhere
  17     "floor": Floor,
  18     "tunnel": Tunnel,
  19     "underground": Underground,
  20 }
  21
  22 class LevelLoader(ResourceLoader):
  23     """ Level loader. """
  24
  25     EXTNS = ['json']
  26     TYPE = 'level'
  27
  28     def _load(self, level_path):
  29         f = open(level_path, 'r')
  30         level_data = json.load(f)
  31         f.close()
  32         self._level = Level()
  33         self._level.height = len(level_data['tiles'])
  34         self._level.width = len(level_data['tiles'][0])
  35         self._level.tiles = level_data['tiles']
  36         self._level.tileset = level_data['tileset']
  37         self._level.start_pos = level_data["starting pos"]
  38         self._level.enemies = []
  39         self._level.friends = []
  40         # Consistency check, so we can assume things are correct
  41         # in the level renderer
  42         for row, row_data in enumerate(self._level.tiles):
  43             if len(row_data) != self._level.width:
  44                 raise RuntimeError("Incorrect len for row %d" % row)
  45         for tile in TILES.values():
  46             tile.TILESET = self._level.tileset
  47         self._load_tile_images()
  48         for door_info in level_data['door_info']:
  49             # Create the doors first
  50             doors = []
  51             for door in door_info['doors']:
  52                 x, y = door
  53                 # is this the right check
  54                 if self._level.can_walk(x + 1, y, 'floor') and self._level.can_walk(x - 1, y, 'floor'):
  55                     # vertical door
  56                     angle = 90
  57                 else:
  58                     angle = 0
  59                 new_door = Door(x, y, angle)
  60                 self._level.doors.append(new_door)
  61                 doors.append(new_door)
  62             # Add the keypads
  63             for keypad in door_info['keypads']:
  64                 new_keypad = Keypad(keypad[0], keypad[1], doors)
  65                 self._level.keypads.append(new_keypad)
  66         for pos in level_data['friendly roaches']:
  67             roach = get_friendly_roach(pos[0], pos[1])
  68             self._level.friends.append(roach)
  69         return self._level
  70
  71     def _load_tile_images(self):
  72         """Load all the tile images"""
  73         for y, row_data in enumerate(self._level.tiles):
  74             for x, tile in enumerate(row_data):
  75                 neighborhood = self._level.get_neighbors(x, y)
  76                 for layer in ['floor', 'tunnels']:
  77                     neighbors = [x[layer] if x else None for x in neighborhood]
  78                     tile['%s image' % layer] = \
  79                             TILES[tile[layer]['base']].image(neighbors)
  80                     # Hack this for now
  81                     if layer == 'floor' and 'crawl' in tile[layer]['behaviour']:
  82                         tile['floor image'] = Grate.image(neighbors)
  83                         self._level.grates.append((x, y))
  84
  85
  86 levels = LevelLoader('levels')