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