Files
librepods/head-tracking/final.py
2024-12-30 20:40:40 +05:30

117 lines
3.9 KiB
Python

# AirPods like Normal (ALN) - Bringing Apple-only features to Linux and Android for seamless AirPods functionality!
#
# Copyright (C) 2024 Kavish Devar
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License.
#
# 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
import bluetooth
import threading
import time
sock = bluetooth.BluetoothSocket(bluetooth.L2CAP)
bt_addr = "28:2D:7F:C2:05:5B"
psm = 0x1001
sock.connect((bt_addr, psm))
running = threading.Event()
data = []
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:
...
start_time = time.time()
timeout = 10 # seconds
vertical_count = 0
horizontal_count = 0
horizontal_fluctuation_count = 0
initial_131_sequence = True
while time.time() - start_time < timeout:
if len(data) > 0:
column_34_data = [row[34] for row in data]
for i in range(len(column_34_data)):
print(column_34_data[i])
if initial_131_sequence:
if column_34_data[i] == 131:
if i >= 4:
continue
else:
initial_131_sequence = False
if column_34_data[i] == 131 and not initial_131_sequence:
vertical_count += 1
horizontal_count = 0
if vertical_count >= 45:
print("vertical")
running.set()
sock.close()
exit()
elif column_34_data[i] == 3:
horizontal_count += 1
vertical_count = 0
if horizontal_count >= 40:
print("horizontal")
running.set()
sock.close()
exit()
else:
vertical_count = 0
horizontal_count = 0
# Check for horizontal nodding (fluctuation between 3 and 67 or higher)
fluctuation_window = 40
if len(column_34_data) >= fluctuation_window:
fluctuation_count = 0
for i in range(len(column_34_data) - fluctuation_window + 1):
segment = column_34_data[i:i+fluctuation_window]
if any(value == 3 for value in segment) and any(value >= 67 for value in segment):
count_67 = segment.count(67)
if count_67 <= 10:
fluctuation_count += 1
if fluctuation_count >= 10:
print("horizontal")
running.set()
sock.close()
exit()
print("unknown")
running.set()
sock.close()