Added epub write support

2026-04-25 16:01:29 +02:00 · 2020-04-13 12:46:37 +02:00
parent 9f18513787
commit 79cad46732
9 changed files with 3049 additions and 0 deletions
@@ -0,0 +1,81 @@
+#!/usr/bin/env python2
+# vim:fileencoding=UTF-8:ts=4:sw=4:sta:et:sts=4:ai
+from __future__ import absolute_import, division, print_function, unicode_literals
+
+__license__   = 'GPL v3'
+__copyright__ = '2009, Kovid Goyal <kovid@kovidgoyal.net>'
+__docformat__ = 'restructuredtext en'
+
+from calibre import fit_image
+
+
+class RescaleImages(object):
+
+    'Rescale all images to fit inside given screen size'
+
+    def __init__(self, check_colorspaces=False):
+        self.check_colorspaces = check_colorspaces
+
+    def __call__(self, oeb, opts):
+        self.oeb, self.opts, self.log = oeb, opts, oeb.log
+        self.rescale()
+
+    def rescale(self):
+        from PIL import Image
+        from io import BytesIO
+
+        is_image_collection = getattr(self.opts, 'is_image_collection', False)
+
+        if is_image_collection:
+            page_width, page_height = self.opts.dest.comic_screen_size
+        else:
+            page_width, page_height = self.opts.dest.width, self.opts.dest.height
+            page_width -= (self.opts.margin_left + self.opts.margin_right) * self.opts.dest.dpi/72
+            page_height -= (self.opts.margin_top + self.opts.margin_bottom) * self.opts.dest.dpi/72
+
+        for item in self.oeb.manifest:
+            if item.media_type.startswith('image'):
+                ext = item.media_type.split('/')[-1].upper()
+                if ext == 'JPG':
+                    ext = 'JPEG'
+                if ext not in ('PNG', 'JPEG', 'GIF'):
+                    ext = 'JPEG'
+
+                raw = item.data
+                if hasattr(raw, 'xpath') or not raw:
+                    # Probably an svg image
+                    continue
+                try:
+                    img = Image.open(BytesIO(raw))
+                except Exception:
+                    continue
+                width, height = img.size
+
+                try:
+                    if self.check_colorspaces and img.mode == 'CMYK':
+                        self.log.warn(
+                            'The image %s is in the CMYK colorspace, converting it '
+                            'to RGB as Adobe Digital Editions cannot display CMYK' % item.href)
+                        img = img.convert('RGB')
+                except Exception:
+                    self.log.exception('Failed to convert image %s from CMYK to RGB' % item.href)
+
+                scaled, new_width, new_height = fit_image(width, height, page_width, page_height)
+                if scaled:
+                    new_width = max(1, new_width)
+                    new_height = max(1, new_height)
+                    self.log('Rescaling image from %dx%d to %dx%d'%(
+                        width, height, new_width, new_height), item.href)
+                    try:
+                        img = img.resize((new_width, new_height))
+                    except Exception:
+                        self.log.exception('Failed to rescale image: %s' % item.href)
+                        continue
+                    buf = BytesIO()
+                    try:
+                        img.save(buf, ext)
+                    except Exception:
+                        self.log.exception('Failed to rescale image: %s' % item.href)
+                    else:
+                        item.data = buf.getvalue()
+                        item.unload_data_from_memory()