#!/usr/bin/env python

################################################################################

# #

# This program is free software: you can redistribute it and/or modify #

# it under the terms of the GNU General Public License as published by #

# the Free Software Foundation, either version 3 of the License, or #

# (at your option) any later version. #

# #

# This program is distributed in the hope that it will be useful, #

# but WITHOUT ANY WARRANTY; without even the implied warranty of #

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #

# GNU General Public License for more details. #

# #

# You should have received a copy of the GNU General Public License #

# along with this program. If not, see <http://www.gnu.org/licenses/>. #

# #

################################################################################

import cwiid

## Configuration

wiimote_hwaddr = '' # Use your address to speed up the connection proccess

#wiimote_hwaddr = '00:19:1D:5D:5D:DC'

last_point = (0,0)

btA = 0

btB = 0

def cback(messages):

'''Wiimote callback managing method

Recieves a message list, each element is different, see the libcwiid docs'''

global btA, btB, last_point

#print "wiimote callback"

for msg in messages:

if msg[0] == cwiid.MESG_IR:

# msg is of the form (cwiid.MESG_IR, (((x, y), size) or None * 4))

for p in msg[1]:

if p:

pos = p['pos'][0], p['pos'][1] # point is mirrored

#s = max(p['size'], 1)

last_point = tuple(pos)

#print "last_point",last_point

elif msg[0] == cwiid.MESG_BTN:

# msg is of the form (cwiid.MESG_BTN, cwiid.BTN_*)

if msg[1] & cwiid.BTN_A:

btA = 1

#print "btA = 1"

else:

btA = 0

#print "btA = 0"

if msg[1] & cwiid.BTN_B:

btB = 1

#print "btB = 1"

else:

btB = 0

#print "btB = 0"

#elif msg[0] == cwiid.MESG_STATUS:

# # msg is of the form (cwiid.MESG_BTN, { 'status' : value, ... })

# print msg[1]

try:

#if False:

wm = cwiid.Wiimote(wiimote_hwaddr)

if wm is not None:

# 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 = cback

# quick check on the wiimote

print "Got Wiimote!"

st = wm.state

for e in st:

print str(e).ljust(8), ">", st[e]

except:

#else:

print "Error with wiimote " + str(wiimote_hwaddr)

def _runtime_cleanup():

print "_runtime_cleanup() Called"

runing = 0

if wm is not None:

wm.close()