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         def __init__(self, config, device, logger):
  20                 def get_objects(names):
  21                         return list(map(lambda x:
  22                                 self.builder.get_object(x), names))
  23
  24                 self.config   = config
  25                 self.device   = device
  26                 self.logger   = logger
  27                 self.builder  = Gtk.Builder()
  28
  29                 self.builder.add_from_file('visual.ui')
  30                 self.builder.connect_signals(self)
  31                 self.window   = self.builder.get_object("window")
  32                 self.settings = self.builder.get_object("settings")
  33
  34                 self.accs     = get_objects(['accx', 'accy', 'accz'])
  35                 self.mags     = get_objects(['magx', 'magy', 'magz'])
  36                 self.lgts     = get_objects(['lgt'])
  37                 self.tchs     = get_objects(['tchp', 'tchd'])
  38                 self.a2ds     = get_objects(['a2d0', 'a2d1', 'a2d2', 'a2d3', 'a2d4', 'a2d5'])
  39
  40         # Signal handlers
  41         def on_hide(self, win, *args):
  42                 self.settings.hide()
  43                 return True
  44
  45         def on_show(self, act):
  46                 self.settings.show()
  47                 return True
  48
  49         def on_serial(self, act):
  50                 if act.get_active():
  51                         self.status(self.device.connect())
  52                 else:
  53                         self.status(self.device.disconnect())
  54                 return True
  55
  56         def on_xively(self, act):
  57                 if act.get_active():
  58                         self.status(self.logger.connect())
  59                 else:
  60                         self.status(self.logger.disconnect())
  61                 return True
  62
  63         def on_flush(self, act):
  64                 self.logger.flush()
  65
  66         def on_enable(self, obj):
  67                 name  = Gtk.Buildable.get_name(obj)
  68                 name  = name.replace('_btn', '')
  69                 state = obj.get_active()
  70                 if self.config.enable[name] != state:
  71                         self.device.set_enable(name, state)
  72                         self.config.enable[name] = state
  73                         self.config.save()
  74
  75         def on_rate(self, obj):
  76                 name  = Gtk.Buildable.get_name(obj)
  77                 name  = name.replace('_spin', '')
  78                 value = obj.get_value()
  79                 if self.config.rate[name] != value:
  80                         self.device.set_rate(name, value)
  81                         self.config.rate[name] = value
  82                         self.config.save()
  83
  84         def on_config(self, obj):
  85                 name = Gtk.Buildable.get_name(obj)
  86                 if isinstance(obj, Gtk.SpinButton):
  87                         value = obj.get_value()
  88                 else:
  89                         value = obj.get_text()
  90                 if getattr(self.config, name) != value:
  91                         setattr(self.config, name, value)
  92                         self.config.save()
  93
  94         def on_key(self, win, ev):
  95                 if ev.string == 'q':
  96                         Gtk.main_quit();
  97
  98         def on_destroy(self, win):
  99                 Gtk.main_quit()
 100
 101         # Methods
 102         def load_config(self):
 103                 def set_val(name, val):
 104                         self.builder.get_object(name).set_value(val)
 105                 def set_txt(name, val):
 106                         self.builder.get_object(name).set_text(val)
 107                 def set_act(name, val):
 108                         self.builder.get_object(name).set_active(val)
 109
 110                 # Serial settings
 111                 set_txt('device',   self.config.device)
 112                 set_val('baudrate', self.config.baudrate)
 113                 set_txt('parity',   self.config.parity)
 114                 set_val('databits', self.config.databits)
 115                 set_val('stopbits', self.config.stopbits)
 116
 117                 # Xively settings
 118                 set_txt('feedid',   self.config.feedid)
 119                 set_txt('apikey',   self.config.apikey)
 120                 set_val('maxrate',  self.config.maxrate)
 121
 122                 # Sensor settings
 123                 for key in list(self.config.enable.keys()):
 124                         set_act(key+'_btn',  self.config.enable[key])
 125                         set_val(key+'_spin', self.config.rate[key])
 126
 127
 128         def update(self, state):
 129                 def setval(objs, vals, lbls):
 130                         for i in range(0,len(objs)):
 131                                 if vals[i] == None:
 132                                         continue
 133                                 if lbls and lbls[i]:
 134                                         objs[i].set_text('%s: %f' % (lbls[i], vals[i]))
 135                                 else:
 136                                         objs[i].set_text('%f' % vals[i])
 137
 138                 setval(self.accs,  state.acc, ['X', 'Y', 'Z'])
 139                 setval(self.mags,  state.mag, ['X', 'Y', 'Z'])
 140                 setval(self.lgts,  state.lgt, [])
 141                 setval(self.tchs,  state.tch, ['P', 'D'])
 142                 setval(self.a2ds,  state.a2d, [])
 143
 144         def debug(self, state):
 145                 print('update: ' + str(state))
 146                 print('\tacc - ' + str(state.acc))
 147                 print('\tmag - ' + str(state.mag))
 148                 print('\tlgt - ' + str(state.lgt))
 149                 print('\ttch - ' + str(state.touch))
 150                 print('\ta2d - ' + str(state.a2d))
 151
 152         def timer(self):
 153                 def check(state, name):
 154                         btn = self.builder.get_object(name + '_btn')
 155                         con = self.builder.get_object(name + '_con')
 156                         img = Gtk.STOCK_YES if state else Gtk.STOCK_NO
 157                         btn.set_active(state)
 158                         con.set_from_stock(img, ICON_SIZE_BUTTON)
 159
 160                 check(self.device.running(), "serial")
 161                 check(self.logger.running(), "xively")
 162
 163                 for item in self.device.process():
 164                         self.update(item)
 165                         self.logger.append(item)
 166                         #self.debug(item)
 167
 168                 return True
 169
 170         def run(self):
 171                 self.load_config()
 172                 self.window.show()
 173                 GLib.timeout_add(1000/60, self.timer)
 174                 Gtk.main()
 175
 176         # Private methods
 177         def status(self, msg):
 178                 if not msg:
 179                         return
 180                 status = self.builder.get_object("status")
 181                 status.push(text=('Error: ' + msg), context_id=0)