From 3783da2bc41012c6ef1c81a8f20578809f65729f Mon Sep 17 00:00:00 2001
From: Hykilpikonna <me@hydev.org>
Date: Thu, 20 Apr 2023 19:56:44 -0400
Subject: [PATCH] [+] Backend reader

---
 taiko/reader.py | 69 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 taiko/reader.py

diff --git a/taiko/reader.py b/taiko/reader.py
new file mode 100644
index 0000000..82daaa6
--- /dev/null
+++ b/taiko/reader.py
@@ -0,0 +1,69 @@
+"""
+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 = ['x', 'z']
+
+
+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.')