Mercurial > rinkhals

annotate gamelib/misc.py @ 553:7963fc09fac2

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
author Neil Muller <drnlmuller@gmail.com>
date Sat, 28 Nov 2009 17:45:28 +0000
parents 27c09c58d89d
children 95c71a13468c
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
69
18db99fda6bd Move spawing code from engine to gameboard - seems more natural.
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
1 # Holder for misc useful classes
18db99fda6bd Move spawing code from engine to gameboard - seems more natural.
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
2
241
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
3 import random
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
4
336
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
5 from pygame.locals import KEYDOWN, K_ESCAPE
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
6 from pgu import gui
534
d16ed2a8a33e Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents: 519
diff changeset
7 from pgu import html
431
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
8 from pgu.algo import getline
336
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
9
427
e89a1afe4e84 Make Position simplifiable. Override unsimplify for animals to update their image after recreating.
Simon Cross <hodgestar@gmail.com>
parents: 395
diff changeset
10 import serializer
e89a1afe4e84 Make Position simplifiable. Override unsimplify for animals to update their image after recreating.
Simon Cross <hodgestar@gmail.com>
parents: 395
diff changeset
11
e89a1afe4e84 Make Position simplifiable. Override unsimplify for animals to update their image after recreating.
Simon Cross <hodgestar@gmail.com>
parents: 395
diff changeset
12 class Position(serializer.Simplifiable):
456
96dbf2c8506e Factor position cache out into its own class, make Position more useful.
Jeremy Thurgood <firxen@gmail.com>
parents: 431
diff changeset
13 """2D/3D position / vector. Assumed immutable."""
69
18db99fda6bd Move spawing code from engine to gameboard - seems more natural.
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
14
427
e89a1afe4e84 Make Position simplifiable. Override unsimplify for animals to update their image after recreating.
Simon Cross <hodgestar@gmail.com>
parents: 395
diff changeset
15 SIMPLIFY = ['x', 'y', 'z']
e89a1afe4e84 Make Position simplifiable. Override unsimplify for animals to update their image after recreating.
Simon Cross <hodgestar@gmail.com>
parents: 395
diff changeset
16
395
2d0ff46118e2 Basic support for z coordinate
Neil Muller <drnlmuller@gmail.com>
parents: 347
diff changeset
17 def __init__(self, x, y, z=0):
100
e90068d1f374 Re-indent to four spaces.
Simon Cross <hodgestar@gmail.com>
parents: 69
diff changeset
18 self.x = x
e90068d1f374 Re-indent to four spaces.
Simon Cross <hodgestar@gmail.com>
parents: 69
diff changeset
19 self.y = y
395
2d0ff46118e2 Basic support for z coordinate
Neil Muller <drnlmuller@gmail.com>
parents: 347
diff changeset
20 self.z = z
69
18db99fda6bd Move spawing code from engine to gameboard - seems more natural.
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
21
395
2d0ff46118e2 Basic support for z coordinate
Neil Muller <drnlmuller@gmail.com>
parents: 347
diff changeset
22 def to_tile_tuple(self):
100
e90068d1f374 Re-indent to four spaces.
Simon Cross <hodgestar@gmail.com>
parents: 69
diff changeset
23 return self.x, self.y
69
18db99fda6bd Move spawing code from engine to gameboard - seems more natural.
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
24
456
96dbf2c8506e Factor position cache out into its own class, make Position more useful.
Jeremy Thurgood <firxen@gmail.com>
parents: 431
diff changeset
25 def to_3d_tuple(self):
96dbf2c8506e Factor position cache out into its own class, make Position more useful.
Jeremy Thurgood <firxen@gmail.com>
parents: 431
diff changeset
26 return self.x, self.y, self.z
96dbf2c8506e Factor position cache out into its own class, make Position more useful.
Jeremy Thurgood <firxen@gmail.com>
parents: 431
diff changeset
27
100
e90068d1f374 Re-indent to four spaces.
Simon Cross <hodgestar@gmail.com>
parents: 69
diff changeset
28 def dist(self, b):
e90068d1f374 Re-indent to four spaces.
Simon Cross <hodgestar@gmail.com>
parents: 69
diff changeset
29 """Gives the distance to another position"""
69
18db99fda6bd Move spawing code from engine to gameboard - seems more natural.
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
30
395
2d0ff46118e2 Basic support for z coordinate
Neil Muller <drnlmuller@gmail.com>
parents: 347
diff changeset
31 return max(abs(self.x - b.x), abs(self.y - b.y), abs(self.z - b.z))
69
18db99fda6bd Move spawing code from engine to gameboard - seems more natural.
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
32
100
e90068d1f374 Re-indent to four spaces.
Simon Cross <hodgestar@gmail.com>
parents: 69
diff changeset
33 def __sub__(self, b):
395
2d0ff46118e2 Basic support for z coordinate
Neil Muller <drnlmuller@gmail.com>
parents: 347
diff changeset
34 return Position(self.x - b.x, self.y - b.y, self.z - b.z)
69
18db99fda6bd Move spawing code from engine to gameboard - seems more natural.
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
35
100
e90068d1f374 Re-indent to four spaces.
Simon Cross <hodgestar@gmail.com>
parents: 69
diff changeset
36 def __add__(self, b):
395
2d0ff46118e2 Basic support for z coordinate
Neil Muller <drnlmuller@gmail.com>
parents: 347
diff changeset
37 return Position(self.x + b.x, self.y + b.y, self.z + b.z)
69
18db99fda6bd Move spawing code from engine to gameboard - seems more natural.
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
38
100
e90068d1f374 Re-indent to four spaces.
Simon Cross <hodgestar@gmail.com>
parents: 69
diff changeset
39 def left_of(self, b):
e90068d1f374 Re-indent to four spaces.
Simon Cross <hodgestar@gmail.com>
parents: 69
diff changeset
40 return self.x < b.x
69
18db99fda6bd Move spawing code from engine to gameboard - seems more natural.
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
41
100
e90068d1f374 Re-indent to four spaces.
Simon Cross <hodgestar@gmail.com>
parents: 69
diff changeset
42 def right_of(self, b):
e90068d1f374 Re-indent to four spaces.
Simon Cross <hodgestar@gmail.com>
parents: 69
diff changeset
43 return self.x > b.x
69
18db99fda6bd Move spawing code from engine to gameboard - seems more natural.
Neil Muller <drnlmuller@gmail.com>
parents:
diff changeset
44
456
96dbf2c8506e Factor position cache out into its own class, make Position more useful.
Jeremy Thurgood <firxen@gmail.com>
parents: 431
diff changeset
45 def __hash__(self):
96dbf2c8506e Factor position cache out into its own class, make Position more useful.
Jeremy Thurgood <firxen@gmail.com>
parents: 431
diff changeset
46 return hash(self.to_3d_tuple())
96dbf2c8506e Factor position cache out into its own class, make Position more useful.
Jeremy Thurgood <firxen@gmail.com>
parents: 431
diff changeset
47
100
e90068d1f374 Re-indent to four spaces.
Simon Cross <hodgestar@gmail.com>
parents: 69
diff changeset
48 def __eq__(self, b):
456
96dbf2c8506e Factor position cache out into its own class, make Position more useful.
Jeremy Thurgood <firxen@gmail.com>
parents: 431
diff changeset
49 return self.to_3d_tuple() == b.to_3d_tuple()
96dbf2c8506e Factor position cache out into its own class, make Position more useful.
Jeremy Thurgood <firxen@gmail.com>
parents: 431
diff changeset
50
96dbf2c8506e Factor position cache out into its own class, make Position more useful.
Jeremy Thurgood <firxen@gmail.com>
parents: 431
diff changeset
51 def __str__(self):
96dbf2c8506e Factor position cache out into its own class, make Position more useful.
Jeremy Thurgood <firxen@gmail.com>
parents: 431
diff changeset
52 return "<Position: %s>" % (self.to_3d_tuple(),)
241
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
53
431
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
54 def intermediate_positions(self, b):
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
55 """Only operates in two dimensions."""
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
56 if max(abs(self.x - b.x), abs(self.y - b.y)) <= 1:
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
57 # pgu gets this case wrong on occasion.
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
58 return [b]
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
59 start = self.to_tile_tuple()
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
60 end = b.to_tile_tuple()
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
61 points = getline(start, end)
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
62 points.remove(start) # exclude start_pos
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
63 if end not in points:
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
64 # Rounding errors in getline cause this
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
65 points.append(end)
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
66 return [Position(p[0], p[1]) for p in points]
129de5883524 Moved intermediate position generation to a more suitable location.
Jeremy Thurgood <firxen@gmail.com>
parents: 427
diff changeset
67
241
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
68 class WeightedSelection(object):
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
69 def __init__(self, weightings=None):
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
70 self.weightings = []
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
71 self.total = 0
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
72 if weightings:
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
73 for item, weight in weightings:
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
74 self.weightings.append((item, weight))
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
75 self.total += weight
456
96dbf2c8506e Factor position cache out into its own class, make Position more useful.
Jeremy Thurgood <firxen@gmail.com>
parents: 431
diff changeset
76
241
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
77 def choose(self):
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
78 roll = random.uniform(0, self.total)
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
79 for item, weight in self.weightings:
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
80 if roll < weight:
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
81 return item
1a7000c8211c Demolition foxes, including better fox selection.
Jeremy Thurgood <firxen@gmail.com>
parents: 107
diff changeset
82 roll -= weight
336
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
83
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
84 class CheckDialog(gui.Dialog):
553
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
85 def __init__(self, sure_func, message, yes_choice, no_choice,
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
86 maybe_choice, **params):
548
27c09c58d89d Remove gameboard and dialog event flow hacks.
Simon Cross <hodgestar@gmail.com>
parents: 534
diff changeset
87 self._sure_func = sure_func
347
35f09e0ccd16 Make 'Esc' cancel the 'Are You Sure?' dialog. Fix dialog title
Neil Muller <drnlmuller@gmail.com>
parents: 336
diff changeset
88 title = gui.Label('Are You Sure?')
336
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
89 tbl = gui.Table()
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
90 tbl.tr()
553
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
91 tbl.td(gui.Label(message), colspan=3)
336
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
92 tbl.tr()
553
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
93 tbl.td(gui.Spacer(0, 15), colspan=3)
336
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
94 tbl.tr()
553
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
95 no_button = gui.Button(no_choice)
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
96 no_button.connect(gui.CLICK, self.clicked, 0)
336
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
97 tbl.td(no_button, align=-1)
553
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
98 if maybe_choice:
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
99 maybe_button = gui.Button(maybe_choice)
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
100 maybe_button.connect(gui.CLICK, self.clicked, 2)
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
101 tbl.td(maybe_button, align=1)
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
102 else:
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
103 tbl.td(gui.Spacer(0, 15))
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
104 yes_button = gui.Button(yes_choice)
7963fc09fac2 Make checkdialog take strings to show as input. Add possiblity of 3rd button on dialog
Neil Muller <drnlmuller@gmail.com>
parents: 548
diff changeset
105 yes_button.connect(gui.CLICK, self.clicked, 1)
336
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
106 tbl.td(yes_button, align=1)
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
107 gui.Dialog.__init__(self, title, tbl, **params)
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
108
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
109 def clicked(self, val):
548
27c09c58d89d Remove gameboard and dialog event flow hacks.
Simon Cross <hodgestar@gmail.com>
parents: 534
diff changeset
110 self._sure_func(val)
336
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
111 self.close()
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
112
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
113 def event(self, e):
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
114 if e.type == KEYDOWN and e.key == K_ESCAPE:
347
35f09e0ccd16 Make 'Esc' cancel the 'Are You Sure?' dialog. Fix dialog title
Neil Muller <drnlmuller@gmail.com>
parents: 336
diff changeset
115 self.clicked(False)
336
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
116 return True
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
117 return gui.Dialog.event(self, e)
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
118
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
119
519
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
120 # Utility layout functions
336
82a18615a0ab Ask 'Are you sure?'
Neil Muller <drnlmuller@gmail.com>
parents: 241
diff changeset
121
534
d16ed2a8a33e Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents: 519
diff changeset
122 def make_box(text, markup=False):
519
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
123 style = {
534
d16ed2a8a33e Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents: 519
diff changeset
124 'border' : 1,
d16ed2a8a33e Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents: 519
diff changeset
125 'padding_left': 2,
d16ed2a8a33e Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents: 519
diff changeset
126 'padding_right': 2,
d16ed2a8a33e Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents: 519
diff changeset
127 'padding_top': 2,
d16ed2a8a33e Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents: 519
diff changeset
128 'padding_bottom': 2,
d16ed2a8a33e Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents: 519
diff changeset
129 }
d16ed2a8a33e Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents: 519
diff changeset
130 if markup:
d16ed2a8a33e Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents: 519
diff changeset
131 word = html.HTML(text, style=style)
d16ed2a8a33e Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents: 519
diff changeset
132 else:
d16ed2a8a33e Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents: 519
diff changeset
133 word = gui.Label(text, style=style)
519
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
134 return word
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
135
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
136 def fix_widths(doc):
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
137 """Loop through all the widgets in the doc, and set the
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
138 width of the labels to max + 10"""
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
139 # We need to do this because of possible font issues
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
140 max_width = 0
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
141 for thing in doc.widgets:
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
142 if hasattr(thing, 'style'):
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
143 # A label
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
144 if thing.style.width > max_width:
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
145 max_width = thing.style.width
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
146 for thing in doc.widgets:
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
147 if hasattr(thing, 'style'):
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
148 thing.style.width = max_width + 10
f84ad10a9625 Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents: 456
diff changeset
149