X-Git-Url: https://scm.cri.ensmp.fr/git/minwii.git/blobdiff_plain/3b16ad256b097d7e910682bf7708bb4f2f135ebe..128d1283372aa86d58d346569a08a97d64f5acf6:/src/pywiiuse/pygame_wiimouse.py?ds=sidebyside diff --git a/src/pywiiuse/pygame_wiimouse.py b/src/pywiiuse/pygame_wiimouse.py index c57cad8..d26ce12 100755 --- a/src/pywiiuse/pygame_wiimouse.py +++ b/src/pywiiuse/pygame_wiimouse.py @@ -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,9 +44,19 @@ class wiimote_thread(Thread): found = wiiuse.find(self.wiimotes, self.nmotes, self.timeout) self.actual_nmotes = wiiuse.connect(self.wiimotes, self.nmotes) + + 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]) - for i in range(self.nmotes): - 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 @@ -48,19 +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 while True: try: func, args = self.queue.get_nowait() except Empty: break - print 'do:', func.__name__, args func(*args) def pause(self) : @@ -68,15 +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] - 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''' @@ -95,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''' @@ -163,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):