Mercurial > mamba

annotate mamba/level.py @ 105:87acd3ddc972

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Tell sprites what their tile char is.
author Jeremy Thurgood <firxen@gmail.com>
date Sun, 11 Sep 2011 19:59:14 +0200
parents 5b92ff941de3
children e6299eb296ce
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
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
104
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
18 def mktile(cls, **kw):
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
19 return (cls, kw)
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
20
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
21
37
9a0ae29c7f5e Tiles *and* things.
Jeremy Thurgood <firxen@gmail.com>
parents: 34
diff changeset
22 TILE_MAP = {
9a0ae29c7f5e Tiles *and* things.
Jeremy Thurgood <firxen@gmail.com>
parents: 34
diff changeset
23 '.': None,
104
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
24 'X': mktile(TileSprite, image_name='wall'),
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
25 'R': mktile(DoorSprite, colour='red'),
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
26 'B': mktile(DoorSprite, colour='blue'),
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
27 '^': mktile(EntrySprite, direction=Snake.UP),
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
28 'v': mktile(EntrySprite, direction=Snake.DOWN),
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
29 '<': mktile(EntrySprite, direction=Snake.LEFT),
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
30 '>': mktile(EntrySprite, direction=Snake.RIGHT),
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
31 'E': mktile(ExitSprite),
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
32 '~': mktile(PuddleSprite),
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 THING_MAP = {
104
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
36 'a': mktile(BaseSprite, image_name='snake/snake-head-r'),
37
9a0ae29c7f5e Tiles *and* things.
Jeremy Thurgood <firxen@gmail.com>
parents: 34
diff changeset
37 }
9a0ae29c7f5e Tiles *and* things.
Jeremy Thurgood <firxen@gmail.com>
parents: 34
diff changeset
38
9a0ae29c7f5e Tiles *and* things.
Jeremy Thurgood <firxen@gmail.com>
parents: 34
diff changeset
39
8
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
40 class Tileset(object):
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
41 def __init__(self, tileset_name):
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
42 self.name = tileset_name
31
67218082cdbb More level non-functionality.
Jeremy Thurgood <firxen@gmail.com>
parents: 21
diff changeset
43 self.load_tiles()
8
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
44
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
45 def load_tiles(self):
31
67218082cdbb More level non-functionality.
Jeremy Thurgood <firxen@gmail.com>
parents: 21
diff changeset
46 self.tiles = {}
104
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
47 self.floor = TileSprite(tileset=self.name, image_name='floor').image
50
1b725035b8ef Levels that are visible.
Jeremy Thurgood <firxen@gmail.com>
parents: 37
diff changeset
48 for name, value in TILE_MAP.items():
1b725035b8ef Levels that are visible.
Jeremy Thurgood <firxen@gmail.com>
parents: 37
diff changeset
49 if value is not None:
104
5b92ff941de3 Refactor sprites.
Jeremy Thurgood <firxen@gmail.com>
parents: 98
diff changeset
50 value[1]['tileset'] = self.name
50
1b725035b8ef Levels that are visible.
Jeremy Thurgood <firxen@gmail.com>
parents: 37
diff changeset
51 self.tiles[name] = value
1b725035b8ef Levels that are visible.
Jeremy Thurgood <firxen@gmail.com>
parents: 37
diff changeset
52 self.tiles.update(THING_MAP)
8
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
53
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
54 def __getitem__(self, key):
84
cced0ddda33f More level validation.
Jeremy Thurgood <firxen@gmail.com>
parents: 83
diff changeset
55 try:
cced0ddda33f More level validation.
Jeremy Thurgood <firxen@gmail.com>
parents: 83
diff changeset
56 tilespec = self.tiles[key]
cced0ddda33f More level validation.
Jeremy Thurgood <firxen@gmail.com>
parents: 83
diff changeset
57 except KeyError:
cced0ddda33f More level validation.
Jeremy Thurgood <firxen@gmail.com>
parents: 83
diff changeset
58 raise InvalidMapError("Unknown tile type: '%s'" % key)
50
1b725035b8ef Levels that are visible.
Jeremy Thurgood <firxen@gmail.com>
parents: 37
diff changeset
59 if not tilespec:
1b725035b8ef Levels that are visible.
Jeremy Thurgood <firxen@gmail.com>
parents: 37
diff changeset
60 return None
105
87acd3ddc972 Tell sprites what their tile char is.
Jeremy Thurgood <firxen@gmail.com>
parents: 104
diff changeset
61 cls, params = tilespec
87acd3ddc972 Tell sprites what their tile char is.
Jeremy Thurgood <firxen@gmail.com>
parents: 104
diff changeset
62 params['tile_char'] = key
87acd3ddc972 Tell sprites what their tile char is.
Jeremy Thurgood <firxen@gmail.com>
parents: 104
diff changeset
63 return cls(**params)
31
67218082cdbb More level non-functionality.
Jeremy Thurgood <firxen@gmail.com>
parents: 21
diff changeset
64
50
1b725035b8ef Levels that are visible.
Jeremy Thurgood <firxen@gmail.com>
parents: 37
diff changeset
65 def get_tile(self, key, tile_pos, *groups):
31
67218082cdbb More level non-functionality.
Jeremy Thurgood <firxen@gmail.com>
parents: 21
diff changeset
66 tile = self[key]
50
1b725035b8ef Levels that are visible.
Jeremy Thurgood <firxen@gmail.com>
parents: 37
diff changeset
67 if tile:
1b725035b8ef Levels that are visible.
Jeremy Thurgood <firxen@gmail.com>
parents: 37
diff changeset
68 tile.add(*groups)
1b725035b8ef Levels that are visible.
Jeremy Thurgood <firxen@gmail.com>
parents: 37
diff changeset
69 tile.set_tile_pos(tile_pos)
31
67218082cdbb More level non-functionality.
Jeremy Thurgood <firxen@gmail.com>
parents: 21
diff changeset
70 return tile
8
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
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
73 class Level(object):
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
74 def __init__(self, level_name):
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
75 self.level_name = level_name
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
76 self.load_level_data()
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
77
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
78 def load_level_data(self):
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
79 """
31
67218082cdbb More level non-functionality.
Jeremy Thurgood <firxen@gmail.com>
parents: 21
diff changeset
80 This file format is potentially yucky.
8
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
81 """
66ae99f6903e First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff changeset
82 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
83 tileset_name = level_data.readline().strip()
21
f76c3f033772 Release early, release broken. Level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 8
diff changeset
84 self.tileset = Tileset(tileset_name)
84
cced0ddda33f More level validation.
Jeremy Thurgood <firxen@gmail.com>
parents: 83
diff changeset
85 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
86 self.setup_tiles(tiles_ascii)
71
58557a89ec86 Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 57
diff changeset
87 self.make_background()
58557a89ec86 Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 57
diff changeset
88
81
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
89 def setup_tiles(self, tiles_ascii):
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
90 self.sprites = RenderUpdates()
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
91 self.tiles = []
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
92 self.entry = None
84
cced0ddda33f More level validation.
Jeremy Thurgood <firxen@gmail.com>
parents: 83
diff changeset
93 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
94 for y, row in enumerate(tiles_ascii):
84
cced0ddda33f More level validation.
Jeremy Thurgood <firxen@gmail.com>
parents: 83
diff changeset
95 if len(row) != self.tile_size[0]:
cced0ddda33f More level validation.
Jeremy Thurgood <firxen@gmail.com>
parents: 83
diff changeset
96 raise InvalidMapError("Map not rectangular.")
81
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
97 tile_row = []
84
cced0ddda33f More level validation.
Jeremy Thurgood <firxen@gmail.com>
parents: 83
diff changeset
98 for x, tile_char in enumerate(row):
81
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
99 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
100 tile_row.append(tile)
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
101 if isinstance(tile, EntrySprite):
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
102 if self.entry is not None:
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
103 raise InvalidMapError("Too many entry points.")
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
104 self.entry = tile
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
105 self.tiles.append(tile_row)
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
106 if self.entry is None:
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
107 raise InvalidMapError("Not enough entry points.")
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
108
71
58557a89ec86 Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 57
diff changeset
109 def get_tile_size(self):
84
cced0ddda33f More level validation.
Jeremy Thurgood <firxen@gmail.com>
parents: 83
diff changeset
110 return self.tile_size
21
f76c3f033772 Release early, release broken. Level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 8
diff changeset
111
f76c3f033772 Release early, release broken. Level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 8
diff changeset
112 def get_size(self):
71
58557a89ec86 Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 57
diff changeset
113 x, y = self.get_tile_size()
58557a89ec86 Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 57
diff changeset
114 return tile_sizify((x, y))
58557a89ec86 Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 57
diff changeset
115
58557a89ec86 Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 57
diff changeset
116 def make_background(self):
58557a89ec86 Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 57
diff changeset
117 sx, sy = self.get_tile_size()
58557a89ec86 Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 57
diff changeset
118 self.background = Surface(self.get_size())
58557a89ec86 Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents: 57
diff changeset
119 [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
120 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
121
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
122 def get_entry(self):
c8d6ca296884 Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents: 71
diff changeset
123 return (self.entry.tile_pos, self.entry.direction)