matrix.py

#!/usr/bin/env python

import sys, os
import cgi, cgitb
import random
import string
from PIL import Image, ImageDraw, ImageFont, ImageFilter
from StringIO import StringIO

TERMINUS = ImageFont.load("terminus-bold.pil")
WHITEMASK = lambda x, y: Image.new("L", (x, y), color=255)
BLACKMASK = lambda x, y: Image.new("L", (x, y), color=0)

ADDFILTER = ImageFilter.MaxFilter(4)
RANDCHARS = string.ascii_letters + string.digits + "!@#$%^&*;',.?/\\"

def matrix(cols, rows):
    show        = [random.randint(128, 255) * (random.randint(0, 7) > 1) for i in range(cols)]
    glow        = [0] * cols
    delayedglow = [0] * cols
    last        = [random.randint(1, 3) for i in range(cols)]

    for i in range(rows):
        ret = []
        char = ""

        for j in range(cols):
            if show[j]:
                char = random.choice(RANDCHARS)
            else:
                char = " "

            if glow[j]:
                show[j] = min(255, show[j] + random.randint(500 / rows, 1000 / rows) )
            else:
                show[j] = min(255, show[j] - random.randint(4, last[j] * (400 / rows)) )

            show[j] = max(show[j], 0)

            glow[j] = max(glow[j] - 1, 0)
            delayedglow[j] = max(delayedglow[j] - 1, 0)

            if show[j] < 32:
                if not random.randint(0, 30000 / rows):
                    glow[j] = 10
                    try:
                        delayedglow[j-1] = 8
                        delayedglow[j+1] = 8
                    except IndexError:
                        pass

                elif not random.randint(0, 3000 / rows):
                    glow[j] = 10

            if delayedglow[j] == 1: glow[j] = 8

            ret.append((char, show[j]))

        yield ret

    raise StopIteration

def generateImage(width, height, font, alpha=128):
    bigwidth, bigheight   = width * 2, height * 2
    fontwidth, fontheight = font.getsize(" ")

    ret  = Image.new("RGBA", (bigwidth, bigheight), color=(0, 0, 0, alpha))
    ret1 = Image.new("RGBA", (bigwidth, bigheight), color=(128, 255, 128, 255))
    ret2 = Image.new("RGBA", (bigwidth, bigheight), color=(0,   255, 0,   255))

    mask = BLACKMASK(bigwidth, bigheight)

    canvas = ImageDraw.Draw(mask)

    for i, layer in enumerate(matrix(bigwidth / fontwidth, (bigheight / fontheight) + 2)):

        for j, (char, alpha) in enumerate(layer):
            canvas.text((2 + (fontwidth * j), 2 + (fontheight * i)), char, font=font, fill=alpha)

    blurred = mask.filter(ImageFilter.BLUR)

    ret1.putalpha(mask)
    ret2.putalpha(blurred)

    ret1flat = ret1.convert("RGB")
    ret2flat = ret2.convert("RGB")

    ret.paste(ret2flat, (0, 0), ret2)
    ret.paste(ret1flat, (0, 0), ret1)

    resized = ret.resize((width, height), Image.BICUBIC)

    #return ret
    return resized

def norMain():
    img = generateImage(200, 200, TERMINUS)
    img.save("test.png")

def cgiMain():
    form = {}
    cgiforms = cgi.FieldStorage()
    size  = 125
    alpha = 256

    for val in cgiforms:
        form[val.lower()] = cgiforms[val].value

    if "alpha" in form:
        try:
            alpha = int(form["alpha"])
        except ValueError:
            pass

    if "size" in form:
        try:
            size = int(form["size"])
        except ValueError:
            pass

    size = max(50, size)
    size = min(size, 300)

    alpha = max(0, alpha)
    alpha = min(alpha, 255)

    output = StringIO()
    img = generateImage(size, size, TERMINUS, alpha=alpha)
    img.save(output, "PNG")

    print("Content-type: image/png\n")
    print(output.getvalue())

if __name__ == "__main__":

    if "REMOTE_ADDR" in os.environ:
        cgitb.enable()
        cgiMain()

    else:
        norMain()