Mercurial > mamba

annotate mamba/widgets/levelbutton.py @ 532:f2f7a91aae95

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Red died text
author Stefano Rivera <stefano@rivera.za.net>
date Sun, 18 Sep 2011 01:14:50 +0200
parents f4b53c56757a
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
1 import pygame
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
2 from pygame.locals import SRCALPHA
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
3
491
ab83fc14df11 First round of level menu background
Stefano Rivera <stefano@rivera.za.net>
parents: 413
diff changeset
4 from mamba.constants import FOCUS_COLOR
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
5 from mamba.data import load_image
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
6 from mamba.widgets.base import Button
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
7 from mamba.widgets.text import TextWidget
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
8
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
9
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
10 class LevelButton(Button):
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
11
314
fba74abad7b7 Pass Level() objects to LevelHabitat and LevelEditorHabitat, in preparation for generating thumbnails from levels in levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents: 310
diff changeset
12 def __init__(self, rect, level, done=False):
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
13 super(LevelButton, self).__init__(rect)
314
fba74abad7b7 Pass Level() objects to LevelHabitat and LevelEditorHabitat, in preparation for generating thumbnails from levels in levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents: 310
diff changeset
14 self.level = level
fba74abad7b7 Pass Level() objects to LevelHabitat and LevelEditorHabitat, in preparation for generating thumbnails from levels in levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents: 310
diff changeset
15 self.text = level.name
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
16 self.done = done
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
17 self.focussable = True
322
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
18 self.border = 2
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
19 self.rect.width = 100
491
ab83fc14df11 First round of level menu background
Stefano Rivera <stefano@rivera.za.net>
parents: 413
diff changeset
20 self.rect.height = 120
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
21 self.prepare()
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
22
413
cbbba6ddc77d Use cached level images for extra blazing fastitude.
Jeremy Thurgood <firxen@gmail.com>
parents: 322
diff changeset
23 def make_thumbnail(self, dest_rect):
cbbba6ddc77d Use cached level images for extra blazing fastitude.
Jeremy Thurgood <firxen@gmail.com>
parents: 322
diff changeset
24 level_surface = pygame.Surface(self.level.get_size(), SRCALPHA)
cbbba6ddc77d Use cached level images for extra blazing fastitude.
Jeremy Thurgood <firxen@gmail.com>
parents: 322
diff changeset
25 self.level.draw(level_surface)
cbbba6ddc77d Use cached level images for extra blazing fastitude.
Jeremy Thurgood <firxen@gmail.com>
parents: 322
diff changeset
26 size = level_surface.get_rect().fit(dest_rect).size
cbbba6ddc77d Use cached level images for extra blazing fastitude.
Jeremy Thurgood <firxen@gmail.com>
parents: 322
diff changeset
27 level_thumbnail = pygame.transform.scale(level_surface, size)
cbbba6ddc77d Use cached level images for extra blazing fastitude.
Jeremy Thurgood <firxen@gmail.com>
parents: 322
diff changeset
28 return level_thumbnail
cbbba6ddc77d Use cached level images for extra blazing fastitude.
Jeremy Thurgood <firxen@gmail.com>
parents: 322
diff changeset
29
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
30 def prepare(self):
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
31 self.surface = pygame.Surface(self.rect.size, SRCALPHA)
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
32 self.surface.fill(0)
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
33
322
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
34 dest_rect = pygame.Rect((self.border, self.border),
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
35 (self.rect.width - self.border,
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
36 self.rect.height - self.border))
413
cbbba6ddc77d Use cached level images for extra blazing fastitude.
Jeremy Thurgood <firxen@gmail.com>
parents: 322
diff changeset
37 if not hasattr(self.level, 'button_thumbnail'):
cbbba6ddc77d Use cached level images for extra blazing fastitude.
Jeremy Thurgood <firxen@gmail.com>
parents: 322
diff changeset
38 self.level.button_thumbnail = self.make_thumbnail(dest_rect)
cbbba6ddc77d Use cached level images for extra blazing fastitude.
Jeremy Thurgood <firxen@gmail.com>
parents: 322
diff changeset
39 self.surface.blit(self.level.button_thumbnail, dest_rect)
322
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
40
310
b371c2ea309e excised level locking
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 293
diff changeset
41 if self.done:
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
42 image = load_image('menus/tick.png')
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
43 self.surface.blit(image, image.get_rect())
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
44
503
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
45 # We only have space for two lines
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
46 self._text_lines = self.wrap_text(self.text)[:2]
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
47
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
48 text_height = sum(line.rect.height for line in self._text_lines)
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
49 text_pos = self.level.button_thumbnail.get_rect().height
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
50 text_pos += (self.rect.height - text_height - text_pos) // 2
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
51 for line in self._text_lines:
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
52 text_rect = pygame.Rect(((self.rect.width - line.rect.width) // 2,
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
53 text_pos),
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
54 line.rect.size)
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
55 text_pos = text_rect.bottom
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
56 self.surface.blit(line.surface, text_rect)
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
57
491
ab83fc14df11 First round of level menu background
Stefano Rivera <stefano@rivera.za.net>
parents: 413
diff changeset
58 color = pygame.Color(FOCUS_COLOR if self.focussed else '#444444')
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
59 pygame.draw.rect(self.surface, color, self.surface.get_rect(),
322
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
60 self.border + 1)
310
b371c2ea309e excised level locking
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 293
diff changeset
61 self._state = (self.done, self.focussed)
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
62
503
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
63 def wrap_text(self, text):
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
64 args = {'rect': (0, 0),
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
65 'text': text,
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
66 'fontsize': 12,
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
67 'color': 'white',
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
68 }
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
69 w = TextWidget(**args)
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
70 w.prepare()
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
71 splitpoint = len(text)
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
72 max_width = self.rect.width - (self.border * 3)
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
73 while w.rect.width > max_width and ' ' in text[:splitpoint]:
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
74 splitpoint = text.rfind(' ', 0, splitpoint)
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
75 args['text'] = text[:splitpoint]
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
76 w = TextWidget(**args)
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
77 w.prepare()
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
78 if splitpoint < len(text):
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
79 return [w] + self.wrap_text(text[splitpoint + 1:])
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
80 else:
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
81 return [w]
f4b53c56757a Multi line level names \o/
Stefano Rivera <stefano@rivera.za.net>
parents: 491
diff changeset
82
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
83 def draw(self, surface):
310
b371c2ea309e excised level locking
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 293
diff changeset
84 if self._state != (self.done, self.focussed):
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
85 self.prepare()
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
86 surface.blit(self.surface, self.rect)