forked from humberry/photo_text
/
photo_text.py
163 lines (139 loc) · 6.12 KB
/
photo_text.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
from __future__ import print_function
import collections
import photos
import scene
import sys
import Image
import ImageFont
import ImageDraw
import clipboard
fonttypes = ('Helvetica', 'Helvetica Bold', 'Arial', 'Avenir Book',
'Times New Roman', 'Baskerville', 'Courier',
'Chalkduster', 'American Typewriter', 'Verdana')
colors_dict = collections.OrderedDict([
('black', (0,0,0)), ('white', (1,1,1)), ('grey', (0.7,0.7,0.7)),
('red', (1,0,0)), ('green', (0,1,0)), ('blue', (0,0,1)),
('cyan', (0,1,1)), ('magenta', (1,0,1)), ('yellow', (1,1,0))])
def color(color_name):
return colors_dict[color_name.lower()]
def color_by_number(color_number):
return colors_dict[colors_dict.keys()[color_number % len(colors_dict)]]
def pic_save(image, width, height, text, font, fontsize, color, x, y, scale):
background = Image.new('RGBA', (width,height), 'white')
background.paste(image, (0, 0))
draw = ImageDraw.Draw(background)
offset = fontsize / 3.2 # offset is relative to the fontsize
fontsize *= scale
y = height - y
f = ImageFont.truetype(font, int(fontsize))
textsize = draw.textsize(text, font=f)
x -= textsize[0]/2
y -= (textsize[1]/2) + offset
draw.text((x, y), text, font=f, fill=color)
clipboard.set_image(background, format='jpeg', jpeg_quality=0.98)
photos.save_image(clipboard.get_image())
class TextButton(scene.Layer):
def __init__(self, inScene, inLoc, inText, inFgColor, inBgColor):
(theImage, theSize) = scene.render_text(inText, font_size=64)
super(self.__class__, self).__init__(scene.Rect(inLoc[0], inLoc[1], *theSize))
inScene.add_layer(self)
self.parent = inScene
self.text = inText
self.image = theImage
self.stroke = inFgColor # border color
self.stroke_weight = 2 # border width
self.background = inBgColor
def touch_began(self, touch):
self.parent.button_pressed(self.text)
class PhotoText(scene.Scene):
def __init__(self):
self.text = raw_input('Text to insert in the picture [Hello]: ') or 'Hello'
self.position = None
self.fontnr = 0 # Helvetica
self.colornr = 3 # red
self.fontsize = 48.0 # 48 point
self.img = photos.pick_image()
self.picsize = scene.Size(*self.img.size)
if self.picsize.h > self.picsize.w:
print('Sorry at the moment only landscape or square pictures are supported!')
self.img = None
self.picborder = None
self.btn_height = 0
if self.img:
scene.run(self)
else:
print('Good bye!')
def increase_font_size(self):
if 2 <= self.fontsize < 16:
self.fontsize += 1.0
elif self.fontsize >= 16:
self.fontsize += 16.0
def decrease_font_size(self):
if 3 < self.fontsize <= 16:
self.fontsize -= 1.0
elif self.fontsize > 16:
self.fontsize -= 16.0
def next_font(self):
self.fontnr += 1
def next_color(self):
self.colornr += 1
def save_image(self):
color = tuple([int(i * 255) for i in self.current_color()]) # covert scene color to PIL color
pic_save(self.img, self.picsize[0], self.picsize[1], self.text, self.current_font(), self.fontsize, color, self.position[0]*self.picscale, (self.position[1]-self.btn_height)*self.picscale, self.picscale)
sys.exit()
def cancel(self):
sys.exit(1)
def setup(self):
self.button_dict = collections.OrderedDict([
('+', self.increase_font_size),
('-', self.decrease_font_size),
('Font', self.next_font),
('Color', self.next_color),
('Save', self.save_image),
('Cancel', self.cancel) ])
fgColor = scene.Color(*color('blue'))
bgColor = scene.Color(*color('grey'))
loc = [0, 0]
for button_text in self.button_dict:
theButton = TextButton(self, loc, button_text, fgColor, bgColor)
self.btn_height = max(self.btn_height, theButton.frame.h)
loc[0] += theButton.frame.w + 4 # 4 pixels between each button
self.picratio = self.picsize.w / (self.picsize.h * 1.0)
x = 668 * self.picratio # where does 668 come from? should it be a constant?
if x <= 1024: # where does 1024 come from? should it be a constant?
y = 668
self.picborder = scene.Rect(x, self.btn_height, 1024, y)
else:
x = 1024
y = 1024 / self.picratio
self.picborder = scene.Rect(0, y+self.btn_height, x, 668-y)
self.position = scene.Size(x/2, y/2+self.btn_height)
self.picscale = self.picsize[0] / (x * 1.0)
self.layer = scene.Layer(scene.Rect(0, self.btn_height, x, y))
self.layer.image = scene.load_pil_image(self.img)
self.add_layer(self.layer)
def button_pressed(self, button_text):
#print('button_pressed({})'.format(button_text))
self.button_dict[button_text]() # call the associated function
def current_color(self):
return color_by_number(self.colornr)
def current_font(self):
return fonttypes[self.fontnr % len(fonttypes)]
def touch_moved(self, touch): # where does 748 come from? should it be a constant?
if ((0 < touch.location[0] < 1024)
and (self.btn_height < touch.location[1] < 748)):
self.position = touch.location
def touch_ended(self, touch):
self.touch_moved(touch)
def draw(self):
self.root_layer.update(self.dt)
self.root_layer.draw()
scene.tint(*self.current_color()) # draw the user's text
scene.text(self.text, self.current_font(), self.fontsize,
self.position[0], self.position[1], 5)
scene.fill(*color('white')) # watch+battery -> white background
scene.rect(0, 748, 1024, 20) # watch+battery
if photos.get_count():
PhotoText()
else:
print('Sorry no access or no pictures.')