X-Git-Url: http://pileus.org/git/?a=blobdiff_plain;f=vis%2Fvisual.py;h=4f0e0e0019cbc3ef859e9cf0c87a04255b3074a0;hb=3912db35028a79fb9b660e8bd372be6abd2b8d8a;hp=0b21349b76bb6a4aa12afef8626b473af3b6a441;hpb=ab2b73de6d936912d16fc75de708d554fa5f4fa6;p=~andy%2Fcsm213a-hw diff --git a/vis/visual.py b/vis/visual.py index 0b21349..4f0e0e0 100644 --- a/vis/visual.py +++ b/vis/visual.py @@ -1,18 +1,31 @@ #!/usr/bin/env python -import pygtk -import gtk +import sys + +if sys.version_info<(3,0,0): + import pygtk as PyGtk + import gtk as Gtk + import gtk as GLib +else: + from gi.repository import Gtk + from gi.repository import GLib + +if sys.version_info<(3,0,0): + ICON_SIZE_BUTTON = Gtk.ICON_SIZE_BUTTON +else: + ICON_SIZE_BUTTON = Gtk.IconSize.BUTTON class Visual: - def __init__(self, config, device, xively): + def __init__(self, config, device, logger): def get_objects(names): - return map(lambda x: - self.builder.get_object(x), names) + return list(map(lambda x: + self.builder.get_object(x), names)) self.config = config self.device = device - self.xively = xively - self.builder = gtk.Builder() + self.logger = logger + self.builder = Gtk.Builder() + self.history = [0.5]*1000 self.builder.add_from_file('visual.ui') self.builder.connect_signals(self) @@ -21,8 +34,8 @@ class Visual: self.accs = get_objects(['accx', 'accy', 'accz']) self.mags = get_objects(['magx', 'magy', 'magz']) - self.touch = get_objects(['touch']) - self.light = get_objects(['light']) + self.lgts = get_objects(['lgt']) + self.tchs = get_objects(['tchp', 'tchd']) self.a2ds = get_objects(['a2d0', 'a2d1', 'a2d2', 'a2d3', 'a2d4', 'a2d5']) # Signal handlers @@ -43,95 +56,143 @@ class Visual: def on_xively(self, act): if act.get_active(): - self.status(self.xively.connect()) + self.status(self.logger.connect()) else: - self.status(self.xively.disconnect()) - return True - - def on_enable(self, _): - print "Unimplemented: on_enable" + self.status(self.logger.disconnect()) return True - def on_rate(self, _): - print "Unimplemented: on_rate" - return True + def on_flush(self, act): + self.logger.flush() + + def on_enable(self, obj): + name = Gtk.Buildable.get_name(obj) + name = name.replace('_btn', '') + state = obj.get_active() + if self.config.enable[name] != state: + self.device.set_enable(name, state) + self.config.enable[name] = state + self.config.save() + + def on_rate(self, obj): + name = Gtk.Buildable.get_name(obj) + name = name.replace('_spin', '') + value = obj.get_value() + if self.config.rate[name] != value: + self.device.set_rate(name, value) + self.config.rate[name] = value + self.config.save() + + def on_config(self, obj): + name = Gtk.Buildable.get_name(obj) + if isinstance(obj, Gtk.SpinButton): + value = obj.get_value() + else: + value = obj.get_text() + if getattr(self.config, name) != value: + setattr(self.config, name, value) + self.config.save() + + def on_expose(self, obj, _): + cairo = obj.window.cairo_create() + count = len(self.history) + extents = cairo.clip_extents() + width = extents[2] - extents[0] + height = extents[3] - extents[1] + + cairo.set_line_width(4) + for i in range(0,len(self.history)): + x = i*width/(count-1) + y = ((1-self.history[i]) * 0.8 + 0.1) * height + if i==0: + cairo.move_to(x, y) + else: + cairo.line_to(x, y) + cairo.stroke() def on_key(self, win, ev): if ev.string == 'q': - gtk.main_quit(); + Gtk.main_quit(); def on_destroy(self, win): - gtk.main_quit() + Gtk.main_quit() # Methods def load_config(self): + def set_val(name, val): + self.builder.get_object(name).set_value(val) + def set_txt(name, val): + self.builder.get_object(name).set_text(val) + def set_act(name, val): + self.builder.get_object(name).set_active(val) + # Serial settings - device = self.builder.get_object("device") - baud = self.builder.get_object("baudrate") - parity = self.builder.get_object("parity") - datab = self.builder.get_object("databits") - stopb = self.builder.get_object("stopbits") - device.set_text(self.config.device) - baud.set_value(self.config.baudrate) - parity.set_text(self.config.parity) - datab.set_value(self.config.databits) - stopb.set_value(self.config.stopbits) + set_txt('device', self.config.device) + set_val('baudrate', self.config.baudrate) + set_txt('parity', self.config.parity) + set_val('databits', self.config.databits) + set_val('stopbits', self.config.stopbits) # Xively settings - user = self.builder.get_object("username") - pwd = self.builder.get_object("password") - api = self.builder.get_object("apikey") - user.set_text(self.config.username) - pwd.set_text(self.config.password) - api.set_text(self.config.apikey) + set_txt('feedid', self.config.feedid) + set_txt('apikey', self.config.apikey) + set_val('maxrate', self.config.maxrate) + + # Sensor settings + for key in list(self.config.enable.keys()): + set_act(key+'_btn', self.config.enable[key]) + set_val(key+'_spin', self.config.rate[key]) - # TODO - send changes back to config def update(self, state): - def setxyz(objs, vals): - if vals[0]: objs[0].set_text('X: %f' % vals[0]) - if vals[1]: objs[1].set_text('Y: %f' % vals[1]) - if vals[2]: objs[2].set_text('Z: %f' % vals[2]) - def setabs(objs, vals): + def setval(objs, vals, lbls): for i in range(0,len(objs)): - if vals[i]: objs[i].set_text('%f' % vals[i]) - - setxyz(self.accs, state.acc) - setxyz(self.mags, state.mag) - setabs(self.touch, state.touch) - setabs(self.light, state.light) - setabs(self.a2ds, state.a2d) + if vals[i] == None: + continue + if lbls and lbls[i]: + objs[i].set_text('%s: %f' % (lbls[i], vals[i])) + else: + objs[i].set_text('%f' % vals[i]) + + setval(self.accs, state.acc, ['X', 'Y', 'Z']) + setval(self.mags, state.mag, ['X', 'Y', 'Z']) + setval(self.lgts, state.lgt, []) + setval(self.tchs, state.tch, ['P', 'D']) + setval(self.a2ds, state.a2d, []) + + self.history = self.history[1:] + [state.a2d[0]] + self.window.queue_draw() def debug(self, state): - print 'update: ' + str(state) - print '\tacc - ' + str(state.acc) - print '\tmag - ' + str(state.mag) - print '\tlgt - ' + str(state.light) - print '\ttch - ' + str(state.touch) - print '\ta2d - ' + str(state.a2d) - + print('update: ' + str(state)) + print('\tacc - ' + str(state.acc)) + print('\tmag - ' + str(state.mag)) + print('\tlgt - ' + str(state.lgt)) + print('\ttch - ' + str(state.touch)) + print('\ta2d - ' + str(state.a2d)) + def timer(self): - serial = self.builder.get_object("serial_btn") - status = self.builder.get_object("conn") + def check(state, name): + btn = self.builder.get_object(name + '_btn') + con = self.builder.get_object(name + '_con') + img = Gtk.STOCK_YES if state else Gtk.STOCK_NO + btn.set_active(state) + con.set_from_stock(img, ICON_SIZE_BUTTON) - if self.device.running(): - serial.set_active(True) - status.set_from_stock(gtk.STOCK_YES, gtk.ICON_SIZE_BUTTON) - else: - serial.set_active(False) - status.set_from_stock(gtk.STOCK_NO, gtk.ICON_SIZE_BUTTON) - if self.device.running(): - for item in self.device.process(): - self.update(item) - #self.debug(item) + check(self.device.running(), "serial") + check(self.logger.running(), "xively") + + for item in self.device.process(): + self.update(item) + self.logger.append(item) + #self.debug(item) return True def run(self): self.load_config() self.window.show() - gtk.timeout_add(1000/60, self.timer) - gtk.main() + GLib.timeout_add(1000/60, self.timer) + Gtk.main() # Private methods def status(self, msg): @@ -139,4 +200,3 @@ class Visual: return status = self.builder.get_object("status") status.push(text=('Error: ' + msg), context_id=0) -