Mercurial > mamba
view mamba/level.py @ 136:00ada2e29798
Somewhat better (but still hideous) image variant support.
author | Jeremy Thurgood <firxen@gmail.com> |
---|---|
date | Sun, 11 Sep 2011 23:56:59 +0200 |
parents | 0abd6e3680b6 |
children | 2c643ec8dc4e |
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), 'l': mktile(PuddleSprite, variant=(1, 1, 0, 0)), 'r': mktile(PuddleSprite, variant=(1, 0, 1, 0)), } THING_MAP = { 'a': mktile(BaseSprite, image_name='red'), } 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 cls, params = tilespec params['tile_char'] = key return cls(**params) 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.tiles_ascii = '' 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,)) self.name = level_data.readline().strip() tileset_name = level_data.readline().strip() self.tileset = Tileset(tileset_name) tiles_ascii = [line.strip() for line in level_data.readlines()] self.tiles_ascii = tiles_ascii 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_orientation = self.get_tile_orientation(y, x, row, tile_char) 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 is_same_tile(self, tile, x, y): """Is there a tile of the same type?""" try: temp_tile = self.tiles_ascii[y][x] except IndexError: return {}#'The map went out of range' try: return temp_tile == tile except KeyError: return {}#'The map tile is not existing' return self.get_bool(x, y, 'wall') #please note: planning on tile tuple to be -1, 0, 1 of x then y def get_tile_orientation(self, map_y, map_x, row, tile_char): #print "y: ", map_y, " x: ", map_x, " row: ", row, " tile_char: ", tile_char #tile_char if self.is_same_tile(tile_char, map_x, map_y): # Draw different tiles depending on neighbourhood if not self.is_same_tile(tile_char, map_x, map_y+1): if self.is_same_tile(tile_char, map_x+1, map_y) and self.is_same_tile(tile_char, map_x-1, map_y): # tile = 1, 2 elif self.is_same_tile(tile_char, map_x+1, map_y): tile = 0, 2 elif self.is_same_tile(tile_char, map_x-1, map_y): tile = 2, 2 else: tile = 3, 2 else: if self.is_same_tile(tile_char, map_x+1, map_y+1) and self.is_same_tile(tile_char, map_x-1, map_y+1): tile = 1, 1 elif self.is_same_tile(tile_char, map_x+1, map_y+1): tile = 0, 1 elif self.is_same_tile(tile_char, map_x-1, map_y+1): tile = 2, 1 else: tile = 3, 1 return tile 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) def draw(self, surface): surface.blit(self.background, (0, 0)) self.sprites.draw(surface)