gifToPF.py

#!/bin/python

import sys
import random
from PIL import Image, ImageSequence

import time

packetSize = 200 #number of pixelflut commands per packet

def main(imgPath, offsetX=0, offsetY=0, algo=0):
    img = Image.open(imgPath)
    duration = img.info.get('duration', -1)
    if not hasattr(img, 'n_frames'):
        img.n_frames = 1
    print("converting image... 0/" + str(img.n_frames) + " done", end='\r')
    imageBuffer = []
    i = 0
    ii = ImageSequence.Iterator(img)
    for frame in ii:
        #frame.save('rendered/test_%02d.png' % i, 'PNG')
        imageBuffer.append(generatePFLines(frame, offsetX, offsetY, algo))
        i += 1
        print("converting image... " + str(i) + "/" + str(img.n_frames) + " done", end='\r')
    print()
    return {'frameBuffer': imageBuffer, 'duration': duration}


def swapRGB(rgb, fromIdx, toIdx):
    valFrom = rgb[fromIdx]
    valTo = rgb[toIdx]
    newRgb = list(rgb)
    for i in range(0, len(rgb)):
        if i == fromIdx:
            newRgb[i] = valTo
        elif i == toIdx:
            newRgb[i] = valFrom
        else:
            newRgb[i] = rgb[i]
    return tuple(newRgb)

#returns array of pixelflut commands for whole line
def generatePFLines(img, offX, offY, algo):
    # convert gif to rgb format (with respect to transparency)
    if img.format == "GIF":
        if img.info.get('transparency', -1) >= 0:
            img = img.convert('RGBA')
        else:
            img = img.convert('RGB')
    # check for existence of alpha channel based on length of pixel array
    hasAlpha = len(img.getpixel((0, 0))) == 4

    if algo == 0:
        # iterate through all pixels in frame
        lines = []
        for y in range(img.size[1]):
            line = ""
            for x in range(img.size[0]):
                rgb = img.getpixel((x, y))
                # rgb = swapRGB(rgb, 0, 2)
                if hasAlpha:
                    hexColor = '%02x%02x%02x%02x' % rgb
                else:
                    hexColor = '%02x%02x%02x' % rgb
                line += 'PX %d %d %s\n' % (x + offX, y + offY, hexColor)
            lines.append(line)
        return lines
    elif algo == 1:
        coordinates = []
        for y in range(img.size[1]):
            for x in range(img.size[0]):
                coordinates.append((x, y))
        random.shuffle(coordinates)

        packets = []
        for i in range(int(len(coordinates) / packetSize) + 1):
            packet = ""
            for j in range(packetSize):
                index = i * packetSize + j
                if index >= len(coordinates):
                    break
                x = coordinates[index][0]
                y = coordinates[index][1]
                rgb = img.getpixel(coordinates[index])
                # rgb = swapRGB(rgb, 0, 2)
                if hasAlpha:
                    hexColor = '%02x%02x%02x%02x' % rgb
                else:
                    hexColor = '%02x%02x%02x' % rgb
                packet += 'PX %d %d %s\n' % (x + offX, y + offY, hexColor)
            packets.append(packet)
        return packets



if __name__ == "__main__":
    main(sys.argv[1])