Mercurial > mamba
view mamba/level.py @ 104:5b92ff941de3
Refactor sprites.
author | Jeremy Thurgood <firxen@gmail.com> |
---|---|
date | Sun, 11 Sep 2011 19:52:49 +0200 |
parents | 7ce2d2d8381a |
children | 87acd3ddc972 |
line wrap: on
line source
""" Level for our shiny game. """ from pygame.surface import Surface from pygame.sprite import RenderUpdates from mamba.data import load_file from mamba.sprites import (tile_sizify, TileSprite, BaseSprite, DoorSprite, EntrySprite, ExitSprite, PuddleSprite) from mamba.snake import Snake class InvalidMapError(Exception): pass def mktile(cls, **kw): return (cls, kw) TILE_MAP = { '.': None, 'X': mktile(TileSprite, image_name='wall'), 'R': mktile(DoorSprite, colour='red'), 'B': mktile(DoorSprite, colour='blue'), '^': mktile(EntrySprite, direction=Snake.UP), 'v': mktile(EntrySprite, direction=Snake.DOWN), '<': mktile(EntrySprite, direction=Snake.LEFT), '>': mktile(EntrySprite, direction=Snake.RIGHT), 'E': mktile(ExitSprite), '~': mktile(PuddleSprite), } THING_MAP = { 'a': mktile(BaseSprite, image_name='snake/snake-head-r'), } class Tileset(object): def __init__(self, tileset_name): self.name = tileset_name self.load_tiles() def load_tiles(self): self.tiles = {} self.floor = TileSprite(tileset=self.name, image_name='floor').image for name, value in TILE_MAP.items(): if value is not None: value[1]['tileset'] = self.name self.tiles[name] = value self.tiles.update(THING_MAP) def __getitem__(self, key): try: tilespec = self.tiles[key] except KeyError: raise InvalidMapError("Unknown tile type: '%s'" % key) if not tilespec: return None return tilespec[0](**tilespec[1]) def get_tile(self, key, tile_pos, *groups): tile = self[key] if tile: tile.add(*groups) tile.set_tile_pos(tile_pos) return tile class Level(object): def __init__(self, level_name): self.level_name = level_name self.load_level_data() def load_level_data(self): """ This file format is potentially yucky. """ level_data = load_file('levels/%s.txt' % (self.level_name,)) tileset_name = level_data.readline().strip() self.tileset = Tileset(tileset_name) tiles_ascii = [line.strip() for line in level_data.readlines()] self.setup_tiles(tiles_ascii) self.make_background() def setup_tiles(self, tiles_ascii): self.sprites = RenderUpdates() self.tiles = [] self.entry = None self.tile_size = (len(tiles_ascii[0]), len(tiles_ascii)) for y, row in enumerate(tiles_ascii): if len(row) != self.tile_size[0]: raise InvalidMapError("Map not rectangular.") tile_row = [] for x, tile_char in enumerate(row): tile = self.tileset.get_tile(tile_char, (x, y), self.sprites) tile_row.append(tile) if isinstance(tile, EntrySprite): if self.entry is not None: raise InvalidMapError("Too many entry points.") self.entry = tile self.tiles.append(tile_row) if self.entry is None: raise InvalidMapError("Not enough entry points.") def get_tile_size(self): return self.tile_size def get_size(self): x, y = self.get_tile_size() return tile_sizify((x, y)) def make_background(self): sx, sy = self.get_tile_size() self.background = Surface(self.get_size()) [self.background.blit(self.tileset.floor, tile_sizify((x, y))) for x in range(sx) for y in range(sy)] def get_entry(self): return (self.entry.tile_pos, self.entry.direction)