fix MRO.
[minwii.git] / src / pywiiuse / pygame_wiimouse.py
index 3f31800..d26ce12 100755 (executable)
@@ -17,16 +17,24 @@ wiiuse = None # import within the thread, why do I have to do this?
 
 class wiimote_thread(Thread):
     '''Manage the wiiuse interface'''
-    def __init__(self, nmotes=1, timeout=5):
+    
+    def __init__(self, nmotes=1, timeout=5, screenResolution=(660, 370), position='ABOVE'):
         Thread.__init__(self, name='wiimote')
         self.queue = Queue()
         self.startup = Queue()
         self.nmotes = nmotes
         self.timeout = timeout
+        self.screenResolution = screenResolution
+        self.position = position
+        self.selectedWiimoteIndex = 0
         self.setDaemon(1)
         self._paused = False
         self.start()
         self.startup.get(True) # wait for the thread to get started and acquire the motes
+        self.eventCallBack = _default_event_cb
+    
+    def setEventCallBack(self, func) :
+        self.eventCallBack = func
 
     def run(self):
         '''This runs in a separate thread'''
@@ -36,11 +44,19 @@ class wiimote_thread(Thread):
         found = wiiuse.find(self.wiimotes, self.nmotes, self.timeout)
         self.actual_nmotes = wiiuse.connect(self.wiimotes, self.nmotes)
         
-        self.previousPositions = {}
-
-        for i in range(self.nmotes):
-            wiiuse.set_leds(self.wiimotes[i], wiiuse.LED[i])
-            self.previousPositions[self.wiimotes[i][0].unid] = (0, 0)
+        
+        if self.nmotes <= 4 :
+            for i in range(self.nmotes):
+                wiiuse.set_leds(self.wiimotes[i], wiiuse.LED[i])
+        else :
+            for i in range(4):
+                wiiuse.set_leds(self.wiimotes[i], wiiuse.LED[i])
+
+            if self.nmotes == 5 :
+                wiiuse.set_leds(self.wiimotes[4], wiiuse.LED_1 | wiiuse.LED_4)
+            if self.nmotes == 6 :
+                wiiuse.set_leds(self.wiimotes[4], wiiuse.LED_1 | wiiuse.LED_4)
+                wiiuse.set_leds(self.wiimotes[5], wiiuse.LED_1 | wiiuse.LED_2 | wiiuse.LED_3 | wiiuse.LED_4)
 
         self.go = self.actual_nmotes != 0
 
@@ -50,25 +66,18 @@ class wiimote_thread(Thread):
             if self._paused : continue
             try :
                 if wiiuse.poll(self.wiimotes, self.nmotes) :
-                    for i in range(self.nmotes):
+                    for i in range(self.nmotes) :
                         m = self.wiimotes[i]
                         if m[0].event == wiiuse.EVENT:
-                            self.event_cb(m)
-            except :
-                pass
+                            self.eventCallBack(self, i, m)
+            except Exception, e:
+                print e
                 
-            #try:
-            #    wiiuse.poll(self.wiimotes, self.nmotes)
-            #except:
-            #    pass
-
-            # allow executing functions in this thread
             while True:
                 try:
                     func, args = self.queue.get_nowait()
                 except Empty:
                     break
-                print 'do:', func.__name__, args
                 func(*args)
     
     def pause(self) :
@@ -76,85 +85,13 @@ class wiimote_thread(Thread):
     
     def resume(self) :
         self._paused = False
+    
+    def selectWiimote(self, wiimoteIndex) :
+        self.selectedWiimoteIndex = wiimoteIndex
 
     def do(self, func, *args):
         '''Run the function in the thread handling the wiimote'''
