annotate mamba/widgets/levelbutton.py @ 322:86c5c6afdac0

Thumbnails on the level screen
author Stefano Rivera <stefano@rivera.za.net>
date Fri, 16 Sep 2011 20:01:13 +0200
parents fba74abad7b7
children cbbba6ddc77d
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
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
4 from mamba.constants import COLOR, FOCUS_COLOR
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
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
20 self.rect.height = 100
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
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
23 def prepare(self):
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
24 self.surface = pygame.Surface(self.rect.size, SRCALPHA)
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
25 self.surface.fill(0)
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
26
322
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
27 # Level Thumbnail
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
28 level_surface = pygame.Surface(self.level.get_size(), SRCALPHA)
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
29 self.level.draw(level_surface)
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
30 dest_rect = pygame.Rect((self.border, self.border),
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
31 (self.rect.width - self.border,
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
32 self.rect.height - self.border))
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
33 size = level_surface.get_rect().fit(dest_rect).size
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
34 level_thumbnail = pygame.transform.scale(level_surface, size)
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
35 self.surface.blit(level_thumbnail, dest_rect)
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
36
310
b371c2ea309e excised level locking
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 293
diff changeset
37 if self.done:
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
38 image = load_image('menus/tick.png')
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
39 self.surface.blit(image, image.get_rect())
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
40
322
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
41 self._text = TextWidget((0, 0), self.text, fontsize=12)
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
42 self._text.prepare()
322
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
43 text_rect = pygame.Rect((0, 0), self.rect.size).inflate(
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
44 self._text.rect.width - self.rect.width,
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
45 self._text.rect.height - self.rect.height)
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
46 text_rect.top = self.rect.height - self._text.rect.height - self.border
86c5c6afdac0 Thumbnails on the level screen
Stefano Rivera <stefano@rivera.za.net>
parents: 314
diff changeset
47 self.surface.blit(self._text.surface, text_rect)
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
48
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
49 color = pygame.Color(FOCUS_COLOR if self.focussed else COLOR)
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
50 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
51 self.border + 1)
310
b371c2ea309e excised level locking
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 293
diff changeset
52 self._state = (self.done, self.focussed)
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
53
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
54 def draw(self, surface):
310
b371c2ea309e excised level locking
Adrianna Pińska <adrianna.pinska@gmail.com>
parents: 293
diff changeset
55 if self._state != (self.done, self.focussed):
293
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
56 self.prepare()
e6a3b00f997b Ticks on the levelmenu
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
57 surface.blit(self.surface, self.rect)