02d76f9a27ac22061732e93d607e78dafa23225e
[librarian.git] / librarian / cover.py
1 # -*- coding: utf-8 -*-
2 #
3 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
5 #
6 import re
7 import Image, ImageFont, ImageDraw, ImageFilter
8 from librarian import get_resource
9
10
11 class TextBox(object):
12     """Creates an Image with a series of centered strings."""
13
14     SHADOW_X = 3
15     SHADOW_Y = 3
16     SHADOW_BLUR = 3
17
18     def __init__(self, max_width, max_height, padding_x=None, padding_y=None):
19         if padding_x is None:
20             padding_x = self.SHADOW_X + self.SHADOW_BLUR
21         if padding_y is None:
22             padding_y = self.SHADOW_Y + self.SHADOW_BLUR
23
24         self.max_width = max_width
25         self.max_text_width = max_width - 2 * padding_x
26         self.padding_y = padding_y
27         self.height = padding_y
28         self.img = Image.new('RGBA', (max_width, max_height))
29         self.draw = ImageDraw.Draw(self.img)
30         self.shadow_img = None
31         self.shadow_draw = None
32
33     def skip(self, height):
34         """Skips some vertical space."""
35         self.height += height
36
37     def text(self, text, color='#000', font=None, line_height=20,
38              shadow_color=None):
39         """Writes some centered text."""
40         text = re.sub(r'\s+', ' ', text)
41         if shadow_color:
42             if not self.shadow_img:
43                 self.shadow_img = Image.new('RGBA', self.img.size)
44                 self.shadow_draw = ImageDraw.Draw(self.shadow_img)
45         while text:
46             line = text
47             line_width = self.draw.textsize(line, font=font)[0]
48             while line_width > self.max_text_width:
49                 parts = line.rsplit(' ', 1)
50                 if len(parts) == 1:
51                     line_width = self.max_text_width
52                     break
53                 line = parts[0]
54                 line_width = self.draw.textsize(line, font=font)[0]
55             line = line.strip() + ' '
56
57             pos_x = (self.max_width - line_width) / 2
58
59             if shadow_color:
60                 self.shadow_draw.text(
61                         (pos_x + self.SHADOW_X, self.height + self.SHADOW_Y),
62                         line, font=font, fill=shadow_color
63                 )
64
65             self.draw.text((pos_x, self.height), line, font=font, fill=color)
66             self.height += line_height
67             # go to next line
68             text = text[len(line):]
69
70     def image(self):
71         """Creates the actual Image object."""
72         image = Image.new('RGBA', (self.max_width,
73                                    self.height + self.padding_y))
74         if self.shadow_img:
75             shadow = self.shadow_img.filter(ImageFilter.BLUR)
76             image.paste(shadow, (0, 0), shadow)
77             image.paste(self.img, (0, 0), self.img)
78         else:
79             image.paste(self.img, (0, 0))
80         return image
81
82
83 class Cover(object):
84     """Abstract base class for cover images generator."""
85     width = 600
86     height = 800
87     background_color = '#fff'
88     background_img = None
89
90     author_top = 100
91     author_margin_left = 20
92     author_margin_right = 20
93     author_lineskip = 40
94     author_color = '#000'
95     author_shadow = None
96     author_font = None
97
98     title_top = 100
99     title_margin_left = 20
100     title_margin_right = 20
101     title_lineskip = 54
102     title_color = '#000'
103     title_shadow = None
104     title_font = None
105
106     logo_bottom = None
107     logo_width = None
108     uses_dc_cover = False
109
110     format = 'JPEG'
111
112     exts = {
113         'JPEG': 'jpg',
114         'PNG': 'png',
115         }
116
117     mime_types = {
118         'JPEG': 'image/jpeg',
119         'PNG': 'image/png',
120         }
121
122     def __init__(self, book_info):
123         self.author = ", ".join(auth.readable() for auth in book_info.authors)
124         self.title = book_info.title
125
126     def pretty_author(self):
127         """Allows for decorating author's name."""
128         return self.author
129
130     def pretty_title(self):
131         """Allows for decorating title."""
132         return self.title
133
134     def image(self):
135         img = Image.new('RGB', (self.width, self.height), self.background_color)
136
137         if self.background_img:
138             background = Image.open(self.background_img)
139             img.paste(background, None, background)
140             del background
141
142         # WL logo
143         if self.logo_width:
144             logo = Image.open(get_resource('res/wl-logo.png'))
145             logo = logo.resize((self.logo_width, logo.size[1] * self.logo_width / logo.size[0]))
146             img.paste(logo, ((self.width - self.logo_width) / 2, img.size[1] - logo.size[1] - self.logo_bottom))
147
148         top = self.author_top
149         tbox = TextBox(
150             self.width - self.author_margin_left - self.author_margin_right,
151             self.height - top,
152             )
153         author_font = self.author_font or ImageFont.truetype(
154             get_resource('fonts/DejaVuSerif.ttf'), 30)
155         tbox.text(self.pretty_author(), self.author_color, author_font,
156             self.author_lineskip, self.author_shadow)
157         text_img = tbox.image()
158         img.paste(text_img, (self.author_margin_left, top), text_img)
159
160         top += text_img.size[1] + self.title_top
161         tbox = TextBox(
162             self.width - self.title_margin_left - self.title_margin_right,
163             self.height - top,
164             )
165         title_font = self.author_font or ImageFont.truetype(
166             get_resource('fonts/DejaVuSerif.ttf'), 40)
167         tbox.text(self.pretty_title(), self.title_color, title_font,
168             self.title_lineskip, self.title_shadow)
169         text_img = tbox.image()
170         img.paste(text_img, (self.title_margin_left, top), text_img)
171
172         return img
173
174     def mime_type(self):
175         return self.mime_types[self.format]
176
177     def ext(self):
178         return self.exts[self.format]
179
180     def save(self, *args, **kwargs):
181         return self.image().save(format=self.format, *args, **kwargs)
182
183
184 class WLCover(Cover):
185     """Default Wolne Lektury cover generator."""
186     width = 600
187     height = 833
188     uses_dc_cover = True
189     author_font = ImageFont.truetype(
190         get_resource('fonts/JunicodeWL-Regular.ttf'), 20)
191     author_lineskip = 30
192     title_font = ImageFont.truetype(
193         get_resource('fonts/DejaVuSerif-Bold.ttf'), 30)
194     title_lineskip = 40
195     title_box_width = 350
196     bar_width = 35
197     background_color = '#444'
198     author_color = '#444'
199     default_background = get_resource('res/cover.png')
200     format = 'JPEG'
201
202     epoch_colors = {
203         u'Starożytność': '#9e3610',
204         u'Średniowiecze': '#564c09',
205         u'Renesans': '#8ca629',
206         u'Barok': '#a6820a',
207         u'Oświecenie': '#f2802e',
208         u'Romantyzm': '#db4b16',
209         u'Pozytywizm': '#961060',
210         u'Modernizm': '#7784e0',
211         u'Dwudziestolecie międzywojenne': '#3044cf',
212         u'Współczesność': '#06393d',
213     }
214
215     def __init__(self, book_info, image_cache=None):
216         super(WLCover, self).__init__(book_info)
217         self.kind = book_info.kind
218         self.epoch = book_info.epoch
219         if book_info.cover_url:
220             from urllib2 import urlopen
221             from StringIO import StringIO
222
223             url = book_info.cover_url
224             bg_src = None
225             if image_cache:
226                 from urllib import quote
227                 try:
228                     bg_src = urlopen(image_cache + quote(url, safe=""))
229                 except:
230                     pass
231             if bg_src is None:
232                 bg_src = urlopen(url)
233             self.background_img = StringIO(bg_src.read())
234             bg_src.close()
235         else:
236             self.background_img = self.default_background
237
238     def pretty_author(self):
239         return self.author.upper()
240
241     def image(self):
242         img = Image.new('RGB', (self.width, self.height), self.background_color)
243         draw = ImageDraw.Draw(img)
244
245         if self.epoch in self.epoch_colors:
246             epoch_color = self.epoch_colors[self.epoch]
247         else:
248             epoch_color = '#000'
249         draw.rectangle((0, 0, self.bar_width, self.height), fill=epoch_color)
250
251         if self.background_img:
252             src = Image.open(self.background_img)
253             trg_size = (self.width - self.bar_width, self.height)
254             if src.size[0] * trg_size[1] < src.size[1] * trg_size[0]:
255                 resized = (
256                     trg_size[0],
257                     src.size[1] * trg_size[0] / src.size[0]
258                 )
259                 cut = (resized[1] - trg_size[1]) / 2
260                 src = src.resize(resized)
261                 src = src.crop((0, cut, src.size[0], src.size[1] - cut))
262             else:
263                 resized = (
264                     src.size[0] * trg_size[1] / src.size[1],
265                     trg_size[1],
266                 )
267                 cut = (resized[0] - trg_size[0]) / 2
268                 src = src.resize(resized)
269                 src = src.crop((cut, 0, src.size[0] - cut, src.size[1]))
270
271             img.paste(src, (self.bar_width, 0))
272             del src
273
274         box = TextBox(self.title_box_width, self.height, padding_y=20)
275         box.text(self.pretty_author(),
276                  font=self.author_font,
277                  line_height=self.author_lineskip,
278                  color=self.author_color,
279                  shadow_color=self.author_shadow,
280                 )
281
282         box.skip(10)
283         box.draw.line((75, box.height, 275, box.height),
284                 fill=self.author_color, width=2)
285         box.skip(15)
286
287         box.text(self.pretty_title(),
288                  line_height=self.title_lineskip,
289                  font=self.title_font,
290                  color=epoch_color,
291                  shadow_color=self.title_shadow,
292                 )
293         box_img = box.image()
294
295         if self.kind == 'Liryka':
296             # top
297             box_top = 100
298         elif self.kind == 'Epika':
299             # bottom
300             box_top = self.height - 100 - box_img.size[1]
301         else:
302             # center
303             box_top = (self.height - box_img.size[1]) / 2
304
305         box_left = self.bar_width + (self.width - self.bar_width -
306                         box_img.size[0]) / 2
307         draw.rectangle((box_left, box_top,
308             box_left + box_img.size[0], box_top + box_img.size[1]),
309             fill='#fff')
310         img.paste(box_img, (box_left, box_top), box_img)
311
312         return img
313
314
315
316 class VirtualoCover(Cover):
317     width = 600
318     height = 730
319     author_top = 73
320     title_top = 73
321     logo_bottom = 25
322     logo_width = 250
323
324
325 class PrestigioCover(Cover):
326     width = 580
327     height = 783
328     background_img = get_resource('res/cover-prestigio.png')
329
330     author_top = 446
331     author_margin_left = 118
332     author_margin_right = 62
333     author_lineskip = 60
334     author_color = '#fff'
335     author_shadow = '#000'
336     author_font = ImageFont.truetype(get_resource('fonts/JunicodeWL-Italic.ttf'), 50)
337
338     title_top = 0
339     title_margin_left = 118
340     title_margin_right = 62
341     title_lineskip = 60
342     title_color = '#fff'
343     title_shadow = '#000'
344     title_font = ImageFont.truetype(get_resource('fonts/JunicodeWL-Italic.ttf'), 50)
345
346     def pretty_title(self):
347         return u"„%s”" % self.title
348
349
350 class BookotekaCover(Cover):
351     width = 2140
352     height = 2733
353     background_img = get_resource('res/cover-bookoteka.png')
354
355     author_top = 480
356     author_margin_left = 307
357     author_margin_right = 233
358     author_lineskip = 156
359     author_color = '#d9d919'
360     author_font = ImageFont.truetype(get_resource('fonts/JunicodeWL-Regular.ttf'), 130)
361
362     title_top = 400
363     title_margin_left = 307
364     title_margin_right = 233
365     title_lineskip = 168
366     title_color = '#d9d919'
367     title_font = ImageFont.truetype(get_resource('fonts/JunicodeWL-Regular.ttf'), 140)
368
369     format = 'PNG'
370
371
372 class GandalfCover(Cover):
373     width = 600
374     height = 730
375     background_img = get_resource('res/cover-gandalf.png')
376     author_font = ImageFont.truetype(get_resource('fonts/JunicodeWL-Regular.ttf'), 30)
377     title_font = ImageFont.truetype(get_resource('fonts/JunicodeWL-Regular.ttf'), 40)
378     logo_bottom = 25
379     logo_width = 250
380     format = 'PNG'