codeklavier.py

#!/usr/bin/env python3
"""
CodeKlavier Masterscript

This script will help you run the Codeklavier
"""

import configparser
import getopt
import sys
import time

import importlib

from CK_Setup import Setup, BColors
from CK_rec import CK_Rec
import CK_config



ck_deltatime_mem = []

VERSIONS = ('ckar', 'ckalculator', 'hybrid')

def doHelp():
    """
    Show the help for running this script.
    """
    print(BColors.HEADER + 'This script will help you to run the 🎹 CodeKlavier 🎹' + BColors.ENDC)
    print('')
    print(BColors.CKGREEN + 'Usage: ./codeklaver.py [OPTION]' + BColors.ENDC)
    print('Where [OPTION] is:')
    print(BColors.BOLD + '-h | --help' + BColors.ENDC)
    print('Show this help text.')

    print('')
    print(BColors.BOLD + '-p | --play' + BColors.WARNING + ' <<name>>' + BColors.ENDC)
    print('Boot CodeKlavier with version <<name>>')
    print('These versions are available: ' + (', ').join(VERSIONS))   
    print('')    
    print(BColors.BOLD + '-i | --ini' + BColors.WARNING + ' <<filename.ini>>' + BColors.ENDC)
    print('Boot CodeKlavier with configuration from <<filename.ini>>')
    print('')     
    print(BColors.BOLD + '-r | --rec' + BColors.ENDC)
    print('Boot CodeKlavier to record MIDI for machine learning')
    print('')        
    print(BColors.BOLD + '-t | --test' + BColors.ENDC)
    print('Test and visualize if Codeklavier is receving MIDI')
    print('')       
    print(BColors.UNDERLINE + 'Example:' + BColors.ENDC)
    print(BColors.CKGREEN + './codeklavier.py -p hybrid' + BColors.ENDC)
    print('')

def miditest(configfile='default_setup.ini'):
    """
    Run a basic miditest to see how the CodeKlavier is receiving your midi.

    :param string configfile: Path to the configuration file (default: default_setup.ini)
    """
    #Read config and settings
    config = configparser.ConfigParser()
    config.read(configfile, encoding='utf8')

    try:
        myPort = config['midi'].getint('port')
        device_id = config['midi'].getint('noteon_id')
        pedal = config['midi'].getint('pedal_id')
        sostenuto = config['midi'].getint('pedal_midi_id')
    except KeyError:
        raise LookupError('Missing key information in the config file.')

    if (myPort == None or device_id == None):
        raise LookupError('Missing key information in the config file.')

    codeK = Setup()
    codeK.open_port(myPort)
    print('your device id is: ', device_id, '\n')
    print("CodeKlavier is ON. Press Control-C to exit.")
    try:
        while True:
            msg = codeK.get_message()

            if msg:
                message, deltatime = msg
                if message[0] in [device_id, pedal, sostenuto]:
                    print('deltatime: ', deltatime, 'msg: ', message)

            time.sleep(0.01)

    except KeyboardInterrupt:
        print('')
    finally:
        print("Bye-Bye :(")
        codeK.end()

def boot(configfile='default_setup.ini', version=None):
    """
    Boot a specific version from the CodeKlavier

    :param string configfile: Path to the configuration file (default: default_setup.ini)
    :param string version: name of the version to boot
    """
    if (version not in VERSIONS):
        raise ValueError(BColors.WARNING + 'This version doesn\'t exist. Please retry.' + BColors.ENDC)
    
    if version == 'ckar':
        version = 'ckalculator'
        module = importlib.import_module(version)
        version = getattr(module, version)  
        eval(version.main(ar_hook=True))
    else:
        module = importlib.import_module(version)
        version = getattr(module, version)
        eval(version.main())        
    
if __name__ == '__main__':
    """
    Catch the arguments that were used to start this function.

    Options are h, c, p, t, r
    """

    showHelp = False
    test = False
    record = False
    config = None
    play = None

    try:
        options, args = getopt.getopt(sys.argv[1:],'h:p:i:rt',['help', 'play=', 'ini=', 'rec', 'test'])
        selected_options = [x[0] for x in options]
        
        for o, a in options:
            if o in ('-h', '--help'):
                showHelp = True
            if o in ('-p', '--play'):
                play = a
            if o in ('-i', '--ini'):
                config = a
            if o in ('-t', '--test'):
                test = True
            if o in ('-r', '--rec'):
                record = True    
                
    except getopt.GetoptError:
        print('Something went wrong parsing the optional arguments')        

    if showHelp:
        doHelp()
        sys.exit(0)

    if config == None:
        CK_config.inifile = 'default_setup.ini'
    else:
        CK_config.inifile = config

    if test:
        miditest(configfile=CK_config.inifile)
        sys.exit(0)
        
    if record:
        rec = CK_Rec(configfile=CK_config.inifile)
        rec.record(framesize=1)
        sys.exit(0)        

    if play:
        boot(configfile=config, version=play)
        sys.exit(0)

    #no arguments -> print help
    doHelp()
    sys.exit(0)