host: Add gifserver

1 files changed, 80 insertions, 0 deletions

diff --git a/host/gifserver.py b/host/gifserver.py
new file mode 100755
index 0000000..7fbbfe3
--- /dev/null
+++ b/host/gifserver.py
@@ -0,0 +1,80 @@
+#!/usr/bin/env python
+
+import argparse, time, argparse, textwrap, threading, io
+from urllib.parse import urlparse
+from PIL import Image, GifImagePlugin, ImageSequence, ImageOps
+from flask import Flask, request
+
+import crap
+
+import config
+
+
+gif = None
+gif_cond = threading.Condition()
+
+def putgif(filelike):
+	global gif
+	with gif_cond:
+		gif = filelike
+		gif_cond.notify()
+
+
+def sendgifs(target):
+	thost, tport = target.split(':')
+	crapthing = crap.CRAPClient(thost, int(tport))
+	last_gif = None
+
+	while True:
+		with gif_cond:
+			gif_cond.wait_for(lambda: last_gif != gif)
+
+		last_gif = gif
+		img = Image.open(last_gif)
+		palette = img.getpalette()
+		last_frame = Image.new("RGBA", img.size)
+		sleepiness = min(img.info.get('duration', 100) / 1000.0, 10.0)
+
+		for frame in ImageSequence.Iterator(img):
+			#This works around a known bug in Pillow
+			#See also: http://stackoverflow.com/questions/4904940/python-converting-gif-frames-to-png
+			frame.putpalette(palette)
+			c = frame.convert("RGBA")
+
+			if 'background' in img.info and 'transparency' in img.info:
+				last_frame.paste(c, c)
+			else:
+				last_frame = c 
+
+			im = last_frame.copy()
+			im = ImageOps.fit(im, (config.display_width, config.display_height), Image.NEAREST, centering=(0.5, 0.5))
+			crapthing.sendframe(im.tobytes())
+			time.sleep(sleepiness)
+			if last_gif != gif:
+				break
+
+
+app = Flask(__name__)
+#app.debug = True
+
+@app.route('/', methods=('POST', 'GET'))
+def index():
+	if request.method == 'GET':
+		return textwrap.dedent('''
+			<form action="#" method="POST" enctype="multipart/form-data">
+				<input name="gif" type="file" accept="image/gif">
+				<input type="submit" value="upload">
+			</form>''')
+	else: #POST
+		putgif(io.BytesIO(request.files['gif'].read()))
+		return 'Done, thanks!'
+
+if __name__ == '__main__':
+	parser = argparse.ArgumentParser(description='Present a web interface to stream .gif files via CRAP to a matelight')
+	parser.add_argument('target', type=str, default='localhost:1337', nargs='?', help='Target matelight, host:port notation')
+	args = parser.parse_args()
+
+	gif_sender = threading.Thread(target=sendgifs, args=(args.target,), daemon=True)
+	gif_sender.start()
+	app.run()
+