99 lines
2.4 KiB
Plaintext
99 lines
2.4 KiB
Plaintext
|
add_library('net')
|
||
|
import math
|
||
|
import jarray
|
||
|
|
||
|
gamma = 1.7
|
||
|
brightness = 4
|
||
|
errorCount = 0
|
||
|
framerate = 30
|
||
|
dimension = 0
|
||
|
|
||
|
# TODO: test with real serial
|
||
|
# https://www.pjrc.com/teensy/td_uart.html
|
||
|
|
||
|
long_line = False
|
||
|
|
||
|
ledTCP = None
|
||
|
data = None
|
||
|
|
||
|
current_px = 0
|
||
|
|
||
|
def TCPConfigure(server, port):
|
||
|
global ledTCP
|
||
|
ledTCP = Client(this, server, port)
|
||
|
|
||
|
def image2data(data):
|
||
|
offset = 1
|
||
|
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)
|
||
|
println(data)
|
||
|
ledTCP.write(data)
|
||
|
|
||
|
def prepare_data():
|
||
|
global data
|
||
|
data = jarray.zeros(dimension * 24 + 1 , "b")
|
||
|
data[0] = ord('*')
|
||
|
|
||
|
def setup():
|
||
|
global gammatable
|
||
|
global dimension
|
||
|
size(5, 8)
|
||
|
dimension = width * height
|
||
|
frameRate(framerate)
|
||
|
TCPConfigure("127.0.0.1", 9999)
|
||
|
if (errorCount > 0):
|
||
|
exit()
|
||
|
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()
|