blob: f6274e16e8f335771191180eefe15775adc3db8e [file] [log] [blame]
The Android Open Source Project52d4c302009-03-03 19:29:09 -08001#!/usr/bin/python
2
3import sys
4import StringIO
5try:
6 import Image, ImageDraw, ImageFont
7except ImportError, e:
8 print str(e)
9 print "Are you missing the Python Imaging Library? (apt-get install python-imaging)"
10 sys.exit(1)
11
12FONT_PATH = "/usr/share/fonts/truetype/msttcorefonts/arial.ttf"
13
14class Args(object):
15 def __init__(self, dest_name, size, circle_color, border_color,
16 letter_color, letter):
17 self.dest_name = dest_name
18 self.size = size
19 self.circle_color = circle_color
20 self.border_color = border_color
21 self.letter_color = letter_color
22 self.letter = letter
23
24def main(args):
25 data = process_args(args)
26 if data:
27 createImage(data)
28
29def process_args(args):
30 if not args or len(args) != 6:
31 usage()
32 return Args(*args)
33
34def usage():
35 print """Usage: %s <file_name> <size> <circle-color> <border-color> <letter-color> <letter>""" % sys.argv[0]
36 sys.exit(1)
37
38def createImage(data):
39 zoom = 4
40 rmin = -zoom/2
41 rmax = zoom/2
42 if zoom > 1:
43 r = range(-zoom/2, zoom/2+1)
44 else:
45 r = [ 0 ]
46 sz = int(data.size)
47 sz4 = sz * zoom
48
49 img = Image.new("RGBA", (sz4, sz4), (255,255,255,0))
50 draw = ImageDraw.Draw(img)
51
52 draw.ellipse((0, 0, sz4-zoom, sz4-zoom),
53 fill=data.circle_color, outline=None)
54 for i in r:
55 draw.ellipse((i, i, sz4-i-zoom, sz4-i-zoom),
56 fill=None, outline=data.border_color)
57
58 font = ImageFont.truetype(FONT_PATH, int(sz4 * .75))
59 tsx, tsy = draw.textsize(data.letter, font=font)
60
61 ptx = (sz4 - tsx) / 2
62 pty = (sz4 - tsy) / 2
63 for i in r:
64 draw.text((ptx + i, pty), data.letter, font=font, fill=data.letter_color)
65
66 img = img.resize((sz, sz), Image.BICUBIC)
67 img.save(data.dest_name, "PNG")
68 print "Saved", data.dest_name
69
70if __name__ == "__main__":
71 main(sys.argv[1:])