5 if sys.version_info<(3,0,0):
10 from gi.repository import Gtk
11 from gi.repository import GLib
13 if sys.version_info<(3,0,0):
14 ICON_SIZE_BUTTON = Gtk.ICON_SIZE_BUTTON
16 ICON_SIZE_BUTTON = Gtk.IconSize.BUTTON
19 FRAMES_PER_SEC = 60 # hz
20 RATES_PER_SEC = 1 # hz
22 def __init__(self, config, device, logger):
23 def get_objects(names):
24 return list(map(lambda x:
25 self.builder.get_object(x), names))
30 self.builder = Gtk.Builder()
31 self.history = [0.5]*1000
34 self.rate_cnt = {'acc': 0,
40 self.builder.add_from_file('visual.ui')
41 self.builder.connect_signals(self)
43 self.window = self.builder.get_object("window")
44 self.settings = self.builder.get_object("settings")
45 self.plotdraw = self.builder.get_object('plot')
46 self.plotsns = self.builder.get_object('sns')
47 self.plotaxis = self.builder.get_object('axis')
49 if sys.version_info<(3,0,0):
50 self.plotdraw.connect('expose-event', self.on_expose)
52 self.plotdraw.connect('draw', self.on_draw)
54 self.accs = get_objects(['accx', 'accy', 'accz'])
55 self.mags = get_objects(['magx', 'magy', 'magz'])
56 self.lgts = get_objects(['lgt'])
57 self.tchs = get_objects(['tchp', 'tchd'])
58 self.a2ds = get_objects(['a2d0', 'a2d1', 'a2d2', 'a2d3', 'a2d4', 'a2d5'])
61 def on_hide(self, win, *args):
65 def on_show(self, act):
69 def on_serial(self, act):
71 self.status(self.device.connect())
73 self.device.disconnect()
76 def on_xively(self, act):
78 self.status(self.logger.connect())
80 self.logger.disconnect()
83 def on_flush(self, act):
86 def on_enable(self, obj):
87 name = Gtk.Buildable.get_name(obj)
88 name = name.replace('_btn', '')
89 state = obj.get_active()
90 if self.config.enable[name] != state:
91 self.device.set_enable(name, state)
92 self.config.enable[name] = state
95 def on_rate(self, obj):
96 name = Gtk.Buildable.get_name(obj)
97 name = name.replace('_spin', '')
98 value = obj.get_value()
99 if self.config.rate[name] != value:
100 self.device.set_rate(name, value)
101 self.config.rate[name] = value
104 def on_config(self, obj):
105 name = Gtk.Buildable.get_name(obj)
106 if isinstance(obj, Gtk.SpinButton):
107 value = obj.get_value()
109 value = obj.get_text()
110 if getattr(self.config, name) != value:
111 setattr(self.config, name, value)
114 def on_draw(self, obj, cairo):
115 count = len(self.history)
116 extents = cairo.clip_extents()
117 width = extents[2] - extents[0]
118 height = extents[3] - extents[1]
120 cairo.set_line_width(4)
121 for i in range(count):
122 x = i*width/(count-1)
123 y = ((1-self.history[i]) * 0.8 + 0.1) * height
130 def on_expose(self, obj, event):
131 cairo = obj.window.cairo_create()
132 self.draw(obj, cairo)
134 def on_key(self, win, ev):
138 def on_destroy(self, win):
142 def load_config(self):
143 def set_val(name, val):
144 self.builder.get_object(name).set_value(val)
145 def set_txt(name, val):
146 self.builder.get_object(name).set_text(val)
147 def set_act(name, val):
148 self.builder.get_object(name).set_active(val)
151 set_txt('device', self.config.device)
152 set_val('baudrate', self.config.baudrate)
153 set_txt('parity', self.config.parity)
154 set_val('databits', self.config.databits)
155 set_val('stopbits', self.config.stopbits)
158 set_txt('feedid', self.config.feedid)
159 set_txt('apikey', self.config.apikey)
160 set_val('maxrate', self.config.maxrate)
163 for key in list(self.config.enable.keys()):
164 set_act(key+'_btn', self.config.enable[key])
165 set_val(key+'_spin', self.config.rate[key])
167 def update(self, state):
168 def setval(objs, vals, lbls):
170 for i in range(0,len(objs)):
175 objs[i].set_text('%s: %f' % (lbls[i], vals[i]))
177 objs[i].set_text('%f' % vals[i])
180 # Dispaly values and sum up rates
181 self.rate_cnt['acc'] += setval(self.accs, state.acc, ['X', 'Y', 'Z'])
182 self.rate_cnt['mag'] += setval(self.mags, state.mag, ['X', 'Y', 'Z'])
183 self.rate_cnt['lgt'] += setval(self.lgts, state.lgt, [])
184 self.rate_cnt['tch'] += setval(self.tchs, state.tch, ['P', 'D'])
185 self.rate_cnt['a2d'] += setval(self.a2ds, state.a2d, [])
188 idxs = ['acc', 'mag', 'tch', 'lgt', 'a2d']
189 sns = idxs[self.plotsns.get_active()]
190 axis = int(self.plotaxis.get_value())
191 if hasattr(state, sns):
192 data = getattr(state, sns)
193 axis = min(axis-1, len(data)-1)
196 self.history = self.history[1:] + [item]
198 def debug(self, state):
199 print('update: ' + str(state))
200 print('\tacc - ' + str(state.acc))
201 print('\tmag - ' + str(state.mag))
202 print('\tlgt - ' + str(state.lgt))
203 print('\ttch - ' + str(state.touch))
204 print('\ta2d - ' + str(state.a2d))
206 def read_timer(self):
207 def check(state, name):
208 btn = self.builder.get_object(name + '_btn')
209 con = self.builder.get_object(name + '_con')
210 img = Gtk.STOCK_YES if state else Gtk.STOCK_NO
211 btn.set_active(state)
212 con.set_from_stock(img, ICON_SIZE_BUTTON)
214 # Update status icons
215 check(self.device.running(), "serial")
216 check(self.logger.running(), "xively")
218 # Read data and update data display
219 for item in self.device.process():
221 self.logger.append(item)
225 self.window.queue_draw()
229 def rate_timer(self):
230 for sns in self.rate_cnt:
231 rate = self.rate_cnt[sns] * Visual.RATES_PER_SEC
232 obj = self.builder.get_object(sns + '_rate')
233 obj.set_text('%d' % rate)
234 self.rate_cnt[sns] = 0
240 GLib.timeout_add(1000/Visual.FRAMES_PER_SEC, self.read_timer)
241 GLib.timeout_add(1000/Visual.RATES_PER_SEC, self.rate_timer)
245 def status(self, msg):
246 status = self.builder.get_object("status")
247 status.pop(context_id=0)
250 status.push(text=('Error: ' + msg), context_id=0)