X-Git-Url: https://scm.cri.ensmp.fr/git/minwii.git/blobdiff_plain/30e36fe2e442924342ce39d3217002ff0d222f35..ae79067c99dcb4473df80ddd5eb24395d90ab6d3:/src/pywiiuse/pygame_wiimouse.py diff --git a/src/pywiiuse/pygame_wiimouse.py b/src/pywiiuse/pygame_wiimouse.py index 8a4f6e0..d26ce12 100755 --- a/src/pywiiuse/pygame_wiimouse.py +++ b/src/pywiiuse/pygame_wiimouse.py @@ -17,17 +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.selectedWiimoteIndex = 0 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''' @@ -37,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) - - for i in range(self.nmotes): - wiiuse.set_leds(self.wiimotes[i], wiiuse.LED[i]) + + 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 @@ -49,18 +66,18 @@ class wiimote_thread(Thread): if self._paused : continue try : if wiiuse.poll(self.wiimotes, self.nmotes) : - m = self.wiimotes[self.selectedWiimoteIndex] - if m[0].event == wiiuse.EVENT: - self.event_cb(m) - except : - pass + for i in range(self.nmotes) : + m = self.wiimotes[i] + if m[0].event == wiiuse.EVENT: + 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) : @@ -74,31 +91,7 @@ class wiimote_thread(Thread): 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] - 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) - - + self.queue.put((func, args)) def control_cb(self, wmp, attachment, speaker, ir, led, battery): '''Could check the battery level and such here''' @@ -117,30 +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, screenResolution=(660, 370)): +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=wiiuse.IR_BELOW) + 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''' @@ -192,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):