1 # -*- coding: utf-8 -*-
3 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
6 from __future__ import unicode_literals
9 from PIL import Image, ImageFont, ImageDraw, ImageFilter
10 from six import BytesIO
11 from librarian import get_resource, OutputFile, URLOpener
15 """Gets metrics from an object, scaling it by a factor."""
16 def __init__(self, obj, scale):
18 self._scale = float(scale)
20 def __getattr__(self, name):
21 src = getattr(self._obj, name)
22 if src and self._scale:
23 return type(src)(self._scale * src)
28 class TextBox(object):
29 """Creates an Image with a series of centered strings."""
35 def __init__(self, max_width, max_height, padding_x=None, padding_y=None, bar_width=0, bar_color=None):
37 padding_x = self.SHADOW_X + self.SHADOW_BLUR
39 padding_y = self.SHADOW_Y + self.SHADOW_BLUR
41 self.max_width = max_width
42 self.bar_width = bar_width
43 self.bar_color = bar_color
44 self.max_text_width = max_width - 2 * padding_x - bar_width
45 self.padding_x = padding_x
46 self.padding_y = padding_y
47 self.height = padding_y
48 self.img = Image.new('RGBA', (max_width, max_height))
49 self.draw = ImageDraw.Draw(self.img)
50 self.shadow_img = None
51 self.shadow_draw = None
53 def skip(self, height):
54 """Skips some vertical space."""
57 def text(self, text, color='#000', font=None, line_height=20,
58 shadow_color=None, centering=True):
59 """Writes some centered text."""
60 text = re.sub(r'\s+', ' ', text)
62 if not self.shadow_img:
63 self.shadow_img = Image.new('RGBA', self.img.size)
64 self.shadow_draw = ImageDraw.Draw(self.shadow_img)
67 line_width = self.draw.textsize(line, font=font)[0]
68 while line_width > self.max_text_width:
69 parts = line.rsplit(' ', 1)
71 line_width = self.max_text_width
78 line_width = self.draw.textsize(line, font=font)[0]
79 line = line.strip() + ' '
82 pos_x = (self.max_width - line_width) // 2
84 pos_x = self.bar_width + self.padding_x
87 self.shadow_draw.text(
88 (pos_x + self.SHADOW_X, self.height + self.SHADOW_Y),
89 line, font=font, fill=shadow_color
92 self.draw.text((pos_x, self.height), line, font=font, fill=color)
93 self.height += line_height
95 text = text[len(line):]
98 """Creates the actual Image object."""
99 image = Image.new('RGBA', (self.max_width,
100 int(round(self.height + self.padding_y))))
103 shadow = self.shadow_img.filter(ImageFilter.BLUR)
104 image.paste(shadow, (0, 0), shadow)
105 image.paste(self.img, (0, 0), self.img)
107 image.paste(self.img, (0, 0))
112 """Abstract base class for cover images generator."""
115 background_color = '#fff'
116 background_img = None
119 author_margin_left = 20
120 author_margin_right = 20
122 author_color = '#000'
124 author_font_ttf = get_resource('fonts/DejaVuSerif.ttf')
125 author_font_size = 30
128 title_margin_left = 20
129 title_margin_right = 20
133 title_font_ttf = get_resource('fonts/DejaVuSerif.ttf')
138 uses_dc_cover = False
150 'JPEG': 'image/jpeg',
154 def __init__(self, book_info, format=None, width=None, height=None, cover_logo=None):
155 self.authors = [auth.readable() for auth in book_info.authors]
156 self.title = book_info.title
157 if format is not None:
160 self.height = int(round(height * self.width / width))
161 scale = max(float(width or 0) / self.width,
162 float(height or 0) / self.height)
166 self.scale_after = scale
168 def pretty_authors(self):
169 """Allows for decorating authors' names."""
170 return [self.authors]
172 def pretty_title(self):
173 """Allows for decorating title."""
177 metr = Metric(self, self.scale)
178 img = Image.new('RGB', (metr.width, metr.height),
179 self.background_color)
181 if self.background_img:
182 background = Image.open(self.background_img)
183 img.paste(background, None, background)
188 logo = Image.open(get_resource('res/wl-logo.png'))
191 int(round(logo.size[1] * metr.logo_width / logo.size[0]))
194 (metr.width - metr.logo_width) // 2,
195 img.size[1] - logo.size[1] - metr.logo_bottom
198 top = metr.author_top
200 metr.width - metr.author_margin_left - metr.author_margin_right,
204 author_font = ImageFont.truetype(
205 self.author_font_ttf, metr.author_font_size,
206 layout_engine=ImageFont.LAYOUT_BASIC)
207 for pa in self.pretty_authors():
208 tbox.text(pa, self.author_color, author_font, metr.author_lineskip,
210 text_img = tbox.image()
211 img.paste(text_img, (metr.author_margin_left, top), text_img)
213 top += text_img.size[1] + metr.title_top
215 metr.width - metr.title_margin_left - metr.title_margin_right,
218 title_font = ImageFont.truetype(
219 self.title_font_ttf, metr.title_font_size,
220 layout_engine=ImageFont.LAYOUT_BASIC)
221 tbox.text(self.pretty_title(), self.title_color, title_font,
222 metr.title_lineskip, self.title_shadow)
223 text_img = tbox.image()
224 img.paste(text_img, (metr.title_margin_left, top), text_img)
228 def final_image(self):
230 if self.scale_after != 1:
232 int(round(img.size[0] * self.scale_after)),
233 int(round(img.size[1] * self.scale_after))),
238 return self.mime_types[self.format]
241 return self.exts[self.format]
243 def save(self, *args, **kwargs):
245 'format': self.format,
248 default_kwargs.update(kwargs)
249 return self.final_image().save(*args, **default_kwargs)
251 def output_file(self, *args, **kwargs):
253 self.save(imgstr, *args, **kwargs)
254 return OutputFile.from_bytes(imgstr.getvalue())
257 class WLCover(Cover):
258 """Wolne Lektury cover without logos."""
262 author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
263 author_font_size = 20
265 author_centering = True
266 title_font_ttf = get_resource('fonts/DejaVuSerif-Bold.ttf')
269 title_box_width = 350
270 title_centering = True
273 box_bottom_margin = 100
288 box_position = 'middle'
289 background_color = '#444'
290 author_color = '#444'
291 background_img = get_resource('res/cover.png')
295 u'Starożytność': '#9e3610',
296 u'Średniowiecze': '#564c09',
297 u'Renesans': '#8ca629',
299 u'Oświecenie': '#f2802e',
300 u'Romantyzm': '#db4b16',
301 u'Pozytywizm': '#961060',
302 u'Modernizm': '#7784e0',
303 u'Dwudziestolecie międzywojenne': '#3044cf',
304 u'Współczesność': '#06393d',
306 set_title_color = True
308 kind_box_position = {
313 def __init__(self, book_info, format=None, width=None, height=None,
314 bleed=0, cover_logo=None):
315 super(WLCover, self).__init__(book_info, format=format, width=width,
318 self.box_position = book_info.cover_box_position or \
319 self.kind_box_position.get(book_info.kind, self.box_position)
321 if book_info.cover_bar_color == 'none':
324 self.bar_color = book_info.cover_bar_color or \
325 self.epoch_colors.get(book_info.epoch, self.bar_color)
327 if self.set_title_color:
328 self.title_color = self.epoch_colors.get(book_info.epoch,
332 self.box_top_margin += bleed
333 self.box_bottom_margin += bleed
334 self.bar_width += bleed
336 if book_info.cover_url:
337 url = book_info.cover_url
340 bg_src = URLOpener().open(url)
341 self.background_img = BytesIO(bg_src.read())
344 def pretty_authors(self):
345 return [a.upper() for a in self.authors]
347 def add_box(self, img):
348 if self.box_position == 'none':
351 metr = Metric(self, self.scale)
354 box = TextBox(metr.title_box_width, metr.height,
355 padding_y=metr.box_padding_y,
356 padding_x=metr.box_padding_x,
357 bar_width=metr.box_bar_width,
358 bar_color=self.bar_color,
360 author_font = ImageFont.truetype(
361 self.author_font_ttf, metr.author_font_size,
362 layout_engine=ImageFont.LAYOUT_BASIC)
363 for pa in self.pretty_authors():
364 box.text(pa, font=author_font, line_height=metr.author_lineskip,
365 color=self.author_color, shadow_color=self.author_shadow,
366 centering=self.author_centering
369 box.skip(metr.box_above_line)
371 (metr.box_line_left, box.height, metr.box_line_right, box.height),
372 fill=self.author_color, width=metr.box_line_width
374 box.skip(metr.box_below_line)
377 title_font = ImageFont.truetype(
378 self.title_font_ttf, metr.title_font_size,
379 layout_engine=ImageFont.LAYOUT_BASIC)
380 box.text(self.pretty_title(),
381 line_height=metr.title_lineskip,
383 color=self.title_color,
384 shadow_color=self.title_shadow,
385 centering=self.title_centering
388 box_img = box.image()
391 if self.box_position == 'bottom' or box_img.size[1] + metr.box_top_margin + metr.box_bottom_margin > metr.height:
392 box_top = metr.height - metr.box_bottom_margin - box_img.size[1]
393 elif self.box_position == 'top':
394 box_top = metr.box_top_margin
396 box_top = (metr.height - box_img.size[1]) // 2
398 box_left = metr.bar_width + (
399 metr.width - metr.bar_width - box_img.size[0]
402 # Draw the white box.
403 img_draw = ImageDraw.Draw(img)
408 box_left + box_img.size[0],
409 box_top + box_img.size[1]
413 # Paste the contents into the white box.
414 img.paste(box_img, (box_left, box_top), box_img)
415 if self.box_bar_width:
420 box_left + metr.box_bar_width,
421 box_top + box_img.size[1]
427 def add_cut_lines(self, img):
431 metr = Metric(self, self.scale)
432 draw = ImageDraw.Draw(img)
433 for corner_x, corner_y in (
434 (0, 0), (metr.width, 0),
435 (0, metr.height), (metr.width, metr.height)
437 dir_x = 1 if corner_x == 0 else -1
438 dir_y = 1 if corner_y == 0 else -1
439 for offset in (-1, 0, 1):
443 corner_y + dir_y * metr.bleed + offset,
444 corner_x + dir_x * metr.bleed * line_ratio,
445 corner_y + dir_y * metr.bleed + offset
447 fill='black' if offset == 0 else 'white',
452 corner_x + dir_x * metr.bleed + offset,
454 corner_x + dir_x * metr.bleed + offset,
455 corner_y + dir_y * metr.bleed * line_ratio
457 fill='black' if offset == 0 else 'white',
463 metr = Metric(self, self.scale)
464 img = Image.new('RGB', (metr.width, metr.height),
465 self.background_color)
466 draw = ImageDraw.Draw(img)
468 draw.rectangle((0, 0, metr.bar_width, metr.height),
471 if self.background_img:
472 src = Image.open(self.background_img)
473 trg_size = (metr.width - metr.bar_width, metr.height)
474 if src.size[0] * trg_size[1] < src.size[1] * trg_size[0]:
477 int(round(src.size[1] * trg_size[0] / src.size[0]))
479 cut = (resized[1] - trg_size[1]) // 2
480 src = src.resize(resized, Image.ANTIALIAS)
481 src = src.crop((0, cut, src.size[0], src.size[1] - cut))
484 int(round(src.size[0] * trg_size[1] / src.size[1])),
487 cut = (resized[0] - trg_size[0]) // 2
488 src = src.resize(resized, Image.ANTIALIAS)
489 src = src.crop((cut, 0, src.size[0] - cut, src.size[1]))
491 img.paste(src, (metr.bar_width, 0))
494 img = self.add_box(img)
496 # img = self.add_cut_lines(img)
501 class WLNoBoxCover(WLCover):
502 def add_box(self, img):
506 class LogoWLCover(WLCover):
509 gradient_logo_height = 57
510 gradient_logo_max_width = 200
511 gradient_logo_margin_right = 30
512 gradient_logo_spacing = 40
513 gradient_color = '#000'
514 gradient_opacity = .6
516 'res/wl-logo-white.png',
517 'res/fnp-logo-white.png',
520 annotation_height = 10
523 gradient_logo_centering = False
526 def __init__(self, book_info, *args, cover_logo=None, **kwargs):
527 super(LogoWLCover, self).__init__(book_info, *args, **kwargs)
528 self.gradient_height += self.bleed
529 self.gradient_logo_margin_right += self.bleed
531 self.additional_cover_logos = [
532 BytesIO(URLOpener().open(cover_logo_url).read())
533 for cover_logo_url in book_info.cover_logo_urls
535 self.end_cover_logos = []
537 self.end_cover_logos.append(
538 open(cover_logo, 'rb')
542 def has_gradient_logos(self):
543 return self.gradient_logos or self.additional_cover_logos or self.end_cover_logos or self.annotation
545 def add_gradient_logos(self, img, metr):
546 gradient = Image.new(
548 (metr.width - metr.bar_width, metr.gradient_height),
551 gradient_mask = Image.new(
553 (metr.width - metr.bar_width, metr.gradient_height)
555 draw = ImageDraw.Draw(gradient_mask)
556 for line in range(0, metr.gradient_easing):
558 (0, line, metr.width - metr.bar_width, line),
560 255 * self.gradient_opacity * line / metr.gradient_easing
564 0, metr.gradient_easing,
565 metr.width - metr.bar_width, metr.gradient_height
566 ), fill=int(255 * self.gradient_opacity))
570 (metr.bar_width, metr.height - metr.gradient_height),
574 cursor = metr.width - metr.gradient_logo_margin_right
576 cursor = metr.gradient_logo_margin_right
579 - metr.gradient_height / 2
580 - metr.gradient_logo_height / 2 - metr.bleed / 2
584 get_resource(logo_path)
585 for logo_path in self.gradient_logos
588 logos = self.additional_cover_logos + logos + self.end_cover_logos
594 Image.open(logo_bytes).convert('RGBA')
595 for logo_bytes in logos
598 # See if logos fit into the gradient. If not, scale down accordingly.
602 - 2 * metr.gradient_logo_margin_right
606 metr.gradient_logo_max_width,
607 logo.size[0] * metr.gradient_logo_height / logo.size[1]
610 taken_space = sum(widths) + (len(logos) - 1) * metr.gradient_logo_spacing
611 if taken_space > space_for_logos:
612 logo_scale = space_for_logos / taken_space
615 if self.gradient_logo_centering:
616 cursor += int((space_for_logos - taken_space) / 2)
618 space_for_logos / taken_space
619 if taken_space > space_for_logos else 1
621 #logo_top += int(metr.gradient_logo_height * (1 - logo_scale) / 2)
623 for i, logo in enumerate(logos):
628 L_top = logo_top + int(metr.gradient_logo_height * (1 - L_scale) / 2)
632 int(round(widths[i] * L_scale)),
634 logo.size[1] * widths[i] / logo.size[0] * L_scale
639 cursor -= logo.size[0]
646 #int(round(logo_top + (metr.gradient_logo_height - logo.size[1]) * L_scale / 2))
651 cursor -= int(round(metr.gradient_logo_spacing * logo_scale))
653 cursor += int(round(metr.gradient_logo_spacing * logo_scale)) + logo.size[0]
656 img2 = Image.new('RGBA', (metr.height, metr.height), color=None)
657 draw = ImageDraw.Draw(img2)
658 author_font = ImageFont.truetype(
659 self.author_font_ttf, metr.annotation_height,
660 layout_engine=ImageFont.LAYOUT_BASIC)
661 draw.text((self.annotation_height, self.annotation_height), self.annotation, font=author_font, fill='#FFFFFF')
663 img2 = img2.rotate(90)
666 img.alpha_composite(img2, (0, 0))
667 img = img.convert('RGB')
672 img = super(LogoWLCover, self).image()
673 metr = Metric(self, self.scale)
674 if self.has_gradient_logos:
675 img = self.add_gradient_logos(img, metr)
679 class LegimiCover(LogoWLCover):
685 author_font_ttf = get_resource('fonts/Lora-Regular.ttf')
686 author_font_size = 15
687 author_lineskip = 19.5
688 author_centering = False
689 title_font_ttf = get_resource('fonts/Lora-Bold.ttf')
691 title_lineskip = 19.5
692 title_centering = False
694 title_box_width = 210
696 box_bottom_margin = 20
698 box_padding_y = 10 # do baseline
712 box_position = 'bottom'
713 background_color = '#444'
714 author_color = '#000'
716 set_title_color = False
718 kind_box_position = {}
720 box_bottom_margin_logos_add = 10
723 gradient_logo_height = 20
724 gradient_logo_max_width = 200
725 gradient_logo_margin_right = 10
726 gradient_logo_spacing = 20
727 gradient_color = '#000'
728 gradient_opacity = .5
730 'res/wl-logo-white.png',
733 gradient_logo_centering = True
735 def __init__(self, *args, **kwargs):
736 super().__init__(*args, **kwargs)
737 if self.has_gradient_logos:
738 self.box_bottom_margin += self.box_bottom_margin_logos_add
741 def pretty_authors(self):
745 class LegimiCornerCover(LegimiCover):
748 corner_image = get_resource('res/book-band.png')
751 image = super().image()
752 metr = Metric(self, self.scale)
754 if self.corner_image:
755 corner = Image.open(self.corner_image).convert('RGBA')
756 corner = corner.resize(
758 int(round(metr.corner_width)),
759 int(round(corner.size[1] / corner.size[0] * metr.corner_width))
762 image.paste(corner, (
763 metr.width - int(round(metr.corner_width)),
768 class LegimiAudiobookCover(LegimiCornerCover):
770 corner_image = get_resource('res/audiobook-sticker.png')
774 class EbookpointCover(LogoWLCover):
775 gradient_logo_height = 58
776 gradient_logo_spacing = 25
778 'res/ebookpoint-logo-white.png',
779 'res/wl-logo-white.png',
780 'res/fnp-logo-white.png',
784 class VirtualoCover(Cover):
793 class PrestigioCover(Cover):
796 background_img = get_resource('res/cover-prestigio.png')
799 author_margin_left = 118
800 author_margin_right = 62
802 author_color = '#fff'
803 author_shadow = '#000'
804 author_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
805 author_font_size = 50
808 title_margin_left = 118
809 title_margin_right = 62
812 title_shadow = '#000'
813 title_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
816 def pretty_title(self):
817 return u"„%s”" % self.title
820 class BookotekaCover(Cover):
823 background_img = get_resource('res/cover-bookoteka.png')
826 author_margin_left = 307
827 author_margin_right = 233
828 author_lineskip = 156
829 author_color = '#d9d919'
830 author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
831 author_font_size = 130
834 title_margin_left = 307
835 title_margin_right = 233
837 title_color = '#d9d919'
838 title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
839 title_font_size = 140
844 class GandalfCover(Cover):
847 background_img = get_resource('res/cover-gandalf.png')
848 author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
849 author_font_size = 30
850 title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
857 class KMLUCover(LogoWLCover):
858 gradient_logo_height = 58
859 gradient_logo_spacing = 25
861 'res/kmlu-logo-white.png',
862 'res/wl-logo-white.png',
863 'res/fnp-logo-white.png',
867 class MPWCover(LogoWLCover):
868 gradient_logo_height = 57
869 gradient_logo_spacing = 25
871 'res/mpw-logo-white.png',
872 'res/wl-logo-white.png',
873 'res/fnp-logo-white.png',
877 class AtriumCover(LogoWLCover):
878 gradient_logo_height = 58
879 gradient_logo_spacing = 25
881 'res/atrium-logo.png',
882 'res/wl-logo-white.png',
883 'res/fnp-logo-white.png',
887 class BNCover(LogoWLCover):
889 'res/dofinansowano.png',
892 'res/wl-logo-white.png',
894 # annotation = 'Zadanie „Udostępnienie publikacji w formatach cyfrowych” w ramach Narodowego Programu Rozwoju Czytelnictwa. Dofinansowano ze środków Ministra Kultury, Dziedzictwa Narodowego i Sportu.'
898 'default': LogoWLCover,
901 'atrium': AtriumCover,
903 'legimi': LegimiCover,
904 'legimi-corner': LegimiCornerCover,
905 'legimi-audiobook': LegimiAudiobookCover,
909 def make_cover(book_info, *args, **kwargs):
910 if 'cover_class' in kwargs:
911 cover_class_name = kwargs.pop('cover_class')
913 cover_class_name = book_info.cover_class
914 cover_class = COVER_CLASSES[cover_class_name]
915 return cover_class(book_info, *args, **kwargs)