mirror of
https://github.com/kavishdevar/librepods.git
synced 2026-03-01 20:23:42 +00:00
head gestures, maybe??
This commit is contained in:
104
head-tracking/tk.py
Normal file
104
head-tracking/tk.py
Normal file
@@ -0,0 +1,104 @@
|
||||
import matplotlib.pyplot as plt
|
||||
import mplcursors
|
||||
import bluetooth
|
||||
import threading
|
||||
import time
|
||||
from matplotlib.animation import FuncAnimation
|
||||
import tkinter as tk
|
||||
from tkinter import ttk
|
||||
|
||||
sock = bluetooth.BluetoothSocket(bluetooth.L2CAP)
|
||||
|
||||
bt_addr = "28:2D:7F:C2:05:5B"
|
||||
psm = 0x1001
|
||||
|
||||
sock.connect((bt_addr, psm))
|
||||
running = threading.Event()
|
||||
|
||||
data = []
|
||||
selected_columns = list(range(80))
|
||||
|
||||
def hex_to_base10(hex_string):
|
||||
hex_values = hex_string.split()
|
||||
base10_values = [int(hex_value, 16) for hex_value in hex_values]
|
||||
return base10_values
|
||||
|
||||
def listen():
|
||||
global running, data
|
||||
while not running.is_set():
|
||||
res = sock.recv(1024)
|
||||
if len(res.hex()) == 162:
|
||||
hex_string = " ".join(res.hex()[i:i+2].upper() for i in range(0, len(res.hex()), 2))
|
||||
base10_values = hex_to_base10(hex_string)
|
||||
data.append(base10_values)
|
||||
|
||||
t = threading.Thread(target=listen)
|
||||
t.start()
|
||||
|
||||
try:
|
||||
byts = bytes(int(b, 16) for b in "00 00 04 00 01 00 02 00 00 00 00 00 00 00 00 00".split(" "))
|
||||
sock.send(byts)
|
||||
byts = bytes(int(b, 16) for b in "04 00 04 00 0F 00 FF FF FE FF".split(" "))
|
||||
sock.send(byts)
|
||||
byts = bytes(int(b, 16) for b in "04 00 04 00 17 00 00 00 10 00 11 00 08 7C 10 02 42 0B 08 4E 10 02 1A 05 01 40 9C 00 00".split(" "))
|
||||
sock.send(byts)
|
||||
except:
|
||||
...
|
||||
|
||||
def update(frame):
|
||||
plt.clf()
|
||||
plt.title('Bluetooth Data')
|
||||
for i in range(len(data[0])):
|
||||
if i in selected_columns:
|
||||
plt.plot([row[i] for row in data], label=f'Column {i}', alpha=1, linewidth=1.2)
|
||||
plt.xlabel('Time')
|
||||
plt.ylabel('Value')
|
||||
mplcursors.cursor(hover=True, highlight=True)
|
||||
|
||||
def on_checkbox_change():
|
||||
global selected_columns
|
||||
selected_columns = [i for i, var in enumerate(checkbox_vars) if var.get()]
|
||||
|
||||
def create_checkbox_window():
|
||||
root = tk.Tk()
|
||||
root.title("Select Columns")
|
||||
frame = ttk.Frame(root)
|
||||
frame.pack(fill=tk.BOTH, expand=True)
|
||||
canvas = tk.Canvas(frame)
|
||||
scrollbar = ttk.Scrollbar(frame, orient="vertical", command=canvas.yview)
|
||||
scrollable_frame = ttk.Frame(canvas)
|
||||
|
||||
scrollable_frame.bind(
|
||||
"<Configure>",
|
||||
lambda e: canvas.configure(
|
||||
scrollregion=canvas.bbox("all")
|
||||
)
|
||||
)
|
||||
|
||||
canvas.create_window((0, 0), window=scrollable_frame, anchor="nw")
|
||||
canvas.configure(yscrollcommand=scrollbar.set)
|
||||
|
||||
global checkbox_vars
|
||||
checkbox_vars = []
|
||||
for i in range(80):
|
||||
var = tk.BooleanVar(value=True)
|
||||
checkbox = ttk.Checkbutton(scrollable_frame, text=f"Column {i}", variable=var, command=on_checkbox_change)
|
||||
checkbox.pack(anchor='w')
|
||||
checkbox_vars.append(var)
|
||||
|
||||
canvas.pack(side="left", fill=tk.BOTH, expand=True)
|
||||
scrollbar.pack(side="right", fill="y")
|
||||
|
||||
root.mainloop()
|
||||
|
||||
checkbox_thread = threading.Thread(target=create_checkbox_window)
|
||||
checkbox_thread.start()
|
||||
|
||||
fig = plt.figure()
|
||||
fig.tight_layout()
|
||||
ani = FuncAnimation(fig, update, interval=500, repeat=True, cache_frame_data=False)
|
||||
|
||||
plt.show()
|
||||
|
||||
running.set()
|
||||
sock.close()
|
||||
Reference in New Issue
Block a user