mategif.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import socket
import colorsys # for HSV-to-RGB-conversion
import time
import sys

from PIL import Image, GifImagePlugin, ImageSequence, ImageOps


# UDP_IP = "10.0.0.200"
UDP_PORT = 1337

ROWS = 16
COLS = 40

BRIGHTNESS = 1.0
GAMMA = 1.0

loop_last_frame = False
    
def send_array(data, hostname):
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    sock.sendto(data, (hostname, UDP_PORT))

def prepare_message(data, unpack=False, gamma=GAMMA):
    """Prepares the pixel data for transmission over UDP
    """
    # 4 bytes for future use as a crc32 checksum in network byte order.
    checksum = bytearray([0,0,0,0])
    data_as_bytes = bytearray()
    if unpack:
        for r, g, b, a in data:
            r = int(((r/255.0) ** gamma) * 255 * BRIGHTNESS)
            g = int(((g/255.0) ** gamma) * 255 * BRIGHTNESS)
            b = int(((b/255.0) ** gamma) * 255 * BRIGHTNESS)
            data_as_bytes += bytearray([r,g,b])
    else:
        data_as_bytes = bytearray(data)
        
    while len(data_as_bytes) < 1920:
        data_as_bytes += bytearray([0,0,0])
    
    message = data_as_bytes + checksum
    return message

def make_gradient(hue):
    """This creates a gradient for on hue value.
    """
    array = []
    num_pixels = ROWS * COLS
    step_size = 255.0 / float(num_pixels)
    for i in range(num_pixels):
        h = float(hue) / 360.0
        s = 1.0
        v = (i * step_size) / 255.0
        color = int(i * step_size)
        (r, g, b) = colorsys.hsv_to_rgb(h, s, v)
        r = r ** GAMMA
        g = g ** GAMMA
        b = b ** GAMMA
        array += [int(r * 255), int(g * 255), int(b * 255)]
    return array
    
def cycle_hue():
    """Cycles through all possible hue values and send it to the Mate 
    Light
    """
    for hue in range(360):
        data = make_gradient(hue)
        message = prepare_message(data)
        send_array(message)
        
def show_gif(filename, hostname, gamma, centering=0.5):
    img = Image.open(filename)
    palette = img.getpalette()
    last_frame = Image.new("RGBA", img.size)
    frames = []
    message = None
    
    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")
        sleep_time = img.info['duration'] / 1000.0
        
        # print img.info['background'], img.info['transparency']
        try:
            if img.info['background'] != img.info['transparency']:
                last_frame.paste(c, c)
            else:
                last_frame.paste(c, c)
                # last_frame = c
        except KeyError:
            last_frame = c 

        im = last_frame.copy()
        tw, th = im.size
        if (tw, th) != (40, 16):
            im = ImageOps.fit(im, (40, 16), Image.NEAREST, centering=(0.5, centering))
        else:
            pass

        data=list(im.getdata())
        message = prepare_message(data, unpack=True, gamma=gamma)
        send_array(message, hostname)     
        time.sleep(sleep_time)         

    if loop_last_frame and message is not None:
        while True:
            message = prepare_message(data, unpack=True, gamma=gamma)
            send_array(message, hostname)
            
    
if __name__ == '__main__':
    if len(sys.argv) < 3:
    	print("Usage: mategif.py HOSTNAME/IPADDRESS FILENAME [GAMMA]")
    else:
        hostname = sys.argv[1]
        filename = sys.argv[2]
        if len(sys.argv) == 4:
            if sys.argv[3] == '-l':
                loop_last_frame = True
                gamma = GAMMA
            else:
                gamma = float(sys.argv[3])
        else:
            gamma = GAMMA
	
        print("Transmitting '%s' to %s (press Ctrl+C to abort) ..." % (filename, hostname))
        try:
            centering = 0.5
            while True:                
                show_gif(filename, hostname, gamma, centering)
                centering += 0.1
                if centering > 1.0:
                    centering = 0.0
        except KeyboardInterrupt:
            print("Goodbye!")