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.selectedWiimoteIndex = 0
  26         self.timeout = timeout
  27         self.setDaemon(1)
  28         self._paused = False
  29         self.start()
  30         self.startup.get(True) # wait for the thread to get started and acquire the motes
  31
  32     def run(self):
  33         '''This runs in a separate thread'''
  34         global wiiuse
  35         import PyWiiUse as wiiuse # import here to avoid thread problems on windows
  36         self.wiimotes = wiiuse.init(self.nmotes)
  37         found = wiiuse.find(self.wiimotes, self.nmotes, self.timeout)
  38         self.actual_nmotes = wiiuse.connect(self.wiimotes, self.nmotes)
  39
  40
  41         for i in range(self.nmotes):
  42             wiiuse.set_leds(self.wiimotes[i], wiiuse.LED[i])
  43
  44         self.go = self.actual_nmotes != 0
  45
  46         self.startup.put(self.go)
  47
  48         while self.go:
  49             if self._paused : continue
  50             try :
  51                 if wiiuse.poll(self.wiimotes, self.nmotes) :
  52                     m = self.wiimotes[self.selectedWiimoteIndex]
  53                     if m[0].event == wiiuse.EVENT:
  54                         self.event_cb(m)
  55             except :
  56                 pass
  57
  58             while True:
  59                 try:
  60                     func, args = self.queue.get_nowait()
  61                 except Empty:
  62                     break
  63                 func(*args)
  64
  65     def pause(self) :
  66         self._paused  = True
  67
  68     def resume(self) :
  69         self._paused = False
  70
  71     def selectWiimote(self, wiimoteIndex) :
  72         self.selectedWiimoteIndex = wiimoteIndex
  73
  74     def do(self, func, *args):
  75         '''Run the function in the thread handling the wiimote'''
  76         self.queue.put((func, args))
  77
  78     def event_cb(self, wmp):
  79         '''Called when the library has some data for the user.'''
  80         wm = wmp[0]
  81         pos = (wm.ir.x, wm.ir.y)
  82         pygame.mouse.set_pos(pos)
  83
  84         eventType = None
  85
  86         if wm.btns and \
  87             wiiuse.is_just_pressed(wm, wiiuse.button['B']) :
  88                 event = pygame.event.Event(pygame.MOUSEBUTTONDOWN,
  89                                            pos = pos,
  90                                            button = 1)
  91                 pygame.event.post(event)
  92
  93         if wm.btns_released and \
  94             wiiuse.is_released(wm, wiiuse.button['B']):
  95                 event = pygame.event.Event(pygame.MOUSEBUTTONUP,
  96                                            pos = pos,
  97                                            button = 1)
  98                 pygame.event.post(event)
  99
 100
 101
 102     def control_cb(self, wmp, attachment, speaker, ir, led, battery):
 103         '''Could check the battery level and such here'''
 104         pygame.event.post(pygame.event.Event(WIIMOTE_STATUS,
 105                                              attachment=attachment,
 106                                              speaker=speaker,
 107                                              ir=ir,
 108                                              led=[led[i] for i in range(4)],
 109                                              battery=battery,
 110                                              id=wmp[0].unid))
 111
 112     def disconnect_cb(self, wmp):
 113         '''What should we do here?'''
 114         pygame.event.post(pygame.event.Event(WIIMOTE_DISCONNECT,
 115                                              id=wmp[0].unid))
 116
 117     def quit(self):
 118         '''Go away.'''
 119         for i in range(self.nmotes):
 120             wiiuse.set_leds(self.wiimotes[i], 0)
 121             wiiuse.disconnect(self.wiimotes[i])
 122         self.go = False
 123
 124     def get_count(self):
 125         return self.actual_nmotes
 126
 127
 128 WT = None
 129
 130 def init(nmotes, timeout, screenResolution=(660, 370), position='ABOVE'):
 131     '''Initialize the module.'''
 132     global WT
 133     if WT:
 134         return
 135     WT = wiimote_thread(nmotes, timeout)
 136
 137     if position == 'ABOVE' :
 138         position = wiiuse.IR_ABOVE
 139     elif position == 'BELOW' :
 140         position = wiiuse.IR_BELOW
 141     else :
 142         position = wiiuse.IR_ABOVE
 143
 144
 145     nmotes = get_count()
 146     for i in range(nmotes) :
 147         wm = Wiimote(i) # access the wiimote object
 148         wm.enable_accels(0) # turn off acceleration reporting
 149         wm.enable_ir(1, vres = screenResolution, position=position)
 150
 151
 152 def get_count():
 153     '''How many Wiimotes were found?'''
 154     return WT.get_count()
 155
 156 def quit():
 157     '''Gracefully shutdown the connection and turn off the wiimote leds'''
 158     WT.quit()
 159     WT.join()
 160
 161 class wiimote(object):
 162     '''Object representing a Wiimote'''
 163     def __init__(self, n):
 164         self.wm = WT.wiimotes[n]
 165
 166     def enable_leds(self, m):
 167         '''Control leds. The lower 4 bits map to the 4 leds'''
 168         WT.do(wiiuse.set_leds, self.wm, sum([wiiuse.LED[i] for i in range(4) if m & (1<<i)]))
 169
 170     def enable_rumble(self, on):
 171         '''Control rumble'''
 172         WT.do(wiiuse.rumble, self.wm, on)
 173
 174     def enable_accels(self, on):
 175         '''Control reporting of accelerometer data.'''
 176         WT.do(wiiuse.motion_sensing, self.wm, on)
 177
 178     def enable_ir(self, on, vres=None, position=None, aspect=None):
 179         '''Control reporting IR data.'''
 180         WT.do(wiiuse.set_ir, self.wm, on)
 181         if vres is not None:
 182             WT.do(wiiuse.set_ir_vres, self.wm, *vres)
 183         if position is not None:
 184             WT.do(wiiuse.set_ir_position, self.wm, position)
 185         if aspect is not None:
 186             WT.do(wiiuse.set_aspect_ratio, self.wm, aspect)
 187
 188     def set_flags(self, smoothing=None, continuous=None, threshold=None):
 189         '''Set flags SMOOTHING, CONTINUOUS, ORIENT_THRESH'''
 190         enable = disable = 0
 191         if smoothing is not None:
 192             if smoothing:
 193                 enable |= wiiuse.SMOOTHING
 194             else:
 195                 disable |= wiiuse.SMOOTHING
 196         if continuous is not None:
 197             if continuous:
 198                 enable |= wiiuse.CONTINUOUS
 199             else:
 200                 disable |= wiiuse.CONTINUOUS
 201         if threshold is not None:
 202             if threshold:
 203                 enable |= wiiuse.ORIENT_THRESH
 204             else:
 205                 disable |= wiiuse.ORIENT_THRESH
 206         WT.do(wiiuse.set_flags, self.wm, enable, disable)
 207
 208     def set_orient_thresh(self, thresh):
 209         '''Set orientation threshold'''
 210         WT.do(wiiuse.set_orient_threshold, self.wm, thresh)
 211
 212     def status(self):
 213         '''Trigger a status callback.'''
 214         WT.do(wiiuse.status, self.wm)
 215
 216     def disconnect(self):
 217         '''Disconnect this Wiimote'''
 218         WT.do(wiiuse.disconnect(self.wm))
 219
 220 def Wiimote(n):
 221     '''Get the object for the nth Wiimote'''
 222     return wiimote(n)
 223