-        self.queue.put((func, args))
-
-    def event_cb(self, wmp):
-        '''Called when the library has some data for the user.'''
-        wm = wmp[0]
-#        if wm.btns:
-#            for name,b in wiiuse.button.items():
-#                if wiiuse.is_just_pressed(wm, b):
-#                    pygame.event.post(pygame.event.Event(WIIMOTE_BUTTON_PRESS, button=name,
-#                                                         time=time.time(),
-#                                                         id=wm.unid))
-#
-#        if wm.btns_released:
-#            for name,b in wiiuse.button.items():
-#                if wiiuse.is_released(wm, b):
-#                    pygame.event.post(pygame.event.Event(WIIMOTE_BUTTON_RELEASE, button=name,
-#                                                         time=time.time(),
-#                                                         id=wm.unid))
-#
-#        if True:
-#            pygame.event.post(pygame.event.Event(WIIMOTE_ACCEL, 
-#                                                 orient=(wm.orient.roll, wm.orient.pitch,
-#                                                         wm.orient.yaw),
-#                                                 accel=(wm.gforce.x, wm.gforce.y, wm.gforce.z),
-#                                                 time=time.time(),
-#                                                 id=wm.unid))
-        if True:
-            #dots = [ (wm.ir.dot[i].visible, wm.ir.dot[i].x, wm.ir.dot[i].y) for i in range(4) ]
-            #pygame.event.post(pygame.event.Event(WIIMOTE_IR,
-            #                                     dots=dots,
-            #                                     cursor=(wm.ir.x, wm.ir.y, wm.ir.z),
-            #                                     time=time.time(),
-            #                                     id=wm.unid))
-
-            dots = [ (wm.ir.dot[i].visible, wm.ir.dot[i].x, wm.ir.dot[i].y) for i in range(4) ]
-            cursor=(wm.ir.x, wm.ir.y, wm.ir.z)
-            pos = cursor[:2]
-            previousPos = self.previousPositions[wm.unid]
-            rel = (pos[0] - previousPos[0], pos[1] - previousPos[1])
-            self.previousPositions[wm.unid] = pos
-            
-            evt = pygame.event.Event(pygame.MOUSEMOTION,
-                                     pos = pos,
-                                     rel = rel,
-                                     buttons = [],
-                                     dots=dots,
-                                     cursor=(wm.ir.x, wm.ir.y, wm.ir.z),
-                                     time=time.time(),
-                                     wiimoteid=wm.unid)
-            pygame.event.post(evt)
-
-
-#        if wm.exp.type == wiiuse.EXP_NUNCHUK:
-#            nc = wm.exp.u.nunchuk
-#            
-#            for name,b in wiiuse.nunchuk_button.items():
-#                if wiiuse.is_just_pressed(nc, b):
-#                    pygame.event.post(pygame.event.Event(NUNCHUK_BUTTON_PRESS, button=name,
-#                                                         time=time.time(),
-#                                                         id=wm.unid))
-#                elif wiiuse.is_released(nc, b):
-#                    pygame.event.post(pygame.event.Event(NUNCHUK_BUTTON_RELEASE, button=name,
-#                                                         time=time.time(),
-#                                                         id=wm.unid))
-#
-#            pygame.event.post(pygame.event.Event(NUNCHUK_ACCEL,
-#                                                 orient=(nc.orient.roll, nc.orient.pitch,
-#                                                         nc.orient.yaw),
-#                                                 accel=(nc.gforce.x, nc.gforce.y, nc.gforce.z),
-#                                                 time=time.time(),
-#                                                 id=wm.unid))
-#            pygame.event.post(pygame.event.Event(NUNCHUK_JOY,
-#                                                 angle=nc.js.ang,
-#                                                 mag=nc.js.mag,
-#                                                 time=time.time(),
-#                                                 id=wm.unid))
+        self.queue.put((func, args))                    
 
     def control_cb(self, wmp, attachment, speaker, ir, led, battery):
         '''Could check the battery level and such here'''
@@ -173,23 +110,103 @@ class wiimote_thread(Thread):
 
     def quit(self):
         '''Go away.'''
