pyfootctrl.py

import argparse
import time

import rtmidi # https://spotlightkid.github.io/python-rtmidi/

from config import behaviour
from chocolate import display

# For windows, get loop midi at https://www.tobias-erichsen.de/software/loopmidi.html
# midiberry for BLE https://apps.microsoft.com/store/detail/midiberry/9N39720H2M05?hl=pt-br&gl=br

midi_in = rtmidi.MidiIn()
in_ports_by_name = midi_in.get_ports()

midi_out = rtmidi.MidiOut()
out_ports_by_name = midi_out.get_ports()

def display_midi_ports():
    print("Available MIDI IN ports:", '\n'.join([ f'{number} -> {name}' for number,name in enumerate(in_ports_by_name)]), sep='\n')
    print()
    print("Available MIDI OUT ports:", '\n'.join([ f'{number} -> {name}' for number,name in enumerate(out_ports_by_name)]), sep='\n')
    print()

def user_input_midi_ports():
    in_port = int(input("Input port number:"))
    out_port = int(input("Output port number:"))
    print(f"Input: {in_ports_by_name[in_port]} | Output: {out_ports_by_name[out_port]}")
    return in_port, out_port

def match_device_name(expected_name_parts, device_name):
    for expected_name_part in expected_name_parts:
        if expected_name_part in device_name:
            return True
    return False

def default_midi_ports():
    """
    Return the default MIDI ports indexes, with MIDI IN as USB-Midi and MIDI OUT as loopMIDI
    """

    DEFAULT_MIDI_IN_PORT_NAMES = ['FootCtrl','USB-Midi']
    DEFAULT_MIDI_OUT_PORT_NAMES = ['Virtual'] #['Virtual','Midi Through','loopMIDI']
    in_port_index = None
    out_port_index = None
    for index, port_name in enumerate(in_ports_by_name):
        if match_device_name(DEFAULT_MIDI_IN_PORT_NAMES, port_name):
            in_port_index = index
            print(f"Using {port_name} as MIDI in")
            break
    else:
        print("MIDI input device not found")
    for index, port_name in enumerate(out_ports_by_name):
        if match_device_name(DEFAULT_MIDI_OUT_PORT_NAMES, port_name):
            out_port_index = index
            print(f"Using {port_name} as MIDI out")
            break
    else:
        print("MIDI output device not found")
    return in_port_index, out_port_index

def start_midi_loop(midi_in, midi_out):
    print("Starting MIDI message processing loop...")
    while True:
        message = midi_in.get_message()
        if message:
            try:
                ([midi_msg_type, midi_msg_data], delta_seconds) = message
                footswitch = display(midi_msg_data)
                print(f"{footswitch} | {midi_msg_type} | {midi_msg_data} | {delta_seconds:0.000f}s")
                behaviour[footswitch](midi_out, delta_seconds)
            except ValueError as e:
                print(e)
                print(message)
        time.sleep(0.001)


if __name__ == '__main__':
    parser = argparse.ArgumentParser(
        prog='PyFootControl',
        description='Expands capabilities of M-Vave Chocolate footswitch controller'
    )
    parser.add_argument('-i', '--input', type=int, help="Number of the input channel (the footswitch controller).")
    parser.add_argument('-o', '--output', type=int, help="Number of the output channel, probably the MIDI loopback or virtual cable.")
    parser.add_argument('-l', '--list', action='store_true', help="List MIDI device indexes and quit")
    args = parser.parse_args()

    in_port, out_port = default_midi_ports()
    if args.input:
        midi_in.open_port(args.input)
    else:
        midi_in.open_port(in_port)

    if args.output:
        midi_out.open_port(args.output)
    else:
        midi_out.open_port(out_port)

    if args.list:
        display_midi_ports()
        exit(0)
    try:
        start_midi_loop(midi_in, midi_out)
    except KeyboardInterrupt:
        print("Bye!")
        exit(0)