Covers with logos.
[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 from PIL import Image, ImageFont, ImageDraw, ImageFilter, ImageEnhance
8 from StringIO import StringIO
9 from librarian import get_resource, OutputFile, URLOpener
10
11
12 class Metric(object):
13     """Gets metrics from an object, scaling it by a factor."""
14     def __init__(self, obj, scale):
15         self._obj = obj
16         self._scale = float(scale)
17
18     def __getattr__(self, name):
19         src = getattr(self._obj, name)
20         if src and self._scale:
21             src = type(src)(self._scale * src)
22         return src
23
24
25 class TextBox(object):
26     """Creates an Image with a series of centered strings."""
27
28     SHADOW_X = 3
29     SHADOW_Y = 3
30     SHADOW_BLUR = 3
31
32     def __init__(self, max_width, max_height, padding_x=None, padding_y=None):
33         if padding_x is None:
34             padding_x = self.SHADOW_X + self.SHADOW_BLUR
35         if padding_y is None:
36             padding_y = self.SHADOW_Y + self.SHADOW_BLUR
37
38         self.max_width = max_width
39         self.max_text_width = max_width - 2 * padding_x
40         self.padding_y = padding_y
41         self.height = padding_y
42         self.img = Image.new('RGBA', (max_width, max_height))
43         self.draw = ImageDraw.Draw(self.img)
44         self.shadow_img = None
45         self.shadow_draw = None
46
47     def skip(self, height):
48         """Skips some vertical space."""
49         self.height += height
50
51     def text(self, text, color='#000', font=None, line_height=20,
52              shadow_color=None):
53         """Writes some centered text."""
54         text = re.sub(r'\s+', ' ', text)
55         if shadow_color:
56             if not self.shadow_img:
57                 self.shadow_img = Image.new('RGBA', self.img.size)
58                 self.shadow_draw = ImageDraw.Draw(self.shadow_img)
59         while text:
60             line = text
61             line_width = self.draw.textsize(line, font=font)[0]
62             while line_width > self.max_text_width:
63                 parts = line.rsplit(' ', 1)
64                 if len(parts) == 1:
65                     line_width = self.max_text_width
66                     break
67                 line = parts[0]
68                 line_width = self.draw.textsize(line, font=font)[0]
69             line = line.strip() + ' '
70
71             pos_x = (self.max_width - line_width) / 2
72
73             if shadow_color:
74                 self.shadow_draw.text(
75                         (pos_x + self.SHADOW_X, self.height + self.SHADOW_Y),
76                         line, font=font, fill=shadow_color
77                 )
78
79             self.draw.text((pos_x, self.height), line, font=font, fill=color)
80             self.height += line_height
81             # go to next line
82             text = text[len(line):]
83
84     def image(self):
85         """Creates the actual Image object."""
86         image = Image.new('RGBA', (self.max_width,
87                                    self.height + self.padding_y))
88         if self.shadow_img:
89             shadow = self.shadow_img.filter(ImageFilter.BLUR)
90             image.paste(shadow, (0, 0), shadow)
91             image.paste(self.img, (0, 0), self.img)
92         else:
93             image.paste(self.img, (0, 0))
94         return image
95
96
97 class Cover(object):
98     """Abstract base class for cover images generator."""
99     width = 600
100     height = 800
101     background_color = '#fff'
102     background_img = None
103
104     author_top = 100
105     author_margin_left = 20
106     author_margin_right = 20
107     author_lineskip = 40
108     author_color = '#000'
109     author_shadow = None
110     author_font_ttf = get_resource('fonts/DejaVuSerif.ttf')
111     author_font_size = 30
112
113     title_top = 100
114     title_margin_left = 20
115     title_margin_right = 20
116     title_lineskip = 54
117     title_color = '#000'
118     title_shadow = None
119     title_font_ttf = get_resource('fonts/DejaVuSerif.ttf')
120     title_font_size = 40
121
122     logo_bottom = None
123     logo_width = None
124     uses_dc_cover = False
125
126     format = 'JPEG'
127     scale = 1
128
129     exts = {
130         'JPEG': 'jpg',
131         'PNG': 'png',
132         }
133
134     mime_types = {
135         'JPEG': 'image/jpeg',
136         'PNG': 'image/png',
137         }
138
139     def __init__(self, book_info, format=None, width=None, height=None):
140         self.author = ", ".join(auth.readable() for auth in book_info.authors)
141         self.title = book_info.title
142         if format is not None:
143             self.format = format
144         scale = max(float(width or 0) / self.width, float(height or 0) / self.height)
145         if scale:
146             self.scale = scale
147
148     def pretty_author(self):
149         """Allows for decorating author's name."""
150         return self.author
151
152     def pretty_title(self):
153         """Allows for decorating title."""
154         return self.title
155
156     def image(self):
157         metr = Metric(self, self.scale)
158         img = Image.new('RGB', (metr.width, metr.height), self.background_color)
159
160         if self.background_img:
161             background = Image.open(self.background_img)
162             img.paste(background, None, background)
163             del background
164
165         # WL logo
166         if metr.logo_width:
167             logo = Image.open(get_resource('res/wl-logo.png'))
168             logo = logo.resize((metr.logo_width, logo.size[1] * metr.logo_width / logo.size[0]))
169             img.paste(logo, ((metr.width - metr.logo_width) / 2, img.size[1] - logo.size[1] - metr.logo_bottom))
170
171         top = metr.author_top
172         tbox = TextBox(
173             metr.width - metr.author_margin_left - metr.author_margin_right,
174             metr.height - top,
175             )
176             
177         author_font = ImageFont.truetype(
178             self.author_font_ttf, metr.author_font_size)
179         tbox.text(self.pretty_author(), self.author_color, author_font,
180             metr.author_lineskip, self.author_shadow)
181         text_img = tbox.image()
182         img.paste(text_img, (metr.author_margin_left, top), text_img)
183
184         top += text_img.size[1] + metr.title_top
185         tbox = TextBox(
186             metr.width - metr.title_margin_left - metr.title_margin_right,
187             metr.height - top,
188             )
189         title_font = ImageFont.truetype(
190             self.title_font_ttf, metr.title_font_size)
191         tbox.text(self.pretty_title(), self.title_color, title_font,
192             metr.title_lineskip, self.title_shadow)
193         text_img = tbox.image()
194         img.paste(text_img, (metr.title_margin_left, top), text_img)
195
196         return img
197
198     def mime_type(self):
199         return self.mime_types[self.format]
200
201     def ext(self):
202         return self.exts[self.format]
203
204     def save(self, *args, **kwargs):
205         return self.image().save(format=self.format, quality=95, *args, **kwargs)
206
207     def output_file(self, *args, **kwargs):
208         imgstr = StringIO()
209         self.save(imgstr, *args, **kwargs)
210         return OutputFile.from_string(imgstr.getvalue())
211
212
213 class WLCover(Cover):
214     """Default Wolne Lektury cover generator."""
215     width = 600
216     height = 833
217     uses_dc_cover = True
218     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
219     author_font_size = 20
220     author_lineskip = 30
221     title_font_ttf = get_resource('fonts/DejaVuSerif-Bold.ttf')
222     title_font_size = 30
223     title_lineskip = 40
224     title_box_width = 350
225     
226     box_top_margin = 100
227     box_bottom_margin = 100
228     box_padding_y = 20
229     box_above_line = 10
230     box_below_line = 15
231     box_line_left = 75
232     box_line_right = 275
233     box_line_width = 2
234
235     logo_top = 15
236     logo_width = 140
237
238     bar_width = 35
239     background_color = '#444'
240     author_color = '#444'
241     default_background = get_resource('res/cover.png')
242     format = 'JPEG'
243
244     epoch_colors = {
245         u'Starożytność': '#9e3610',
246         u'Średniowiecze': '#564c09',
247         u'Renesans': '#8ca629',
248         u'Barok': '#a6820a',
249         u'Oświecenie': '#f2802e',
250         u'Romantyzm': '#db4b16',
251         u'Pozytywizm': '#961060',
252         u'Modernizm': '#7784e0',
253         u'Dwudziestolecie międzywojenne': '#3044cf',
254         u'Współczesność': '#06393d',
255     }
256
257     def __init__(self, book_info, format=None, width=None, height=None, with_logo=False):
258         super(WLCover, self).__init__(book_info, format=format, width=width, height=height)
259         self.kind = book_info.kind
260         self.epoch = book_info.epoch
261         self.with_logo = with_logo
262         if book_info.cover_url:
263             url = book_info.cover_url
264             bg_src = None
265             if bg_src is None:
266                 bg_src = URLOpener().open(url)
267             self.background_img = StringIO(bg_src.read())
268             bg_src.close()
269         else:
270             self.background_img = self.default_background
271
272     def pretty_author(self):
273         return self.author.upper()
274
275     def image(self):
276         metr = Metric(self, self.scale)
277         img = Image.new('RGB', (metr.width, metr.height), self.background_color)
278         draw = ImageDraw.Draw(img)
279
280         if self.epoch in self.epoch_colors:
281             epoch_color = self.epoch_colors[self.epoch]
282         else:
283             epoch_color = '#000'
284         draw.rectangle((0, 0, metr.bar_width, metr.height), fill=epoch_color)
285
286         if self.background_img:
287             src = Image.open(self.background_img)
288             trg_size = (metr.width - metr.bar_width, metr.height)
289             if src.size[0] * trg_size[1] < src.size[1] * trg_size[0]:
290                 resized = (
291                     trg_size[0],
292                     src.size[1] * trg_size[0] / src.size[0]
293                 )
294                 cut = (resized[1] - trg_size[1]) / 2
295                 src = src.resize(resized, Image.ANTIALIAS)
296                 src = src.crop((0, cut, src.size[0], src.size[1] - cut))
297             else:
298                 resized = (
299                     src.size[0] * trg_size[1] / src.size[1],
300                     trg_size[1],
301                 )
302                 cut = (resized[0] - trg_size[0]) / 2
303                 src = src.resize(resized, Image.ANTIALIAS)
304                 src = src.crop((cut, 0, src.size[0] - cut, src.size[1]))
305
306             img.paste(src, (metr.bar_width, 0))
307             del src
308
309         box = TextBox(metr.title_box_width, metr.height, padding_y=metr.box_padding_y)
310         author_font = ImageFont.truetype(
311             self.author_font_ttf, metr.author_font_size)
312         box.text(self.pretty_author(),
313                  font=author_font,
314                  line_height=metr.author_lineskip,
315                  color=self.author_color,
316                  shadow_color=self.author_shadow,
317                 )
318
319         box.skip(metr.box_above_line)
320         box.draw.line((metr.box_line_left, box.height, metr.box_line_right, box.height),
321                 fill=self.author_color, width=metr.box_line_width)
322         box.skip(metr.box_below_line)
323
324         title_font = ImageFont.truetype(
325             self.title_font_ttf, metr.title_font_size)
326         box.text(self.pretty_title(),
327                  line_height=metr.title_lineskip,
328                  font=title_font,
329                  color=epoch_color,
330                  shadow_color=self.title_shadow,
331                 )
332
333         if self.with_logo:
334             logo = Image.open(get_resource('res/wl-logo-mono.png'))
335             logo = logo.resize((metr.logo_width, logo.size[1] * metr.logo_width / logo.size[0]), Image.ANTIALIAS)
336             alpha = logo.split()[3]
337             alpha = ImageEnhance.Brightness(alpha).enhance(.75)
338             logo.putalpha(alpha)
339             box.skip(metr.logo_top + logo.size[1])
340
341         box_img = box.image()
342
343         if self.kind == 'Liryka':
344             # top
345             box_top = metr.box_top_margin
346         elif self.kind == 'Epika':
347             # bottom
348             box_top = metr.height - metr.box_bottom_margin - box_img.size[1]
349         else:
350             # center
351             box_top = (metr.height - box_img.size[1]) / 2
352
353         box_left = metr.bar_width + (metr.width - metr.bar_width -
354                         box_img.size[0]) / 2
355         draw.rectangle((box_left, box_top,
356             box_left + box_img.size[0], box_top + box_img.size[1]),
357             fill='#fff')
358         img.paste(box_img, (box_left, box_top), box_img)
359
360         if self.with_logo:
361             img.paste(logo, 
362                 (box_left + (box_img.size[0] - logo.size[0]) / 2,
363                     box_top + box_img.size[1] - metr.box_padding_y - logo.size[1]), mask=logo)
364
365         return img
366
367
368 class LogoWLCover(WLCover):
369     gradient_height = 90
370     gradient_logo_height = 60
371     gradient_logo_margin_right = 30
372     gradient_logo_spacing = 40
373     gradient_color = '#000'
374     gradient_opacity = .6
375     gradient_logos = [
376         'res/wl-logo-white.png',
377         'res/fnp-logo-white.png',
378     ]
379
380     def image(self):
381         img = super(LogoWLCover, self).image()
382         metr = Metric(self, self.scale)
383         gradient = Image.new('RGBA', (metr.width - metr.bar_width, metr.gradient_height), self.gradient_color)
384         gradient_mask = Image.new('L', (metr.width - metr.bar_width, metr.gradient_height))
385         draw = ImageDraw.Draw(gradient_mask)
386         for line in range(0, metr.gradient_height):
387             draw.line((0, line, metr.width - metr.bar_width, line), fill=int(255 * self.gradient_opacity * line / metr.gradient_height))
388         img.paste(gradient, 
389             (metr.bar_width, metr.height - metr.gradient_height), mask=gradient_mask)
390
391         cursor = metr.width - metr.gradient_logo_margin_right
392         logo_top = metr.height - metr.gradient_height / 2  - metr.gradient_logo_height / 2
393         for logo_path in self.gradient_logos[::-1]:
394             logo = Image.open(get_resource(logo_path))
395             logo = logo.resize(
396                 (logo.size[0] * metr.gradient_logo_height / logo.size[1], metr.gradient_logo_height),
397                 Image.ANTIALIAS)
398             cursor -= logo.size[0]
399             img.paste(logo, (cursor, logo_top), mask=logo)
400             cursor -= metr.gradient_logo_spacing
401
402         return img
403
404
405 class EbookpointCover(LogoWLCover):
406     gradient_logo_height = 58
407     gradient_logo_spacing = 25
408     gradient_logos = [
409         'res/ebookpoint-logo-white.png',
410         'res/wl-logo-white.png',
411         'res/fnp-logo-white.png',
412     ]
413
414
415 class VirtualoCover(Cover):
416     width = 600
417     height = 730
418     author_top = 73
419     title_top = 73
420     logo_bottom = 25
421     logo_width = 250
422
423
424 class PrestigioCover(Cover):
425     width = 580
426     height = 783
427     background_img = get_resource('res/cover-prestigio.png')
428
429     author_top = 446
430     author_margin_left = 118
431     author_margin_right = 62
432     author_lineskip = 60
433     author_color = '#fff'
434     author_shadow = '#000'
435     author_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
436     author_font_size = 50
437
438     title_top = 0
439     title_margin_left = 118
440     title_margin_right = 62
441     title_lineskip = 60
442     title_color = '#fff'
443     title_shadow = '#000'
444     title_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
445     title_font_size = 50
446
447     def pretty_title(self):
448         return u"„%s”" % self.title
449
450
451 class BookotekaCover(Cover):
452     width = 2140
453     height = 2733
454     background_img = get_resource('res/cover-bookoteka.png')
455
456     author_top = 480
457     author_margin_left = 307
458     author_margin_right = 233
459     author_lineskip = 156
460     author_color = '#d9d919'
461     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
462     author_font_size = 130
463
464     title_top = 400
465     title_margin_left = 307
466     title_margin_right = 233
467     title_lineskip = 168
468     title_color = '#d9d919'
469     title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
470     title_font_size = 140
471
472     format = 'PNG'
473
474
475 class GandalfCover(Cover):
476     width = 600
477     height = 730
478     background_img = get_resource('res/cover-gandalf.png')
479     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
480     author_font_size = 30
481     title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
482     title_font_size = 40
483     logo_bottom = 25
484     logo_width = 250
485     format = 'PNG'