Mercurial > mamba
annotate mamba/snake.py @ 304:49fe66680ff8
And even death shall lose it's mysteries
| author | Neil Muller <drnlmuller@gmail.com> |
|---|---|
| date | Fri, 16 Sep 2011 17:37:28 +0200 |
| parents | 43ac21d498a1 |
| children | 0ed07dcebff4 |
| rev | line source |
|---|---|
| 78 | 1 """The player snake object.""" |
| 2 | |
| 221 | 3 import random |
| 4 | |
|
188
d76c069164c0
Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents:
187
diff
changeset
|
5 from pygame.sprite import Group, spritecollide |
| 110 | 6 |
|
199
986e72d2cb4d
Rejiggered entrances and shifted directions around.
Jeremy Thurgood <firxen@gmail.com>
parents:
198
diff
changeset
|
7 from mamba.constants import TILE_SIZE, UP, DOWN, LEFT, RIGHT |
|
252
8198492745b1
Try not to let head wander during teleporting (it's bad for your health).
Simon Cross <hodgestar@gmail.com>
parents:
232
diff
changeset
|
8 from mamba.sprites import BaseSprite, tile_sizify |
| 213 | 9 from mamba.engine import SnakeDiedEvent, LevelCompletedEvent |
| 282 | 10 from mamba.sound import load_sound, play_sound |
| 98 | 11 from mamba import mutators |
| 78 | 12 |
| 13 | |
|
217
874eb26aad10
No more mile-long snake to start with.
Jeremy Thurgood <firxen@gmail.com>
parents:
213
diff
changeset
|
14 INITIAL_SEGMENT_COUNT = 4 |
|
874eb26aad10
No more mile-long snake to start with.
Jeremy Thurgood <firxen@gmail.com>
parents:
213
diff
changeset
|
15 |
|
874eb26aad10
No more mile-long snake to start with.
Jeremy Thurgood <firxen@gmail.com>
parents:
213
diff
changeset
|
16 |
| 110 | 17 class Snake(object): |
| 78 | 18 |
| 19 def __init__(self, tile_pos, orientation): | |
| 282 | 20 load_sound('crash', 'sounds/crash.ogg') |
| 110 | 21 self.segments = self.create_segments(tile_pos, orientation) |
|
201
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
22 self.pending_segments = [] # segments waiting to be added |
| 110 | 23 self.segment_group = Group() |
| 178 | 24 self.segment_group.add(*reversed(self.segments)) |
|
232
ec23ecbbd1c6
Move U-turn checking to orientation-setting time, and only update the head's orientaiton at tile shift
Stefano Rivera <stefano@rivera.za.net>
parents:
225
diff
changeset
|
25 self.orientation = orientation |
| 178 | 26 self.speed = 120.0 # pixel / s |
|
177
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
27 self.frac_ds = 0.0 |
|
181
061d711ba570
Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents:
178
diff
changeset
|
28 self.mutation = None |
|
225
af95fbc8ef5c
Snakes should be impermeable. So now they are.
Jeremy Thurgood <firxen@gmail.com>
parents:
223
diff
changeset
|
29 self.coiled = True |
| 78 | 30 |
| 110 | 31 head = property(fget=lambda self: self.segments[0]) |
| 32 tail = property(fget=lambda self: self.segments[-1]) | |
| 33 | |
| 34 def create_segments(self, tile_pos, orientation): | |
|
177
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
35 tx, ty = tile_pos |
| 110 | 36 dx, dy = orientation |
| 178 | 37 segments = [] |
|
217
874eb26aad10
No more mile-long snake to start with.
Jeremy Thurgood <firxen@gmail.com>
parents:
213
diff
changeset
|
38 for cls in [Head] + [Body] * INITIAL_SEGMENT_COUNT + [Tail]: |
| 178 | 39 segments.append(cls((tx, ty))) |
| 40 return segments | |
| 110 | 41 |
|
201
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
42 def add_segment(self, segment=None): |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
43 if segment is None: |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
44 segment = Body((0, 0)) |
|
205
2b5956bedf64
Set colour of default new segment to colour of snake head.
Simon Cross <hodgestar@gmail.com>
parents:
203
diff
changeset
|
45 segment.set_colour(self.head.colour) |
|
201
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
46 self.pending_segments.append(segment) |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
47 |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
48 def remove_segment(self, segment=None): |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
49 if segment is None: |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
50 if len(self.segments) < 3: |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
51 return |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
52 segment = self.segments[-2] |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
53 try: |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
54 idx = self.segments.index(segment) |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
55 except IndexError: |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
56 return |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
57 self.shiftup_segments(idx, segment.get_tile_state()) |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
58 del self.segments[idx] |
|
202
498c1d659a94
Remove a segment when driving over the small rat.
Simon Cross <hodgestar@gmail.com>
parents:
201
diff
changeset
|
59 segment.kill() |
|
201
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
60 |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
61 def shiftup_segments(self, idx, state): |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
62 for segment in self.segments[idx:]: |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
63 next_state = segment.get_tile_state() |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
64 segment.shift_tile(state) |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
65 state = next_state |
|
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
66 |
| 110 | 67 def draw(self, surface): |
|
177
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
68 self.segment_group.draw(surface) |
| 98 | 69 |
|
181
061d711ba570
Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents:
178
diff
changeset
|
70 def update(self, dt, world): |
|
177
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
71 ds = dt * self.speed + self.frac_ds |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
72 ds, self.frac_ds = divmod(ds, 1) |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
73 ds = int(ds) |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
74 while True: |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
75 tile_state = self.head.get_tile_state() |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
76 shifted, ds = self.head.shift_head(ds) |
|
218
10804ae58066
Open mouth when eating things
Stefano Rivera <stefano@rivera.za.net>
parents:
217
diff
changeset
|
77 if shifted: |
|
225
af95fbc8ef5c
Snakes should be impermeable. So now they are.
Jeremy Thurgood <firxen@gmail.com>
parents:
223
diff
changeset
|
78 self.coiled = False |
| 221 | 79 self.head.shifted_tile() |
|
232
ec23ecbbd1c6
Move U-turn checking to orientation-setting time, and only update the head's orientaiton at tile shift
Stefano Rivera <stefano@rivera.za.net>
parents:
225
diff
changeset
|
80 self.head.set_orientation(self.orientation) |
|
218
10804ae58066
Open mouth when eating things
Stefano Rivera <stefano@rivera.za.net>
parents:
217
diff
changeset
|
81 else: |
|
177
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
82 break |
|
203
8009742103e7
Add a segment when driving over the big rat.
Simon Cross <hodgestar@gmail.com>
parents:
202
diff
changeset
|
83 if self.pending_segments: |
|
8009742103e7
Add a segment when driving over the big rat.
Simon Cross <hodgestar@gmail.com>
parents:
202
diff
changeset
|
84 new_segment = self.pending_segments.pop(0) |
|
8009742103e7
Add a segment when driving over the big rat.
Simon Cross <hodgestar@gmail.com>
parents:
202
diff
changeset
|
85 self.segments.insert(1, new_segment) |
|
8009742103e7
Add a segment when driving over the big rat.
Simon Cross <hodgestar@gmail.com>
parents:
202
diff
changeset
|
86 self.segment_group.add(new_segment) |
|
8009742103e7
Add a segment when driving over the big rat.
Simon Cross <hodgestar@gmail.com>
parents:
202
diff
changeset
|
87 new_segment.shift_tile(tile_state) |
| 206 | 88 for segment in self.segments[2:]: |
| 89 segment.shift_tile(segment.get_tile_state()) | |
|
203
8009742103e7
Add a segment when driving over the big rat.
Simon Cross <hodgestar@gmail.com>
parents:
202
diff
changeset
|
90 else: |
|
8009742103e7
Add a segment when driving over the big rat.
Simon Cross <hodgestar@gmail.com>
parents:
202
diff
changeset
|
91 self.shiftup_segments(1, tile_state) |
|
119
119c0fb758c2
Move key handling into GameWidget and GameWidget into widgets.game.
Simon Cross <hodgestar@gmail.com>
parents:
118
diff
changeset
|
92 |
|
225
af95fbc8ef5c
Snakes should be impermeable. So now they are.
Jeremy Thurgood <firxen@gmail.com>
parents:
223
diff
changeset
|
93 self.check_self_crash() |
|
177
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
94 for segment in self.segments: |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
95 segment.shift_pixels(ds) |
|
188
d76c069164c0
Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents:
187
diff
changeset
|
96 world.interact(segment) |
|
145
20b2cedf0f1d
Snake movement now turn
Gideon Visser <gideon@gideonvisser.com>
parents:
141
diff
changeset
|
97 |
|
201
78f9cafb0819
Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents:
199
diff
changeset
|
98 def set_orientation(self, orientation): |
|
232
ec23ecbbd1c6
Move U-turn checking to orientation-setting time, and only update the head's orientaiton at tile shift
Stefano Rivera <stefano@rivera.za.net>
parents:
225
diff
changeset
|
99 # Don't allow the snake to go back on itself |
|
ec23ecbbd1c6
Move U-turn checking to orientation-setting time, and only update the head's orientaiton at tile shift
Stefano Rivera <stefano@rivera.za.net>
parents:
225
diff
changeset
|
100 # immediately. More creative self-inflicted |
|
ec23ecbbd1c6
Move U-turn checking to orientation-setting time, and only update the head's orientaiton at tile shift
Stefano Rivera <stefano@rivera.za.net>
parents:
225
diff
changeset
|
101 # deaths are allowed though. |
|
ec23ecbbd1c6
Move U-turn checking to orientation-setting time, and only update the head's orientaiton at tile shift
Stefano Rivera <stefano@rivera.za.net>
parents:
225
diff
changeset
|
102 if (abs(self.head.orientation[0] - orientation[0]) < 2 |
|
ec23ecbbd1c6
Move U-turn checking to orientation-setting time, and only update the head's orientaiton at tile shift
Stefano Rivera <stefano@rivera.za.net>
parents:
225
diff
changeset
|
103 and abs(self.head.orientation[1] - orientation[1]) < 2): |
|
ec23ecbbd1c6
Move U-turn checking to orientation-setting time, and only update the head's orientaiton at tile shift
Stefano Rivera <stefano@rivera.za.net>
parents:
225
diff
changeset
|
104 self.orientation = orientation |
| 98 | 105 |
|
225
af95fbc8ef5c
Snakes should be impermeable. So now they are.
Jeremy Thurgood <firxen@gmail.com>
parents:
223
diff
changeset
|
106 def check_self_crash(self): |
|
af95fbc8ef5c
Snakes should be impermeable. So now they are.
Jeremy Thurgood <firxen@gmail.com>
parents:
223
diff
changeset
|
107 if self.coiled: |
|
af95fbc8ef5c
Snakes should be impermeable. So now they are.
Jeremy Thurgood <firxen@gmail.com>
parents:
223
diff
changeset
|
108 return |
|
af95fbc8ef5c
Snakes should be impermeable. So now they are.
Jeremy Thurgood <firxen@gmail.com>
parents:
223
diff
changeset
|
109 collides = spritecollide(self.head, self.segment_group, False) |
|
af95fbc8ef5c
Snakes should be impermeable. So now they are.
Jeremy Thurgood <firxen@gmail.com>
parents:
223
diff
changeset
|
110 if [s for s in collides if s not in self.segments[:2]]: |
|
304
49fe66680ff8
And even death shall lose it's mysteries
Neil Muller <drnlmuller@gmail.com>
parents:
295
diff
changeset
|
111 self.crash('You hit yourself!') |
|
225
af95fbc8ef5c
Snakes should be impermeable. So now they are.
Jeremy Thurgood <firxen@gmail.com>
parents:
223
diff
changeset
|
112 |
|
304
49fe66680ff8
And even death shall lose it's mysteries
Neil Muller <drnlmuller@gmail.com>
parents:
295
diff
changeset
|
113 def crash(self, reason='You hit something'): |
| 282 | 114 play_sound('crash') |
|
304
49fe66680ff8
And even death shall lose it's mysteries
Neil Muller <drnlmuller@gmail.com>
parents:
295
diff
changeset
|
115 SnakeDiedEvent.post(reason) |
|
181
061d711ba570
Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents:
178
diff
changeset
|
116 |
| 213 | 117 def exit_level(self): |
| 118 LevelCompletedEvent.post() | |
| 119 | |
|
181
061d711ba570
Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents:
178
diff
changeset
|
120 def mutate(self, mutation): |
|
061d711ba570
Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents:
178
diff
changeset
|
121 self.mutation = mutation |
|
261
d0ed88799c44
mutation indicators
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
255
diff
changeset
|
122 self.tail.show_mutation(mutation) |
|
181
061d711ba570
Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents:
178
diff
changeset
|
123 |
|
061d711ba570
Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents:
178
diff
changeset
|
124 def can_swim(self): |
|
061d711ba570
Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents:
178
diff
changeset
|
125 return self.mutation == 'amphibious' |
|
061d711ba570
Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents:
178
diff
changeset
|
126 |
| 255 | 127 def flame_retardant(self): |
| 128 return self.mutation == 'flame-retardant' | |
| 129 | |
|
263
c7c9d71ac6ea
Lizards and snails are extra delicious.
Jeremy Thurgood <firxen@gmail.com>
parents:
262
diff
changeset
|
130 def adjust_speed(self, delta): |
|
c7c9d71ac6ea
Lizards and snails are extra delicious.
Jeremy Thurgood <firxen@gmail.com>
parents:
262
diff
changeset
|
131 self.speed += delta |
|
c7c9d71ac6ea
Lizards and snails are extra delicious.
Jeremy Thurgood <firxen@gmail.com>
parents:
262
diff
changeset
|
132 self.speed = max(self.speed, 60) |
|
c7c9d71ac6ea
Lizards and snails are extra delicious.
Jeremy Thurgood <firxen@gmail.com>
parents:
262
diff
changeset
|
133 self.speed = min(self.speed, 180) |
|
c7c9d71ac6ea
Lizards and snails are extra delicious.
Jeremy Thurgood <firxen@gmail.com>
parents:
262
diff
changeset
|
134 |
| 98 | 135 |
| 110 | 136 class Segment(BaseSprite): |
| 137 | |
|
195
d488731886dc
Wean colour overlays off image colour patches.
Simon Cross <hodgestar@gmail.com>
parents:
188
diff
changeset
|
138 GREEN = mutators.SNAKE_GREEN |
|
125
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
139 BLUE = mutators.BLUE |
|
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
140 RED = mutators.RED |
|
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
141 YELLOW = mutators.YELLOW |
|
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
142 |
| 131 | 143 _detail_mutators = () |
| 144 | |
|
182
0a955d2536f0
Painted Jezebel! (Assuming you name your snake Jezebel.)
Jeremy Thurgood <firxen@gmail.com>
parents:
181
diff
changeset
|
145 is_head = False |
|
0a955d2536f0
Painted Jezebel! (Assuming you name your snake Jezebel.)
Jeremy Thurgood <firxen@gmail.com>
parents:
181
diff
changeset
|
146 |
| 110 | 147 def __init__(self, image_name, tile_pos): |
| 148 super(Segment, self).__init__() | |
|
153
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
149 self.set_base_image(image_name) |
|
205
2b5956bedf64
Set colour of default new segment to colour of snake head.
Simon Cross <hodgestar@gmail.com>
parents:
203
diff
changeset
|
150 self.colour = self.GREEN |
|
199
986e72d2cb4d
Rejiggered entrances and shifted directions around.
Jeremy Thurgood <firxen@gmail.com>
parents:
198
diff
changeset
|
151 self.orientation = UP |
|
125
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
152 |
| 131 | 153 self.make_images() |
|
125
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
154 self.update_image() |
|
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
155 self.set_tile_pos(tile_pos) |
|
188
d76c069164c0
Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents:
187
diff
changeset
|
156 self.on_tiles = [] |
|
d76c069164c0
Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents:
187
diff
changeset
|
157 |
|
d76c069164c0
Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents:
187
diff
changeset
|
158 def filter_collisions(self, group): |
|
d76c069164c0
Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents:
187
diff
changeset
|
159 collide = [] |
|
d76c069164c0
Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents:
187
diff
changeset
|
160 tiles = spritecollide(self, group, False) |
|
d76c069164c0
Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents:
187
diff
changeset
|
161 for tile in tiles: |
|
d76c069164c0
Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents:
187
diff
changeset
|
162 if tile not in self.on_tiles: |
|
d76c069164c0
Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents:
187
diff
changeset
|
163 collide.append(tile) |
|
d76c069164c0
Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents:
187
diff
changeset
|
164 self.on_tiles = tiles |
|
d76c069164c0
Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents:
187
diff
changeset
|
165 return collide |
|
125
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
166 |
|
153
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
167 def set_base_image(self, image_name): |
|
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
168 self._base_image = "/".join(["snake", image_name]) |
|
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
169 |
| 131 | 170 def make_images(self): |
| 110 | 171 self._images = {} |
| 172 for orientation, muts in [ | |
|
199
986e72d2cb4d
Rejiggered entrances and shifted directions around.
Jeremy Thurgood <firxen@gmail.com>
parents:
198
diff
changeset
|
173 (RIGHT, (mutators.RIGHT,)), |
|
986e72d2cb4d
Rejiggered entrances and shifted directions around.
Jeremy Thurgood <firxen@gmail.com>
parents:
198
diff
changeset
|
174 (LEFT, (mutators.LEFT,)), |
|
986e72d2cb4d
Rejiggered entrances and shifted directions around.
Jeremy Thurgood <firxen@gmail.com>
parents:
198
diff
changeset
|
175 (UP, (mutators.UP,)), |
|
986e72d2cb4d
Rejiggered entrances and shifted directions around.
Jeremy Thurgood <firxen@gmail.com>
parents:
198
diff
changeset
|
176 (DOWN, (mutators.DOWN,)), |
| 110 | 177 ]: |
|
205
2b5956bedf64
Set colour of default new segment to colour of snake head.
Simon Cross <hodgestar@gmail.com>
parents:
203
diff
changeset
|
178 all_muts = (self.colour,) + self._detail_mutators + muts |
| 141 | 179 self._images[orientation] = self.load_image(self._base_image, |
| 180 all_muts) | |
|
125
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
181 |
|
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
182 def update_image(self): |
|
177
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
183 self.image = self._images[self.orientation] |
| 110 | 184 |
| 185 def set_orientation(self, orientation): | |
|
177
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
186 self.orientation = orientation |
|
125
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
187 self.update_image() |
|
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
188 |
|
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
189 def set_colour(self, colour_overlay): |
|
205
2b5956bedf64
Set colour of default new segment to colour of snake head.
Simon Cross <hodgestar@gmail.com>
parents:
203
diff
changeset
|
190 self.colour = colour_overlay |
| 131 | 191 self.make_images() |
|
125
625b22f92efa
the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
122
diff
changeset
|
192 self.update_image() |
| 110 | 193 |
|
177
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
194 def get_tile_state(self): |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
195 return self.tile_pos, self.orientation |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
196 |
|
252
8198492745b1
Try not to let head wander during teleporting (it's bad for your health).
Simon Cross <hodgestar@gmail.com>
parents:
232
diff
changeset
|
197 def get_distance(self): |
|
8198492745b1
Try not to let head wander during teleporting (it's bad for your health).
Simon Cross <hodgestar@gmail.com>
parents:
232
diff
changeset
|
198 rx, ry = self.rect.topleft |
|
8198492745b1
Try not to let head wander during teleporting (it's bad for your health).
Simon Cross <hodgestar@gmail.com>
parents:
232
diff
changeset
|
199 x, y = tile_sizify(self.tile_pos) |
|
8198492745b1
Try not to let head wander during teleporting (it's bad for your health).
Simon Cross <hodgestar@gmail.com>
parents:
232
diff
changeset
|
200 return max(abs(rx - x), abs(ry - y)) |
|
8198492745b1
Try not to let head wander during teleporting (it's bad for your health).
Simon Cross <hodgestar@gmail.com>
parents:
232
diff
changeset
|
201 |
|
177
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
202 def shift_tile(self, tile_state): |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
203 """Shift this segment to the tile the other one was on. |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
204 |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
205 Also reset the position to be the center of the tile. |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
206 """ |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
207 tile_pos, orientation = tile_state |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
208 self.set_tile_pos(tile_pos) |
|
295
43ac21d498a1
Fix segment orientation.
Jeremy Thurgood <firxen@gmail.com>
parents:
282
diff
changeset
|
209 self.set_orientation(orientation) |
|
177
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
210 |
|
252
8198492745b1
Try not to let head wander during teleporting (it's bad for your health).
Simon Cross <hodgestar@gmail.com>
parents:
232
diff
changeset
|
211 def shift_tile_and_pixels(self, tile_state): |
|
8198492745b1
Try not to let head wander during teleporting (it's bad for your health).
Simon Cross <hodgestar@gmail.com>
parents:
232
diff
changeset
|
212 ds = self.get_distance() |
|
8198492745b1
Try not to let head wander during teleporting (it's bad for your health).
Simon Cross <hodgestar@gmail.com>
parents:
232
diff
changeset
|
213 self.shift_tile(tile_state) |
|
8198492745b1
Try not to let head wander during teleporting (it's bad for your health).
Simon Cross <hodgestar@gmail.com>
parents:
232
diff
changeset
|
214 self.shift_pixels(ds) |
|
8198492745b1
Try not to let head wander during teleporting (it's bad for your health).
Simon Cross <hodgestar@gmail.com>
parents:
232
diff
changeset
|
215 |
| 221 | 216 def shifted_tile(self): |
| 217 pass | |
| 218 | |
|
177
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
219 def shift_pixels(self, distance): |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
220 """Shift this segment a number of pixels.""" |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
221 dx, dy = self.orientation |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
222 dx, dy = distance * dx, distance * dy |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
223 self.rect = self.rect.move(dx, dy) |
|
e0573297b17c
Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents:
153
diff
changeset
|
224 |
|
223
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
225 def shift_head(self, ds): |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
226 """Shift the head a number of pixels in the direction of it |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
227 orientation. |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
228 """ |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
229 dx, dy = self.orientation |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
230 TX, TY = TILE_SIZE |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
231 rx, ry = self.rect.left, self.rect.top |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
232 tx, ty = self.tile_pos |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
233 # WARNING: Tri-state logic ahead |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
234 # (Tri-state logic is the mamba's natural habitat) |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
235 if dx != 0: |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
236 newdx = rx + ds * dx |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
237 newtx = newdx / TX |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
238 if newtx > tx: |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
239 self.set_tile_pos((tx + 1, ty)) |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
240 ds = newdx - self.rect.left |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
241 return True, ds |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
242 elif newtx < tx - 1: |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
243 self.set_tile_pos((tx - 1, ty)) |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
244 ds = self.rect.left - newdx |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
245 return True, ds |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
246 else: |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
247 newdy = ry + ds * dy |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
248 newty = newdy / TY |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
249 if newty > ty: |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
250 self.set_tile_pos((tx, ty + 1)) |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
251 ds = newdy - self.rect.top |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
252 return True, ds |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
253 elif newty < ty - 1: |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
254 self.set_tile_pos((tx, ty - 1)) |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
255 ds = self.rect.top - newdy |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
256 return True, ds |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
257 return False, ds |
|
c3018764942a
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
Simon Cross <hodgestar@gmail.com>
parents:
221
diff
changeset
|
258 |
| 110 | 259 |
| 260 class Head(Segment): | |
|
153
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
261 CLOSED = "snake-head" |
|
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
262 OPEN = "snake-head-mouth-open-r" |
|
212
e768056b0588
Refactor tile image loading stuff. More magic is good, right?
Jeremy Thurgood <firxen@gmail.com>
parents:
206
diff
changeset
|
263 EYE = mutators.Overlay("snake/snake-head-eye-r") |
|
e768056b0588
Refactor tile image loading stuff. More magic is good, right?
Jeremy Thurgood <firxen@gmail.com>
parents:
206
diff
changeset
|
264 TONGUE = mutators.Overlay("snake/snake-head-tongue-r") |
|
153
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
265 |
|
182
0a955d2536f0
Painted Jezebel! (Assuming you name your snake Jezebel.)
Jeremy Thurgood <firxen@gmail.com>
parents:
181
diff
changeset
|
266 is_head = True |
|
0a955d2536f0
Painted Jezebel! (Assuming you name your snake Jezebel.)
Jeremy Thurgood <firxen@gmail.com>
parents:
181
diff
changeset
|
267 |
| 110 | 268 def __init__(self, tile_pos): |
|
153
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
269 self._detail_mutators = (self.EYE,) |
|
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
270 super(Head, self).__init__(self.CLOSED, tile_pos) |
|
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
271 |
|
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
272 def mouth_open(self): |
|
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
273 self.set_base_image(self.OPEN) |
|
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
274 self.make_images() |
|
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
275 self.update_image() |
|
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
276 |
|
218
10804ae58066
Open mouth when eating things
Stefano Rivera <stefano@rivera.za.net>
parents:
217
diff
changeset
|
277 def mouth_close(self): |
|
153
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
278 self.set_base_image(self.CLOSED) |
|
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
279 self.make_images() |
|
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
280 self.update_image() |
| 110 | 281 |
| 131 | 282 def tongue_out(self): |
|
153
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
283 self._detail_mutators = (self.EYE, self.TONGUE) |
| 131 | 284 self.make_images() |
| 285 self.update_image() | |
| 286 | |
| 287 def tongue_in(self): | |
|
153
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
288 self._detail_mutators = (self.EYE,) |
| 131 | 289 self.make_images() |
| 290 self.update_image() | |
| 291 | |
| 221 | 292 def shifted_tile(self): |
| 293 if random.random() < 0.02: | |
| 294 self.mouth_open() | |
| 295 self.tongue_out() | |
| 296 else: | |
| 297 self.mouth_close() | |
| 298 self.tongue_in() | |
| 299 | |
| 110 | 300 |
| 301 class Body(Segment): | |
| 302 def __init__(self, tile_pos): | |
|
153
9c47bf162ea1
snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
145
diff
changeset
|
303 super(Body, self).__init__("snake-body", tile_pos) |
| 110 | 304 |
| 305 | |
| 306 class Tail(Segment): | |
|
261
d0ed88799c44
mutation indicators
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
255
diff
changeset
|
307 INDICATORS = { |
|
262
9f3b4e16372e
renamed tail images for consistency
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
261
diff
changeset
|
308 "flame-retardant": mutators.Overlay("snake/snake-tail-fire-r"), |
|
9f3b4e16372e
renamed tail images for consistency
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
261
diff
changeset
|
309 "amphibious": mutators.Overlay("snake/snake-tail-puddle-r"), |
|
261
d0ed88799c44
mutation indicators
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
255
diff
changeset
|
310 } |
|
d0ed88799c44
mutation indicators
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
255
diff
changeset
|
311 |
| 110 | 312 def __init__(self, tile_pos): |
|
262
9f3b4e16372e
renamed tail images for consistency
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
261
diff
changeset
|
313 super(Tail, self).__init__("snake-tail-r", tile_pos) |
|
261
d0ed88799c44
mutation indicators
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
255
diff
changeset
|
314 |
|
d0ed88799c44
mutation indicators
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
255
diff
changeset
|
315 def show_mutation(self, mutation): |
|
d0ed88799c44
mutation indicators
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
255
diff
changeset
|
316 self._detail_mutators = (self.INDICATORS[mutation],) |
|
d0ed88799c44
mutation indicators
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
255
diff
changeset
|
317 self.make_images() |
|
d0ed88799c44
mutation indicators
Adrianna Pińska <adrianna.pinska@gmail.com>
parents:
255
diff
changeset
|
318 self.update_image() |