qmo.py

import rtmidi
import sys
import qmomidi
import qmolights
import qmodisplay

# range: 0 .. len(get_kights()) - 1
light_idx = 0


def go():
    qmolights.send_message("/go", [])

def record_new_cue():
    qmolights.send_message("/dashboard/newCueWithAll", [])

# v = 0 .. 100
def update_fader(fader, v, l):
    if l != "all":
        qmodisplay.setLcdPanel(fader, f'{v}', l)
    v = int(v * 16383 / 100)
    msg = rtmidi.MidiMessage.pitchWheel(fader, v)
    qmomidi.midiout().sendMessage(msg)

def update_faders():
    global light_idx
    light_list = qmolights.get_lights()
    i = 1
    for l in range(light_idx, light_idx + 8): # qmolights.get_lights():
        if l < len(light_list):
            name = light_list[l]
            v = qmolights.get_light(name)
            update_fader(i, v, name)
            i = i + 1
    if "all" in light_list:
        v = qmolights.get_light("all")
        update_fader(9, v, "all")

def pageLeft(n):
    global light_idx

    if light_idx < n:
        light_idx = 0
    else:
        light_idx = light_idx - n
    update_faders()

def pageRight(n):
    global light_idx

    # If there's only 8 or fewer lights, can't move...
    l = len(qmolights.get_lights())
    if l <= 8:
        return
    light_idx = light_idx + n
    if light_idx + 8 >= l:
        light_idx = l - 8
    update_faders()


def main():
    if not qmomidi.midiinit():
        return 0

    qmolights.init_lights()

    # indicate QLAB mode!
    qmodisplay.setBars("qla")
    qmodisplay.setBeats("b ")

    # Get the first 8 lights
    update_faders()

    while True:
        m = qmomidi.midiin().getMessage(100) # Timeout = 100
        if m:
            if m.isPitchWheel():
                c = m.getChannel()
                if c == 9:
                    v = m.getPitchWheelValue()
                    v = int(v * 100 / 16383)
                    update_fader(c, v, "all")
                    qmolights.set_light("all", v, True)
                    qmodisplay.setAssignment(str(v))
                else:
                    v = m.getPitchWheelValue()
                    v = int(v * 100 / 16383)
                    l = qmolights.get_lights()[light_idx + c - 1]
                    update_fader(c, v, l)
                    qmolights.set_light(l, v, True)
            elif m.isNoteOn():
                if m.getNoteNumber() == 46:
                    pageLeft(8)
                elif m.getNoteNumber() == 47:
                    pageRight(8)
                elif m.getNoteNumber() == 48:
                    pageLeft(1)
                elif m.getNoteNumber() == 49:
                    pageRight(1)
                elif m.getNoteNumber() == 51:
                    qmolights.update_lights()
                    update_faders()
                elif m.getNoteNumber() == 95:
                    record_new_cue()
                elif m.getNoteNumber() == 94:
                    go()

if __name__ == "__main__":
    sys.exit(main())