You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

241 lines
7.6 KiB

'''Wiiboard driver
Nedim Jackman December 2008
No liability held for any use of this software.
More information at http://code.google.com/p/wiiboard-simple/
'''
import bluetooth
import sys
import _thread
import time
import binascii
CONTINUOUS_REPORTING = 0x04
COMMAND_LIGHT = 0x11
COMMAND_REPORTING = 0x12
COMMAND_REQUEST_STATUS = 0x15
COMMAND_REGISTER = 0x16
COMMAND_READ_REGISTER = 0x17
#input is Wii device to host
INPUT_STATUS = 0x20
INPUT_READ_DATA = 0x21
EXTENSION_8BYTES = 0x32
BUTTON_DOWN_MASK = 8
TOP_RIGHT = 0
BOTTOM_RIGHT = 1
TOP_LEFT = 2
BOTTOM_LEFT = 3
BLUETOOTH_NAME = "Nintendo RVL-WBC-01"
class BoardEvent:
def __init__(self, topLeft,topRight,bottomLeft,bottomRight, buttonPressed, buttonReleased):
self.topLeft = topLeft
self.topRight = topRight
self.bottomLeft = bottomLeft
self.bottomRight = bottomRight
self.buttonPressed = buttonPressed
self.buttonReleased = buttonReleased
#convenience value
self.totalWeight = topLeft + topRight + bottomLeft + bottomRight
class Wiiboard:
def __init__(self):
self.datasocket = None
self.controlsocket = None
self.calibration = []
self.calibrationRequested = False
self.LED = False
self.address = None
self.buttonDown = False
for i in range(3):
self.calibration.append([])
for j in range(4):
self.calibration[i].append(10000) #high dummy value so events with it don't register
self.status = "Disconnected"
self.lastEvent = BoardEvent(0,0,0,0,False,False)
self.mass = self.lastEvent
try:
self.datasocket = bluetooth.BluetoothSocket(bluetooth.L2CAP)
self.datasocket.settimeout(2)
self.controlsocket = bluetooth.BluetoothSocket(bluetooth.L2CAP)
self.controlsocket.settimeout(2)
except ValueError:
raise Exception("Error: Bluetooth not found")
def isConnected(self):
if self.status == "Connected":
return True
else:
return False
# Connect to the Wiiboard at bluetooth address <address>
def connect(self, address):
if address == None:
print("Non existant address")
return
self.datasocket.connect((address, 0x13))
self.controlsocket.connect((address, 0x11))
if self.datasocket and self.controlsocket:
print("Connected to Wiiboard at address " + address)
self.status = "Connected"
self.address = address
_thread.start_new_thread(self.receivethread, ())
self.calibrate()
useExt = bytes([COMMAND_REGISTER, 0x04, 0xA4, 0x00, 0x40, 0x00])
self.send(useExt)
self.setReportingType()
else:
print("Could not connect to Wiiboard at address " + address)
# Disconnect from the Wiiboard
def disconnect(self):
if self.status == "Connected":
self.status = "Disconnecting"
while self.status == "Disconnecting":
time.sleep(0.001)
try:
self.datasocket.close()
self.controlsocket.close()
except:
pass
print("WiiBoard disconnected")
# Try to discover a Wiiboard
def discover(self):
#print "Press the red sync button on the board now"
address = None
bluetoothdevices = bluetooth.discover_devices(duration = 6, lookup_names = True)
for addr, name in bluetoothdevices:
if name == BLUETOOTH_NAME:
address = addr
print("Found Wiiboard at address " + address)
#if address == None:
#print "No Wiiboards discovered."
return address
def createBoardEvent(self, data):
buttonBytes = data[0:2]
data = data[2:12]
buttonPressed = False
buttonReleased = False
state = (buttonBytes[0] << 8) | buttonBytes[1]
if state == BUTTON_DOWN_MASK:
buttonPressed = True
if not self.buttonDown:
self.buttonDown = True
if buttonPressed == False:
if self.lastEvent.buttonPressed == True:
buttonReleased = True
self.buttonDown = False
rawTR = (data[0] << 8) + data[1]
rawBR = (data[2] << 8) + data[3]
rawTL = (data[4] << 8) + data[5]
rawBL = (data[6] << 8) + data[7]
topLeft = self.calcMass(rawTL, TOP_LEFT)
topRight = self.calcMass(rawTR, TOP_RIGHT)
bottomLeft = self.calcMass(rawBL, BOTTOM_LEFT)
bottomRight = self.calcMass(rawBR, BOTTOM_RIGHT)
boardEvent = BoardEvent(topLeft,topRight,bottomLeft,bottomRight,buttonPressed,buttonReleased)
return boardEvent
def calcMass(self, raw, pos):
val = 0.0
#calibration[0] is calibration values for 0kg
#calibration[1] is calibration values for 17kg
#calibration[2] is calibration values for 34kg
if raw < self.calibration[0][pos]:
return val
elif raw < self.calibration[1][pos]:
val = 17 * ((raw - self.calibration[0][pos]) / float((self.calibration[1][pos] - self.calibration[0][pos])))
elif raw > self.calibration[1][pos]:
val = 17 + 17 * ((raw - self.calibration[1][pos]) / float((self.calibration[2][pos] - self.calibration[1][pos])))
return val
def getEvent(self):
return self.lastEvent
def getLED(self):
return self.LED
# Thread that listens for incoming data
def receivethread(self):
while self.status == "Connected":
if True:
data = self.datasocket.recv(25)
intype = data[1]
if intype == INPUT_STATUS:
self.setReportingType()
elif intype == INPUT_READ_DATA:
if self.calibrationRequested == True:
packetLength = data[4]//16 + 1
self.parseCalibrationResponse(data[7:(7+packetLength)])
if packetLength < 16:
self.calibrationRequested = False
elif intype == EXTENSION_8BYTES:
self.lastEvent = self.createBoardEvent(data[2:12])
self.mass = self.lastEvent
else:
print("ACK to data write received")
self.status = "Disconnected"
self.disconnect()
def parseCalibrationResponse(self, bytes):
index = 0
if len(bytes) == 16:
for i in range(2):
for j in range(4):
self.calibration[i][j] = (bytes[index] << 8) + bytes[index+1]
index += 2
elif len(bytes) < 16:
for i in range(4):
self.calibration[2][i] = (bytes[index] << 8) + bytes[index+1]
index += 2
# Send <data> to the Wiiboard
# <data> should be an array of strings, each string representing a single hex byte
def send(self, data):
if self.status != "Connected":
return
senddata = b'0xa2'+data
self.datasocket.send(senddata)
def setLight(self, light):
val = 0x00
if light == True:
val = 0x10
msg = bytes([COMMAND_LIGHT, val])
self.send(msg)
self.LED = light
def calibrate(self):
msg = bytes([COMMAND_READ_REGISTER, 0x04, 0xA4, 0x00, 0x24, 0x00, 0x18])
self.send_legacy(msg)
self.calibrationRequested = True
def setReportingType(self):
msg = bytes([COMMAND_REPORTING, CONTINUOUS_REPORTING, EXTENSION_8BYTES])
self.send_legacy(bytearr)