]> Pileus Git - ~andy/csm213a-hw/blobdiff - vis/visual.py
Fix whitespace error
[~andy/csm213a-hw] / vis / visual.py
index 4f0e0e0019cbc3ef859e9cf0c87a04255b3074a0..6fab6306caf63626d76ae0636a863e178d8794d1 100644 (file)
@@ -16,6 +16,9 @@ else:
        ICON_SIZE_BUTTON = Gtk.IconSize.BUTTON
 
 class Visual:
+       FRAMES_PER_SEC = 60 # hz
+       RATES_PER_SEC  = 1  # hz
+
        def __init__(self, config, device, logger):
                def get_objects(names):
                        return list(map(lambda x:
@@ -27,10 +30,26 @@ class Visual:
                self.builder  = Gtk.Builder()
                self.history  = [0.5]*1000
 
+               self.rate_due = 0
+               self.rate_cnt = {'acc': 0,
+                                'mag': 0,
+                                'lgt': 0,
+                                'tch': 0,
+                                'a2d': 0}
+
                self.builder.add_from_file('visual.ui')
                self.builder.connect_signals(self)
+
                self.window   = self.builder.get_object("window")
                self.settings = self.builder.get_object("settings")
+               self.plotdraw = self.builder.get_object('plot')
+               self.plotsns  = self.builder.get_object('sns')
+               self.plotaxis = self.builder.get_object('axis')
+
+               if sys.version_info<(3,0,0):
+                       self.plotdraw.connect('expose-event', self.on_expose)
+               else:
+                       self.plotdraw.connect('draw', self.on_draw)
 
                self.accs     = get_objects(['accx', 'accy', 'accz'])
                self.mags     = get_objects(['magx', 'magy', 'magz'])
@@ -51,14 +70,14 @@ class Visual:
                if act.get_active():
                        self.status(self.device.connect())
                else:
-                       self.status(self.device.disconnect())
+                       self.device.disconnect()
                return True
 
        def on_xively(self, act):
                if act.get_active():
                        self.status(self.logger.connect())
                else:
-                       self.status(self.logger.disconnect())
+                       self.logger.disconnect()
                return True
 
        def on_flush(self, act):
@@ -92,15 +111,14 @@ class Visual:
                        setattr(self.config, name, value)
                        self.config.save()
 
-       def on_expose(self, obj, _):
-               cairo   = obj.window.cairo_create()
+       def on_draw(self, obj, cairo):
                count   = len(self.history)
                extents = cairo.clip_extents()
-               width   = extents[2] - extents[0] 
-               height  = extents[3] - extents[1] 
+               width   = extents[2] - extents[0]
+               height  = extents[3] - extents[1]
 
                cairo.set_line_width(4)
-               for i in range(0,len(self.history)):
+               for i in range(count):
                        x = i*width/(count-1)
                        y = ((1-self.history[i]) * 0.8 + 0.1) * height
                        if i==0:
@@ -109,6 +127,10 @@ class Visual:
                                cairo.line_to(x, y)
                cairo.stroke()
 
+       def on_expose(self, obj, event):
+               cairo = obj.window.cairo_create()
+               self.draw(obj, cairo)
+
        def on_key(self, win, ev):
                if ev.string == 'q':
                        Gtk.main_quit();
@@ -116,7 +138,7 @@ class Visual:
        def on_destroy(self, win):
                Gtk.main_quit()
 
-        # Methods
+       # Methods
        def load_config(self):
                def set_val(name, val):
                        self.builder.get_object(name).set_value(val)
@@ -142,25 +164,36 @@ class Visual:
                        set_act(key+'_btn',  self.config.enable[key])
                        set_val(key+'_spin', self.config.rate[key])
 
-
        def update(self, state):
                def setval(objs, vals, lbls):
+                       found = 0
                        for i in range(0,len(objs)):
                                if vals[i] == None:
                                        continue
+                               found = 1
                                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()
+                       return found
+
+               # Dispaly values and sum up rates
+               self.rate_cnt['acc'] += setval(self.accs, state.acc, ['X', 'Y', 'Z'])
+               self.rate_cnt['mag'] += setval(self.mags, state.mag, ['X', 'Y', 'Z'])
+               self.rate_cnt['lgt'] += setval(self.lgts, state.lgt, [])
+               self.rate_cnt['tch'] += setval(self.tchs, state.tch, ['P', 'D'])
+               self.rate_cnt['a2d'] += setval(self.a2ds, state.a2d, [])
+
+               # Save history
+               idxs = ['acc', 'mag', 'tch', 'lgt', 'a2d']
+               sns  = idxs[self.plotsns.get_active()]
+               axis = int(self.plotaxis.get_value())
+               if hasattr(state, sns):
+                       data = getattr(state, sns)
+                       axis = min(axis-1, len(data)-1)
+                       item = data[axis]
+                       if item != None:
+                               self.history = self.history[1:] + [item]
 
        def debug(self, state):
                print('update: ' + str(state))
@@ -170,7 +203,7 @@ class Visual:
                print('\ttch - ' + str(state.touch))
                print('\ta2d - ' + str(state.a2d))
 
-       def timer(self):
+       def read_timer(self):
                def check(state, name):
                        btn = self.builder.get_object(name + '_btn')
                        con = self.builder.get_object(name + '_con')
@@ -178,25 +211,40 @@ class Visual:
                        btn.set_active(state)
                        con.set_from_stock(img, ICON_SIZE_BUTTON)
 
+               # Update status icons
                check(self.device.running(), "serial")
                check(self.logger.running(), "xively")
 
+               # Read data and update data display
                for item in self.device.process():
                        self.update(item)
                        self.logger.append(item)
                        #self.debug(item)
 
+               # Refresh window
+               self.window.queue_draw()
+
+               return True
+
+       def rate_timer(self):
+               for sns in self.rate_cnt:
+                       rate  = self.rate_cnt[sns] * Visual.RATES_PER_SEC
+                       obj   = self.builder.get_object(sns + '_rate')
+                       obj.set_text('%d' % rate)
+                       self.rate_cnt[sns] = 0
                return True
 
        def run(self):
                self.load_config()
                self.window.show()
-               GLib.timeout_add(1000/60, self.timer)
+               GLib.timeout_add(1000/Visual.FRAMES_PER_SEC, self.read_timer)
+               GLib.timeout_add(1000/Visual.RATES_PER_SEC,  self.rate_timer)
                Gtk.main()
 
        # Private methods
        def status(self, msg):
+               status = self.builder.get_object("status")
+               status.pop(context_id=0)
                if not msg:
                        return
-               status = self.builder.get_object("status")
                status.push(text=('Error: ' + msg), context_id=0)