Adding the tester.py

.gitignore

@@ -1 +1,6 @@
-__pycache__
+__pycache__
+lib
+bin
+pyvenv.cfg
+include
+

requirement.txt

@@ -0,0 +1,3 @@
+mido
+pygame
+pygame.midi

tester.py

@@ -0,0 +1,132 @@
+#!/usr/bin/env python3
+
+import sys
+import os
+from typing import List
+
+import pygame as pg
+from pygame.constants import KEYDOWN
+import pygame.midi
+from mido import MidiFile
+
+# Status definitions
+TOUCH_DOWN = 144
+TOUCH_UP   = 128
+
+
+class Key:
+    def __init__(self, key: int, start: int, duration: int):
+        self.key = key
+        self.start = start
+        self.duration = duration
+
+
+def read_midi(file):
+    notes = []
+    notes_on = []
+    s = 0
+    for msg in MidiFile(file):
+        d = msg.dict()
+        s += d['time'] * 1000
+        if d["type"] == "note_on":
+            notes_on[d["note"]] = s
+        if d["type"] == "note_off":
+            duration = s - notes_on[d["note"]]
+            notes_on[d["note"]] = s
+            notes.append(Key(d["note"], s, duration))
+
+
+keys_to_play = read_midi(sys.argv[1])
+tempo = 1000 # The duration of a black key in ms.
+
+# List of keys currently holded. Format: (key, timestamp)
+keys_down = []
+
+points = 0
+
+
+def print_device_info():
+    pygame.midi.init()
+    _print_device_info()
+    pygame.midi.quit()
+
+
+def _print_device_info():
+    for i in range(pygame.midi.get_count()):
+        r = pygame.midi.get_device_info(i)
+        (interf, name, input, output, opened) = r
+
+        in_out = ""
+        if input:
+            in_out = "(input)"
+        if output:
+            in_out = "(output)"
+
+        print(
+            "%2i: interface :%s:, name :%s:, opened :%s:  %s"
+            % (i, interf, name, opened, in_out)
+        )
+
+def poll(midi):
+    if midi.poll():
+        [((status, key, intensity, data3), timestamp)] = midi.read(1)
+        # For status, see STATUS DEFINITIONS up there (either TOUCH_DOWN, TOUCH_UP or others for pedals)
+        # The key is between 21 and 108, C5 is 60
+        # The itensity is how strong the key got struck (between 1 and 130ish)
+        # data3 seems to always be 0
+        # timestamp seems to be a unix timestamp since the midi has been oppened.
+        if status == TOUCH_DOWN:
+            keys_down.append((key, timestamp))
+            # print(f"Midi: {status} - {key} - {intensity} - {data3} at {timestamp}")
+        elif status == TOUCH_UP:
+            down_since = next(since for (h_key, since) in keys_down if h_key == key)
+            keys_down.remove((key, down_since))
+            return Key(key, down_since, (timestamp - down_since) / tempo)
+
+def run(midi):
+    global points
+    while keys_to_play:
+        key =  poll(midi)
+        if key is None:
+            continue
+        to_play = keys_to_play.pop()
+        if key.key == to_play.key:
+            tempo_percent = 100 - abs(key.duration - to_play.duration) * 100
+            if tempo_percent < 80:
+                points += tempo_percent / 2
+                print("Too short" if key.duration < to_play.duration else "Too long")
+            elif tempo_percent < 90:
+                points += tempo_percent
+                print(f"GREAT. {int(tempo_percent)}pts")
+            else:
+                points += tempo_percent * 2
+                print(f"EXCELLENT. {int(tempo_percent * 2)}pts")
+        else:
+            print(f"Invalid key. Got {key.key} expected {to_play.key}.")
+
+
+def input_main(device_id=None):
+    pg.init()
+    pygame.midi.init()
+
+    _print_device_info()
+
+    if device_id is None:
+        input_id = pygame.midi.get_default_input_id()
+    else:
+        input_id = device_id
+
+    print("using input_id :%s:" % input_id)
+    i = pygame.midi.Input(input_id)
+
+    pg.display.set_mode((1, 1))
+    try:
+        run(i)
+    except KeyboardInterrupt:
+        pass
+    print(f"You got: {int(points)}pts")
+    pygame.midi.quit()
+
+
+if __name__ == '__main__':
+    exit(input_main(int(sys.argv[2]) if len(sys.argv) == 3 else None))