135 lines
3.5 KiB
Python
135 lines
3.5 KiB
Python
add_library('net')
|
|
import math
|
|
import jarray
|
|
import time
|
|
import struct
|
|
|
|
# Config, will be checked upstream
|
|
height = 5
|
|
width = 8
|
|
framerate = 30
|
|
#####################################
|
|
|
|
gamma = 1.7
|
|
brightness = 4
|
|
dimension = 0
|
|
|
|
# TODO: test with real serial
|
|
# https://www.pjrc.com/teensy/td_uart.html
|
|
|
|
long_line = True
|
|
|
|
ledTCP = None
|
|
data = None
|
|
|
|
current_px = 0
|
|
|
|
def TCPConfigure(server, port):
|
|
global ledTCP
|
|
ledTCP = Client(this, server, port)
|
|
|
|
def image2data(data):
|
|
offset = 0
|
|
pixel_nb = 0
|
|
for x in range(0, height):
|
|
pixel_line = pixels[pixel_nb:pixel_nb+width]
|
|
if long_line and pixel_nb/width%2 == 1:
|
|
pixel_line = reversed(pixel_line)
|
|
for px in pixel_line:
|
|
pixel = [colorWiring(px) for i in range(0, 8)]
|
|
imgmask = 0x800000
|
|
while imgmask != 0:
|
|
b = 0
|
|
for i in range(0, 8):
|
|
if ((pixel[i] & imgmask) != 0):
|
|
b |= (1 << i)
|
|
if b > 127:
|
|
# Convert to signed bytes (expected by jarray)
|
|
b -= 2**8
|
|
data[offset] = b
|
|
else:
|
|
data[offset] = b
|
|
offset += 1
|
|
imgmask >>= 1
|
|
pixel_nb +=1
|
|
|
|
def colorWiring(c):
|
|
red = (c & 0xFF0000) >> 16
|
|
green = (c & 0x00FF00) >> 8
|
|
blue = (c & 0x0000FF)
|
|
red = gammatable[red] >> 8
|
|
green = gammatable[green] >> 8
|
|
blue = gammatable[blue] >> 8
|
|
return (green << 16) | (red << 8) | (blue)
|
|
|
|
def send_TCP():
|
|
image2data(data)
|
|
ledTCP.write(data)
|
|
|
|
def prepare_data():
|
|
global data
|
|
data = jarray.zeros(dimension * 24, "b")
|
|
|
|
def receive_config():
|
|
max_height = jarray.zeros(1, "b")
|
|
max_width = jarray.zeros(1, "b")
|
|
max_framerate = jarray.zeros(1, "b")
|
|
while True:
|
|
available_bytes = ledTCP.available()
|
|
if available_bytes > 0:
|
|
break
|
|
time.sleep(1)
|
|
ledTCP.readBytes(max_height)
|
|
ledTCP.readBytes(max_width)
|
|
ledTCP.readBytes(max_framerate)
|
|
return max_height[0], max_width[0], max_framerate[0]
|
|
|
|
def check_config(max_height, max_width, max_framerate):
|
|
if height > max_height:
|
|
return False, "height cannot be higher than {}".format(max_height)
|
|
if width > max_width:
|
|
return False, "width cannot be higher than {}".format(max_width)
|
|
if framerate > max_framerate:
|
|
return False, "framerate cannot be higher than {}".format(max_framerate)
|
|
return True, None
|
|
|
|
def send_config():
|
|
ledTCP.write(height)
|
|
ledTCP.write(width)
|
|
ledTCP.write(framerate)
|
|
|
|
def setup():
|
|
global gammatable
|
|
global dimension
|
|
TCPConfigure("127.0.0.1", 9999)
|
|
max_height, max_width, max_framerate = receive_config()
|
|
|
|
good, reason = check_config(max_height, max_width, max_framerate)
|
|
if not good:
|
|
raise Exception(reason)
|
|
send_config()
|
|
size(width, height)
|
|
dimension = width * height
|
|
frameRate(framerate)
|
|
gammatable = [int((math.pow(i / 255.0, gamma) * 255.0 + 0.5) * brightness) for i in range(0, 256)]
|
|
prepare_data()
|
|
loadPixels()
|
|
for i in range(dimension):
|
|
pixels[i] = color(0, 0, 0)
|
|
updatePixels()
|
|
send_TCP()
|
|
|
|
def draw():
|
|
global current_px
|
|
pixels[current_px] = color(0, 255, 0)
|
|
if current_px == 0:
|
|
pixels[len(pixels) - 1] = color(0, 0, 0)
|
|
else:
|
|
pixels[current_px - 1] = color(0, 0, 0)
|
|
updatePixels()
|
|
if current_px == len(pixels) - 1:
|
|
current_px = 0
|
|
else:
|
|
current_px += 1
|
|
send_TCP()
|