Mercurial > rinkhals
annotate gamelib/misc.py @ 536:1a224ba50edf
Correctly set the position of newly hatched chickens in buildings
author | Neil Muller <drnlmuller@gmail.com> |
---|---|
date | Fri, 27 Nov 2009 23:29:32 +0000 |
parents | d16ed2a8a33e |
children | 27c09c58d89d |
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 | 5 from pygame.locals import KEYDOWN, K_ESCAPE |
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 | 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 | 18 self.x = x |
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 | 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 | 28 def dist(self, b): |
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 | 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 | 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 | 39 def left_of(self, b): |
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 | 42 def right_of(self, b): |
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 | 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 | 83 |
84 class CheckDialog(gui.Dialog): | |
85 def __init__(self, **params): | |
347
35f09e0ccd16
Make 'Esc' cancel the 'Are You Sure?' dialog. Fix dialog title
Neil Muller <drnlmuller@gmail.com>
parents:
336
diff
changeset
|
86 title = gui.Label('Are You Sure?') |
336 | 87 self.do_quit = False |
88 self.running = True | |
89 tbl = gui.Table() | |
90 tbl.tr() | |
91 tbl.td(gui.Label("Do you REALLY want to exit this game?"), colspan=2) | |
92 tbl.tr() | |
93 tbl.td(gui.Spacer(0, 15), colspan=2) | |
94 tbl.tr() | |
95 yes_button = gui.Button('Yes') | |
96 yes_button.connect(gui.CLICK, self.clicked, True) | |
97 no_button = gui.Button('No') | |
98 no_button.connect(gui.CLICK, self.clicked, False) | |
99 tbl.td(no_button, align=-1) | |
100 tbl.td(yes_button, align=1) | |
101 gui.Dialog.__init__(self, title, tbl, **params) | |
102 | |
103 def clicked(self, val): | |
104 self.do_quit = val | |
105 self.running = False | |
106 self.close() | |
107 | |
108 def event(self, e): | |
109 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
|
110 self.clicked(False) |
336 | 111 return True |
112 return gui.Dialog.event(self, e) | |
113 | |
114 def check_exit(): | |
115 dialog = CheckDialog() | |
116 dialog.open() | |
117 return dialog | |
118 | |
519
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
119 # Utility layout functions |
336 | 120 |
534
d16ed2a8a33e
Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents:
519
diff
changeset
|
121 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
|
122 style = { |
534
d16ed2a8a33e
Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents:
519
diff
changeset
|
123 'border' : 1, |
d16ed2a8a33e
Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents:
519
diff
changeset
|
124 'padding_left': 2, |
d16ed2a8a33e
Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents:
519
diff
changeset
|
125 'padding_right': 2, |
d16ed2a8a33e
Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents:
519
diff
changeset
|
126 'padding_top': 2, |
d16ed2a8a33e
Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents:
519
diff
changeset
|
127 'padding_bottom': 2, |
d16ed2a8a33e
Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents:
519
diff
changeset
|
128 } |
d16ed2a8a33e
Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents:
519
diff
changeset
|
129 if markup: |
d16ed2a8a33e
Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents:
519
diff
changeset
|
130 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
|
131 else: |
d16ed2a8a33e
Make price and control reference dialogs a little shinier.
Simon Cross <hodgestar@gmail.com>
parents:
519
diff
changeset
|
132 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
|
133 return word |
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
134 |
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
135 def fix_widths(doc): |
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
136 """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
|
137 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
|
138 # 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
|
139 max_width = 0 |
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
140 for thing in doc.widgets: |
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
141 if hasattr(thing, 'style'): |
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
142 # A label |
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
143 if thing.style.width > max_width: |
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
144 max_width = thing.style.width |
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
145 for thing in doc.widgets: |
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
146 if hasattr(thing, 'style'): |
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
147 thing.style.width = max_width + 10 |
f84ad10a9625
Remove move tool. Add basic controls reference dialog
Neil Muller <drnlmuller@gmail.com>
parents:
456
diff
changeset
|
148 |