lucasoskorep/kano-wand-smarthome
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactoring code from original POC phase - integers have been decoded roughly successfully so its time to start building a huge library for this bad boy

Browse Source
This commit is contained in:
Lucas Oskorep
2019-11-08 12:58:37 -06:00
parent 48866f6af2
commit 564eed46d5
14 changed files with 223 additions and 246 deletions
Download Patch File Download Diff File Expand all files Collapse all files
collect_data.py
+33
View File
@@ -0,0 +1,33 @@
import asyncio
import json
import platform
"""
This is used for reading and decoding values from t he Kano Harry Potter Coding Wand
- Wand sends 25 updates to gyro and accel every second
- Wand sends 3 Dimensional data as 48 bit reverse marshalled integers
- Wand seems to send smaller single dimensional data in 16bit unsigned integers
"""
SPELLS = json.load("spells.json")
CURR_SPELL = 0
# TODO: RUMBLE
# TODO: RGB
device_address = "D8:9B:12:D1:08:80"
if __name__ == "__main__":
import os
os.environ["PYTHONASYNCIODEBUG"] = str(1)
address = (
device_address # <--- Change to your device's address here if you are using Windows or Linux
if platform.system() != "Darwin"
else "243E23AE-4A99-406C-B317-18F1BD7B4CBE" # <--- Change to your device's address here if you are using macOS
)
loop = asyncio.get_event_loop()
loop.run_until_complete(run(address, loop, True))
kano_wand/__init__.py
View File
kano_wand/ble_client.py
+64
View File
@@ -0,0 +1,64 @@
import logging
import struct
from converters import BinToInt, BinToFloat
from .sensor_tracker import *
from bleak import BleakClient
from bleak import _logger as logger
from .utils import *
class KanoBLEClient(object):
def __init__(self, wand_address, spells=None):
self.wand_address = wand_address
self.int_decoder = BinToInt(48)
self.float_decoder = BinToFloat(16, 31)
self.client = None
async def connect(self, loop, debug=False):
if debug:
import sys
l = logging.getLogger("asyncio")
l.setLevel(logging.DEBUG)
h = logging.StreamHandler(sys.stdout)
h.setLevel(logging.DEBUG)
l.addHandler(h)
logger.addHandler(h)
async with BleakClient(self.wand_address, loop=loop) as client:
self.client = client
x = await client.is_connected()
logger.info("Connected: {0}".format(x))
async def start_recieving_data(self):
await start_notify(self.client)
async def stop_recieving_data(self):
await stop_notify(self.client)
def sensor_handling(self, sender, data):
"""Simple notification handler which prints the data received."""
# TODO: Convert to a dictionary and a single decode command
sender = CHARACTERISTIC_UUIDS[sender]
if sender == BUTTON:
self.decode_button(data)
elif sender == NINE_AXIS:
#TODO: refactor the gyro to be local to the sensors file only
gyro = Gyro()
gyro.x, gyro.y, gyro.z = self.decode_nine_axis(data)
sensors.set_gyro(gyro)
sensors.append_to_dataframe()
elif sender == ACCELEROMETER:
accel = Accel()
accel.x, accel.y, accel.z = self.decode_nine_axis(data)
sensors.set_accel(accel)
sensors.append_to_dataframe()
elif sender == BATTERY:
self.decode_battery(data)
elif sender == TEMPERATURE:
self.decode_temp(data)
elif sender == MAGNETOMETER:
self.decode_magnet(data)
kano_wand/constants.py
+15
View File
@@ -0,0 +1,15 @@
BUTTON = 1
NINE_AXIS = 2
ACCELEROMETER = 3
BATTERY = 4
TEMPERATURE = 5
MAGNETOMETER = 6
CHARACTERISTIC_UUIDS = {
("64a7000d-f691-4b93-a6f4-0968f5b648f8"): BUTTON, # Button
("64a7000a-f691-4b93-a6f4-0968f5b648f8"): NINE_AXIS, # 9 axis
("64a7000c-f691-4b93-a6f4-0968f5b648f8"): ACCELEROMETER, # Accel
("64a70007-f691-4b93-a6f4-0968f5b648f8"): BATTERY,
("64a70014-f691-4b93-a6f4-0968f5b648f8"): TEMPERATURE,
("64a70021-f691-4b93-a6f4-0968f5b648f8"):MAGNETOMETER
} # <--- Change to the characteristic you want to enable notifications from.
kano_wand/sensor_decoders.py
+33
View File
@@ -0,0 +1,33 @@
import struct
from converters import BinToInt, BinToFloat
class 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):
if int.from_bytes(data, byteorder='big'):
print("Button Pressed")
else:
print("Button Released")
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):
int.from_bytes(data, byteorder='big')
def decode_temp(self, data):
print(len(data))
print(data)
print(struct.unpack("h", data))
def decode_magnet(self, data):
print(data)
print(len(data))
kano_wand/sensor_tracker.py
+54
View File
@@ -0,0 +1,54 @@
import pandas as pd
class Gyro(object):
def __init__(self):
self.x = 0
self.y = 0
self.z = 0
def __str__(self):
print(self.x, self.y, self.z)
class Accel(object):
def __init__(self):
self.x = 0
self.y = 0
self.z = 0
def __str__(self):
print(self.x, self.y, self.z)
class SensorTracker(object):
def __init__(self):
self.gyro = None
self.gyro_updates = 0
self.accel = None
self.accel_updates = 0
self.dataframe = pd.DataFrame(columns=["gyro_x", "gyro_y", "gyro_z", "accel_x", "accel_y", "accel_z"])
def append_to_dataframe(self):
if self.gyro_updates == self.accel_updates:
self.dataframe = self.dataframe.append(
pd.DataFrame({
"gyro_x": self.gyro.x,
"gyro_y": self.gyro.y,
"gyro_z": self.gyro.z,
"accel_x": self.accel.x,
"accel_y": self.accel.y,
"accel_z": self.accel.z,
}, index=[0])
)
def save_df_to_file(self, filename):
self.dataframe.to_csv(filename, index=False)
def set_gyro(self, gyro):
self.gyro = gyro
self.gyro_updates += 1
def set_accel(self, accel):
self.accel = accel
self.accel_updates += 1
kano_wand/testing/__init__.py
View File
accelerometer.data → kano_wand/testing/accelerometer.data
View File
byteDecodingPlayground.py → kano_wand/testing/byteDecodingPlayground.py
View File
discover_and_probe_characteristics.py → kano_wand/testing/discover_and_probe_characteristics.py
+4 -4
View File
@@ -1,21 +1,21 @@
import asyncio
from bleak import discover
from bleak import discover, BleakClient
from pprint import pprint
async def run():
devices = await discover()
for d in devices:
print(d)
loop = asyncio.get_event_loop()
loop.run_until_complete(run())
import asyncio
from bleak import BleakClient
address = "D8:9B:12:D1:08:80"
MODEL_NBR_UUID = "64a7000f-f691-4b93-a6f4-0968f5b648f8"
async def run(address, loop):
async with BleakClient(address, loop=loop) as client:
# await client.connect()
sensor_information → kano_wand/testing/sensor_information
View File
kano_wand/utils.py
+13
View File
@@ -0,0 +1,13 @@
from kano_wand.constants import *
async def start_notify(client, notification_handler):
for char in CHARACTERISTIC_UUIDS.keys():
await client.start_notify(char, notification_handler)
async def stop_notify(client):
for char in CHARACTERISTIC_UUIDS.keys():
await client.stop_notify(char)
def chunker(seq, size):
return (seq[pos:pos + size] for pos in range(0, len(seq), size))
spells.json
+6
View File
@@ -0,0 +1,6 @@
{
"0": "unknown",
"1": "lumos",
"2": "nox",
"3": "accio"
}
subscribe_to_notification.py
-241
View File
@@ -1,241 +0,0 @@
import logging
import asyncio
import platform
import struct
import time
import pandas as pd
from bleak import BleakClient
from bleak import _logger as logger
from converters import BinToInt, BinToFloat
"""
This is used for reading and decoding values from t he Kano Harry Potter Coding Wand
- Wand sends 25 updates to gyro and accel every second
- Wand sends 3 Dimensional data as 48 bit reverse marshalled integers
- Wand seems to send smaller single dimensional data in 16bit unsigned integers
"""
BUTTON = 1
NINE_AXIS = 2
ACCELEROMETER = 3
BATTERY = 4
TEMPERATURE = 5
MAGNETOMETER = 6
INT_DECODER = BinToInt(48)
FLOAT_DECODER = BinToFloat(16, 31)
BUTTON_STATUS = None
CHARACTERISTIC_UUIDS = {
("64a7000d-f691-4b93-a6f4-0968f5b648f8"): BUTTON, # Button
("64a7000a-f691-4b93-a6f4-0968f5b648f8"): NINE_AXIS, # 9 axis
("64a7000c-f691-4b93-a6f4-0968f5b648f8"): ACCELEROMETER, # Accel
# ("64a70007-f691-4b93-a6f4-0968f5b648f8"):BATTERY,
# ("64a70014-f691-4b93-a6f4-0968f5b648f8"):TEMPERATURE,
# ("64a70014-f691-4b93-a6f4-0968f5b648f8"):MAGNETOMETER
} # <--- Change to the characteristic you want to enable notifications from.
# TODO: RUMBLE
# TODO: RGB
device_address = "D8:9B:12:D1:08:80"
class Gyro(object):
def __init__(self):
self.x = 0
self.y = 0
self.z = 0
def __str__(self):
print(self.x, self.y, self.z)
class Accel(object):
def __init__(self):
self.x = 0
self.y = 0
self.z = 0
def __str__(self):
print(self.x, self.y, self.z)
class Sensors(object):
def __init__(self):
self.gyro = None
self.gyro_updates = 0
self.accel = None
self.accel_updates = 0
self.dataframe = pd.DataFrame(columns=["gyro_x", "gyro_y", "gyro_z", "accel_x", "accel_y", "accel_z"])
def append_to_dataframe(self):
# print("Printing the gyro and accel", self.gyro, self.accel)
# print(self.dataframe)
# print(self.gyro_updates, self.accel_updates)
if self.gyro_updates == self.accel_updates:
# print({
# "gyro_x": self.gyro.x,
# "gyro_y": self.gyro.y,
# "gyro_z": self.gyro.z,
# "accel_x": self.accel.x,
# "accel_y": self.accel.y,
# "accel_z": self.accel.z,
# })
# try:
#
# except Exception as e:
# print("ERROR HAS OCCURRED")
# print(e)
self.dataframe = self.dataframe.append(
pd.DataFrame({
"gyro_x": self.gyro.x,
"gyro_y": self.gyro.y,
"gyro_z": self.gyro.z,
"accel_x": self.accel.x,
"accel_y": self.accel.y,
"accel_z": self.accel.z,
}, index=[0])
)
def save_df_to_file(self, filename):
self.dataframe.to_csv(filename, index=False)
def set_gyro(self, gyro):
self.gyro = gyro
self.gyro_updates += 1
def set_accel(self, accel):
self.accel = accel
self.accel_updates += 1
def chunker(seq, size):
return (seq[pos:pos + size] for pos in range(0, len(seq), size))
count = 0
def decode_button(data):
global sensors, count
if int.from_bytes(data, byteorder='big'):
print("Button Pressed")
sensors = Sensors()
count += 1
else:
print("Button Released")
sensors.save_df_to_file(f"dataset{count}.csv")
def decode_nine_axis(data):
converted = [INT_DECODER.process(x, True) / 10 ** 14 for x in chunker(data, 6)]
# print(converted, sum(converted))
return converted
def decode_accelerometer(data):
converted = [INT_DECODER.process(x, True) / 10 ** 14 for x in chunker(data, 6)]
# print(converted, sum(converted))
return converted
def decode_battery(data):
global BUTTON_STATUS
BUTTON_STATUS = int.from_bytes(data, byteorder='big')
def decode_temp(data):
print(len(data))
print(data)
print(struct.unpack("h", data))
def decode_magnet(data):
print(data)
print(len(data))
count = 0
sensors = Sensors()
def notification_handler(sender, data):
global count, sensors
"""Simple notification handler which prints the data received."""
# TODO: Convert to a dictionary and a single decode command
sender = CHARACTERISTIC_UUIDS[sender]
if sender == BUTTON:
decode_button(data)
elif sender == NINE_AXIS:
gyro = Gyro()
# print("getting gyro values")
gyro.x, gyro.y, gyro.z = decode_nine_axis(data)
# print("got gyro values")
# print(gyro)
sensors.set_gyro(gyro)
sensors.append_to_dataframe()
elif sender == ACCELEROMETER:
accel = Accel()
accel.x, accel.y, accel.z = decode_nine_axis(data)
sensors.set_accel(accel)
sensors.append_to_dataframe()
# decode_accelerometer(data)
elif sender == BATTERY:
decode_battery(data)
elif sender == TEMPERATURE:
decode_temp(data)
elif sender == MAGNETOMETER:
decode_magnet(data)
async def start_notify(client):
for char in CHARACTERISTIC_UUIDS.keys():
await client.start_notify(char, notification_handler)
async def stop_notify(client):
for char in CHARACTERISTIC_UUIDS.keys():
await client.stop_notify(char)
async def run(address, loop, debug=False):
if debug:
import sys
# loop.set_debug(True)
l = logging.getLogger("asyncio")
l.setLevel(logging.DEBUG)
h = logging.StreamHandler(sys.stdout)
h.setLevel(logging.DEBUG)
l.addHandler(h)
logger.addHandler(h)
async with BleakClient(address, loop=loop) as client:
x = await client.is_connected()
logger.info("Connected: {0}".format(x))
await start_notify(client)
# await asyncio.sleep(60.0, loop=loop)
while True:
time.sleep(1)
await stop_notify(client)
if __name__ == "__main__":
import os
os.environ["PYTHONASYNCIODEBUG"] = str(1)
address = (
device_address # <--- Change to your device's address here if you are using Windows or Linux
if platform.system() != "Darwin"
else "243E23AE-4A99-406C-B317-18F1BD7B4CBE" # <--- Change to your device's address here if you are using macOS
)
loop = asyncio.get_event_loop()
loop.run_until_complete(run(address, loop, True))