Hykilpikonna
70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
"""
|
|
This script reads the Arduino's serial output and convert it to keyboard input.
|
|
|
|
Requirements: pip install pyserial keyboard matplotlib
|
|
"""
|
|
import time
|
|
|
|
import keyboard
|
|
import serial.tools.list_ports
|
|
|
|
TAIKO_KEYS = ['f', 'd']
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# List serial ports
|
|
ports = serial.tools.list_ports.comports()
|
|
|
|
# If there are no ports, exit
|
|
if not ports:
|
|
print('No serial ports found!')
|
|
exit(1)
|
|
|
|
# If there are more than one serial port, print them out
|
|
if len(ports) > 1:
|
|
print('Multiple serial ports found:')
|
|
for i, port in enumerate(ports):
|
|
print(f'[{i}] {port.device}')
|
|
print()
|
|
|
|
# Ask the user to select one
|
|
inp = input('Select a port: ')
|
|
port = ports[int(inp)].device
|
|
|
|
# If there is only one serial port, use it
|
|
else:
|
|
port = ports[0].device
|
|
|
|
# Open the serial port
|
|
print(f'Opening serial port {port}...')
|
|
ser = serial.Serial(port, 115200)
|
|
|
|
# Read the serial port
|
|
try:
|
|
start_time = time.time()
|
|
while True:
|
|
line = ser.readline().decode('utf-8', errors='ignore').strip()
|
|
print(line)
|
|
|
|
# If the line starts with ";", it's a plot point
|
|
if line.startswith(';'):
|
|
line = line[1:].split(';')
|
|
ys = [float(y) / 1024.0 for y in line]
|
|
yleft, yright = ys
|
|
|
|
# If the line starts with ":", it's a keyboard input
|
|
elif line.startswith('Hit:'):
|
|
# Index of the drumstick that hit the piezo
|
|
line = line[4:]
|
|
cmd = line[0]
|
|
key = TAIKO_KEYS[int(line[1:])]
|
|
|
|
if cmd == "+":
|
|
keyboard.press(key)
|
|
elif cmd == "-":
|
|
keyboard.release(key)
|
|
|
|
except KeyboardInterrupt:
|
|
ser.close()
|
|
print('Closed serial port.')
|