New mobi builder.
[librarian.git] / src / 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 from __future__ import unicode_literals
7
8 import re
9 from PIL import Image, ImageFont, ImageDraw, ImageFilter
10 from six import BytesIO
11 from librarian import get_resource, OutputFile, URLOpener
12
13
14 class Metric(object):
15     """Gets metrics from an object, scaling it by a factor."""
16     def __init__(self, obj, scale):
17         self._obj = obj
18         self._scale = float(scale)
19
20     def __getattr__(self, name):
21         src = getattr(self._obj, name)
22         if src and self._scale:
23             return type(src)(self._scale * src)
24         else:
25             return src
26
27
28 class TextBox(object):
29     """Creates an Image with a series of centered strings."""
30
31     SHADOW_X = 3
32     SHADOW_Y = 3
33     SHADOW_BLUR = 3
34
35     def __init__(self, max_width, max_height, padding_x=None, padding_y=None, bar_width=0, bar_color=None):
36         if padding_x is None:
37             padding_x = self.SHADOW_X + self.SHADOW_BLUR
38         if padding_y is None:
39             padding_y = self.SHADOW_Y + self.SHADOW_BLUR
40
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
52
53     def skip(self, height):
54         """Skips some vertical space."""
55         self.height += height
56
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)
61         if shadow_color:
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)
65         while text:
66             line = text
67             line_width = self.draw.textsize(line, font=font)[0]
68             while line_width > self.max_text_width:
69                 parts = line.rsplit(' ', 1)
70                 if len(parts) == 1:
71                     line_width = self.max_text_width
72                     break
73                 line = parts[0]
74
75                 if line[-2] == ' ':
76                     line = line[:-2]
77
78                 line_width = self.draw.textsize(line, font=font)[0]
79             line = line.strip() + ' '
80
81             if centering:
82                 pos_x = (self.max_width - line_width) // 2
83             else:
84                 pos_x = self.bar_width + self.padding_x
85
86             if shadow_color:
87                 self.shadow_draw.text(
88                         (pos_x + self.SHADOW_X, self.height + self.SHADOW_Y),
89                         line, font=font, fill=shadow_color
90                 )
91
92             self.draw.text((pos_x, self.height), line, font=font, fill=color)
93             self.height += line_height
94             # go to next line
95             text = text[len(line):]
96
97     def image(self):
98         """Creates the actual Image object."""
99         image = Image.new('RGBA', (self.max_width,
100                                    int(round(self.height + self.padding_y))))
101
102         if self.shadow_img:
103             shadow = self.shadow_img.filter(ImageFilter.BLUR)
104             image.paste(shadow, (0, 0), shadow)
105             image.paste(self.img, (0, 0), self.img)
106         else:
107             image.paste(self.img, (0, 0))
108         return image
109
110
111 class Cover(object):
112     """Abstract base class for cover images generator."""
113     width = 600
114     height = 800
115     background_color = '#fff'
116     background_img = None
117
118     author_top = 100
119     author_margin_left = 20
120     author_margin_right = 20
121     author_lineskip = 40
122     author_color = '#000'
123     author_shadow = None
124     author_font_ttf = get_resource('fonts/DejaVuSerif.ttf')
125     author_font_size = 30
126
127     title_top = 100
128     title_margin_left = 20
129     title_margin_right = 20
130     title_lineskip = 54
131     title_color = '#000'
132     title_shadow = None
133     title_font_ttf = get_resource('fonts/DejaVuSerif.ttf')
134     title_font_size = 40
135
136     logo_bottom = None
137     logo_width = None
138     uses_dc_cover = False
139
140     format = 'JPEG'
141     scale = 1
142     scale_after = 1
143
144     exts = {
145         'JPEG': 'jpg',
146         'PNG': 'png',
147         }
148
149     mime_types = {
150         'JPEG': 'image/jpeg',
151         'PNG': 'image/png',
152         }
153
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:
158             self.format = format
159         if width and height:
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)
163         if scale >= 1:
164             self.scale = scale
165         elif scale:
166             self.scale_after = scale
167
168     def pretty_authors(self):
169         """Allows for decorating authors' names."""
170         return [self.authors]
171
172     def pretty_title(self):
173         """Allows for decorating title."""
174         return self.title
175
176     def image(self):
177         metr = Metric(self, self.scale)
178         img = Image.new('RGB', (metr.width, metr.height),
179                         self.background_color)
180
181         if self.background_img:
182             background = Image.open(self.background_img)
183             img.paste(background, None, background)
184             del background
185
186         # WL logo
187         if metr.logo_width:
188             logo = Image.open(get_resource('res/wl-logo.png'))
189             logo = logo.resize((
190                 metr.logo_width,
191                 int(round(logo.size[1] * metr.logo_width / logo.size[0]))
192             ))
193             img.paste(logo, (
194                 (metr.width - metr.logo_width) // 2,
195                 img.size[1] - logo.size[1] - metr.logo_bottom
196             ))
197
198         top = metr.author_top
199         tbox = TextBox(
200             metr.width - metr.author_margin_left - metr.author_margin_right,
201             metr.height - top,
202             )
203
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,
209                       self.author_shadow)
210         text_img = tbox.image()
211         img.paste(text_img, (metr.author_margin_left, top), text_img)
212
213         top += text_img.size[1] + metr.title_top
214         tbox = TextBox(
215             metr.width - metr.title_margin_left - metr.title_margin_right,
216             metr.height - top,
217             )
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)
225
226         return img
227
228     def final_image(self):
229         img = self.image()
230         if self.scale_after != 1:
231             img = img.resize((
232                     int(round(img.size[0] * self.scale_after)),
233                     int(round(img.size[1] * self.scale_after))),
234                 Image.ANTIALIAS)
235         return img
236
237     def mime_type(self):
238         return self.mime_types[self.format]
239
240     def ext(self):
241         return self.exts[self.format]
242
243     def save(self, *args, **kwargs):
244         default_kwargs = {
245                 'format': self.format,
246                 'quality': 95,
247         }
248         default_kwargs.update(kwargs)
249         return self.final_image().save(*args, **default_kwargs)
250
251     def output_file(self, *args, **kwargs):
252         imgstr = BytesIO()
253         self.save(imgstr, *args, **kwargs)
254         return OutputFile.from_bytes(imgstr.getvalue())
255
256
257 class WLCover(Cover):
258     """Wolne Lektury cover without logos."""
259     width = 600
260     height = 833
261     uses_dc_cover = True
262     author_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
263     author_font_size = 20
264     author_lineskip = 30
265     author_centering = True
266     title_font_ttf = get_resource('fonts/DejaVuSerif-Bold.ttf')
267     title_font_size = 30
268     title_lineskip = 40
269     title_box_width = 350
270     title_centering = True
271
272     box_top_margin = 100
273     box_bottom_margin = 100
274     box_padding_y = 20
275     box_above_line = 10
276     box_below_line = 15
277     box_line_left = 75
278     box_line_right = 275
279     box_line_width = 2
280     box_padding_x = 0
281     box_bar_width = 0
282
283     logo_top = 15
284     logo_width = 140
285
286     bar_width = 35
287     bar_color = '#000'
288     box_position = 'middle'
289     background_color = '#444'
290     author_color = '#444'
291     background_img = get_resource('res/cover.png')
292     format = 'JPEG'
293
294     epoch_colors = {
295         u'Starożytność': '#9e3610',
296         u'Średniowiecze': '#564c09',
297         u'Renesans': '#8ca629',
298         u'Barok': '#a6820a',
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',
305     }
306     set_title_color = True
307
308     kind_box_position = {
309         u'Liryka': 'top',
310         u'Epika': 'bottom',
311     }
312
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,
316                                       height=height)
317         # Set box position.
318         self.box_position = book_info.cover_box_position or \
319             self.kind_box_position.get(book_info.kind, self.box_position)
320         # Set bar color.
321         if book_info.cover_bar_color == 'none':
322             self.bar_width = 0
323         else:
324             self.bar_color = book_info.cover_bar_color or \
325                 self.epoch_colors.get(book_info.epoch, self.bar_color)
326         # Set title color.
327         if self.set_title_color:
328             self.title_color = self.epoch_colors.get(book_info.epoch,
329                                                      self.title_color)
330
331         self.bleed = bleed
332         self.box_top_margin += bleed
333         self.box_bottom_margin += bleed
334         self.bar_width += bleed
335
336         if book_info.cover_url:
337             url = book_info.cover_url
338             bg_src = None
339             if bg_src is None:
340                 bg_src = URLOpener().open(url)
341             self.background_img = BytesIO(bg_src.read())
342             bg_src.close()
343
344     def pretty_authors(self):
345         return [a.upper() for a in self.authors]
346
347     def add_box(self, img):
348         if self.box_position == 'none':
349             return img
350
351         metr = Metric(self, self.scale)
352
353         # Write author name.
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,
359                       )
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
367                      )
368
369         box.skip(metr.box_above_line)
370         box.draw.line(
371             (metr.box_line_left, box.height, metr.box_line_right, box.height),
372             fill=self.author_color, width=metr.box_line_width
373         )
374         box.skip(metr.box_below_line)
375
376         # Write title.
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,
382                  font=title_font,
383                  color=self.title_color,
384                  shadow_color=self.title_shadow,
385                  centering=self.title_centering
386                  )
387
388         box_img = box.image()
389
390         # Find box position.
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
395         else:   # Middle.
396             box_top = (metr.height - box_img.size[1]) // 2
397
398         box_left = metr.bar_width + (
399             metr.width - metr.bar_width - box_img.size[0]
400         ) // 2
401
402         # Draw the white box.
403         img_draw = ImageDraw.Draw(img)
404         img_draw.rectangle(
405             (
406                 box_left,
407                 box_top,
408                 box_left + box_img.size[0],
409                 box_top + box_img.size[1]
410             ),
411             fill='#fff'
412         )
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:
416             img_draw.rectangle(
417                 (
418                     box_left,
419                     box_top,
420                     box_left + metr.box_bar_width,
421                     box_top + box_img.size[1]
422                 ),
423                 fill=self.bar_color
424             )
425         return img
426
427     def add_cut_lines(self, img):
428         line_ratio = 0.5
429         if self.bleed == 0:
430             return 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)
436                 ):
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):
440                 draw.line(
441                     (
442                         corner_x,
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
446                     ),
447                     fill='black' if offset == 0 else 'white',
448                     width=1
449                 )
450                 draw.line(
451                     (
452                         corner_x + dir_x * metr.bleed + offset,
453                         corner_y,
454                         corner_x + dir_x * metr.bleed + offset,
455                         corner_y + dir_y * metr.bleed * line_ratio
456                     ),
457                     fill='black' if offset == 0 else 'white',
458                     width=1
459                 )
460         return img
461
462     def image(self):
463         metr = Metric(self, self.scale)
464         img = Image.new('RGB', (metr.width, metr.height),
465                         self.background_color)
466         draw = ImageDraw.Draw(img)
467
468         draw.rectangle((0, 0, metr.bar_width, metr.height),
469                        fill=self.bar_color)
470
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]:
475                 resized = (
476                     trg_size[0],
477                     int(round(src.size[1] * trg_size[0] / src.size[0]))
478                 )
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))
482             else:
483                 resized = (
484                     int(round(src.size[0] * trg_size[1] / src.size[1])),
485                     trg_size[1],
486                 )
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]))
490
491             img.paste(src, (metr.bar_width, 0))
492             del src
493
494         img = self.add_box(img)
495
496         # img = self.add_cut_lines(img)
497
498         return img
499
500
501 class WLNoBoxCover(WLCover):
502     def add_box(self, img):
503         return img
504
505
506 class LogoWLCover(WLCover):
507     gradient_height = 90
508     gradient_easing = 90
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
515     gradient_logos = [
516         'res/wl-logo-white.png',
517         'res/fnp-logo-white.png',
518     ]
519     annotation = None
520     annotation_height = 10
521
522     logos_right = True
523     gradient_logo_centering = False
524
525
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
530
531         self.additional_cover_logos = [
532             BytesIO(URLOpener().open(cover_logo_url).read())
533             for cover_logo_url in book_info.cover_logo_urls
534         ]
535         self.end_cover_logos = []
536         if cover_logo:
537             self.end_cover_logos.append(
538                     open(cover_logo, 'rb')
539                     )
540
541     @property
542     def has_gradient_logos(self):
543         return self.gradient_logos or self.additional_cover_logos or self.end_cover_logos or self.annotation
544
545     def add_gradient_logos(self, img, metr):
546         gradient = Image.new(
547             'RGBA',
548             (metr.width - metr.bar_width, metr.gradient_height),
549             self.gradient_color
550         )
551         gradient_mask = Image.new(
552             'L',
553             (metr.width - metr.bar_width, metr.gradient_height)
554         )
555         draw = ImageDraw.Draw(gradient_mask)
556         for line in range(0, metr.gradient_easing):
557             draw.line(
558                 (0, line, metr.width - metr.bar_width, line),
559                 fill=int(
560                     255 * self.gradient_opacity * line / metr.gradient_easing
561                 )
562             )
563         draw.rectangle((
564             0, metr.gradient_easing,
565             metr.width - metr.bar_width, metr.gradient_height
566         ), fill=int(255 * self.gradient_opacity))
567             
568             
569         img.paste(gradient,
570                   (metr.bar_width, metr.height - metr.gradient_height),
571                   mask=gradient_mask)
572
573         if self.logos_right:
574             cursor = metr.width - metr.gradient_logo_margin_right
575         else:
576             cursor = metr.gradient_logo_margin_right
577         logo_top = int(
578             metr.height
579             - metr.gradient_height / 2
580             - metr.gradient_logo_height / 2 - metr.bleed / 2
581         )
582
583         logos = [
584             get_resource(logo_path)
585             for logo_path in self.gradient_logos
586         ]
587
588         logos = self.additional_cover_logos + logos + self.end_cover_logos
589
590         if self.logos_right:
591             logos.reverse()
592
593         logos = [
594             Image.open(logo_bytes).convert('RGBA')
595             for logo_bytes in logos
596         ]
597
598         # See if logos fit into the gradient. If not, scale down accordingly.
599         space_for_logos = (
600             metr.width
601             - metr.bar_width
602             - 2 * metr.gradient_logo_margin_right
603         )
604         widths = [
605             min(
606                 metr.gradient_logo_max_width,
607                 logo.size[0] * metr.gradient_logo_height / logo.size[1]
608             )
609             for logo in logos]
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
613         else:
614             logo_scale = 1
615             if self.gradient_logo_centering:
616                 cursor += int((space_for_logos - taken_space) / 2)
617         logo_scale = (
618             space_for_logos / taken_space
619             if taken_space > space_for_logos else 1
620         )
621         #logo_top += int(metr.gradient_logo_height * (1 - logo_scale) / 2)
622
623         for i, logo in enumerate(logos):
624             if i == -1:
625                 L_scale = 1
626             else:
627                 L_scale = logo_scale
628             L_top = logo_top + int(metr.gradient_logo_height * (1 - L_scale) / 2)
629
630             logo = logo.resize(
631                 (
632                     int(round(widths[i] * L_scale)),
633                     int(round(
634                         logo.size[1] * widths[i] / logo.size[0] * L_scale
635                     ))
636                 ),
637                 Image.ANTIALIAS)
638             if self.logos_right:
639                 cursor -= logo.size[0]
640
641             img.paste(
642                 logo,
643                 (
644                     cursor,
645                     L_top
646                     #int(round(logo_top + (metr.gradient_logo_height - logo.size[1]) * L_scale / 2))
647                 ),
648                 mask=logo
649             )
650             if self.logos_right:
651                 cursor -= int(round(metr.gradient_logo_spacing * logo_scale))
652             else:
653                 cursor += int(round(metr.gradient_logo_spacing * logo_scale)) + logo.size[0]
654
655         if self.annotation:
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')
662             img2.show()
663             img2 = img2.rotate(90)
664             img2.show()
665             img.putalpha(0)
666             img.alpha_composite(img2, (0, 0))
667             img = img.convert('RGB')
668
669         return img
670     
671     def image(self):
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)
676         return img
677
678
679 class LegimiCover(LogoWLCover):
680     width = 210
681     height = 297
682     bar_width = 0
683     # Other bar
684
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')
690     title_font_size = 15
691     title_lineskip = 19.5
692     title_centering = False
693     
694     title_box_width = 210
695
696     box_bottom_margin = 20
697     box_padding_x = 20
698     box_padding_y = 10   # do baseline
699     box_above_line = 6
700     box_below_line = 0
701     box_line_left = 0
702     box_line_right = 0
703
704     box_line_width = 0
705
706     box_bar_width = 20
707
708     #logo_top = 15
709     #logo_width = 140
710
711     bar_color = '#000'
712     box_position = 'bottom'
713     background_color = '#444'
714     author_color = '#000'
715     title_color = '#000'
716     set_title_color = False
717
718     kind_box_position = {}
719
720     box_bottom_margin_logos_add = 10
721     gradient_height = 30
722     gradient_easing = 0
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
729     gradient_logos = [
730         'res/wl-logo-white.png',
731     ]
732     logos_right = False
733     gradient_logo_centering = True
734
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
739         
740     
741     def pretty_authors(self):
742         return self.authors
743
744
745 class LegimiCornerCover(LegimiCover):
746     gradient_logos = []
747     corner_width = 120
748     corner_image = get_resource('res/book-band.png')
749     
750     def image(self):
751         image = super().image()
752         metr = Metric(self, self.scale)
753
754         if self.corner_image:
755             corner = Image.open(self.corner_image).convert('RGBA')
756             corner = corner.resize(
757                 (
758                     int(round(metr.corner_width)),
759                     int(round(corner.size[1] / corner.size[0] * metr.corner_width))
760                 )
761             )
762             image.paste(corner, (
763                 metr.width - int(round(metr.corner_width)),
764                 0,
765             ), corner)
766         return image
767
768 class LegimiAudiobookCover(LegimiCornerCover):
769     corner_width = 82.5
770     corner_image = get_resource('res/audiobook-sticker.png')
771     height = 210
772
773
774 class EbookpointCover(LogoWLCover):
775     gradient_logo_height = 58
776     gradient_logo_spacing = 25
777     gradient_logos = [
778         'res/ebookpoint-logo-white.png',
779         'res/wl-logo-white.png',
780         'res/fnp-logo-white.png',
781     ]
782
783
784 class VirtualoCover(Cover):
785     width = 600
786     height = 730
787     author_top = 73
788     title_top = 73
789     logo_bottom = 25
790     logo_width = 250
791
792
793 class PrestigioCover(Cover):
794     width = 580
795     height = 783
796     background_img = get_resource('res/cover-prestigio.png')
797
798     author_top = 446
799     author_margin_left = 118
800     author_margin_right = 62
801     author_lineskip = 60
802     author_color = '#fff'
803     author_shadow = '#000'
804     author_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
805     author_font_size = 50
806
807     title_top = 0
808     title_margin_left = 118
809     title_margin_right = 62
810     title_lineskip = 60
811     title_color = '#fff'
812     title_shadow = '#000'
813     title_font_ttf = get_resource('fonts/JunicodeWL-Italic.ttf')
814     title_font_size = 50
815
816     def pretty_title(self):
817         return u"„%s”" % self.title
818
819
820 class BookotekaCover(Cover):
821     width = 2140
822     height = 2733
823     background_img = get_resource('res/cover-bookoteka.png')
824
825     author_top = 480
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
832
833     title_top = 400
834     title_margin_left = 307
835     title_margin_right = 233
836     title_lineskip = 168
837     title_color = '#d9d919'
838     title_font_ttf = get_resource('fonts/JunicodeWL-Regular.ttf')
839     title_font_size = 140
840
841     format = 'PNG'
842
843
844 class GandalfCover(Cover):
845     width = 600
846     height = 730
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')
851     title_font_size = 40
852     logo_bottom = 25
853     logo_width = 250
854     format = 'PNG'
855
856
857 class KMLUCover(LogoWLCover):
858     gradient_logo_height = 58
859     gradient_logo_spacing = 25
860     gradient_logos = [
861         'res/kmlu-logo-white.png',
862         'res/wl-logo-white.png',
863         'res/fnp-logo-white.png',
864     ]
865
866
867 class MPWCover(LogoWLCover):
868     gradient_logo_height = 57
869     gradient_logo_spacing = 25
870     gradient_logos = [
871         'res/mpw-logo-white.png',
872         'res/wl-logo-white.png',
873         'res/fnp-logo-white.png',
874     ]
875
876
877 class AtriumCover(LogoWLCover):
878     gradient_logo_height = 58
879     gradient_logo_spacing = 25
880     gradient_logos = [
881         'res/atrium-logo.png',
882         'res/wl-logo-white.png',
883         'res/fnp-logo-white.png',
884     ]
885
886
887 class BNCover(LogoWLCover):
888     gradient_logos = [
889         'res/dofinansowano.png',
890         'res/MKIDN.jpg',
891         'res/BN.png',
892         'res/wl-logo-white.png',
893     ]
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.'
895
896
897 COVER_CLASSES = {
898     'default': LogoWLCover,
899     'kmlu': KMLUCover,
900     'mpw': MPWCover,
901     'atrium': AtriumCover,
902     'bn': BNCover,
903     'legimi': LegimiCover,
904     'legimi-corner': LegimiCornerCover,
905     'legimi-audiobook': LegimiAudiobookCover,
906 }
907
908
909 def make_cover(book_info, *args, **kwargs):
910     if 'cover_class' in kwargs:
911         cover_class_name = kwargs.pop('cover_class')
912     else:
913         cover_class_name = book_info.cover_class
914     cover_class = COVER_CLASSES[cover_class_name]
915     return cover_class(book_info, *args, **kwargs)