syndilights/v2/backend/forwarding/forward.py

#!/usr/bin/env python

import redis
import time
from serial import Serial, SerialException
import sys

height = 50
width = 20


def send(r, s):
    print(r.llen('new'))
    data = r.rpop('new')
    if data is not None and len(data) > 0:
        a = bytes([ord('*')]) + bytearray(data)
        #la = len(a)
        s.write(a)
        #time.sleep(.05)
        #data = s.read(la)
        #print(data)
        # size = s.write(data)
        # print('Data sent ({} bytes)'.format(size))


def serialConfigure(port_name, baudrate=9600):
    '''
        We use a very low baudrate by default because the USB port on the teensy
        enforce this value: http://www.pjrc.com/teensy/td_serial.html
    '''
    ser = Serial()
    ser.port = port_name
    ser.baudrate = baudrate
    ser.timeout = 5
    try:
        ser.open()
    except SerialException as e:
        sys.stderr.write("Could not open serial port %s: %s\n" % (ser.portstr, e))
        return

    ledsPerStrip = height * width
    print(ledsPerStrip)
    ser.write(ledsPerStrip.to_bytes(4, byteorder='little'))

    data = ser.read(4)
    print(data, int.from_bytes(data, byteorder='little'))

    ser.timeout = 1
    return ser


def serialDataConfigure(port_name, baudrate=115200):
    ser = Serial()
    ser.port = port_name
    ser.baudrate = baudrate
    try:
        ser.open()
    except SerialException as e:
        sys.stderr.write("Could not open serial port %s: %s\n" % (ser.portstr, e))
        return

    ser.timeout = 1
    return ser

if __name__ == "__main__":
    r = redis.Redis()
    r.hset('config', 'imgsize', height * width * 24)
    s = serialConfigure('/dev/ttyACM0')
    #s_data = serialDataConfigure('/dev/ttyUSB0')
    while True:
        while r.llen('new') > 0:
            send(r, s)
        time.sleep(1)