mqtt_rgb.py

#! /usr/bin/env python3
import paho.mqtt.client as mqtt
import time
import board
import neopixel
import socket
import random
import subprocess
from datetime import datetime, timedelta

morse = {
    "a": [1,2],
    "b": [2,1,1,1],
    "c": [2,1,2,1],
    "d": [2,1,1],
    "e": [1],
    "f": [1,1,2,1],
    "g": [2,2,1],
    "h": [1,1,1,1],
    "i": [1,1],
    "j": [1,2,2,2],
    "k": [2,1,2],
    "l": [1,2,1,1],
    "m": [2,2],
    "n": [2,1],
    "o": [2,2,2],
    "p": [1,2,2,1],
    "q": [2,2,1,2],
    "r": [1,2,1],
    "s": [1,1,1],
    "t": [2],
    "u": [1,1,2],
    "v": [1,1,1,2],
    "w": [1,2,2],
    "x": [2,1,1,2],
    "y": [2,1,2,2],
    "z": [2,2,1,1],
    " ": [0],
    "1": [1,2,2,2,2],
    "2": [1,1,2,2,2],
    "3": [1,1,1,2,2],
    "4": [1,1,1,1,2],
    "5": [1,1,1,1,1],
    "6": [2,1,1,1,1],
    "7": [2,2,1,1,1],
    "8": [2,2,2,1,1],
    "9": [2,2,2,2,1],
    "0": [2,2,2,2,2]
}

myip = subprocess.check_output(['hostname', '-I'])
try:
    import keyboard
    print("keyboard module present")
except:
    keyboard = None
    print("keyboard module not present. Please run sudo pip3 install keyboard")

############# CONFIG #############
uot = 0.05
pixel_pin = board.D18
num_pixels = 8
ORDER = neopixel.GRB
rainbow_cycle_delay = 0.001
mqtt_enabled = True
delay = 0.0333
##################################

modes = ["0","1","2","3","4","5","6","7"]
myname = socket.gethostname()
current_colors = []
current_color = (0,0,0)
wait_till = time.time()
key_down = False
last_time_update = datetime.now()

if myname == "addresspi":
    num_pixels = 32
    rainbow_cycle_delay = 0.001
elif myname == "canvaspi":
    num_pixels = 50
    rainbow_cycle_delay = 0.001
elif myname == "clockpi":
    num_pixels = 60
    rainbow_cycle_delay = 0.001
elif myname == "windowpi":
    num_pixels = 16
    rainbow_cycle_delay = 0.001
elif myname == "namepi":
    num_pixels = 8
    rainbow_cycle_delay = 0.05
    mqtt_enabled = False

for i in range(0,num_pixels):
    current_colors.append((0,0,0))

pixels = neopixel.NeoPixel(
    pixel_pin, num_pixels, brightness=1.0, auto_write=False, pixel_order=ORDER
)
running = False
if mqtt_enabled is True:
    client = mqtt.Client()
    print("mqtt enabled")
else:
    client = None
    print("mqtt disabled")
mode = 3
# modes: 0-fill, 1-single pixel, 2-range, 3-gradient, 4-rainbow chase, 5-random pixels, 6-change random, 7-change swipe
j = 0

def wheel(pos):
    # Input a value 0 to 255 to get a color value.
    # The colours are a transition r - g - b - back to r.
    if pos < 0 or pos > 255:
        r = g = b = 0
    elif pos < 85:
        r = int(pos * 3)
        g = int(255 - pos * 3)
        b = 0
    elif pos < 170:
        pos -= 85
        r = int(255 - pos * 3)
        g = 0
        b = int(pos * 3)
    else:
        pos -= 170
        r = 0
        g = int(pos * 3)
        b = int(255 - pos * 3)
    return (r, g, b) if ORDER in (neopixel.RGB, neopixel.GRB) else (r, g, b, 0)

def all_same():
    check = True
    for i in range(0,num_pixels):
        if current_colors[i] != current_color:
            check = False
    return check

def fill(color):
    current_color = color
    for i in range(0,num_pixels):
        current_colors[i] = color
    pixels.fill(color)
    pixels.show()

