1 | # -*- coding: utf-8 -*- |
---|
2 | """ |
---|
3 | pieces_test |
---|
4 | ~~~~~~~ |
---|
5 | |
---|
6 | |
---|
7 | :copyright: © 2011 UfSoft.org - :email:`Pedro Algarvio (pedro@algarvio.me)` |
---|
8 | :license: BSD, see LICENSE for more details. |
---|
9 | """ |
---|
10 | import pygtk |
---|
11 | pygtk.require('2.0') |
---|
12 | import gtk |
---|
13 | |
---|
14 | class PiecesBar(gtk.DrawingArea): |
---|
15 | # Draw in response to an expose-event |
---|
16 | __gsignals__ = { "expose-event": "override" } |
---|
17 | |
---|
18 | def __init__(self): |
---|
19 | gtk.DrawingArea.__init__(self) |
---|
20 | self.width = 0 |
---|
21 | self.height = 0 |
---|
22 | self.pieces = [] |
---|
23 | self.colors = {} |
---|
24 | |
---|
25 | self.connect('size-allocate', self.on_size_allocate) |
---|
26 | self.connect('realize', self.on_realize) |
---|
27 | |
---|
28 | self.show() |
---|
29 | |
---|
30 | def on_realize(self, widget): |
---|
31 | map = widget.get_colormap() |
---|
32 | done_color = map.alloc_color("#4883b5", True, True) |
---|
33 | down_color = map.alloc_color("#fff600", True, True) |
---|
34 | wait_color = map.alloc_color("#3dff58", True, True) |
---|
35 | miss_color = map.alloc_color("#ff0000", True, True) |
---|
36 | outline_color = map.alloc_color("#888888") |
---|
37 | |
---|
38 | self.colors = { |
---|
39 | 0: (miss_color.red/65535.0, miss_color.green/65535.0, miss_color.blue/65535.0), |
---|
40 | 1: (wait_color.red/65535.0, wait_color.green/65535.0, wait_color.blue/65535.0), |
---|
41 | 2: (down_color.red/65535.0, down_color.green/65535.0, down_color.blue/65535.0), |
---|
42 | 3: (done_color.red/65535.0, done_color.green/65535.0, done_color.blue/65535.0), |
---|
43 | } |
---|
44 | self.gc_outline = widget.window.new_gc(foreground=outline_color) |
---|
45 | |
---|
46 | def on_size_allocate(self, widget, size): |
---|
47 | self.width = size.width |
---|
48 | self.height = size.height |
---|
49 | |
---|
50 | # Handle the expose-event by drawing |
---|
51 | def do_expose_event(self, event=None): |
---|
52 | |
---|
53 | num_pieces = len(self.pieces) |
---|
54 | |
---|
55 | if num_pieces < 1: |
---|
56 | self.clear() |
---|
57 | return None |
---|
58 | |
---|
59 | # Create the cairo context |
---|
60 | cr = self.window.cairo_create() |
---|
61 | # Restrict Cairo to the exposed area; avoid extra work |
---|
62 | if event: |
---|
63 | cr.rectangle(event.area.x, event.area.y, |
---|
64 | event.area.width, event.area.height) |
---|
65 | cr.clip() |
---|
66 | |
---|
67 | # Fill the background |
---|
68 | cr.set_source_rgb(136, 136, 136) |
---|
69 | cr.rectangle(0, 0, self.width-1, self.height-1) |
---|
70 | cr.fill() |
---|
71 | |
---|
72 | width = self.width - 2 |
---|
73 | pieces = self.collapse_pieces() |
---|
74 | |
---|
75 | start_pos = 1 |
---|
76 | |
---|
77 | for state, wpercent in pieces: |
---|
78 | pwidth = width*wpercent |
---|
79 | cr.set_source_rgb(*self.colors[state]) |
---|
80 | cr.rectangle(start_pos, 1, pwidth, self.height-2) |
---|
81 | cr.fill() |
---|
82 | start_pos += pwidth |
---|
83 | |
---|
84 | def collapse_pieces(self): |
---|
85 | num_pieces = len(self.pieces)*1.0 |
---|
86 | opieces = self.pieces[:] |
---|
87 | npieces = [] |
---|
88 | v = None |
---|
89 | n = None |
---|
90 | while True: |
---|
91 | if not opieces: |
---|
92 | break |
---|
93 | if v is None: |
---|
94 | v = opieces.pop(0) |
---|
95 | n = 1 |
---|
96 | else: |
---|
97 | vv = opieces.pop(0) |
---|
98 | if v == vv: |
---|
99 | n += 1 |
---|
100 | else: |
---|
101 | npieces.append((v, n/num_pieces)) |
---|
102 | v = vv |
---|
103 | n = 1 |
---|
104 | return npieces |
---|
105 | |
---|
106 | def draw_piece(self, piece, start_x, start_y, width, height): |
---|
107 | self.window.draw_rectangle( |
---|
108 | self.colors[piece], True, start_x, start_y, width, height |
---|
109 | ) |
---|
110 | |
---|
111 | def clear(self): |
---|
112 | self.pieces = [0] |
---|
113 | self.update() |
---|
114 | |
---|
115 | def get_text(self): |
---|
116 | return "" |
---|
117 | |
---|
118 | def set_text(self, text): |
---|
119 | pass |
---|
120 | |
---|
121 | pieces = PiecesBar() |
---|
122 | pieces.pieces = [3, 2, 1, 1, 1, 0, 0, 2, 0, 1, 0, 2, 2, 2, 1, 2, 2, 0, 0, 0, 2, |
---|
123 | 1, 1, 1, 2, 0, 0, 2, 2, 1, 0, 2, 1, 0, 1, 1, 0, 3, 3, 3, 3, 3, |
---|
124 | 3, 3, 0, 3, 3, 0, 0, 2, 0, 2, 2, 2, 0, 2, 2, 1]*100 |
---|
125 | |
---|
126 | window = gtk.Window(gtk.WINDOW_TOPLEVEL) |
---|
127 | window.set_default_size(500, 200) |
---|
128 | def destroy(widget, data=None): |
---|
129 | gtk.main_quit() |
---|
130 | window.connect("destroy", destroy) |
---|
131 | vbox = gtk.VBox() |
---|
132 | vbox.add(pieces) |
---|
133 | |
---|
134 | hbox1 = gtk.HBox() |
---|
135 | hbox1.pack_start(gtk.Label("Missing:"), False, False) |
---|
136 | missing_color = gtk.ColorButton(window.get_colormap().alloc_color("#ff0000", True, True)) |
---|
137 | def on_missing_collor_changed(button): |
---|
138 | color = button.get_color() |
---|
139 | r, g, b = color.red, color.green, color.blue |
---|
140 | pieces.colors[0] = r/65535.0, g/65535.0, b/65535.0 |
---|
141 | print 'Miss color RGB', pieces.colors[0] |
---|
142 | pieces.do_expose_event() |
---|
143 | missing_color.connect("color-set", on_missing_collor_changed) |
---|
144 | hbox1.pack_start(missing_color, False, False) |
---|
145 | vbox.pack_start(hbox1, False, False) |
---|
146 | |
---|
147 | hbox2 = gtk.HBox() |
---|
148 | hbox2.pack_start(gtk.Label("Waiting:"), False, False) |
---|
149 | waiting_color = gtk.ColorButton(window.get_colormap().alloc_color("#3dff58", True, True)) |
---|
150 | def on_waiting_collor_changed(button): |
---|
151 | color = button.get_color() |
---|
152 | r, g, b = color.red, color.green, color.blue |
---|
153 | pieces.colors[1] = r/65535.0, g/65535.0, b/65535.0 |
---|
154 | print 'Wait color RGB', pieces.colors[1] |
---|
155 | pieces.do_expose_event() |
---|
156 | waiting_color.connect("color-set", on_waiting_collor_changed) |
---|
157 | hbox2.pack_start(waiting_color, False, False) |
---|
158 | vbox.pack_start(hbox2, False, False) |
---|
159 | |
---|
160 | hbox3 = gtk.HBox() |
---|
161 | hbox3.pack_start(gtk.Label("Downloading:"), False, False) |
---|
162 | down_color = gtk.ColorButton(window.get_colormap().alloc_color("#fff600", True, True)) |
---|
163 | def on_down_collor_changed(button): |
---|
164 | color = button.get_color() |
---|
165 | r, g, b = color.red, color.green, color.blue |
---|
166 | pieces.colors[2] = r/65535.0, g/65535.0, b/65535.0 |
---|
167 | print 'Down color RGB', pieces.colors[2] |
---|
168 | pieces.do_expose_event() |
---|
169 | down_color.connect("color-set", on_down_collor_changed) |
---|
170 | hbox3.pack_start(down_color, False, False) |
---|
171 | vbox.pack_start(hbox3, False, False) |
---|
172 | |
---|
173 | hbox4 = gtk.HBox() |
---|
174 | hbox4.pack_start(gtk.Label("Completed:"), False, False) |
---|
175 | done_color = gtk.ColorButton(window.get_colormap().alloc_color("#4883b5", True, True)) |
---|
176 | def on_done_collor_changed(button): |
---|
177 | color = button.get_color() |
---|
178 | r, g, b = color.red, color.green, color.blue |
---|
179 | pieces.colors[3] = r/65535.0, g/65535.0, b/65535.0 |
---|
180 | print 'Done color RGB', pieces.colors[3] |
---|
181 | pieces.do_expose_event() |
---|
182 | done_color.connect("color-set", on_done_collor_changed) |
---|
183 | hbox4.pack_start(done_color, False, False) |
---|
184 | vbox.pack_start(hbox4, False, False) |
---|
185 | |
---|
186 | window.add(vbox) |
---|
187 | window.show_all() |
---|
188 | gtk.main() |
---|
189 | |
---|