0266b4ed0ac67f8eb00973ec94c449f8eecd319a
[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     scale_after = 1
129
130     exts = {
131         'JPEG': 'jpg',
132         'PNG': 'png',
133         }
134
135     mime_types = {
136         'JPEG': 'image/jpeg',
137         'PNG': 'image/png',
138         }
139
140     def __init__(self, book_info, format=None, width=None, height=None):
141         self.author = ", ".join(auth.readable() for auth in book_info.authors)
142         self.title = book_info.title
143         if format is not None:
144             self.format = format
145         if width and height:
146             self.height = height * self.width / width
147         scale = max(float(width or 0) / self.width, float(height or 0) / self.height)
148         if scale >= 1:
149             self.scale = scale
150         elif scale:
151             self.scale_after = scale
152
153     def pretty_author(self):
154         """Allows for decorating author's name."""
155         return self.author
156
157     def pretty_title(self):
158         """Allows for decorating title."""
159         return self.title
160
161     def image(self):
162         metr = Metric(self, self.scale)
163         img = Image.new('RGB', (metr.width, metr.height), self.background_color)
164
165         if self.background_img:
166             background = Image.open(self.background_img)
167             img.paste(background, None, background)
168             del background
169
170         # WL logo
171         if metr.logo_width:
172             logo = Image.open(get_resource('res/wl-logo.png'))
173             logo = logo.resize((metr.logo_width, logo.size[1] * metr.logo_width / logo.size[0]))
174             img.paste(logo, ((metr.width - metr.logo_width) / 2, img.size[1] - logo.size[1] - metr.logo_bottom))
175
176         top = metr.author_top
177         tbox = TextBox(
178             metr.width - metr.author_margin_left - metr.author_margin_right,
179             metr.height - top,
180             )
181             
182         author_font = ImageFont.truetype(
183             self.author_font_ttf, metr.author_font_size)
184         tbox.text(self.pretty_author(), self.author_color, author_font,
185             metr.author_lineskip, self.author_shadow)
186         text_img = tbox.image()
187         img.paste(text_img, (metr.author_margin_left, top), text_img)
188
189         top += text_img.size[1] + metr.title_top
190         tbox = TextBox(
191             metr.width - metr.title_margin_left - metr.title_margin_right,
192             metr.height - top,
193             )
194         title_font = ImageFont.truetype(
195             self.title_font_ttf, metr.title_font_size)
196         tbox.text(self.pretty_title(), self.title_color, title_font,
197             metr.title_lineskip, self.title_shadow)
198         text_img = tbox.image()
199         img.paste(text_img, (metr.title_margin_left, top), text_img)
200
201         return img
202
203     def final_image(self):
204         img = self.image()
205         if self.scale_after != 1:
206             img = img.resize((
207                     int(round(img.size[0] * self.scale_after)),
208                     int(round(img.size[1] * self.scale_after))),
209                 Image.ANTIALIAS)
210         return img
211
212     def mime_type(self):
213         return self.mime_types[self.format]
214
215     def ext(self):
216         return self.exts[self.format]
217
218     def save(self, *args, **kwargs):
219         default_kwargs = {
220                 'format': self.format,
221                 'quality': 95,
222         }
223         default_kwargs.update(kwargs)
224         return self.final_image().save(*args, **default_kwargs)
225
226     def output_file(self, *args, **kwargs):
227         imgstr = StringIO()
228         self.save(imgstr, *args, **kwargs)
229         return OutputFile.from_string(imgstr.getvalue())
230
231
232 class WLCover(Cover):
233     """Wolne Lektury cover without logos."""
234     width = 600
235     height = 833
236     uses_dc_cover = True
237     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
238     author_font_size = 20
239     author_lineskip = 30
240     title_font_ttf = get_resource('fonts/DejaVuSerif-Bold.ttf')
241     title_font_size = 30
242     title_lineskip = 40
243     title_box_width = 350
244     
245     box_top_margin = 100
246     box_bottom_margin = 100
247     box_padding_y = 20
248     box_above_line = 10
249     box_below_line = 15
250     box_line_left = 75
251     box_line_right = 275
252     box_line_width = 2
253
254     logo_top = 15
255     logo_width = 140
256
257     bar_width = 35
258     bar_color = '#000'
259     box_position = 'middle'
260     background_color = '#444'
261     author_color = '#444'
262     background_img = get_resource('res/cover.png')
263     format = 'JPEG'
264
265     epoch_colors = {
266         u'Starożytność': '#9e3610',
267         u'Średniowiecze': '#564c09',
268         u'Renesans': '#8ca629',
269         u'Barok': '#a6820a',
270         u'Oświecenie': '#f2802e',
271         u'Romantyzm': '#db4b16',
272         u'Pozytywizm': '#961060',
273         u'Modernizm': '#7784e0',
274         u'Dwudziestolecie międzywojenne': '#3044cf',
275         u'Współczesność': '#06393d',
276     }
277
278     kind_box_position = {
279         u'Liryka': 'top',
280         u'Epika': 'bottom',
281     }
282
283     def __init__(self, book_info, format=None, width=None, height=None):
284         super(WLCover, self).__init__(book_info, format=format, width=width, height=height)
285         # Set box position.
286         self.box_position = book_info.cover_box_position or \
287             self.kind_box_position.get(book_info.kind, self.box_position)
288         # Set bar color.
289         if book_info.cover_bar_color == 'none':
290             self.bar_width = 0
291         else:
292             self.bar_color = book_info.cover_bar_color or \
293                 self.epoch_colors.get(book_info.epoch, self.bar_color)
294         # Set title color.
295         self.title_color = self.epoch_colors.get(book_info.epoch, self.title_color)
296
297         if book_info.cover_url:
298             url = book_info.cover_url
299             bg_src = None
300             if bg_src is None:
301                 bg_src = URLOpener().open(url)
302             self.background_img = StringIO(bg_src.read())
303             bg_src.close()
304
305     def pretty_author(self):
306         return self.author.upper()
307
308     def add_box(self, img):
309         if self.box_position == 'none':
310             return img
311
312         metr = Metric(self, self.scale)
313
314         # Write author name.
315         box = TextBox(metr.title_box_width, metr.height, padding_y=metr.box_padding_y)
316         author_font = ImageFont.truetype(
317             self.author_font_ttf, metr.author_font_size)
318         box.text(self.pretty_author(),
319                  font=author_font,
320                  line_height=metr.author_lineskip,
321                  color=self.author_color,
322                  shadow_color=self.author_shadow,
323                 )
324
325         box.skip(metr.box_above_line)
326         box.draw.line((metr.box_line_left, box.height, metr.box_line_right, box.height),
327                 fill=self.author_color, width=metr.box_line_width)
328         box.skip(metr.box_below_line)
329
330         # Write title.
331         title_font = ImageFont.truetype(
332             self.title_font_ttf, metr.title_font_size)
333         box.text(self.pretty_title(),
334                  line_height=metr.title_lineskip,
335                  font=title_font,
336                  color=self.title_color,
337                  shadow_color=self.title_shadow,
338                 )
339
340         box_img = box.image()
341
342         # Find box position.
343         if self.box_position == 'top':
344             box_top = metr.box_top_margin
345         elif self.box_position == 'bottom':
346             box_top = metr.height - metr.box_bottom_margin - box_img.size[1]
347         else:   # Middle.
348             box_top = (metr.height - box_img.size[1]) / 2
349
350         box_left = metr.bar_width + (metr.width - metr.bar_width -
351                         box_img.size[0]) / 2
352
353         # Draw the white box.
354         ImageDraw.Draw(img).rectangle((box_left, box_top,
355             box_left + box_img.size[0], box_top + box_img.size[1]),
356             fill='#fff')
357         # Paste the contents into the white box.
358         img.paste(box_img, (box_left, box_top), box_img)
359         return img
360
361     def image(self):
362         metr = Metric(self, self.scale)
363         img = Image.new('RGB', (metr.width, metr.height), self.background_color)
364         draw = ImageDraw.Draw(img)
365
366         draw.rectangle((0, 0, metr.bar_width, metr.height), fill=self.bar_color)
367
368         if self.background_img:
369             src = Image.open(self.background_img)
370             trg_size = (metr.width - metr.bar_width, metr.height)
371             if src.size[0] * trg_size[1] < src.size[1] * trg_size[0]:
372                 resized = (
373                     trg_size[0],
374                     src.size[1] * trg_size[0] / src.size[0]
375                 )
376                 cut = (resized[1] - trg_size[1]) / 2
377                 src = src.resize(resized, Image.ANTIALIAS)
378                 src = src.crop((0, cut, src.size[0], src.size[1] - cut))
379             else:
380                 resized = (
381                     src.size[0] * trg_size[1] / src.size[1],
382                     trg_size[1],
383                 )
384                 cut = (resized[0] - trg_size[0]) / 2
385                 src = src.resize(resized, Image.ANTIALIAS)
386                 src = src.crop((cut, 0, src.size[0] - cut, src.size[1]))
387
388             img.paste(src, (metr.bar_width, 0))
389             del src
390
391         img = self.add_box(img)
392
393         return img
394
395
396 class LogoWLCover(WLCover):
397     gradient_height = 90
398     gradient_logo_height = 60
399     gradient_logo_margin_right = 30
400     gradient_logo_spacing = 40
401     gradient_color = '#000'
402     gradient_opacity = .6
403     gradient_logos = [
404         'res/wl-logo-white.png',
405         'res/fnp-logo-white.png',
406     ]
407
408     def image(self):
409         img = super(LogoWLCover, self).image()
410         metr = Metric(self, self.scale)
411         gradient = Image.new('RGBA', (metr.width - metr.bar_width, metr.gradient_height), self.gradient_color)
412         gradient_mask = Image.new('L', (metr.width - metr.bar_width, metr.gradient_height))
413         draw = ImageDraw.Draw(gradient_mask)
414         for line in range(0, metr.gradient_height):
415             draw.line((0, line, metr.width - metr.bar_width, line), fill=int(255 * self.gradient_opacity * line / metr.gradient_height))
416         img.paste(gradient, 
417             (metr.bar_width, metr.height - metr.gradient_height), mask=gradient_mask)
418
419         cursor = metr.width - metr.gradient_logo_margin_right
420         logo_top = metr.height - metr.gradient_height / 2  - metr.gradient_logo_height / 2
421         for logo_path in self.gradient_logos[::-1]:
422             logo = Image.open(get_resource(logo_path))
423             logo = logo.resize(
424                 (logo.size[0] * metr.gradient_logo_height / logo.size[1], metr.gradient_logo_height),
425                 Image.ANTIALIAS)
426             cursor -= logo.size[0]
427             img.paste(logo, (cursor, logo_top), mask=logo)
428             cursor -= metr.gradient_logo_spacing
429
430         return img
431
432
433 class EbookpointCover(LogoWLCover):
434     gradient_logo_height = 58
435     gradient_logo_spacing = 25
436     gradient_logos = [
437         'res/ebookpoint-logo-white.png',
438         'res/wl-logo-white.png',
439         'res/fnp-logo-white.png',
440     ]
441
442
443 class VirtualoCover(Cover):
444     width = 600
445     height = 730
446     author_top = 73
447     title_top = 73
448     logo_bottom = 25
449     logo_width = 250
450
451
452 class PrestigioCover(Cover):
453     width = 580
454     height = 783
455     background_img = get_resource('res/cover-prestigio.png')
456
457     author_top = 446
458     author_margin_left = 118
459     author_margin_right = 62
460     author_lineskip = 60
461     author_color = '#fff'
462     author_shadow = '#000'
463     author_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
464     author_font_size = 50
465
466     title_top = 0
467     title_margin_left = 118
468     title_margin_right = 62
469     title_lineskip = 60
470     title_color = '#fff'
471     title_shadow = '#000'
472     title_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
473     title_font_size = 50
474
475     def pretty_title(self):
476         return u"„%s”" % self.title
477
478
479 class BookotekaCover(Cover):
480     width = 2140
481     height = 2733
482     background_img = get_resource('res/cover-bookoteka.png')
483
484     author_top = 480
485     author_margin_left = 307
486     author_margin_right = 233
487     author_lineskip = 156
488     author_color = '#d9d919'
489     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
490     author_font_size = 130
491
492     title_top = 400
493     title_margin_left = 307
494     title_margin_right = 233
495     title_lineskip = 168
496     title_color = '#d9d919'
497     title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
498     title_font_size = 140
499
500     format = 'PNG'
501
502
503 class GandalfCover(Cover):
504     width = 600
505     height = 730
506     background_img = get_resource('res/cover-gandalf.png')
507     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
508     author_font_size = 30
509     title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
510     title_font_size = 40
511     logo_bottom = 25
512     logo_width = 250
513     format = 'PNG'
514
515
516 DefaultEbookCover = LogoWLCover
517