-        for i in range(self.nmotes):
-            wiiuse.set_leds(self.wiimotes[i], 0)
-            wiiuse.disconnect(self.wiimotes[i])
+        # TODO will crash if self.nmotes > 4
+        # for i in range(self.nmotes):
+        #     wiiuse.set_leds(self.wiimotes[i], 0)
+        #     wiiuse.disconnect(self.wiimotes[i])
         self.go = False
 
+    def get_count(self):
+        return self.actual_nmotes
+
+
+def _default_event_cb(self, id, wmp):
+    ''' default callback that emulate a one button mouse '''
+    if id != self.selectedWiimoteIndex : return
+    wm = wmp[0]
+    pos = (wm.ir.x, wm.ir.y)
+    pygame.mouse.set_pos(pos)
+
+    eventType = None
+
+    if wm.btns and \
+        wiiuse.is_just_pressed(wm, wiiuse.button['B']) :
+            event = pygame.event.Event(pygame.MOUSEBUTTONDOWN,
+                                       pos = pos,
+                                       button = 1)
+            pygame.event.post(event)
+
+    if wm.btns_released and \
+        wiiuse.is_released(wm, wiiuse.button['B']):
+            event = pygame.event.Event(pygame.MOUSEBUTTONUP,
+                                       pos = pos,
+                                       button = 1)
+            pygame.event.post(event)
+
+def _full_mouse_event_cb(self, id, wmp):
+    ''' callback that emulate a 2 buttons mouse with wheel '''
+    if id != self.selectedWiimoteIndex : return
+    wm = wmp[0]
+    pos = (wm.ir.x, wm.ir.y)
+    pygame.mouse.set_pos(pos)
+
+    eventType = None
+
+    if wm.btns :
+        button = 0
+        if wiiuse.is_just_pressed(wm, wiiuse.button['B']) :
+            button = 1
+        elif wiiuse.is_just_pressed(wm, wiiuse.button['A']) :
+            button = 2
+        elif wiiuse.is_just_pressed(wm, wiiuse.button['Up']) :
+            button = 4
+        elif wiiuse.is_just_pressed(wm, wiiuse.button['Down']) :
+            button = 5
+
+        if button :
+            event = pygame.event.Event(pygame.MOUSEBUTTONDOWN,
+                                       pos = pos,
+                                       button = button)
+            pygame.event.post(event)
+
+    if wm.btns_released :
+        button = 0
+        if wiiuse.is_released(wm, wiiuse.button['B']) :
+            button = 1
+        elif wiiuse.is_released(wm, wiiuse.button['A']) :
+            button = 2
+        elif wiiuse.is_released(wm, wiiuse.button['Up']) :
+            button = 4
+        elif wiiuse.is_released(wm, wiiuse.button['Down']) :
+            button = 5
+        
+        if button :
+            event = pygame.event.Event(pygame.MOUSEBUTTONUP,
+                                       pos = pos,
+                                       button = button)
+            pygame.event.post(event)
+
+
 WT = None
 
-def init(nmotes, timeout):
+def init(nmotes, timeout, screenResolution=(660, 370), position='ABOVE'):
     '''Initialize the module.'''
     global WT
     if WT:
         return
-    WT = wiimote_thread(nmotes, timeout)
+    
+    WT = wiimote_thread(nmotes, timeout, screenResolution, position)
+    
+    nmotes = get_count()
+    for i in range(nmotes) :
+        wm = Wiimote(i) # access the wiimote object
+        wm.enable_accels(0) # turn off acceleration reporting
+        wm.enable_ir(1, vres = screenResolution, position=position)
+
 
 def get_count():
     '''How many Wiimotes were found?'''
-    return WT.actual_nmotes
+    return WT.get_count()
 
 def quit():
     '''Gracefully shutdown the connection and turn off the wiimote leds'''
@@ -241,7 +258,6 @@ class wiimote(object):
                 enable |= wiiuse.ORIENT_THRESH
             else:
                 disable |= wiiuse.ORIENT_THRESH
-        print enable, disable
         WT.do(wiiuse.set_flags, self.wm, enable, disable)
 
     def set_orient_thresh(self, thresh):