Mercurial > mamba
annotate mamba/snake.py @ 118:1b9b17eb896b
Change default snake orientation to UP for now.
| author | Simon Cross <hodgestar@gmail.com> |
|---|---|
| date | Sun, 11 Sep 2011 20:32:18 +0200 |
| parents | 0c1005c76c87 |
| children | 119c0fb758c2 |
| rev | line source |
|---|---|
| 78 | 1 """The player snake object.""" |
| 2 | |
| 110 | 3 from pygame.sprite import Group |
| 4 | |
|
79
f602a8c3d173
Fix bugs. Stubbed out get_entry for now.
Simon Cross <hodgestar@gmail.com>
parents:
78
diff
changeset
|
5 from mamba.sprites import BaseSprite |
| 98 | 6 from mamba import mutators |
| 78 | 7 |
| 8 | |
| 110 | 9 class Snake(object): |
| 78 | 10 |
| 110 | 11 UP, DOWN, LEFT, RIGHT = [(0, -1), (0, 1), (-1, 0), (1, 0)] |
| 78 | 12 |
| 13 def __init__(self, tile_pos, orientation): | |
| 110 | 14 self.segments = self.create_segments(tile_pos, orientation) |
| 15 self.segment_group = Group() | |
| 16 self.segment_group.add(*self.segments) | |
| 78 | 17 self.set_orientation(orientation) |
| 18 | |
| 110 | 19 head = property(fget=lambda self: self.segments[0]) |
| 20 tail = property(fget=lambda self: self.segments[-1]) | |
| 21 | |
| 22 def create_segments(self, tile_pos, orientation): | |
| 23 x, y = tile_pos | |
| 24 dx, dy = orientation | |
| 25 return [Head((x, y)), | |
| 26 Body((x + dx, y + dy)), | |
| 27 Tail((x + 2 * dx, y + 2 * dy)), | |
| 28 ] | |
| 29 | |
| 30 def draw(self, surface): | |
| 111 | 31 self.segment_group.draw(surface) |
| 98 | 32 |
| 78 | 33 def set_orientation(self, orientation): |
| 110 | 34 #self.orientation = orientation |
| 35 #for segment in self.segments: | |
| 36 # segment.set_orientation(self) | |
| 37 print orientation | |
| 98 | 38 |
| 39 | |
| 110 | 40 class Segment(BaseSprite): |
| 41 | |
| 42 def __init__(self, image_name, tile_pos): | |
| 43 super(Segment, self).__init__() | |
| 44 image_name = "/".join(["snake", image_name]) | |
| 45 self._images = {} | |
| 46 for orientation, muts in [ | |
| 47 (Snake.RIGHT, (mutators.RIGHT,)), | |
| 48 (Snake.LEFT, (mutators.LEFT,)), | |
| 49 (Snake.UP, (mutators.UP,)), | |
| 50 (Snake.DOWN, (mutators.DOWN,)), | |
| 51 ]: | |
| 52 self._images[orientation] = self.load_image(image_name, muts) | |
|
118
1b9b17eb896b
Change default snake orientation to UP for now.
Simon Cross <hodgestar@gmail.com>
parents:
111
diff
changeset
|
53 self.set_orientation(Snake.UP) |
| 110 | 54 self.set_tile_pos(tile_pos) |
| 55 | |
| 56 def set_orientation(self, orientation): | |
| 57 self.image = self._images[orientation] | |
| 58 | |
| 59 | |
| 60 class Head(Segment): | |
| 61 def __init__(self, tile_pos): | |
| 62 super(Head, self).__init__("snake-head-mouth-open-r", tile_pos) | |
| 63 | |
| 64 | |
| 65 class Body(Segment): | |
| 66 def __init__(self, tile_pos): | |
| 67 super(Body, self).__init__("snake-body-r", tile_pos) | |
| 68 | |
| 69 | |
| 70 class Tail(Segment): | |
| 71 def __init__(self, tile_pos): | |
| 72 super(Tail, self).__init__("snake-tail-r", tile_pos) |