X-Git-Url: https://scm.cri.ensmp.fr/git/minwii.git/blobdiff_plain/3b16ad256b097d7e910682bf7708bb4f2f135ebe..1ccd4f7141e49707ec82ef1fc39904951250d05c:/src/pywiiuse/pygame_wiimouse.py?ds=sidebyside diff --git a/src/pywiiuse/pygame_wiimouse.py b/src/pywiiuse/pygame_wiimouse.py index c57cad8..3328eff 100755 --- a/src/pywiiuse/pygame_wiimouse.py +++ b/src/pywiiuse/pygame_wiimouse.py @@ -22,11 +22,16 @@ class wiimote_thread(Thread): self.queue = Queue() self.startup = Queue() self.nmotes = nmotes + self.selectedWiimoteIndex = 0 self.timeout = timeout 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''' @@ -48,19 +53,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 while True: try: func, args = self.queue.get_nowait() except Empty: break - print 'do:', func.__name__, args func(*args) def pause(self) : @@ -68,15 +72,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] - pygame.mouse.set_pos((wm.ir.x, wm.ir.y)) + self.queue.put((func, args)) def control_cb(self, wmp, attachment, speaker, ir, led, battery): '''Could check the battery level and such here''' @@ -100,18 +102,104 @@ class wiimote_thread(Thread): 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) + + if position == 'ABOVE' : + position = wiiuse.IR_ABOVE + elif position == 'BELOW' : + position = wiiuse.IR_BELOW + else : + position = wiiuse.IR_ABOVE + + + 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''' @@ -163,7 +251,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):