Mercurial > mamba
annotate mamba/level.py @ 146:5bade7867d04
Remove useless debug print.
author | Jeremy Thurgood <firxen@gmail.com> |
---|---|
date | Tue, 13 Sep 2011 00:00:22 +0200 |
parents | b292370c4548 |
children | 7fbbe27120a9 |
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 | 6 from pygame.sprite import RenderUpdates |
8
66ae99f6903e
First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff
changeset
|
7 |
50 | 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 | 18 def mktile(cls, **kw): |
19 return (cls, kw) | |
20 | |
21 | |
37 | 22 TILE_MAP = { |
23 '.': None, | |
144
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
24 'X': mktile(TileSprite, image_name='wall', solid=True), |
104 | 25 'R': mktile(DoorSprite, colour='red'), |
26 'B': mktile(DoorSprite, colour='blue'), | |
27 '^': mktile(EntrySprite, direction=Snake.UP), | |
28 'v': mktile(EntrySprite, direction=Snake.DOWN), | |
29 '<': mktile(EntrySprite, direction=Snake.LEFT), | |
30 '>': mktile(EntrySprite, direction=Snake.RIGHT), | |
31 'E': mktile(ExitSprite), | |
32 '~': mktile(PuddleSprite), | |
37 | 33 } |
34 | |
35 THING_MAP = { | |
136
00ada2e29798
Somewhat better (but still hideous) image variant support.
Jeremy Thurgood <firxen@gmail.com>
parents:
132
diff
changeset
|
36 'a': mktile(BaseSprite, image_name='red'), |
37 | 37 } |
38 | |
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 | 47 self.floor = TileSprite(tileset=self.name, image_name='floor').image |
50 | 48 for name, value in TILE_MAP.items(): |
49 if value is not None: | |
104 | 50 value[1]['tileset'] = self.name |
50 | 51 self.tiles[name] = value |
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 | 55 try: |
56 tilespec = self.tiles[key] | |
57 except KeyError: | |
58 raise InvalidMapError("Unknown tile type: '%s'" % key) | |
50 | 59 if not tilespec: |
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 | 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 | 67 if tile: |
68 tile.add(*groups) | |
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 |
123
915fd0ee28f5
Added test tile orientation
Gideon Visser <gideon@gideonvisser.com>
parents:
112
diff
changeset
|
76 self.tiles_ascii = '' |
8
66ae99f6903e
First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff
changeset
|
77 self.load_level_data() |
66ae99f6903e
First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff
changeset
|
78 |
66ae99f6903e
First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff
changeset
|
79 def load_level_data(self): |
66ae99f6903e
First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff
changeset
|
80 """ |
31
67218082cdbb
More level non-functionality.
Jeremy Thurgood <firxen@gmail.com>
parents:
21
diff
changeset
|
81 This file format is potentially yucky. |
8
66ae99f6903e
First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff
changeset
|
82 """ |
66ae99f6903e
First stab at tile/level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
diff
changeset
|
83 level_data = load_file('levels/%s.txt' % (self.level_name,)) |
136
00ada2e29798
Somewhat better (but still hideous) image variant support.
Jeremy Thurgood <firxen@gmail.com>
parents:
132
diff
changeset
|
84 self.name = level_data.readline().strip() |
31
67218082cdbb
More level non-functionality.
Jeremy Thurgood <firxen@gmail.com>
parents:
21
diff
changeset
|
85 tileset_name = level_data.readline().strip() |
21
f76c3f033772
Release early, release broken. Level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
8
diff
changeset
|
86 self.tileset = Tileset(tileset_name) |
84 | 87 tiles_ascii = [line.strip() for line in level_data.readlines()] |
123
915fd0ee28f5
Added test tile orientation
Gideon Visser <gideon@gideonvisser.com>
parents:
112
diff
changeset
|
88 self.tiles_ascii = tiles_ascii |
81
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
89 self.setup_tiles(tiles_ascii) |
71
58557a89ec86
Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
57
diff
changeset
|
90 self.make_background() |
58557a89ec86
Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
57
diff
changeset
|
91 |
81
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
92 def setup_tiles(self, tiles_ascii): |
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
93 self.sprites = RenderUpdates() |
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
94 self.tiles = [] |
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
95 self.entry = None |
84 | 96 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
|
97 for y, row in enumerate(tiles_ascii): |
84 | 98 if len(row) != self.tile_size[0]: |
99 raise InvalidMapError("Map not rectangular.") | |
81
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
100 tile_row = [] |
84 | 101 for x, tile_char in enumerate(row): |
141 | 102 #tile_orientation = self.get_tile_orientation(y, x, row, |
103 # tile_char) | |
81
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
104 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
|
105 tile_row.append(tile) |
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
106 if isinstance(tile, EntrySprite): |
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
107 if self.entry is not None: |
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
108 raise InvalidMapError("Too many entry points.") |
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
109 self.entry = tile |
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
110 self.tiles.append(tile_row) |
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
111 if self.entry is None: |
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
112 raise InvalidMapError("Not enough entry points.") |
144
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
113 self.set_tile_orientations() |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
114 |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
115 def set_tile_orientations(self): |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
116 tiles = [tile # This is a scary listcomp. It makes me happy. |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
117 for row in self.tiles |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
118 for tile in row |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
119 if tile is not None] |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
120 for tile in tiles: |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
121 orientation = self.get_tile_orientation(tile) |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
122 tile.use_variant(*orientation) |
81
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
123 |
123
915fd0ee28f5
Added test tile orientation
Gideon Visser <gideon@gideonvisser.com>
parents:
112
diff
changeset
|
124 def is_same_tile(self, tile, x, y): |
915fd0ee28f5
Added test tile orientation
Gideon Visser <gideon@gideonvisser.com>
parents:
112
diff
changeset
|
125 """Is there a tile of the same type?""" |
144
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
126 if tile.tile_char is None: |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
127 # This isn't really a tile, so bail |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
128 return False |
123
915fd0ee28f5
Added test tile orientation
Gideon Visser <gideon@gideonvisser.com>
parents:
112
diff
changeset
|
129 try: |
144
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
130 other_tile = self.tiles[y][x] |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
131 except IndexError: |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
132 # We're over the edge of the map |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
133 return False |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
134 if other_tile is None: |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
135 # Emptiness. |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
136 return False |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
137 return tile.tile_char == other_tile.tile_char |
123
915fd0ee28f5
Added test tile orientation
Gideon Visser <gideon@gideonvisser.com>
parents:
112
diff
changeset
|
138 |
144
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
139 def get_tile_orientation(self, tile): |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
140 if tile is None: |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
141 return (False, False, False, False) |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
142 map_x, map_y = tile.tile_pos |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
143 return ( |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
144 self.is_same_tile(tile, map_x, map_y - 1), # above |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
145 self.is_same_tile(tile, map_x, map_y + 1), # below |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
146 self.is_same_tile(tile, map_x - 1, map_y), # left |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
147 self.is_same_tile(tile, map_x + 1, map_y), # right |
b292370c4548
Rearranged the orientation stuff. :-)
Jeremy Thurgood <firxen@gmail.com>
parents:
143
diff
changeset
|
148 ) |
123
915fd0ee28f5
Added test tile orientation
Gideon Visser <gideon@gideonvisser.com>
parents:
112
diff
changeset
|
149 |
71
58557a89ec86
Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
57
diff
changeset
|
150 def get_tile_size(self): |
84 | 151 return self.tile_size |
21
f76c3f033772
Release early, release broken. Level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
8
diff
changeset
|
152 |
f76c3f033772
Release early, release broken. Level stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
8
diff
changeset
|
153 def get_size(self): |
71
58557a89ec86
Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
57
diff
changeset
|
154 x, y = self.get_tile_size() |
58557a89ec86
Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
57
diff
changeset
|
155 return tile_sizify((x, y)) |
58557a89ec86
Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
57
diff
changeset
|
156 |
58557a89ec86
Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
57
diff
changeset
|
157 def make_background(self): |
58557a89ec86
Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
57
diff
changeset
|
158 sx, sy = self.get_tile_size() |
58557a89ec86
Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
57
diff
changeset
|
159 self.background = Surface(self.get_size()) |
58557a89ec86
Some tile improvements and stuff.
Jeremy Thurgood <firxen@gmail.com>
parents:
57
diff
changeset
|
160 [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
|
161 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
|
162 |
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
163 def get_entry(self): |
c8d6ca296884
Better level entry points.
Jeremy Thurgood <firxen@gmail.com>
parents:
71
diff
changeset
|
164 return (self.entry.tile_pos, self.entry.direction) |
110 | 165 |
166 def draw(self, surface): | |
112
62a705f4e876
Blit the background onto the screen, not the screen onto the background.
Simon Cross <hodgestar@gmail.com>
parents:
110
diff
changeset
|
167 surface.blit(self.background, (0, 0)) |
110 | 168 self.sprites.draw(surface) |