Lucas Oskorep
31 lines
955 B
Python
31 lines
955 B
Python
import struct
|
|
from .wand_utils import chunker
|
|
from converters import BinToInt, BinToFloat
|
|
|
|
class wand_decoder(object):
|
|
def __init__(self, int_decoder=BinToInt(48), float_decoder= BinToFloat(16, 31)):
|
|
self.int_decoder = int_decoder
|
|
self.float_decoder = float_decoder
|
|
|
|
def decode_button(self, data):
|
|
return int.from_bytes(data, byteorder='big')
|
|
|
|
def decode_nine_axis(self, data):
|
|
converted = [self.int_decoder.process(x, True) / 10 ** 14 for x in chunker(data, 6)]
|
|
return converted
|
|
|
|
def decode_accelerometer(self, data):
|
|
converted = [self.int_decoder.process(x, True) / 10 ** 14 for x in chunker(data, 6)]
|
|
return converted
|
|
|
|
def decode_battery(self, data):
|
|
return int.from_bytes(data, byteorder='big')
|
|
|
|
def decode_temp(self, data):
|
|
return struct.unpack("h", data)
|
|
|
|
def decode_magnet(self, data):
|
|
print(data)
|
|
print(len(data))
|
|
return 0
|