Pileus Git - ~andy/csm213a-hw/blob - vis/visual.py

   1 #!/usr/bin/env python
   2
   3 import sys
   4
   5 if sys.version_info<(3,0,0):
   6         import pygtk as PyGtk
   7         import gtk   as Gtk
   8         import gtk   as GLib
   9 else:
  10         from gi.repository import Gtk
  11         from gi.repository import GLib
  12
  13 if sys.version_info<(3,0,0):
  14         ICON_SIZE_BUTTON = Gtk.ICON_SIZE_BUTTON
  15 else:
  16         ICON_SIZE_BUTTON = Gtk.IconSize.BUTTON
  17
  18 class Visual:
  19         FRAMES_PER_SEC = 60 # hz
  20         RATES_PER_SEC  = 1  # hz
  21
  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))
  26
  27                 self.config   = config
  28                 self.device   = device
  29                 self.logger   = logger
  30                 self.builder  = Gtk.Builder()
  31                 self.history  = [0.5]*1000
  32
  33                 self.rate_due = 0
  34                 self.rate_cnt = {'acc': 0,
  35                                  'mag': 0,
  36                                  'lgt': 0,
  37                                  'tch': 0,
  38                                  'a2d': 0}
  39
  40                 self.builder.add_from_file('visual.ui')
  41                 self.builder.connect_signals(self)
  42
  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')
  48
  49                 if sys.version_info<(3,0,0):
  50                         self.plotdraw.connect('expose-event', self.on_expose)
  51                 else:
  52                         self.plotdraw.connect('draw', self.on_draw)
  53
  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'])
  59
  60         # Signal handlers
  61         def on_hide(self, win, *args):
  62                 self.settings.hide()
  63                 return True
  64
  65         def on_show(self, act):
  66                 self.settings.show()
  67                 return True
  68
  69         def on_serial(self, act):
  70                 if act.get_active():
  71                         self.status(self.device.connect())
  72                 else:
  73                         self.device.disconnect()
  74                 return True
  75
  76         def on_xively(self, act):
  77                 if act.get_active():
  78                         self.status(self.logger.connect())
  79                 else:
  80                         self.logger.disconnect()
  81                 return True
  82
  83         def on_flush(self, act):
  84                 self.logger.flush()
  85
  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
  93                         self.config.save()
  94
  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
 102                         self.config.save()
 103
 104         def on_config(self, obj):
 105                 name = Gtk.Buildable.get_name(obj)
 106                 if isinstance(obj, Gtk.SpinButton):
 107                         value = obj.get_value()
 108                 else:
 109                         value = obj.get_text()
 110                 if getattr(self.config, name) != value:
 111                         setattr(self.config, name, value)
 112                         self.config.save()
 113
 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]
 119
 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
 124                         if i==0:
 125                                 cairo.move_to(x, y)
 126                         else:
 127                                 cairo.line_to(x, y)
 128                 cairo.stroke()
 129
 130         def on_expose(self, obj, event):
 131                 cairo = obj.window.cairo_create()
 132                 self.draw(obj, cairo)
 133
 134         def on_key(self, win, ev):
 135                 if ev.string == 'q':
 136                         Gtk.main_quit();
 137
 138         def on_destroy(self, win):
 139                 Gtk.main_quit()
 140
 141         # Methods
 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)
 149
 150                 # Serial settings
 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)
 156
 157                 # Xively settings
 158                 set_txt('feedid',   self.config.feedid)
 159                 set_txt('apikey',   self.config.apikey)
 160                 set_val('maxrate',  self.config.maxrate)
 161
 162                 # Sensor settings
 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])
 166
 167         def update(self, state):
 168                 def setval(objs, vals, lbls):
 169                         found = 0
 170                         for i in range(0,len(objs)):
 171                                 if vals[i] == None:
 172                                         continue
 173                                 found = 1
 174                                 if lbls and lbls[i]:
 175                                         objs[i].set_text('%s: %f' % (lbls[i], vals[i]))
 176                                 else:
 177                                         objs[i].set_text('%f' % vals[i])
 178                         return found
 179
 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, [])
 186
 187                 # Save history
 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)
 194                         item = data[axis]
 195                         if item != None:
 196                                 self.history = self.history[1:] + [item]
 197
 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))
 205
 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)
 213
 214                 # Update status icons
 215                 check(self.device.running(), "serial")
 216                 check(self.logger.running(), "xively")
 217
 218                 # Read data and update data display
 219                 for item in self.device.process():
 220                         self.update(item)
 221                         self.logger.append(item)
 222                         #self.debug(item)
 223
 224                 # Refresh window
 225                 self.window.queue_draw()
 226
 227                 return True
 228
 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
 235                 return True
 236
 237         def run(self):
 238                 self.load_config()
 239                 self.window.show()
 240                 GLib.timeout_add(1000/Visual.FRAMES_PER_SEC, self.read_timer)
 241                 GLib.timeout_add(1000/Visual.RATES_PER_SEC,  self.rate_timer)
 242                 Gtk.main()
 243
 244         # Private methods
 245         def status(self, msg):
 246                 status = self.builder.get_object("status")
 247                 status.pop(context_id=0)
 248                 if not msg:
 249                         return
 250                 status.push(text=('Error: ' + msg), context_id=0)