Mercurial > mamba
view mamba/level.py @ 88:63d290cbc661
doors and things that make you a colour now to be composed with overlays
author | Adrianna Pińska <adrianna.pinska@gmail.com> |
---|---|
date | Sun, 11 Sep 2011 19:01:07 +0200 |
parents | cced0ddda33f |
children | 7ce2d2d8381a |
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 TILE_MAP = { '.': None, 'X': (TileSprite, 'wall'), 'R': (DoorSprite, 'red'), 'B': (DoorSprite, 'blue'), '^': (EntrySprite, Snake.UP), 'v': (EntrySprite, Snake.DOWN), '<': (EntrySprite, Snake.LEFT), '>': (EntrySprite, Snake.RIGHT), 'E': (ExitSprite,), '~': (PuddleSprite,), } THING_MAP = { 'a': (BaseSprite, '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(self.name, 'floor').image for name, value in TILE_MAP.items(): if value is not None: value = (value[0], self.name) + value[1:] 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)