|
0
|
1 from __future__ import nested_scopes
|
|
|
2 import sys, time
|
|
|
3 sys.path.append('..')
|
|
|
4 from Widgets.Fadable import Fadable
|
|
|
5
|
|
|
6 from Tix import *
|
|
|
7 import math, atexit, pickle
|
|
|
8 from Submaster import Submasters, sub_maxes
|
|
|
9 import dmxclient
|
|
|
10 from uihelpers import toplevelat
|
|
|
11
|
|
|
12 nudge_keys = {
|
|
|
13 'up' : list('qwertyuiop'),
|
|
|
14 'down' : list('asdfghjkl')
|
|
|
15 }
|
|
|
16 nudge_keys['down'].append('semicolon')
|
|
|
17
|
|
|
18 class SubScale(Scale, Fadable):
|
|
|
19 def __init__(self, master, *args, **kw):
|
|
|
20 self.scale_var = kw.get('variable') or DoubleVar()
|
|
|
21 kw.update({'variable' : self.scale_var,
|
|
|
22 'from' : 1, 'to' : 0, 'showvalue' : 0,
|
|
|
23 'sliderlength' : 15, 'res' : 0.01,
|
|
|
24 'width' : 40, 'troughcolor' : 'black', 'bg' : 'grey40',
|
|
|
25 'highlightthickness' : 1, 'bd' : 1,
|
|
|
26 'highlightcolor' : 'red', 'highlightbackground' : 'black',
|
|
|
27 'activebackground' : 'red'})
|
|
|
28 Scale.__init__(self, master, *args, **kw)
|
|
|
29 Fadable.__init__(self, var=self.scale_var, wheel_step=0.05)
|
|
|
30 self.draw_indicator_colors()
|
|
|
31 def draw_indicator_colors(self):
|
|
|
32 if self.scale_var.get() == 0:
|
|
|
33 self['troughcolor'] = 'black'
|
|
|
34 else:
|
|
|
35 self['troughcolor'] = 'blue'
|
|
|
36
|
|
|
37 class SubmasterTk(Frame):
|
|
|
38 def __init__(self, master, name, current_level):
|
|
|
39 Frame.__init__(self, master, bd=1, relief='raised', bg='black')
|
|
|
40 self.slider_var = DoubleVar()
|
|
|
41 self.slider_var.set(current_level)
|
|
|
42 self.scale = SubScale(self, variable=self.slider_var, width=20)
|
|
|
43 namelabel = Label(self, text=name, font="Arial 8", bg='black',
|
|
|
44 fg='white')
|
|
|
45 namelabel.pack(side=TOP)
|
|
|
46 levellabel = Label(self, textvariable=self.slider_var, font="Arial 8",
|
|
|
47 bg='black', fg='white')
|
|
|
48 levellabel.pack(side=TOP)
|
|
|
49 self.scale.pack(side=BOTTOM, expand=1, fill=BOTH)
|
|
|
50
|
|
|
51 class KeyboardComposer(Frame):
|
|
|
52 def __init__(self, root, submasters, current_sub_levels=None, dmxdummy=0):
|
|
|
53 Frame.__init__(self, root, bg='black')
|
|
|
54 self.submasters = submasters
|
|
|
55 self.dmxdummy = dmxdummy
|
|
|
56
|
|
|
57 self.current_sub_levels = {}
|
|
|
58 if current_sub_levels:
|
|
|
59 self.current_sub_levels = current_sub_levels
|
|
|
60 else:
|
|
|
61 try:
|
|
|
62 self.current_sub_levels = \
|
|
|
63 pickle.load(file('.keyboardcomposer.savedlevels'))
|
|
|
64 except IOError:
|
|
|
65 pass
|
|
|
66
|
|
|
67 self.draw_ui()
|
|
|
68 self.send_levels_loop()
|
|
|
69 def draw_ui(self):
|
|
|
70 self.rows = [] # this holds Tk Frames for each row
|
|
|
71 self.slider_vars = {} # this holds subname:sub Tk vars
|
|
|
72 self.slider_table = {} # this holds coords:sub Tk vars
|
|
|
73 self.current_row = 0
|
|
|
74
|
|
|
75 self.make_key_hints()
|
|
|
76 self.draw_sliders()
|
|
|
77 self.highlight_row(self.current_row)
|
|
|
78 self.rows[self.current_row].focus()
|
|
|
79
|
|
|
80 self.buttonframe = Frame(self, bg='black')
|
|
|
81 self.buttonframe.pack(side=BOTTOM)
|
|
|
82 self.refreshbutton = Button(self.buttonframe, text="Refresh",
|
|
|
83 command=self.refresh, bg='black', fg='white')
|
|
|
84 self.refreshbutton.pack(side=LEFT)
|
|
|
85 self.save_stage_button = Button(self.buttonframe, text="Save",
|
|
|
86 command=lambda: self.save_current_stage(self.sub_name.get()),
|
|
|
87 bg='black', fg='white')
|
|
|
88 self.save_stage_button.pack(side=LEFT)
|
|
|
89 self.sub_name = Entry(self.buttonframe, bg='black', fg='white')
|
|
|
90 self.sub_name.pack(side=LEFT)
|
|
|
91 self.stop_frequent_update_time = 0
|
|
|
92 def make_key_hints(self):
|
|
|
93 keyhintrow = Frame(self)
|
|
|
94
|
|
|
95 col = 0
|
|
|
96 for upkey, downkey in zip(nudge_keys['up'],
|
|
|
97 nudge_keys['down']):
|
|
|
98 # what a hack!
|
|
|
99 downkey = downkey.replace('semicolon', ';')
|
|
|
100 upkey, downkey = (upkey.upper(), downkey.upper())
|
|
|
101
|
|
|
102 # another what a hack!
|
|
|
103 keylabel = Label(keyhintrow, text='%s\n%s' % (upkey, downkey),
|
|
|
104 width=1, font=('Arial', 10), bg='red', fg='white', anchor='c')
|
|
|
105 keylabel.pack(side=LEFT, expand=1, fill=X)
|
|
|
106 col += 1
|
|
|
107
|
|
|
108 keyhintrow.pack(fill=X, expand=0)
|
|
|
109 self.keyhints = keyhintrow
|
|
|
110 def setup_key_nudgers(self, tkobject):
|
|
|
111 for d, keys in nudge_keys.items():
|
|
|
112 for key in keys:
|
|
|
113 # lowercase makes full=0
|
|
|
114 keysym = "<KeyPress-%s>" % key
|
|
|
115 tkobject.bind(keysym, \
|
|
|
116 lambda evt, num=keys.index(key), d=d: \
|
|
|
117 self.got_nudger(num, d))
|
|
|
118
|
|
|
119 # uppercase makes full=1
|
|
|
120 keysym = "<KeyPress-%s>" % key.upper()
|
|
|
121 keysym = keysym.replace('SEMICOLON', 'colon')
|
|
|
122 tkobject.bind(keysym, \
|
|
|
123 lambda evt, num=keys.index(key), d=d: \
|
|
|
124 self.got_nudger(num, d, full=1))
|
|
|
125
|
|
|
126 # page up and page down change the row
|
|
|
127 for key in '<Prior> <Next> <Control-n> <Control-p>'.split():
|
|
|
128 tkobject.bind(key, self.change_row)
|
|
|
129
|
|
|
130 def change_row(self, event):
|
|
|
131 diff = 1
|
|
|
132 if event.keysym in ('Prior', 'p'):
|
|
|
133 diff = -1
|
|
|
134 old_row = self.current_row
|
|
|
135 self.current_row += diff
|
|
|
136 self.current_row = max(0, self.current_row)
|
|
|
137 self.current_row = min(len(self.rows) - 1, self.current_row)
|
|
|
138 self.unhighlight_row(old_row)
|
|
|
139 self.highlight_row(self.current_row)
|
|
|
140 row = self.rows[self.current_row]
|
|
|
141 self.keyhints.pack_configure(before=row)
|
|
|
142 def got_nudger(self, number, direction, full=0):
|
|
|
143 subtk = self.slider_table[(self.current_row, number)]
|
|
|
144 if direction == 'up':
|
|
|
145 if full:
|
|
|
146 subtk.scale.fade(1)
|
|
|
147 else:
|
|
|
148 subtk.scale.increase()
|
|
|
149 else:
|
|
|
150 if full:
|
|
|
151 subtk.scale.fade(0)
|
|
|
152 else:
|
|
|
153 subtk.scale.decrease()
|
|
|
154 def draw_sliders(self):
|
|
|
155 self.tk_focusFollowsMouse()
|
|
|
156
|
|
|
157 rowcount = -1
|
|
|
158 col = 0
|
|
|
159 for sub in self.submasters.get_all_subs():
|
|
|
160 if col == 0: # make new row
|
|
|
161 row = self.make_row()
|
|
|
162 rowcount += 1
|
|
|
163 current_level = self.current_sub_levels.get(sub.name, 0)
|
|
|
164 subtk = self.draw_sub_slider(row, col, sub.name, current_level)
|
|
|
165 self.slider_table[(rowcount, col)] = subtk
|
|
|
166 col += 1
|
|
|
167 col %= 10
|
|
|
168
|
|
|
169 def slider_changed(x, y, z, subtk=subtk):
|
|
|
170 subtk.scale.draw_indicator_colors()
|
|
|
171 self.send_levels()
|
|
|
172
|
|
|
173 subtk.slider_var.trace('w', slider_changed)
|
|
|
174 def make_row(self):
|
|
|
175 row = Frame(self, bd=2, bg='black')
|
|
|
176 row.pack(expand=1, fill=BOTH)
|
|
|
177 self.setup_key_nudgers(row)
|
|
|
178 self.rows.append(row)
|
|
|
179 return row
|
|
|
180 def draw_sub_slider(self, row, col, name, current_level):
|
|
|
181 subtk = SubmasterTk(row, name, current_level)
|
|
|
182 subtk.place(relx=col * 0.1, rely=0, relwidth=0.1, relheight=1)
|
|
|
183 self.setup_key_nudgers(subtk.scale)
|
|
|
184
|
|
|
185 self.slider_vars[name] = subtk.slider_var
|
|
|
186 return subtk
|
|
|
187 def highlight_row(self, row):
|
|
|
188 row = self.rows[row]
|
|
|
189 row['bg'] = 'red'
|
|
|
190 def unhighlight_row(self, row):
|
|
|
191 row = self.rows[row]
|
|
|
192 row['bg'] = 'black'
|
|
|
193 def get_levels(self):
|
|
|
194 return dict([(name, slidervar.get())
|
|
|
195 for name, slidervar in self.slider_vars.items()])
|
|
|
196 def get_levels_as_sub(self):
|
|
|
197 scaledsubs = [self.submasters.get_sub_by_name(sub) * level \
|
|
|
198 for sub, level in self.get_levels().items()]
|
|
|
199
|
|
|
200 maxes = sub_maxes(*scaledsubs)
|
|
|
201 return maxes
|
|
|
202 def save_current_stage(self, subname):
|
|
|
203 print "saving current levels as", subname
|
|
|
204 sub = self.get_levels_as_sub()
|
|
|
205 sub.name = subname
|
|
|
206 sub.save()
|
|
|
207
|
|
|
208 def save(self):
|
|
|
209 pickle.dump(self.get_levels(),
|
|
|
210 file('.keyboardcomposer.savedlevels', 'w'))
|
|
|
211 def send_frequent_updates(self):
|
|
|
212 """called when we get a fade -- send events as quickly as possible"""
|
|
|
213 if time.time() <= self.stop_frequent_update_time:
|
|
|
214 self.send_levels()
|
|
|
215 self.after(10, self.send_frequent_updates)
|
|
|
216
|
|
|
217 def get_dmx_list(self):
|
|
|
218 maxes = self.get_levels_as_sub()
|
|
|
219 return maxes.get_dmx_list()
|
|
|
220 def send_levels(self):
|
|
|
221 if not self.dmxdummy:
|
|
|
222 levels = self.get_dmx_list()
|
|
|
223 dmxclient.outputlevels(levels)
|
|
|
224 # print "sending levels", levels
|
|
|
225 def send_levels_loop(self):
|
|
|
226 self.send_levels()
|
|
|
227 self.after(1000, self.send_levels_loop)
|
|
|
228 def refresh(self):
|
|
|
229 self.save()
|
|
|
230 self.submasters = Submasters()
|
|
|
231 self.current_sub_levels = \
|
|
|
232 pickle.load(file('.keyboardcomposer.savedlevels'))
|
|
|
233 for r in self.rows:
|
|
|
234 r.destroy()
|
|
|
235 self.keyhints.destroy()
|
|
|
236 self.buttonframe.destroy()
|
|
|
237 self.draw_ui()
|
|
|
238
|
|
|
239 if __name__ == "__main__":
|
|
|
240 s = Submasters()
|
|
|
241
|
|
|
242 root = Tk()
|
|
|
243 tl = toplevelat("Keyboard Composer", existingtoplevel=root)
|
|
|
244 kc = KeyboardComposer(tl, s, dmxdummy=0)
|
|
|
245 kc.pack(fill=BOTH, expand=1)
|
|
|
246 atexit.register(kc.save)
|
|
|
247 try:
|
|
|
248 mainloop()
|
|
|
249 except KeyboardInterrupt:
|
|
|
250 tl.destroy()
|
|
|
251 sys.exit()
|
