]> Pileus Git - ~andy/csm213a-hw/blobdiff - vis/visual.py
Update from Yue, and add rate column
[~andy/csm213a-hw] / vis / visual.py
index f07ea0d11c2b0a4858803892992c997f60392c01..2ad6db69324ab31f6ea82686d9089a9592f31b19 100644 (file)
@@ -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,9 +34,10 @@ 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'])
+               self.smpl     = get_objects(['sampling'])
 
        # Signal handlers
        def on_hide(self, win, *args):
@@ -43,93 +57,144 @@ 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
-               baud   = self.builder.get_object("baudrate")
-               parity = self.builder.get_object("parity")
-               datab  = self.builder.get_object("databits")
-               stopb  = self.builder.get_object("stopbits")
-               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, [])
+               #setval(self.smpl,  state.)
+
+               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):
@@ -137,4 +202,3 @@ class Visual:
                        return
                status = self.builder.get_object("status")
                status.push(text=('Error: ' + msg), context_id=0)
-