def rainbow_cycle(wait):
    for i in range(num_pixels):
        pixel_index = (i * 256 // num_pixels) + j
        pixels[i] = wheel(pixel_index & 255)
    pixels.show()
    time.sleep(wait)

def on_message(client, userdata, message):
    global running
    global mode
    global delay
    result = str(message.payload.decode("utf-8"))
    print("Received: "+result)
    client.publish('pi/' + myname + '/received', "Received: "+result)
    if result == "stop":
        running = False
    args = result.split(':')
    if args[0] == "setdelay":
        delay = float(args[1])
        return
    mode = int(args[0])
    if mode <= 2:
        r = int(args[1])
        g = int(args[2])
        b = int(args[3])
        if mode == 0:
            fill((r,g,b))
        elif mode == 1:
            i = int(args[4])
            c = (r,g,b)
            pixels[i] = c
            current_colors[i] = c
            pixels.show()
        elif mode == 2:
            s = int(args[4])
            e = int(args[5])
            while s <= e:
                c = (r,g,b)
                pixels[s] = c
                current_colors[s] = c
                s = s + 1
            pixels.show()

def on_key_down():
    global running
    global mode
    global key_down
    global current_color
    if keyboard is None:
        return
    if key_down is True:
        check = False
        for m in modes:
            if keyboard.is_pressed(m):
                check = True
        key_down = check
        return
    do_stuff = False
    for m in modes:
        if keyboard.is_pressed(m):
            do_stuff = True
            mode = int(m)
    if do_stuff is False:
        return
    if mode == 0:
        fill(current_color)
    else:
        current_color = wheel(random.randint(1,8) * 32 - 1)
        fill(current_color)
    key_down = True

def translate_name_address(a):
    if num_pixels > 8:
        return a
    if a == 0:
        return 5
    if a == 1:
        return 6
    if a == 2:
        return 7
    if a == 3:
        return 4
    if a == 4:
        return 3
    if a == 5:
        return 2
    if a == 6:
        return 1
    return 0

def all_on():
    for i in range(8):
        pixels[i] = (255,255,255)
    pixels.show()

def all_off():
    for i in range(8):
        pixels[i] = (0,0,0)
    pixels.show()

def dit():
    all_on()
    time.sleep(uot)
    all_off()
    time.sleep(uot)

def da():
    all_on()
    time.sleep(uot*3)
    all_off()
    time.sleep(uot)

def pause():
    time.sleep(1)

def do_letter(letter):
    l = letter.lower()
    if l not in morse.keys():
        pause()
    code = morse[l]
    for c in code:
        if c == 0:
            pause()
        elif c == 1:
            dit()
        elif c == 2:
            da()
    pause()

def do_message(message):
    for l in message:
        do_letter(l)

if __name__ == "__main__":
    if mqtt_enabled is True:
        try:
            client.on_message = on_message
            client.connect('192.168.2.200')
        except:
            client = None
    if client is not None:
        topic = 'pi/' + myname + '/commands'
        print('subscribing to '+topic)
        client.subscribe(topic)
        client.loop_start()
    running = True
    while running is True:
        now = time.time()
        dtnow = datetime.now()
        if myname == "namepi" and dtnow.minute in [0,15,30,45] and dtnow > last_time_update + timedelta(minutes=14):
            last_time_update = dtnow
            timestr = dtnow.strftime("%H %M")
            do_message(timestr)
            continue
        on_key_down()
        if now < wait_till:
            continue
        if mode <= 2:
            wait_till = time.time() + 5.0
            #time.sleep(5)
            #mosquittoMessage("mqtt_rgb "+str(myip).split(' ')[0].replace("b'","")+" alive at "+datetime.now().strftime("%m/%d/%Y, %H:%M:%S"))
        elif mode == 3 or mode == 4:
            if j + 1 > 255:
                j = 0
                #mosquittoMessage("mqtt_rgb "+str(myip).split(' ')[0].replace("b'","")+" alive at "+datetime.now().strftime("%m/%d/%Y, %H:%M:%S"))
            else:
                j = j + 1
            if mode == 3:
                fill(wheel(j))
                wait_till = time.time() + 0.066
                continue
                #time.sleep(0.066)
            elif mode == 4:
                rainbow_cycle(rainbow_cycle_delay)
        elif mode == 5:
            r = random.randint(1,8) * 32 - 1
            a = random.randint(0, num_pixels-1)
            pixels[a] = wheel(r)
            pixels.show()
            wait_till = time.time() + 1.0
            continue
            #time.sleep(1)
        elif mode == 6:
            if all_same() is True:
                original_color = current_color
                current_color = wheel(random.randint(1,8) * 32 - 1)
                while current_color == original_color:
                    current_color = wheel(random.randint(1,8) * 32 - 1)
                #time.sleep(random.randint(5,30))
                wait_till = time.time() + random.randint(5,30)
                continue
            a = random.randint(0, num_pixels-1)
            while current_colors[a] == current_color:
                a = random.randint(0, num_pixels-1)
            pixels[a] = current_color
            current_colors[a] = current_color
            pixels.show()
            #time.sleep(0.25)
            wait_till = time.time() + 0.25
            continue
        elif mode == 7:
            if all_same() is True:
                original_color = current_color
                current_color = wheel(random.randint(1,8) * 32 - 1)
                while current_color == original_color:
                    current_color = wheel(random.randint(1,8) * 32 - 1)
                #time.sleep(random.randint(5,30))
                wait_till = time.time() + random.randint(5,30)
                continue
            a = 0
            while current_colors[a] == current_color:
                a = a + 1
            if a >= num_pixels:
                a = 0
            x = translate_name_address(a)
            pixels[x] = current_color
            current_colors[a] = current_color
            pixels.show()
            #time.sleep(0.1)
            wait_till = time.time() + 0.1
            continue
        elif mode == 8:
            last = num_pixels - 1
            current_colors[last] = current_colors[0]
            pixels[last] = current_colors[last]
            for i in range(last):
                current_colors[i] = current_colors[i+1]
                pixels[i] = current_colors[i]
            pixels.show()
            time.sleep(delay)
    client.loop_stop()
    client.disconnect()