summaryrefslogtreecommitdiff
path: root/content/resources/ir_control_0.py
diff options
context:
space:
mode:
authorSébastien Dailly <sebastien@chimrod.com>2014-10-04 16:59:06 +0200
committerSébastien Dailly <sebastien@chimrod.com>2014-10-04 16:59:06 +0200
commit2c3cf5179f16e8303ece57e3dbb191a46f3dcb85 (patch)
tree3d4ea31cd91388149ba43d3c8698c90532a51a59 /content/resources/ir_control_0.py
parent19e71ce6cd7ca7540e0e004d180998e051c57402 (diff)
Moved resources to the blog site
Diffstat (limited to 'content/resources/ir_control_0.py')
-rw-r--r--content/resources/ir_control_0.py118
1 files changed, 118 insertions, 0 deletions
diff --git a/content/resources/ir_control_0.py b/content/resources/ir_control_0.py
new file mode 100644
index 0000000..dfce76a
--- /dev/null
+++ b/content/resources/ir_control_0.py
@@ -0,0 +1,118 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import cwiid
+
+## Configuration
+#wiimote_hwaddr = '' # Use your address to speed up the connection proccess
+wiimote_hwaddr = '00:1F:C5:5A:8C:69'
+
+def connect(wm_addr = ''):
+ '''Connects and syncs with a wiimote
+ wm_addr - Is a string representing the hwaddr of the wiimote we will try to
+ connect, or none to try to connect with any discoverable device
+ for example "00:19:1D:5D:5D:DC"'''
+
+ pygame.display.flip() # called now because cwiid.Wiimote is a blocking call
+
+ # This could be done in a thread to allow pygame to draw while searching
+ # but this is only a test
+ try:
+ return cwiid.Wiimote(wm_addr)
+ except:
+ print "Error conectando con wiimote " + str(wm_addr)
+
+
+def wmplugin_init(id, wiimote_arg):
+ #global wiimote
+
+ #wiimote = wiimote_arg
+ wmplugin.set_rpt_mode(id, cwiid.RPT_IR | cwiid.RPT_BTN)
+ return
+
+def wmplugin_info():
+ return [], \
+ [("X", wmplugin.REL | wmplugin.ABS, 1024, 0, 0, 0), \
+ ("Y", wmplugin.REL | wmplugin.ABS, 768, 0, 0, 0)], \
+ []
+
+def wmplugin_exec(messages, buton=[0]):
+ '''Wiimote callback managing method
+ Recieves a message list, each element is different, see the libcwiid docs'''
+ x = y = 0
+
+ for msg in messages:
+
+ if msg[0] == cwiid.MESG_BTN:
+ # msg is of the form (cwiid.MESG_BTN, cwiid.BTN_*)
+ buton[0] = msg[1]
+
+ if msg[0] == cwiid.MESG_IR:
+ if screen:
+ # red dot for the sources
+ [screen.fill((255, 0, 0), (point, (4, 4))) \
+ for point in ir_sensor.get_positions_points(msg)]
+
+ if buton[0] & cwiid.BTN_B == 0 :
+ x, y = (0, 0)
+
+ return [], (x, y)
+
+if __name__ == '__main__':
+ import pygame
+ from pygame.locals import *
+ runing = 0
+ def handle_events():
+ '''Typical event handling via pygame'''
+ for event in pygame.event.get():
+ if event.type == QUIT:
+ return 0
+ elif event.type == KEYUP:
+ if event.key == K_ESCAPE:
+ return 0
+ elif event.key == K_SPACE:
+ canvas.fill((0, 0, 0), ((0, 0), canvas.get_size()))
+ return 1
+
+ pygame.init()
+ pygame.display.set_caption('Wiimote IR test')
+ window = pygame.display.set_mode((cwiid.IR_X_MAX, cwiid.IR_Y_MAX), DOUBLEBUF)
+ screen = pygame.display.get_surface()
+ canvas = pygame.Surface(screen.get_size()) # persistent drawing here
+ canvas = canvas.convert()
+
+ wm = None # our wiimote
+ clock = pygame.time.Clock()
+ runing = 1
+ while(runing):
+ clock.tick(100)
+ runing = handle_events()
+ if not wm:
+ wm = connect(wiimote_hwaddr)
+ if not wm:
+ continue
+ # each message will contain info about ir and buttons
+ wm.rpt_mode = cwiid.RPT_IR | cwiid.RPT_BTN # | cwiid.RPT_STATUS
+ # tell cwiid to use the callback interface and allways send button events
+ wm.enable(cwiid.FLAG_MESG_IFC
+ #| cwiid.FLAG_NONBLOCK
+ | cwiid.FLAG_REPEAT_BTN)
+
+ # specify wich function will manage messages AFTER the other settings
+ wm.mesg_callback = wmplugin_exec
+
+ # quick check on the wiimote
+ print "Got Wiimote!"
+ st = wm.state
+ for e in st:
+ print str(e).ljust(8), ">", st[e]
+
+ screen.blit(canvas, (0, 0))
+ pygame.display.flip()
+
+ if wm:
+ wm.close()
+else:
+ #This code is used as a plugin for cwiid, we import the wminput system
+ import wmplugin
+ screen = None