diff options
Diffstat (limited to 'host/matelight')
| -rw-r--r-- | host/matelight/config.py | 9 | ||||
| -rw-r--r-- | host/matelight/host.py | 2 | ||||
| -rwxr-xr-x[-rw-r--r--] | host/matelight/listeners.py | 80 | ||||
| -rwxr-xr-x | host/matelight/nyancat-test.py | 2 | ||||
| -rw-r--r-- | host/matelight/renderers.py | 55 |
5 files changed, 104 insertions, 44 deletions
diff --git a/host/matelight/config.py b/host/matelight/config.py index 2bc97c1..b68bee5 100644 --- a/host/matelight/config.py +++ b/host/matelight/config.py @@ -1,4 +1,3 @@ -import bdflib # Used to read the bitmap font # Hard timeout in seconds after which (approximately) the rendering of a single item will be cut off RENDERER_TIMEOUT = 20.0 @@ -7,13 +6,7 @@ DEFAULT_IMAGE_DURATION = 10.0 # Default scrolling speed in pixels/second DEFAULT_SCROLL_SPEED = 4 # Pixels to leave blank between two letters -LETTER_SPACING = 1 - -#FONT = bdflib.reader.read_bdf(open('fonts/5x8.bdf').readlines()) -#FONT_WIDTH = 5 - -# Computed value -#FONT_PADDED_BINARY = ('{:0'+str(FONT_WIDTH+'b}').format +LETTER_SPACING = 0 # Display geometry # ┌─────────┐ ┌───┬───┬ ⋯ ┬───┬───┐ diff --git a/host/matelight/host.py b/host/matelight/host.py index 394c299..4fa8578 100644 --- a/host/matelight/host.py +++ b/host/matelight/host.py @@ -9,7 +9,7 @@ dev = usb.core.find(idVendor=0x1cbe, idProduct=0x0003) def sendframe(framedata): # not isinstance(framedata, np.array) or if framedata.shape != (DISPLAY_HEIGHT, DISPLAY_WIDTH, 3) or framedata.dtype != np.uint8: - raise ValueError('framedata must be a ({}, {}, 3)-numpy array of int8s. Got a {}-numpy array of {}'.format(DISPLAY_WIDTH, DISPLAY_HEIGHT, framedata.shape, framedata.dtype)) + raise ValueError('framedata must be a ({}, {}, 3)-numpy array of uint8s. Got a {}-numpy array of {}'.format(DISPLAY_HEIGHT, DISPLAY_WIDTH, framedata.shape, framedata.dtype)) for cy, cx in itertools.product(range(CRATES_Y), range(CRATES_X)): cratedata = framedata[cy*CRATE_HEIGHT:(cy+1)*CRATE_HEIGHT, cx*CRATE_WIDTH:(cx+1)*CRATE_WIDTH] diff --git a/host/matelight/listeners.py b/host/matelight/listeners.py index bb0ce72..91175cf 100644..100755 --- a/host/matelight/listeners.py +++ b/host/matelight/listeners.py @@ -1,19 +1,79 @@ +#!/usr/bin/env python3 from socketserver import * +import threading import zlib import struct +import host +import numpy as np +import time +import renderers +from PIL import Image, ImageSequence +from config import * +# Loading frame (for the big font file) +img = Image.open(open('../nyancat.png', 'rb')) +frame = np.array(img.convert('RGB').getdata(), dtype=np.uint8).reshape((DISPLAY_HEIGHT, DISPLAY_WIDTH, 3)) +host.sendframe(frame) +from font import * class ThreadingTCPServer(ThreadingMixIn, TCPServer): pass class ThreadingUDPServer(ThreadingMixIn, UDPServer): pass +default_renderer = renderers.TextRenderer('\x1B[91mFeed \x1B[92mme via \x1B[93mTCP on \x1B[94;101mml.jaseg.net:1337\x1B[0;91m ! \x1B[95mI can\x1B[96m parse\x1B[97m ANSI\x1B[92m color\x1B[93m codes\x1B[91m!\x1B[38;5;207m http://github.com/jaseg/matelight') +global renderer, count +renderer = default_renderer +count = 0 + class MateLightUDPHandler(BaseRequestHandler): - def handle(self): - data = self.request[0].strip() - if len(data) != FRAME_SIZE+4: - raise ValueError('Invalid frame size: Expected {}, got {}'.format(FRAME_SIZE+4, len(frame))) - frame = data[:-4] - crc1, = struct.unpack('!I', data[-4:]) - crc2 = zlib.crc32(frame), - if crc1 != crc2: - raise ValueError('Invalid frame CRC checksum: Expected {}, got {}'.format(crc2, crc1)) - socket.sendto(b'ACK', self.client_address) + def handle(self): + data = self.request[0].strip() + if len(data) != FRAME_SIZE: #+4 + raise ValueError('Invalid frame size: Expected {}, got {}'.format(FRAME_SIZE, len(frame))) #+4 + frame = data[:-4] + #crc1, = struct.unpack('!I', data[-4:]) + crc2 = zlib.crc32(frame), + if crc1 != crc2: + raise ValueError('Invalid frame CRC checksum: Expected {}, got {}'.format(crc2, crc1)) + #socket.sendto(b'ACK', self.client_address) + a = np.array(frame, dtype=np.uint8) + a.reshape((DISPLAY_HEIGHT, DISPLAY_WIDTH, 3)) + host.sendframe(a) + +class MateLightTCPTextHandler(BaseRequestHandler): + def handle(self): + try: + data = str(self.request.recv(1024).strip(), 'UTF-8') + if len(data) > 140: + self.request.sendall('TOO MUCH INFORMATION!\n') + return + global renderer, count + print(data+'\x1B[0m') + renderer = renderers.TextRenderer(data) + count = 3 + self.request.sendall(b'KTHXBYE!\n') + except: + pass + +TCPServer.allow_reuse_address = True +server = TCPServer(('', 1337), MateLightTCPTextHandler) +t = threading.Thread(target=server.serve_forever) +t.daemon = True +t.start() + +UDPServer.allow_reuse_address = True +userver = UDPServer(('', 1337), MateLightUDPHandler) +t = threading.Thread(target=userver.serve_forever) +t.daemon = True +t.start() + +while True: + global renderer, count + foo = renderer + if count == 0: + renderer = default_renderer + else: + count = count - 1 + for frame, delay in foo.frames(): + #print(list(frame.flatten())) + host.sendframe(np.swapaxes(frame, 0, 1)) + #time.sleep(delay) diff --git a/host/matelight/nyancat-test.py b/host/matelight/nyancat-test.py index aaabe85..73fd28e 100755 --- a/host/matelight/nyancat-test.py +++ b/host/matelight/nyancat-test.py @@ -20,7 +20,7 @@ bar = np.array(im.getdata(), dtype=np.uint8) foo = bar.reshape((DISPLAY_HEIGHT, 300, 3)) while True: - for i in range(20): + for i in range(60): for data in datas: host.sendframe(data) time.sleep(0.1) diff --git a/host/matelight/renderers.py b/host/matelight/renderers.py index 0e223a2..f507513 100644 --- a/host/matelight/renderers.py +++ b/host/matelight/renderers.py @@ -1,11 +1,13 @@ -import numpy as np +import time try: import re2 as re except ImportError: import re +import numpy as np from PIL import Image from pixelterm import xtermcolors from config import * +from font import * default_palette = [ (0x00, 0x00, 0x00), # 0 normal colors @@ -24,17 +26,19 @@ default_palette = [ (0xff, 0x00, 0xff), # 13 (0x00, 0xff, 0xff), # 14 (0xff, 0xff, 0xff)] # 15 +default_colors = (default_palette[8], default_palette[0]) class CharGenerator: def __init__(self, seq=None, lg=None, text=''): - settings = False, False, False, default_palette[8], default_palette[0] + settings = False, False, False, default_colors if lg: - settings = lg,bold, lg.blink, lg.underscore, lg.fg, lg.bg - self.bold, self.blink, self.underscore, self.fg, self.bg = settings + settings = lg.bold, lg.blink, lg.underscore, (lg.fg, lg.bg) + self.bold, self.blink, self.underscore, (self.fg, self.bg) = settings self.text = text - self.parse_escape_sequence(seq) + if seq: + self.parse_escape_sequence(seq) - def parse_escape_sequence(seq): + def parse_escape_sequence(self, seq): codes = list(map(int, seq[2:-1].split(';'))) fg, bg, reverse, i = self.fg, self.bg, False, 0 while i<len(codes): @@ -49,7 +53,7 @@ class CharGenerator: elif a == 49: bg = (0,0,0) elif a == 0: - fg, bg = (0,0,0), (0,0,0) + fg, bg = default_colors self.bold, self.blink, self.underscore = False, False, False elif a in range(30, 38): fg = default_palette[a-30] @@ -57,7 +61,7 @@ class CharGenerator: fg = default_palette[a-90+8] elif a in range(40, 48): bg = default_palette[a-40] - elif a in range(100, 108): + elif a in range(101, 108): bg = default_palette[a-100+8] elif a == 7: reverse = True @@ -68,20 +72,22 @@ class CharGenerator: elif a == 1: # Literally "bright", not bold. self.bold = True i += 1 - fg, bg = bg, fg if reverse else fg, bg + fg, bg = (bg, fg) if reverse else (fg, bg) self.fg, self.bg = fg, bg def generate_char(self, c, now): - fg, bg = self.bg, self.bg if self.blink and now%1.0 < 0.3 else self.fg, self.bg - glyph = font.glyphs_by_codepoint[c] + fg, bg = (self.bg, self.fg) if self.blink and now%1.0 < 0.3 else (self.fg, self.bg) + glyph = FONT.glyphs_by_codepoint[ord(c)] # Please forgive the string manipulation below. lookup = {'0': bg, '1': fg} - return [ list(map(lookup.get, FONT_PADDED_BINARY(int(row, 16)))) for row in glyph.get_data() ] + FONT_PADDED_BINARY = ('{:0'+str(glyph.bbW)+'b}').format + FONT_Y_PAD = [[bg]*glyph.bbW]*(DISPLAY_HEIGHT-FONT_HEIGHT) + return np.swapaxes(np.array([ list(map(lookup.get, FONT_PADDED_BINARY(int(row, 16))[:glyph.bbW])) for row in glyph.get_data() ] + FONT_Y_PAD, dtype=np.uint8), 0, 1) def generate(self, now): chars = [self.generate_char(c, now) for c in self.text] # This refers to inter-letter spacing - space = np.zeros((LETTER_SPACING, DISPLAY_HEIGHT, 3)) + space = np.zeros((LETTER_SPACING, DISPLAY_HEIGHT, 3), dtype=np.uint8) spaces = [space]*(len(chars)-1) everything = chars + spaces everything[::2] = chars @@ -96,25 +102,26 @@ class TextRenderer: """ generators = [] current_generator = CharGenerator() - for esc, char in r'(\x1B\[[0-9;]+m)|(.)'.finditer(text): + for match in re.finditer('(\x1B\[[0-9;]+m)|(.)', text): + esc, char = match.groups() if esc: if current_generator.text != '': generators.append(current_generator) current_generator = CharGenerator(esc, current_generator) elif char: current_generator.text += char - self.generators = generators + [current_generator] - - def frames(self, start): - now = time.time() - zeros = [np.zeros((DISPLAY_WIDTH, DISPLAY_HEIGHT, 3))] + generators = generators + [current_generator] + # Generate the actual frame buffer + zeros = [np.zeros((DISPLAY_WIDTH, DISPLAY_HEIGHT, 3), dtype=np.uint8)] # Pad the array with one screen's worth of zeros on both sides so the text fully scrolls through. - raw = np.concatenate([zeros]+[g.generate(now) for g in generators]+[zeros]) - w,h,_ = raw.size - for i in range(DISPLAY_WIDTH+w, 0, -1): - frame = raw[i:i+DISPLAY_WIDTH, :, :] + now = time.time() + self.raw = np.concatenate(zeros+[g.generate(now) for g in generators]+zeros) + + def frames(self): + w,h,_ = self.raw.shape + for i in range(0, w-DISPLAY_WIDTH, 2): + frame = self.raw[i:i+DISPLAY_WIDTH, :, :] yield frame, 1/DEFAULT_SCROLL_SPEED - raw = np.concatenate([zeros]+[g.generate(now) for g in generators]+[zeros]) class ImageRenderer: def __new__(cls, image_data): |