mirror of
https://github.com/gryf/ebook-converter.git
synced 2026-03-17 07:13:42 +01:00
gryf
0f9792df36
Here is the first batch of modules, which are needed for converting several formats to LRF. Some of the logic has been change, more cleanups will follow.
159 lines
6.1 KiB
Python
159 lines
6.1 KiB
Python
#!/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__ = '2012, Kovid Goyal <kovid@kovidgoyal.net>'
|
|
__docformat__ = 'restructuredtext en'
|
|
|
|
import os
|
|
|
|
from ebook_converter.ebooks.mobi import MAX_THUMB_DIMEN, MAX_THUMB_SIZE
|
|
from ebook_converter.ebooks.mobi.utils import (rescale_image, mobify_image,
|
|
write_font_record)
|
|
from ebook_converter.ebooks import generate_masthead
|
|
from ebook_converter.ebooks.oeb.base import OEB_RASTER_IMAGES
|
|
from ebook_converter.ptempfile import PersistentTemporaryFile
|
|
from ebook_converter.utils.imghdr import what
|
|
from ebook_converter.polyglot.builtins import iteritems, unicode_type
|
|
|
|
PLACEHOLDER_GIF = b'GIF89a\x01\x00\x01\x00\xf0\x00\x00\x00\x00\x00\xff\xff\xff!\xf9\x04\x01\x00\x00\x00\x00!\xfe calibre-placeholder-gif-for-azw3\x00,\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02D\x01\x00;' # noqa
|
|
|
|
|
|
class Resources(object):
|
|
|
|
def __init__(self, oeb, opts, is_periodical, add_fonts=False,
|
|
process_images=True):
|
|
self.oeb, self.log, self.opts = oeb, oeb.log, opts
|
|
self.is_periodical = is_periodical
|
|
self.process_images = process_images
|
|
|
|
self.item_map = {}
|
|
self.records = []
|
|
self.mime_map = {}
|
|
self.masthead_offset = 0
|
|
self.used_image_indices = set()
|
|
self.image_indices = set()
|
|
self.cover_offset = self.thumbnail_offset = None
|
|
self.has_fonts = False
|
|
|
|
self.add_resources(add_fonts)
|
|
|
|
def process_image(self, data):
|
|
if not self.process_images:
|
|
return data
|
|
func = mobify_image if self.opts.mobi_keep_original_images else rescale_image
|
|
try:
|
|
return func(data)
|
|
except Exception:
|
|
if 'png' != what(None, data):
|
|
raise
|
|
with PersistentTemporaryFile(suffix='.png') as pt:
|
|
pt.write(data)
|
|
try:
|
|
from ebook_converter.utils.img import optimize_png
|
|
optimize_png(pt.name)
|
|
data = lopen(pt.name, 'rb').read()
|
|
finally:
|
|
os.remove(pt.name)
|
|
return func(data)
|
|
|
|
def add_resources(self, add_fonts):
|
|
oeb = self.oeb
|
|
oeb.logger.info('Serializing resources...')
|
|
index = 1
|
|
|
|
mh_href = None
|
|
if 'masthead' in oeb.guide and oeb.guide['masthead'].href:
|
|
mh_href = oeb.guide['masthead'].href
|
|
self.records.append(None)
|
|
index += 1
|
|
self.used_image_indices.add(0)
|
|
self.image_indices.add(0)
|
|
elif self.is_periodical:
|
|
# Generate a default masthead
|
|
data = generate_masthead(unicode_type(self.oeb.metadata['title'][0]))
|
|
self.records.append(data)
|
|
self.used_image_indices.add(0)
|
|
self.image_indices.add(0)
|
|
index += 1
|
|
|
|
cover_href = self.cover_offset = self.thumbnail_offset = None
|
|
if (oeb.metadata.cover and
|
|
unicode_type(oeb.metadata.cover[0]) in oeb.manifest.ids):
|
|
cover_id = unicode_type(oeb.metadata.cover[0])
|
|
item = oeb.manifest.ids[cover_id]
|
|
cover_href = item.href
|
|
|
|
for item in self.oeb.manifest.values():
|
|
if item.media_type not in OEB_RASTER_IMAGES:
|
|
continue
|
|
try:
|
|
data = self.process_image(item.data)
|
|
except:
|
|
self.log.warn('Bad image file %r' % item.href)
|
|
continue
|
|
else:
|
|
if mh_href and item.href == mh_href:
|
|
self.records[0] = data
|
|
continue
|
|
|
|
self.image_indices.add(len(self.records))
|
|
self.records.append(data)
|
|
self.item_map[item.href] = index
|
|
self.mime_map[item.href] = 'image/%s'%what(None, data)
|
|
index += 1
|
|
|
|
if cover_href and item.href == cover_href:
|
|
self.cover_offset = self.item_map[item.href] - 1
|
|
self.used_image_indices.add(self.cover_offset)
|
|
try:
|
|
data = rescale_image(item.data, dimen=MAX_THUMB_DIMEN,
|
|
maxsizeb=MAX_THUMB_SIZE)
|
|
except:
|
|
self.log.warn('Failed to generate thumbnail')
|
|
else:
|
|
self.image_indices.add(len(self.records))
|
|
self.records.append(data)
|
|
self.thumbnail_offset = index - 1
|
|
self.used_image_indices.add(self.thumbnail_offset)
|
|
index += 1
|
|
finally:
|
|
item.unload_data_from_memory()
|
|
|
|
if add_fonts:
|
|
for item in self.oeb.manifest.values():
|
|
if item.href and item.href.rpartition('.')[-1].lower() in {
|
|
'ttf', 'otf'} and isinstance(item.data, bytes):
|
|
self.records.append(write_font_record(item.data))
|
|
self.item_map[item.href] = len(self.records)
|
|
self.has_fonts = True
|
|
|
|
def add_extra_images(self):
|
|
'''
|
|
Add any images that were created after the call to add_resources()
|
|
'''
|
|
for item in self.oeb.manifest.values():
|
|
if (item.media_type not in OEB_RASTER_IMAGES or item.href in self.item_map):
|
|
continue
|
|
try:
|
|
data = self.process_image(item.data)
|
|
except:
|
|
self.log.warn('Bad image file %r' % item.href)
|
|
else:
|
|
self.records.append(data)
|
|
self.item_map[item.href] = len(self.records)
|
|
finally:
|
|
item.unload_data_from_memory()
|
|
|
|
def serialize(self, records, used_images):
|
|
used_image_indices = self.used_image_indices | {
|
|
v-1 for k, v in iteritems(self.item_map) if k in used_images}
|
|
for i in self.image_indices-used_image_indices:
|
|
self.records[i] = PLACEHOLDER_GIF
|
|
records.extend(self.records)
|
|
|
|
def __bool__(self):
|
|
return bool(self.records)
|
|
__nonzero__ = __bool__
|