X-Git-Url: https://scm.cri.ensmp.fr/git/minwii.git/blobdiff_plain/f7094076a9f447c33341612d1a530141d35d1a3a..f5301e6f8629f30e872e20133844975404691a99:/src/pywiiuse/pygame_wiimouse.py diff --git a/src/pywiiuse/pygame_wiimouse.py b/src/pywiiuse/pygame_wiimouse.py index 3f31800..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,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):