src/pywiiuse/pygame_wiimouse.py

   1 # -*- coding: utf-8 -*-
   2 '''
   3 wiimote -> mouse interface
   4 $Id$
   5 $URL$
   6 '''
   7
   8 import pygame
   9 from threading import Thread
  10 from Queue import Queue, Empty
  11 import time
  12
  13 # events to use. Is there a way to get ones known to be unused?
  14
  15
  16 wiiuse = None # import within the thread, why do I have to do this?
  17
  18 class wiimote_thread(Thread):
  19     '''Manage the wiiuse interface'''
  20     def __init__(self, nmotes=1, timeout=5):
  21         Thread.__init__(self, name='wiimote')
  22         self.queue = Queue()
  23         self.startup = Queue()
  24         self.nmotes = nmotes
  25         self.timeout = timeout
  26         self.setDaemon(1)
  27         self._paused = False
  28         self.start()
  29         self.startup.get(True) # wait for the thread to get started and acquire the motes
  30
  31     def run(self):
  32         '''This runs in a separate thread'''
  33         global wiiuse
  34         import PyWiiUse as wiiuse # import here to avoid thread problems on windows
  35         self.wiimotes = wiiuse.init(self.nmotes)
  36         found = wiiuse.find(self.wiimotes, self.nmotes, self.timeout)
  37         self.actual_nmotes = wiiuse.connect(self.wiimotes, self.nmotes)
  38
  39
  40         for i in range(self.nmotes):
  41             wiiuse.set_leds(self.wiimotes[i], wiiuse.LED[i])
  42
  43         self.go = self.actual_nmotes != 0
  44
  45         self.startup.put(self.go)
  46
  47         while self.go:
  48             if self._paused : continue
  49             try :
  50                 if wiiuse.poll(self.wiimotes, self.nmotes) :
  51                     for i in range(self.nmotes):
  52                         m = self.wiimotes[i]
  53                         if m[0].event == wiiuse.EVENT:
  54                             self.event_cb(m)
  55             except :
  56                 pass
  57
  58             #try:
  59             #    wiiuse.poll(self.wiimotes, self.nmotes)
  60             #except:
  61             #    pass
  62
  63             # allow executing functions in this thread
  64             while True:
  65                 try:
  66                     func, args = self.queue.get_nowait()
  67                 except Empty:
  68                     break
  69                 print 'do:', func.__name__, args
  70                 func(*args)
  71
  72     def pause(self) :
  73         self._paused  = True
  74
  75     def resume(self) :
  76         self._paused = False
  77
  78     def do(self, func, *args):
  79         '''Run the function in the thread handling the wiimote'''
  80         self.queue.put((func, args))
  81
  82     def event_cb(self, wmp):
  83         '''Called when the library has some data for the user.'''
  84         wm = wmp[0]
  85         pygame.mouse.set_pos((wm.ir.x, wm.ir.y))
  86
  87     def control_cb(self, wmp, attachment, speaker, ir, led, battery):
  88         '''Could check the battery level and such here'''
  89         pygame.event.post(pygame.event.Event(WIIMOTE_STATUS,
  90                                              attachment=attachment,
  91                                              speaker=speaker,
  92                                              ir=ir,
  93                                              led=[led[i] for i in range(4)],
  94                                              battery=battery,
  95                                              id=wmp[0].unid))
  96
  97     def disconnect_cb(self, wmp):
  98         '''What should we do here?'''
  99         pygame.event.post(pygame.event.Event(WIIMOTE_DISCONNECT,
 100                                              id=wmp[0].unid))
 101
 102     def quit(self):
 103         '''Go away.'''
 104         for i in range(self.nmotes):
 105             wiiuse.set_leds(self.wiimotes[i], 0)
 106             wiiuse.disconnect(self.wiimotes[i])
 107         self.go = False
 108
 109 WT = None
 110
 111 def init(nmotes, timeout):
 112     '''Initialize the module.'''
 113     global WT
 114     if WT:
 115         return
 116     WT = wiimote_thread(nmotes, timeout)
 117
 118 def get_count():
 119     '''How many Wiimotes were found?'''
 120     return WT.actual_nmotes
 121
 122 def quit():
 123     '''Gracefully shutdown the connection and turn off the wiimote leds'''
 124     WT.quit()
 125     WT.join()
 126
 127 class wiimote(object):
 128     '''Object representing a Wiimote'''
 129     def __init__(self, n):
 130         self.wm = WT.wiimotes[n]
 131
 132     def enable_leds(self, m):
 133         '''Control leds. The lower 4 bits map to the 4 leds'''
 134         WT.do(wiiuse.set_leds, self.wm, sum([wiiuse.LED[i] for i in range(4) if m & (1<<i)]))
 135
 136     def enable_rumble(self, on):
 137         '''Control rumble'''
 138         WT.do(wiiuse.rumble, self.wm, on)
 139
 140     def enable_accels(self, on):
 141         '''Control reporting of accelerometer data.'''
 142         WT.do(wiiuse.motion_sensing, self.wm, on)
 143
 144     def enable_ir(self, on, vres=None, position=None, aspect=None):
 145         '''Control reporting IR data.'''
 146         WT.do(wiiuse.set_ir, self.wm, on)
 147         if vres is not None:
 148             WT.do(wiiuse.set_ir_vres, self.wm, *vres)
 149         if position is not None:
 150             WT.do(wiiuse.set_ir_position, self.wm, position)
 151         if aspect is not None:
 152             WT.do(wiiuse.set_aspect_ratio, self.wm, aspect)
 153
 154     def set_flags(self, smoothing=None, continuous=None, threshold=None):
 155         '''Set flags SMOOTHING, CONTINUOUS, ORIENT_THRESH'''
 156         enable = disable = 0
 157         if smoothing is not None:
 158             if smoothing:
 159                 enable |= wiiuse.SMOOTHING
 160             else:
 161                 disable |= wiiuse.SMOOTHING
 162         if continuous is not None:
 163             if continuous:
 164                 enable |= wiiuse.CONTINUOUS
 165             else:
 166                 disable |= wiiuse.CONTINUOUS
 167         if threshold is not None:
 168             if threshold:
 169                 enable |= wiiuse.ORIENT_THRESH
 170             else:
 171                 disable |= wiiuse.ORIENT_THRESH
 172         print enable, disable
 173         WT.do(wiiuse.set_flags, self.wm, enable, disable)
 174
 175     def set_orient_thresh(self, thresh):
 176         '''Set orientation threshold'''
 177         WT.do(wiiuse.set_orient_threshold, self.wm, thresh)
 178
 179     def status(self):
 180         '''Trigger a status callback.'''
 181         WT.do(wiiuse.status, self.wm)
 182
 183     def disconnect(self):
 184         '''Disconnect this Wiimote'''
 185         WT.do(wiiuse.disconnect(self.wm))
 186
 187 def Wiimote(n):
 188     '''Get the object for the nth Wiimote'''
 189     return wiimote(n)
 190