Mercurial > mamba

annotate mamba/snake.py @ 223:c3018764942a

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Move shift_head onto base segment so that head is less special (in preparation for snake disappearing down the exit).
author Simon Cross <hodgestar@gmail.com>
date Wed, 14 Sep 2011 22:29:57 +0200
parents 3a652f048f57
children af95fbc8ef5c
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
78
b1019923cb12 Direction control for snake.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
1 """The player snake object."""
b1019923cb12 Direction control for snake.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
2
221
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
3 import random
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
4
188
d76c069164c0 Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents: 187
diff changeset
5 from pygame.sprite import Group, spritecollide
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
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
177
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
8 from mamba.sprites import BaseSprite
213
7a9be9add4c7 Hook up exit.
Simon Cross <hodgestar@gmail.com>
parents: 212
diff changeset
9 from mamba.engine import SnakeDiedEvent, LevelCompletedEvent
98
7ce2d2d8381a Refactor BaseSprite.
Simon Cross <hodgestar@gmail.com>
parents: 79
diff changeset
10 from mamba import mutators
78
b1019923cb12 Direction control for snake.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
11
b1019923cb12 Direction control for snake.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
12
217
874eb26aad10 No more mile-long snake to start with.
Jeremy Thurgood <firxen@gmail.com>
parents: 213
diff changeset
13 INITIAL_SEGMENT_COUNT = 4
874eb26aad10 No more mile-long snake to start with.
Jeremy Thurgood <firxen@gmail.com>
parents: 213
diff changeset
14
874eb26aad10 No more mile-long snake to start with.
Jeremy Thurgood <firxen@gmail.com>
parents: 213
diff changeset
15
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
16 class Snake(object):
78
b1019923cb12 Direction control for snake.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
17
b1019923cb12 Direction control for snake.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
18 def __init__(self, tile_pos, orientation):
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
19 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
20 self.pending_segments = [] # segments waiting to be added
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
21 self.segment_group = Group()
178
74f8d8cbc51d Longer, speedier mamba.
Simon Cross <hodgestar@gmail.com>
parents: 177
diff changeset
22 self.segment_group.add(*reversed(self.segments))
78
b1019923cb12 Direction control for snake.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
23 self.set_orientation(orientation)
178
74f8d8cbc51d Longer, speedier mamba.
Simon Cross <hodgestar@gmail.com>
parents: 177
diff changeset
24 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
25 self.frac_ds = 0.0
181
061d711ba570 Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents: 178
diff changeset
26 self.mutation = None
78
b1019923cb12 Direction control for snake.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
27
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
28 head = property(fget=lambda self: self.segments[0])
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
29 tail = property(fget=lambda self: self.segments[-1])
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
30
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
31 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
32 tx, ty = tile_pos
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
33 dx, dy = orientation
178
74f8d8cbc51d Longer, speedier mamba.
Simon Cross <hodgestar@gmail.com>
parents: 177
diff changeset
34 segments = []
217
874eb26aad10 No more mile-long snake to start with.
Jeremy Thurgood <firxen@gmail.com>
parents: 213
diff changeset
35 for cls in [Head] + [Body] * INITIAL_SEGMENT_COUNT + [Tail]:
178
74f8d8cbc51d Longer, speedier mamba.
Simon Cross <hodgestar@gmail.com>
parents: 177
diff changeset
36 segments.append(cls((tx, ty)))
74f8d8cbc51d Longer, speedier mamba.
Simon Cross <hodgestar@gmail.com>
parents: 177
diff changeset
37 return segments
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
38
201
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
39 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
40 if segment is None:
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
41 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
42 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
43 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
44
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
45 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
46 if segment is None:
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
47 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
48 return
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
49 segment = self.segments[-2]
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
50 try:
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
51 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
52 except IndexError:
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
53 return
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
54 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
55 del self.segments[idx]
202
498c1d659a94 Remove a segment when driving over the small rat.
Simon Cross <hodgestar@gmail.com>
parents: 201
diff changeset
56 segment.kill()
201
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
57
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
58 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
59 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
60 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
61 segment.shift_tile(state)
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
62 state = next_state
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
63
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
64 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
65 self.segment_group.draw(surface)
98
7ce2d2d8381a Refactor BaseSprite.
Simon Cross <hodgestar@gmail.com>
parents: 79
diff changeset
66
181
061d711ba570 Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents: 178
diff changeset
67 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
68 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
69 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
70 ds = int(ds)
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
71 while True:
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
72 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
73 shifted, ds = self.head.shift_head(ds)
218
10804ae58066 Open mouth when eating things
Stefano Rivera <stefano@rivera.za.net>
parents: 217
diff changeset
74 if shifted:
221
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
75 self.head.shifted_tile()
218
10804ae58066 Open mouth when eating things
Stefano Rivera <stefano@rivera.za.net>
parents: 217
diff changeset
76 else:
177
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
77 break
203
8009742103e7 Add a segment when driving over the big rat.
Simon Cross <hodgestar@gmail.com>
parents: 202
diff changeset
78 if self.pending_segments:
8009742103e7 Add a segment when driving over the big rat.
Simon Cross <hodgestar@gmail.com>
parents: 202
diff changeset
79 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
80 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
81 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
82 new_segment.shift_tile(tile_state)
206
bf661acd34ee Fix segment adding.
Simon Cross <hodgestar@gmail.com>
parents: 205
diff changeset
83 for segment in self.segments[2:]:
bf661acd34ee Fix segment adding.
Simon Cross <hodgestar@gmail.com>
parents: 205
diff changeset
84 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
85 else:
8009742103e7 Add a segment when driving over the big rat.
Simon Cross <hodgestar@gmail.com>
parents: 202
diff changeset
86 self.shiftup_segments(1, tile_state)
201
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
87 if not self.check_uturn():
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
88 self.head.set_orientation(self.orientation)
119
119c0fb758c2 Move key handling into GameWidget and GameWidget into widgets.game.
Simon Cross <hodgestar@gmail.com>
parents: 118
diff changeset
89
177
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
90 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
91 segment.shift_pixels(ds)
188
d76c069164c0 Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents: 187
diff changeset
92 world.interact(segment)
145
20b2cedf0f1d Snake movement now turn
Gideon Visser <gideon@gideonvisser.com>
parents: 141
diff changeset
93
201
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
94 def check_uturn(self):
198
2876658d0b98 Don't allow the snake to reverse onto itself immediately.
Simon Cross <hodgestar@gmail.com>
parents: 195
diff changeset
95 if len(self.segments) > 1:
2876658d0b98 Don't allow the snake to reverse onto itself immediately.
Simon Cross <hodgestar@gmail.com>
parents: 195
diff changeset
96 neck_x, neck_y = self.segments[1].tile_pos
2876658d0b98 Don't allow the snake to reverse onto itself immediately.
Simon Cross <hodgestar@gmail.com>
parents: 195
diff changeset
97 head_x, head_y = self.head.tile_pos
201
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
98 if (head_x + self.orientation[0] == neck_x and
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
99 head_y + self.orientation[1] == neck_y):
198
2876658d0b98 Don't allow the snake to reverse onto itself immediately.
Simon Cross <hodgestar@gmail.com>
parents: 195
diff changeset
100 # Don't allow the snake to go back on itself
2876658d0b98 Don't allow the snake to reverse onto itself immediately.
Simon Cross <hodgestar@gmail.com>
parents: 195
diff changeset
101 # immediately. More creative self-inflicted
2876658d0b98 Don't allow the snake to reverse onto itself immediately.
Simon Cross <hodgestar@gmail.com>
parents: 195
diff changeset
102 # deaths are allowed though.
201
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
103 return True
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
104 return False
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
105
78f9cafb0819 Fix uturns. Start of work on adding and removing segments.
Simon Cross <hodgestar@gmail.com>
parents: 199
diff changeset
106 def set_orientation(self, orientation):
119
119c0fb758c2 Move key handling into GameWidget and GameWidget into widgets.game.
Simon Cross <hodgestar@gmail.com>
parents: 118
diff changeset
107 self.orientation = orientation
98
7ce2d2d8381a Refactor BaseSprite.
Simon Cross <hodgestar@gmail.com>
parents: 79
diff changeset
108
181
061d711ba570 Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents: 178
diff changeset
109 def crash(self):
186
275f0be6946c Reincarnating Snakes
Neil Muller <drnlmuller@gmail.com>
parents: 182
diff changeset
110 SnakeDiedEvent.post()
181
061d711ba570 Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents: 178
diff changeset
111
213
7a9be9add4c7 Hook up exit.
Simon Cross <hodgestar@gmail.com>
parents: 212
diff changeset
112 def exit_level(self):
7a9be9add4c7 Hook up exit.
Simon Cross <hodgestar@gmail.com>
parents: 212
diff changeset
113 LevelCompletedEvent.post()
7a9be9add4c7 Hook up exit.
Simon Cross <hodgestar@gmail.com>
parents: 212
diff changeset
114
181
061d711ba570 Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents: 178
diff changeset
115 def mutate(self, mutation):
061d711ba570 Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents: 178
diff changeset
116 self.mutation = mutation
061d711ba570 Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents: 178
diff changeset
117
061d711ba570 Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents: 178
diff changeset
118 def can_swim(self):
061d711ba570 Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents: 178
diff changeset
119 return self.mutation == 'amphibious'
061d711ba570 Interact with the world.
Jeremy Thurgood <firxen@gmail.com>
parents: 178
diff changeset
120
98
7ce2d2d8381a Refactor BaseSprite.
Simon Cross <hodgestar@gmail.com>
parents: 79
diff changeset
121
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
122 class Segment(BaseSprite):
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
123
195
d488731886dc Wean colour overlays off image colour patches.
Simon Cross <hodgestar@gmail.com>
parents: 188
diff changeset
124 GREEN = mutators.SNAKE_GREEN
125
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
125 BLUE = mutators.BLUE
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
126 RED = mutators.RED
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
127 YELLOW = mutators.YELLOW
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
128
131
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
129 _detail_mutators = ()
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
130
182
0a955d2536f0 Painted Jezebel! (Assuming you name your snake Jezebel.)
Jeremy Thurgood <firxen@gmail.com>
parents: 181
diff changeset
131 is_head = False
0a955d2536f0 Painted Jezebel! (Assuming you name your snake Jezebel.)
Jeremy Thurgood <firxen@gmail.com>
parents: 181
diff changeset
132
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
133 def __init__(self, image_name, tile_pos):
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
134 super(Segment, self).__init__()
153
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
135 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
136 self.colour = self.GREEN
199
986e72d2cb4d Rejiggered entrances and shifted directions around.
Jeremy Thurgood <firxen@gmail.com>
parents: 198
diff changeset
137 self.orientation = UP
125
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
138
131
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
139 self.make_images()
125
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
140 self.update_image()
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
141 self.set_tile_pos(tile_pos)
188
d76c069164c0 Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents: 187
diff changeset
142 self.on_tiles = []
d76c069164c0 Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents: 187
diff changeset
143
d76c069164c0 Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents: 187
diff changeset
144 def filter_collisions(self, group):
d76c069164c0 Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents: 187
diff changeset
145 collide = []
d76c069164c0 Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents: 187
diff changeset
146 tiles = spritecollide(self, group, False)
d76c069164c0 Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents: 187
diff changeset
147 for tile in tiles:
d76c069164c0 Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents: 187
diff changeset
148 if tile not in self.on_tiles:
d76c069164c0 Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents: 187
diff changeset
149 collide.append(tile)
d76c069164c0 Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents: 187
diff changeset
150 self.on_tiles = tiles
d76c069164c0 Much better interaction detection.
Jeremy Thurgood <firxen@gmail.com>
parents: 187
diff changeset
151 return collide
125
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
152
153
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
153 def set_base_image(self, image_name):
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
154 self._base_image = "/".join(["snake", image_name])
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
155
131
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
156 def make_images(self):
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
157 self._images = {}
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
158 for orientation, muts in [
199
986e72d2cb4d Rejiggered entrances and shifted directions around.
Jeremy Thurgood <firxen@gmail.com>
parents: 198
diff changeset
159 (RIGHT, (mutators.RIGHT,)),
986e72d2cb4d Rejiggered entrances and shifted directions around.
Jeremy Thurgood <firxen@gmail.com>
parents: 198
diff changeset
160 (LEFT, (mutators.LEFT,)),
986e72d2cb4d Rejiggered entrances and shifted directions around.
Jeremy Thurgood <firxen@gmail.com>
parents: 198
diff changeset
161 (UP, (mutators.UP,)),
986e72d2cb4d Rejiggered entrances and shifted directions around.
Jeremy Thurgood <firxen@gmail.com>
parents: 198
diff changeset
162 (DOWN, (mutators.DOWN,)),
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
163 ]:
205
2b5956bedf64 Set colour of default new segment to colour of snake head.
Simon Cross <hodgestar@gmail.com>
parents: 203
diff changeset
164 all_muts = (self.colour,) + self._detail_mutators + muts
141
2c643ec8dc4e pep8 & pyflakes cleanup
Neil Muller <drnlmuller@gmail.com>
parents: 131
diff changeset
165 self._images[orientation] = self.load_image(self._base_image,
2c643ec8dc4e pep8 & pyflakes cleanup
Neil Muller <drnlmuller@gmail.com>
parents: 131
diff changeset
166 all_muts)
125
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
167
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
168 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
169 self.image = self._images[self.orientation]
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
170
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
171 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
172 self.orientation = orientation
125
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
173 self.update_image()
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
174
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
175 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
176 self.colour = colour_overlay
131
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
177 self.make_images()
125
625b22f92efa the snake is green
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 122
diff changeset
178 self.update_image()
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
179
177
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
180 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
181 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
182
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
183 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
184 """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
185
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
186 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
187 """
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
188 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
189 self.set_tile_pos(tile_pos)
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
190 self.orientation = orientation
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
191
221
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
192 def shifted_tile(self):
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
193 pass
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
194
177
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
195 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
196 """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
197 dx, dy = self.orientation
e0573297b17c Snake that moves (my god that was an epic struggle).
Simon Cross <hodgestar@gmail.com>
parents: 153
diff changeset
198 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
199 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
200
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
201 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
202 """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
203 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
204 """
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
205 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
206 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
207 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
208 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
209 # 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
210 # (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
211 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
212 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
213 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
214 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
215 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
216 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
217 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
218 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
219 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
220 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
221 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
222 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
223 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
224 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
225 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
226 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
227 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
228 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
229 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
230 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
231 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
232 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
233 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
234
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
235
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
236 class Head(Segment):
153
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
237 CLOSED = "snake-head"
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
238 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
239 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
240 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
241
182
0a955d2536f0 Painted Jezebel! (Assuming you name your snake Jezebel.)
Jeremy Thurgood <firxen@gmail.com>
parents: 181
diff changeset
242 is_head = True
0a955d2536f0 Painted Jezebel! (Assuming you name your snake Jezebel.)
Jeremy Thurgood <firxen@gmail.com>
parents: 181
diff changeset
243
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
244 def __init__(self, tile_pos):
153
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
245 self._detail_mutators = (self.EYE,)
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
246 super(Head, self).__init__(self.CLOSED, tile_pos)
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
247
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
248 def mouth_open(self):
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
249 self.set_base_image(self.OPEN)
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
250 self.make_images()
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
251 self.update_image()
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
252
218
10804ae58066 Open mouth when eating things
Stefano Rivera <stefano@rivera.za.net>
parents: 217
diff changeset
253 def mouth_close(self):
153
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
254 self.set_base_image(self.CLOSED)
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
255 self.make_images()
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
256 self.update_image()
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
257
131
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
258 def tongue_out(self):
153
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
259 self._detail_mutators = (self.EYE, self.TONGUE)
131
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
260 self.make_images()
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
261 self.update_image()
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
262
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
263 def tongue_in(self):
153
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
264 self._detail_mutators = (self.EYE,)
131
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
265 self.make_images()
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
266 self.update_image()
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 129
diff changeset
267
221
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
268 def shifted_tile(self):
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
269 if random.random() < 0.02:
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
270 self.mouth_open()
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
271 self.tongue_out()
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
272 else:
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
273 self.mouth_close()
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
274 self.tongue_in()
3a652f048f57 Tssss, I smell rats
Stefano Rivera <stefano@rivera.za.net>
parents: 218
diff changeset
275
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
276
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
277 class Body(Segment):
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
278 def __init__(self, tile_pos):
153
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
279 super(Body, self).__init__("snake-body", tile_pos)
110
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
280
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
281
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
282 class Tail(Segment):
e6299eb296ce Draw the world!
Simon Cross <hodgestar@gmail.com>
parents: 98
diff changeset
283 def __init__(self, tile_pos):
153
9c47bf162ea1 snake has plastic surgery
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 145
diff changeset
284 super(Tail, self).__init__("snake-tail", tile_pos)