mamba: mamba/level.py annotate

annotate mamba/level.py @ 84:cced0ddda33f

More level validation.

author	Jeremy Thurgood <firxen@gmail.com>
date	Sun, 11 Sep 2011 18:43:21 +0200
parents	30c3bbaf7a28
children	7ce2d2d8381a

rev	line source
8 66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	1 """
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	2 Level for our shiny game.
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	3 """
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	4
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	5 from pygame.surface import Surface
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	6 from pygame.sprite import RenderUpdates
8 66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	7
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	8 from mamba.data import load_file
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	9 from mamba.sprites import (tile_sizify, TileSprite, BaseSprite, DoorSprite,
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	10 EntrySprite, ExitSprite, PuddleSprite)
83 30c3bbaf7a28 Use Snake direction constants in level.py. Simon Cross <hodgestar@gmail.com> parents: 81 diff changeset	11 from mamba.snake import Snake
31 67218082cdbb More level non-functionality. Jeremy Thurgood <firxen@gmail.com> parents: 21 diff changeset	12
21 f76c3f033772 Release early, release broken. Level stuff. Jeremy Thurgood <firxen@gmail.com> parents: 8 diff changeset	13
81 c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	14 class InvalidMapError(Exception):
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	15 pass
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	16
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	17
37 9a0ae29c7f5e Tiles and things. Jeremy Thurgood <firxen@gmail.com> parents: 34 diff changeset	18 TILE_MAP = {
9a0ae29c7f5e Tiles and things. Jeremy Thurgood <firxen@gmail.com> parents: 34 diff changeset	19 '.': None,
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	20 'X': (TileSprite, 'wall'),
56 e730ec8b82e8 Some doors. Jeremy Thurgood <firxen@gmail.com> parents: 50 diff changeset	21 'R': (DoorSprite, 'red'),
e730ec8b82e8 Some doors. Jeremy Thurgood <firxen@gmail.com> parents: 50 diff changeset	22 'B': (DoorSprite, 'blue'),
83 30c3bbaf7a28 Use Snake direction constants in level.py. Simon Cross <hodgestar@gmail.com> parents: 81 diff changeset	23 '^': (EntrySprite, Snake.UP),
30c3bbaf7a28 Use Snake direction constants in level.py. Simon Cross <hodgestar@gmail.com> parents: 81 diff changeset	24 'v': (EntrySprite, Snake.DOWN),
30c3bbaf7a28 Use Snake direction constants in level.py. Simon Cross <hodgestar@gmail.com> parents: 81 diff changeset	25 '<': (EntrySprite, Snake.LEFT),
30c3bbaf7a28 Use Snake direction constants in level.py. Simon Cross <hodgestar@gmail.com> parents: 81 diff changeset	26 '>': (EntrySprite, Snake.RIGHT),
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	27 'E': (ExitSprite,),
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	28 '~': (PuddleSprite,),
37 9a0ae29c7f5e Tiles and things. Jeremy Thurgood <firxen@gmail.com> parents: 34 diff changeset	29 }
9a0ae29c7f5e Tiles and things. Jeremy Thurgood <firxen@gmail.com> parents: 34 diff changeset	30
9a0ae29c7f5e Tiles and things. Jeremy Thurgood <firxen@gmail.com> parents: 34 diff changeset	31 THING_MAP = {
57 cd5ad2c45e2d Minor level fiddlery. Jeremy Thurgood <firxen@gmail.com> parents: 56 diff changeset	32 'a': (BaseSprite, 'snake/snake-head-r'),
37 9a0ae29c7f5e Tiles and things. Jeremy Thurgood <firxen@gmail.com> parents: 34 diff changeset	33 }
9a0ae29c7f5e Tiles and things. Jeremy Thurgood <firxen@gmail.com> parents: 34 diff changeset	34
9a0ae29c7f5e Tiles and things. Jeremy Thurgood <firxen@gmail.com> parents: 34 diff changeset	35
8 66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	36 class Tileset(object):
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	37 def __init__(self, tileset_name):
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	38 self.name = tileset_name
31 67218082cdbb More level non-functionality. Jeremy Thurgood <firxen@gmail.com> parents: 21 diff changeset	39 self.load_tiles()
8 66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	40
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	41 def load_tiles(self):
31 67218082cdbb More level non-functionality. Jeremy Thurgood <firxen@gmail.com> parents: 21 diff changeset	42 self.tiles = {}
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	43 self.floor = TileSprite(self.name, 'floor').image
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	44 for name, value in TILE_MAP.items():
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	45 if value is not None:
56 e730ec8b82e8 Some doors. Jeremy Thurgood <firxen@gmail.com> parents: 50 diff changeset	46 value = (value[0], self.name) + value[1:]
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	47 self.tiles[name] = value
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	48 self.tiles.update(THING_MAP)
8 66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	49
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	50 def __getitem__(self, key):
84 cced0ddda33f More level validation. Jeremy Thurgood <firxen@gmail.com> parents: 83 diff changeset	51 try:
cced0ddda33f More level validation. Jeremy Thurgood <firxen@gmail.com> parents: 83 diff changeset	52 tilespec = self.tiles[key]
cced0ddda33f More level validation. Jeremy Thurgood <firxen@gmail.com> parents: 83 diff changeset	53 except KeyError:
cced0ddda33f More level validation. Jeremy Thurgood <firxen@gmail.com> parents: 83 diff changeset	54 raise InvalidMapError("Unknown tile type: '%s'" % key)
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	55 if not tilespec:
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	56 return None
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	57 return tilespec[0](*tilespec[1:])
31 67218082cdbb More level non-functionality. Jeremy Thurgood <firxen@gmail.com> parents: 21 diff changeset	58
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	59 def get_tile(self, key, tile_pos, *groups):
31 67218082cdbb More level non-functionality. Jeremy Thurgood <firxen@gmail.com> parents: 21 diff changeset	60 tile = self[key]
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	61 if tile:
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	62 tile.add(*groups)
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: 37 diff changeset	63 tile.set_tile_pos(tile_pos)
31 67218082cdbb More level non-functionality. Jeremy Thurgood <firxen@gmail.com> parents: 21 diff changeset	64 return tile
8 66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	65
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	66
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	67 class Level(object):
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	68 def __init__(self, level_name):
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	69 self.level_name = level_name
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	70 self.load_level_data()
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	71
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	72 def load_level_data(self):
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	73 """
31 67218082cdbb More level non-functionality. Jeremy Thurgood <firxen@gmail.com> parents: 21 diff changeset	74 This file format is potentially yucky.
8 66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	75 """
66ae99f6903e First stab at tile/level stuff. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	76 level_data = load_file('levels/%s.txt' % (self.level_name,))
31 67218082cdbb More level non-functionality. Jeremy Thurgood <firxen@gmail.com> parents: 21 diff changeset	77 tileset_name = level_data.readline().strip()
21 f76c3f033772 Release early, release broken. Level stuff. Jeremy Thurgood <firxen@gmail.com> parents: 8 diff changeset	78 self.tileset = Tileset(tileset_name)
84 cced0ddda33f More level validation. Jeremy Thurgood <firxen@gmail.com> parents: 83 diff changeset	79 tiles_ascii = [line.strip() for line in level_data.readlines()]
81 c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	80 self.setup_tiles(tiles_ascii)
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	81 self.make_background()
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	82
81 c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	83 def setup_tiles(self, tiles_ascii):
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	84 self.sprites = RenderUpdates()
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	85 self.tiles = []
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	86 self.entry = None
84 cced0ddda33f More level validation. Jeremy Thurgood <firxen@gmail.com> parents: 83 diff changeset	87 self.tile_size = (len(tiles_ascii[0]), len(tiles_ascii))
81 c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	88 for y, row in enumerate(tiles_ascii):
84 cced0ddda33f More level validation. Jeremy Thurgood <firxen@gmail.com> parents: 83 diff changeset	89 if len(row) != self.tile_size[0]:
cced0ddda33f More level validation. Jeremy Thurgood <firxen@gmail.com> parents: 83 diff changeset	90 raise InvalidMapError("Map not rectangular.")
81 c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	91 tile_row = []
84 cced0ddda33f More level validation. Jeremy Thurgood <firxen@gmail.com> parents: 83 diff changeset	92 for x, tile_char in enumerate(row):
81 c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	93 tile = self.tileset.get_tile(tile_char, (x, y), self.sprites)
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	94 tile_row.append(tile)
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	95 if isinstance(tile, EntrySprite):
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	96 if self.entry is not None:
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	97 raise InvalidMapError("Too many entry points.")
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	98 self.entry = tile
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	99 self.tiles.append(tile_row)
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	100 if self.entry is None:
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	101 raise InvalidMapError("Not enough entry points.")
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	102
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	103 def get_tile_size(self):
84 cced0ddda33f More level validation. Jeremy Thurgood <firxen@gmail.com> parents: 83 diff changeset	104 return self.tile_size
21 f76c3f033772 Release early, release broken. Level stuff. Jeremy Thurgood <firxen@gmail.com> parents: 8 diff changeset	105
f76c3f033772 Release early, release broken. Level stuff. Jeremy Thurgood <firxen@gmail.com> parents: 8 diff changeset	106 def get_size(self):
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	107 x, y = self.get_tile_size()
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	108 return tile_sizify((x, y))
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	109
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	110 def make_background(self):
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	111 sx, sy = self.get_tile_size()
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	112 self.background = Surface(self.get_size())
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	113 [self.background.blit(self.tileset.floor, tile_sizify((x, y)))
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	114 for x in range(sx) for y in range(sy)]
81 c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	115
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	116 def get_entry(self):
c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	117 return (self.entry.tile_pos, self.entry.direction)

Mercurial > mamba

annotate mamba/level.py @ 84:cced0ddda33f