X-Git-Url: https://scm.cri.ensmp.fr/git/minwii.git/blobdiff_plain/f04df7410041506425a042a6397385de2e1abf53..7ce928e114b798a45a94f933c8eb69b96009f989:/src/pywiiuse/pygame_wiimouse.py?ds=sidebyside diff --git a/src/pywiiuse/pygame_wiimouse.py b/src/pywiiuse/pygame_wiimouse.py index 55a9c99..f77856c 100755 --- a/src/pywiiuse/pygame_wiimouse.py +++ b/src/pywiiuse/pygame_wiimouse.py @@ -127,18 +127,26 @@ class wiimote_thread(Thread): 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) + 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=wiiuse.IR_ABOVE) + wm.enable_ir(1, vres = screenResolution, position=position) def get_count():