1

#!/usr/bin/python2.5

     5

     6

from PIL import Image, ImageDraw, ImageFont, ImageEnhance

     7

from cStringIO import StringIO

     8

import random

     9

    10

    18

    20

    33

    34

fonts = {

    38

    42

    43

def randomize (seq) :

    44

    """

    45

        Returns the given sequence in random order as a list

    46

    """

    47

    48

    # copy

    49

    l = list(seq)

    50

    51

    # rearrange

    52

    random.shuffle(l)

    53

    54

    return l

    55

    56

def build_data (text, chars, line_colors) :

    57

    """

    58

        Returns a matrix of (text, color) tuples representing the data to render

    59

    60

        [ [ (str, str) ] ]

    61

    62

            text        - list of lines

    63

            chars       - list of random chars to interpse

    64

            line_colors - list of colors to draw the chars in

    65

    """

    66

    67

    data = []

    68

    69

    for line, char, color in zip(text, chars, line_colors) :

    70

        # pick position to place char

    71

        pos = random.randint(1, len(line) - 1)

    72

    73

        # split into three parts

    74

        data.append([

    75

                (line[:pos], "#000000"),

    76

                (char, color),

    77

                (line[pos:], "#000000"),

    78

            ])

    79

    80

    return data

    81

    82

def load_font (font_name, font_size) :

    83

    """

    84

        Load a font by name

    85

    """

    86

    87

    # load font

    88

    font_path = fonts[font_name]

    89

    font = ImageFont.truetype(font_path, font_size)

    90

    91

    return font

    92

    93

def render_img (data, font, background_color="#ffffff", line_spacing=0) :

    94

    """

    95

        Render the data (as from build_data) as an image, using the given PIL.ImageFont, and return the PIL Image object

    96

    """

    97

    98

    img_width = img_height = 0

    99

   100

    img_data = []

   101

   102

    # compute image width/height

   103

    for segments in data :

   104

        line_width = line_height = 0

   105

   106

        # build a new list of segments with additional info

   107

        line_segments = []

   108

   109

        for seg_text, seg_color in segments :

   110

            # compute rendered text size

   111

            seg_width, seg_height = font.getsize(seg_text)

   112

   113

            # update line_*

   114

            line_width += seg_width

   115

            line_height = max(line_height, seg_height)

   116

   117

            # build the new segments list

   118

            line_segments.append((seg_text, seg_color, seg_width))

   119

   120

        # update img_*

   121

        img_width = max(img_width, line_width)

   122

        img_height += line_height

   123

        img_data.append((line_segments, line_height))

   124

   125

    # calculate height needed for line spacing

   126

    img_height += (len(img_data) - 1) * line_spacing

   127

   128

    # create image

   129

    img = Image.new("RGB", (img_width, img_height), background_color)

   130

    draw = ImageDraw.Draw(img)

   131

   132

    # draw text

   133

    img_y = 0

   134

    for segments, line_height in img_data :

   135

        img_x = 0

   136

   137

        # draw each segment build above, incremeing along img_x

   138

        for seg_text, seg_color, seg_width in segments :

   139

            draw.text((img_x, img_y), seg_text, font=font, fill=seg_color)

   140

   141

            img_x += seg_width

   142

   143

        img_y += line_height + line_spacing

   144

   145

    return img

   146

   148

    """

   150

    """

   151

   152

    return ImageEnhance.Sharpness(img).enhance(factor)

   153

   154

def build_png (img) :

   155

    """

   156

        Write the given PIL.Image as a string, returning the raw binary data

   157

    """

   158

   159

    # render PNG output

   160

    buf = StringIO()

   161

    img.save(buf, "png")

   162

    data = buf.getvalue()

   163

   164

    return data

   165

   228

    """

   230

    """

   231

   235

   241

   242

def main () :

   249

   251

   252

if __name__ == '__main__' :

   253

    main()

   254