Mercurial > pyntnclick

annotate tools/rect_drawer.py @ 238:2510e0b0e901

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add detail view support to helper
author Neil Muller <neil@dip.sun.ac.za>
date Fri, 27 Aug 2010 09:18:27 +0200
parents 20998c650ce1
children 0ea4661d134c
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
165
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
1 # Quickly hacked together helper for working out
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
2 # interactive regions in Suspended Sentence
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
3
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
4 import sys
174
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
5 import os.path
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
6
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
7 script_path = os.path.realpath(os.path.dirname(os.path.dirname(__file__)))
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
8 sys.path.append(script_path)
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
9
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
10 from albow.root import RootWidget
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
11 from albow.utils import frame_rect
174
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
12 from albow.widget import Widget
200
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
13 from albow.controls import Button, Image
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
14 from albow.palette_view import PaletteView
199
4821c290286d Image loading
Neil Muller <neil@dip.sun.ac.za>
parents: 198
diff changeset
15 from albow.file_dialogs import request_old_filename
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
16 from pygame.locals import SWSURFACE
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
17 import pygame
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
18 from pygame.colordict import THECOLORS
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
19
174
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
20 from gamelib.state import initial_state
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
21 from gamelib import constants
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
22
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
23 constants.DEBUG = True
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
24
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
25
165
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
26
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
27 class AppPalette(PaletteView):
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
28
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
29 sel_width = 5
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
30
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
31 colors = [
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
32 'red', 'maroon1', 'palevioletred1', 'moccasin', 'orange',
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
33 'honeydew', 'yellow', 'gold', 'goldenrod', 'brown',
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
34 'blue', 'purple', 'darkorchid4', 'thistle', 'skyblue1',
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
35 'green', 'palegreen1', 'darkgreen', 'aquamarine', 'darkolivegreen',
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
36 ]
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
37
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
38 def __init__(self, app_image):
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
39 self.image = app_image
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
40 super(AppPalette, self).__init__((35, 35), 5, 5, margin=2)
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
41 self.selection = 0
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
42 self.image.rect_color = pygame.color.Color(self.colors[self.selection])
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
43
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
44 def num_items(self):
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
45 return len(self.colors)
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
46
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
47 def draw_item(self, surface, item_no, rect):
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
48 d = -2 * self.sel_width
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
49 r = rect.inflate(d, d)
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
50 surface.fill(pygame.color.Color(self.colors[item_no]), r)
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
51
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
52 def click_item(self, item_no, event):
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
53 self.selection = item_no
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
54 self.image.rect_color = pygame.color.Color(self.colors[item_no])
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
55
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
56 def item_is_selected(self, item_no):
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
57 return self.selection == item_no
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
58
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
59
174
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
60 class AppImage(Widget):
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
61
181
f98bc17f5e67 Add attributes for drawing rects
Neil Muller <neil@dip.sun.ac.za>
parents: 179
diff changeset
62 rect_thick = 3
f98bc17f5e67 Add attributes for drawing rects
Neil Muller <neil@dip.sun.ac.za>
parents: 179
diff changeset
63 draw_thick = 1
f98bc17f5e67 Add attributes for drawing rects
Neil Muller <neil@dip.sun.ac.za>
parents: 179
diff changeset
64
174
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
65 def __init__(self, state):
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
66 self.state = state
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
67 super(AppImage, self).__init__(pygame.rect.Rect(0, 0, 800, 600))
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
68 self.mode = 'draw'
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
69 self.rects = []
200
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
70 self.images = []
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
71 self.start_pos = None
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
72 self.end_pos = None
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
73 self.draw_color = pygame.color.Color('white')
165
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
74 self.rect_color = pygame.color.Color('white')
200
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
75 self.current_image = None
199
4821c290286d Image loading
Neil Muller <neil@dip.sun.ac.za>
parents: 198
diff changeset
76 self.place_image_menu = None
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
77
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
78 def draw_mode(self):
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
79 self.mode = 'draw'
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
80
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
81 def del_mode(self):
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
82 self.mode = 'del'
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
83 self.start_pos = None
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
84 self.end_pos = None
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
85
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
86 def draw(self, surface):
238
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
87 if self.state.current_detail:
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
88 w, h = self.state.current_detail.get_detail_size()
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
89 self.state.draw_detail(surface, None)
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
90 else:
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
91 self.state.draw(surface, None)
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
92 if self.mode == 'draw' and self.start_pos:
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
93 rect = pygame.rect.Rect(self.start_pos[0], self.start_pos[1],
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
94 self.end_pos[0] - self.start_pos[0],
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
95 self.end_pos[1] - self.start_pos[1])
181
f98bc17f5e67 Add attributes for drawing rects
Neil Muller <neil@dip.sun.ac.za>
parents: 179
diff changeset
96 frame_rect(surface, self.draw_color, rect, self.draw_thick)
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
97 for (col, rect) in self.rects:
181
f98bc17f5e67 Add attributes for drawing rects
Neil Muller <neil@dip.sun.ac.za>
parents: 179
diff changeset
98 frame_rect(surface, col, rect, self.rect_thick)
200
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
99 for image in self.images:
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
100 if image.rect.colliderect(surface.get_rect()):
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
101 cropped_rect = image.rect.clip(surface.get_rect())
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
102 sub = surface.subsurface(cropped_rect)
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
103 image.draw(sub)
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
104 else:
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
105 print 'image outside surface', image
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
106 if self.current_image and self.mode == 'image':
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
107 if self.current_image.rect.colliderect(surface.get_rect()):
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
108 cropped_rect = self.current_image.rect.clip(surface.get_rect())
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
109 sub = surface.subsurface(cropped_rect)
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
110 self.current_image.draw(sub)
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
111
165
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
112 def _make_dict(self):
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
113 d = {}
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
114 for col, rect in self.rects:
213
20998c650ce1 Fixed rect_drawer tool and adjusted JIM message background.
Jeremy Thurgood <firxen@gmail.com>
parents: 200
diff changeset
115 col = (col.r, col.g, col.b)
165
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
116 d.setdefault(col, [])
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
117 d[col].append(rect)
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
118 return d
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
119
198
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
120 def print_objs(self):
165
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
121 d = self._make_dict()
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
122 for (num, col) in enumerate(d):
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
123 print 'Rect %d : ' % num
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
124 for r in d[col]:
179
39f75faf16cb Add trailing , for copy-n-paste
Neil Muller <neil@dip.sun.ac.za>
parents: 174
diff changeset
125 print ' (%d, %d, %d, %d),' % (r.x, r.y, r.w, r.h)
165
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
126 print
200
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
127 for i, image in enumerate(self.images):
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
128 print 'Image %d' % i
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
129 r = image.rect
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
130 print ' (%d, %d, %d, %d),' % (r.x, r.y, r.w, r.h)
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
131 print
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
132
199
4821c290286d Image loading
Neil Muller <neil@dip.sun.ac.za>
parents: 198
diff changeset
133 def image_load(self):
4821c290286d Image loading
Neil Muller <neil@dip.sun.ac.za>
parents: 198
diff changeset
134 image_path= '%s/Resources/images/%s' % (script_path, self.state.current_scene.FOLDER)
4821c290286d Image loading
Neil Muller <neil@dip.sun.ac.za>
parents: 198
diff changeset
135 imagename = request_old_filename(directory=image_path)
4821c290286d Image loading
Neil Muller <neil@dip.sun.ac.za>
parents: 198
diff changeset
136 try:
200
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
137 image_data = pygame.image.load(imagename)
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
138 self.current_image = Image(image_data)
199
4821c290286d Image loading
Neil Muller <neil@dip.sun.ac.za>
parents: 198
diff changeset
139 self.place_image_menu.enabled = True
200
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
140 # ensure we're off screen to start
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
141 self.current_image.rect = image_data.get_rect().move(1000, 600)
199
4821c290286d Image loading
Neil Muller <neil@dip.sun.ac.za>
parents: 198
diff changeset
142 except pygame.error, e:
4821c290286d Image loading
Neil Muller <neil@dip.sun.ac.za>
parents: 198
diff changeset
143 print 'Unable to load image %s' % e
198
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
144
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
145 def image_mode(self):
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
146 self.mode = 'image'
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
147 self.start_pos = None
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
148 self.end_pos = None
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
149
200
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
150 def mouse_move(self, e):
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
151 if self.mode == 'image' and self.current_image:
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
152 self.current_image.rect.topleft = e.pos
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
153 self.invalidate()
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
154
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
155 def mouse_down(self, e):
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
156 if self.mode == 'del':
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
157 pos = e.pos
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
158 cand = None
200
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
159 # Images are drawn above rectangles, so search those first
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
160 for image in self.images:
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
161 if image.rect.collidepoint(pos):
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
162 cand = image
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
163 break
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
164 if cand:
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
165 self.images.remove(cand)
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
166 self.invalidate()
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
167 return
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
168 for (col, rect) in self.rects:
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
169 if rect.collidepoint(pos):
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
170 cand = (col, rect)
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
171 break
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
172 if cand:
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
173 self.rects.remove(cand)
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
174 self.invalidate()
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
175 elif self.mode == 'draw':
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
176 self.start_pos = e.pos
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
177 self.end_pos = e.pos
200
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
178 elif self.mode == 'image':
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
179 if self.current_image:
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
180 self.images.append(self.current_image)
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
181 self.current_image = None
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
182 self.invalidate()
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
183 else:
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
184 cand = None
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
185 for image in self.images:
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
186 if image.rect.collidepoint(e.pos):
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
187 cand = image
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
188 break
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
189 if cand:
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
190 self.images.remove(cand)
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
191 self.current_image = cand
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
192 self.invalidate()
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
193
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
194 def mouse_up(self, e):
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
195 if self.mode == 'draw':
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
196 rect = pygame.rect.Rect(self.start_pos[0], self.start_pos[1],
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
197 self.end_pos[0] - self.start_pos[0],
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
198 self.end_pos[1] - self.start_pos[1])
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
199 rect.normalize()
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
200 self.rects.append((self.rect_color, rect))
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
201 self.start_pos = self.end_pos = None
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
202
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
203 def mouse_drag(self, e):
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
204 if self.mode == 'draw':
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
205 self.end_pos = e.pos
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
206 self.invalidate()
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
207
197
640044d7ddda Tweak layout
Neil Muller <neil@dip.sun.ac.za>
parents: 196
diff changeset
208 def make_button(text, action, ypos):
640044d7ddda Tweak layout
Neil Muller <neil@dip.sun.ac.za>
parents: 196
diff changeset
209 button = Button(text, action=action)
640044d7ddda Tweak layout
Neil Muller <neil@dip.sun.ac.za>
parents: 196
diff changeset
210 button.align = 'l'
640044d7ddda Tweak layout
Neil Muller <neil@dip.sun.ac.za>
parents: 196
diff changeset
211 button.rect = pygame.rect.Rect(0, 0, 200, 40)
640044d7ddda Tweak layout
Neil Muller <neil@dip.sun.ac.za>
parents: 196
diff changeset
212 button.rect.move_ip(805, ypos)
640044d7ddda Tweak layout
Neil Muller <neil@dip.sun.ac.za>
parents: 196
diff changeset
213 return button
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
214
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
215 if __name__ == "__main__":
165
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
216 # FIXME: should load an actual scene with current things, not just a
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
217 # background image
174
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
218 if len(sys.argv) < 2:
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
219 print 'Please provide a scene name'
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
220 sys.exit(0)
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
221 pygame.display.init()
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
222 pygame.font.init()
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
223 display = pygame.display.set_mode((1000, 600))
196
b63ecd5b9a45 Unbeak helper
Neil Muller <neil@dip.sun.ac.za>
parents: 181
diff changeset
224 state = initial_state()
238
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
225 if len(sys.argv) < 3:
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
226 try:
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
227 state.set_current_scene(sys.argv[1])
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
228 state.do_check = None
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
229 except KeyError:
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
230 print 'Invalid scene name'
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
231 sys.exit(1)
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
232 else:
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
233 try:
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
234 state.set_current_scene(sys.argv[1])
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
235 state.set_current_detail(sys.argv[2])
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
236 state.do_check = None
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
237 except KeyError:
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
238 print 'Invalid scene name'
2510e0b0e901 Add detail view support to helper
Neil Muller <neil@dip.sun.ac.za>
parents: 213
diff changeset
239 sys.exit(1)
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
240 app = RootWidget(display)
174
a2d041e0ab83 Make tool use scenes, not images
Neil Muller <neil@dip.sun.ac.za>
parents: 165
diff changeset
241 image = AppImage(state)
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
242 app.add(image)
197
640044d7ddda Tweak layout
Neil Muller <neil@dip.sun.ac.za>
parents: 196
diff changeset
243 draw = make_button('Draw Rect', image.draw_mode, 0)
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
244 app.add(draw)
198
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
245 load_image = make_button("Load image", image.image_load, 40)
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
246 app.add(load_image)
200
7bedca2376f5 Image placement
Neil Muller <neil@dip.sun.ac.za>
parents: 199
diff changeset
247 add_image = make_button("Place/Move images", image.image_mode, 80)
199
4821c290286d Image loading
Neil Muller <neil@dip.sun.ac.za>
parents: 198
diff changeset
248 add_image.enabled = False
198
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
249 app.add(add_image)
199
4821c290286d Image loading
Neil Muller <neil@dip.sun.ac.za>
parents: 198
diff changeset
250 image.place_image_menu = add_image
198
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
251 delete = make_button('Delete Objects', image.del_mode, 120)
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
252 app.add(delete)
165
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
253 palette = AppPalette(image)
198
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
254 palette.rect.move_ip(810, 160)
165
9b3bba5e65f3 add some features to rect helper. now vaguely useful
Neil Muller <neil@dip.sun.ac.za>
parents: 162
diff changeset
255 app.add(palette)
198
e73d78d6dd83 Stubs for ftuture functionality
Neil Muller <neil@dip.sun.ac.za>
parents: 197
diff changeset
256 print_rects = make_button("Print objects", image.print_objs, 320)
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
257 app.add(print_rects)
197
640044d7ddda Tweak layout
Neil Muller <neil@dip.sun.ac.za>
parents: 196
diff changeset
258 quit_but = make_button("Quit", app.quit, 560)
162
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
259 app.add(quit_but)
225e3a4b1e85 Start of a tool to help construct interact rectangles
Neil Muller <neil@dip.sun.ac.za>
parents:
diff changeset
260 app.run()