mamba: mamba/sprites.py annotate

annotate mamba/sprites.py @ 136:00ada2e29798

Somewhat better (but still hideous) image variant support.

author	Jeremy Thurgood <firxen@gmail.com>
date	Sun, 11 Sep 2011 23:56:59 +0200
parents	e0b4bfc51336
children	b292370c4548

rev	line source
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	1 from pygame.sprite import Sprite
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	2
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	3 from mamba.data import load_image
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	4 from mamba.constants import TILE_SIZE
130 9bef49d6db86 Ugly yucky half-done tile orientation magic. Jeremy Thurgood <firxen@gmail.com> parents: 121 diff changeset	5 from mamba import mutators
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	6
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	7
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	8 def tile_sizify(pos):
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	9 ts_x, ts_y = TILE_SIZE
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	10 p_x, p_y = pos
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	11 return (ts_x * p_x, ts_y * p_y)
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	12
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	13
136 00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	14 class SpriteImageVariants(object):
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	15 VARIANTS = {
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	16 '....': ('-0', ()),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	17
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	18 'X...': ('-1', (mutators.R90,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	19 '.X..': ('-1', (mutators.R270,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	20 '..X.': ('-1', (mutators.NULL,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	21 '...X': ('-1', (mutators.R180,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	22
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	23 'XX..': ('-2o', (mutators.NULL,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	24 '..XX': ('-2o', (mutators.R90,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	25
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	26 'X.X.': ('-2a', (mutators.R180,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	27 'X..X': ('-2a', (mutators.R270,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	28 '.XX.': ('-2a', (mutators.R90,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	29 '.X.X': ('-2a', (mutators.NULL,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	30
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	31 '.XXX': ('-3', (mutators.R90,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	32 'X.XX': ('-3', (mutators.R270,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	33 'XX.X': ('-3', (mutators.NULL,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	34 'XXX.': ('-3', (mutators.R180,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	35
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	36 'XXXX': ('', ()),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	37 }
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	38
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	39 def __init__(self, base_image_name):
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	40 self.base_image_name = base_image_name
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	41
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	42 def __call__(self, top, bottom, left, right):
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	43 variant_str = ''.join('X' if d else '.'
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	44 for d in [top, bottom, left, right])
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	45 variant_suffix, mutators = self.VARIANTS[variant_str]
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	46 return (self.base_image_name + variant_suffix, mutators)
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	47
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	48
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	49 class SolidSpriteImageVariants(SpriteImageVariants):
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	50 VARIANTS = {
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	51 '....': ('-0', ()),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	52
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	53 'X...': ('-1', (mutators.R90,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	54 '.X..': ('-1', (mutators.R270,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	55 '..X.': ('-1', (mutators.NULL,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	56 '...X': ('-1', (mutators.R180,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	57
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	58 'XX..': ('', ()),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	59 '..XX': ('', ()),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	60
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	61 'X.X.': ('-2a', (mutators.R180,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	62 'X..X': ('-2a', (mutators.R270,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	63 '.XX.': ('-2a', (mutators.R90,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	64 '.X.X': ('-2a', (mutators.NULL,)),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	65
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	66 '.XXX': ('', ()),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	67 'X.XX': ('', ()),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	68 'XX.X': ('', ()),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	69 'XXX.': ('', ()),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	70
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	71 'XXXX': ('', ()),
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	72 }
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	73
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	74
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	75 class BaseSprite(Sprite):
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	76 tileset = 'common'
136 00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	77 variants_class = None
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	78 variants = None
130 9bef49d6db86 Ugly yucky half-done tile orientation magic. Jeremy Thurgood <firxen@gmail.com> parents: 121 diff changeset	79
9bef49d6db86 Ugly yucky half-done tile orientation magic. Jeremy Thurgood <firxen@gmail.com> parents: 121 diff changeset	80 def __init__(self, tile_char=None, tileset=None, image_name=None,
9bef49d6db86 Ugly yucky half-done tile orientation magic. Jeremy Thurgood <firxen@gmail.com> parents: 121 diff changeset	81 mutators=()):
98 7ce2d2d8381a Refactor BaseSprite. Simon Cross <hodgestar@gmail.com> parents: 81 diff changeset	82 super(BaseSprite, self).__init__()
105 87acd3ddc972 Tell sprites what their tile char is. Jeremy Thurgood <firxen@gmail.com> parents: 104 diff changeset	83 self.tile_char = tile_char
98 7ce2d2d8381a Refactor BaseSprite. Simon Cross <hodgestar@gmail.com> parents: 81 diff changeset	84 if tileset is not None:
7ce2d2d8381a Refactor BaseSprite. Simon Cross <hodgestar@gmail.com> parents: 81 diff changeset	85 self.tileset = tileset
7ce2d2d8381a Refactor BaseSprite. Simon Cross <hodgestar@gmail.com> parents: 81 diff changeset	86 if image_name is not None:
130 9bef49d6db86 Ugly yucky half-done tile orientation magic. Jeremy Thurgood <firxen@gmail.com> parents: 121 diff changeset	87 self.image = self.load_image(image_name, mutators=mutators)
56 e730ec8b82e8 Some doors. Jeremy Thurgood <firxen@gmail.com> parents: 50 diff changeset	88
98 7ce2d2d8381a Refactor BaseSprite. Simon Cross <hodgestar@gmail.com> parents: 81 diff changeset	89 def load_image(self, image_name, mutators=()):
7ce2d2d8381a Refactor BaseSprite. Simon Cross <hodgestar@gmail.com> parents: 81 diff changeset	90 return load_image('tiles/%s/%s.png' % (self.tileset, image_name),
7ce2d2d8381a Refactor BaseSprite. Simon Cross <hodgestar@gmail.com> parents: 81 diff changeset	91 mutators)
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	92
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	93 def set_tile_pos(self, tile_pos):
81 c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	94 self.tile_pos = tile_pos
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	95 self.rect = self.image.get_rect().move(tile_sizify(tile_pos))
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	96
136 00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	97 def get_variant(self, top, bottom, left, right):
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	98 if (self.variants_class is not None) and (self.variants is None):
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	99 self.variants = self.variants_class(self.image_name)
00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	100 return self.variants(top, bottom, left, right)
130 9bef49d6db86 Ugly yucky half-done tile orientation magic. Jeremy Thurgood <firxen@gmail.com> parents: 121 diff changeset	101
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	102
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	103 class TileSprite(BaseSprite):
104 5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	104 def __init__(self, tileset, **kw):
5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	105 kw['tileset'] = tileset
5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	106 super(TileSprite, self).__init__(**kw)
50 1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	107
1b725035b8ef Levels that are visible. Jeremy Thurgood <firxen@gmail.com> parents: diff changeset	108
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	109 class SingleImageTileSprite(TileSprite):
121 f670ef33ae2e Comment SingleImageTileSprite convention Neil Muller <drnlmuller@gmail.com> parents: 105 diff changeset	110
f670ef33ae2e Comment SingleImageTileSprite convention Neil Muller <drnlmuller@gmail.com> parents: 105 diff changeset	111 image_name = None # subclasses will override this
f670ef33ae2e Comment SingleImageTileSprite convention Neil Muller <drnlmuller@gmail.com> parents: 105 diff changeset	112
104 5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	113 def __init__(self, **kw):
5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	114 kw.setdefault('image_name', self.image_name)
5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	115 super(SingleImageTileSprite, self).__init__(**kw)
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	116
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	117
56 e730ec8b82e8 Some doors. Jeremy Thurgood <firxen@gmail.com> parents: 50 diff changeset	118 class DoorSprite(TileSprite):
104 5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	119 def __init__(self, colour, **kw):
57 cd5ad2c45e2d Minor level fiddlery. Jeremy Thurgood <firxen@gmail.com> parents: 56 diff changeset	120 self.colour = colour
56 e730ec8b82e8 Some doors. Jeremy Thurgood <firxen@gmail.com> parents: 50 diff changeset	121 image_name = 'door_%s' % (colour,)
104 5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	122 kw.setdefault('image_name', image_name)
5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	123 super(DoorSprite, self).__init__(**kw)
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	124
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	125
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	126 class EntrySprite(SingleImageTileSprite):
104 5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	127 image_name = 'entrance-1'
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	128
104 5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	129 def __init__(self, direction, **kw):
81 c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	130 self.direction = direction
104 5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	131 super(EntrySprite, self).__init__(**kw)
81 c8d6ca296884 Better level entry points. Jeremy Thurgood <firxen@gmail.com> parents: 71 diff changeset	132
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	133
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	134 class ExitSprite(SingleImageTileSprite):
104 5b92ff941de3 Refactor sprites. Jeremy Thurgood <firxen@gmail.com> parents: 98 diff changeset	135 image_name = 'exit-1'
71 58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	136
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	137
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	138 class PuddleSprite(SingleImageTileSprite):
58557a89ec86 Some tile improvements and stuff. Jeremy Thurgood <firxen@gmail.com> parents: 57 diff changeset	139 image_name = 'puddle'
136 00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	140 variants_class = SolidSpriteImageVariants
130 9bef49d6db86 Ugly yucky half-done tile orientation magic. Jeremy Thurgood <firxen@gmail.com> parents: 121 diff changeset	141
9bef49d6db86 Ugly yucky half-done tile orientation magic. Jeremy Thurgood <firxen@gmail.com> parents: 121 diff changeset	142 def __init__(self, variant=(0, 0, 0, 0), **kw):
9bef49d6db86 Ugly yucky half-done tile orientation magic. Jeremy Thurgood <firxen@gmail.com> parents: 121 diff changeset	143 super(PuddleSprite, self).__init__(**kw)
136 00ada2e29798 Somewhat better (but still hideous) image variant support. Jeremy Thurgood <firxen@gmail.com> parents: 135 diff changeset	144 self.image = self.load_image(self.get_variant(variant))

Mercurial > mamba

annotate mamba/sprites.py @ 136:00ada2e29798