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