mirror of
https://github.com/kovidgoyal/calibre.git
synced 2025-07-09 03:04:10 -04:00
RTF Input: Speed up processing and add support for small-caps and colors
This commit is contained in:
Kovid Goyal
parent
4c6bcf512f
commit
2fafc08735
@ -1,27 +1,10 @@
|
|||||||
#########################################################################
|
|
||||||
# #
|
|
||||||
# #
|
|
||||||
# copyright 2002 Paul Henry Tremblay #
|
|
||||||
# #
|
|
||||||
# This program is distributed in the hope that it will be useful, #
|
|
||||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU #
|
|
||||||
# General Public License for more details. #
|
|
||||||
# #
|
|
||||||
# You should have received a copy of the GNU General Public License #
|
|
||||||
# along with this program; if not, write to the Free Software #
|
|
||||||
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA #
|
|
||||||
# 02111-1307 USA #
|
|
||||||
# #
|
|
||||||
# #
|
|
||||||
#########################################################################
|
|
||||||
|
|
||||||
xhtml = '''\
|
|
||||||
<?xml version="1.0"?>
|
<?xml version="1.0"?>
|
||||||
<xsl:stylesheet version="1.0"
|
<xsl:stylesheet version="1.0"
|
||||||
xmlns:xsl="http://www.w3.org/1999/XSL/Transform"
|
xmlns:xsl="http://www.w3.org/1999/XSL/Transform"
|
||||||
xmlns:html="http://www.w3.org/1999/xhtml"
|
xmlns:html="http://www.w3.org/1999/xhtml"
|
||||||
xmlns:rtf="http://rtf2xml.sourceforge.net/"
|
xmlns:rtf="http://rtf2xml.sourceforge.net/"
|
||||||
|
xmlns:c="calibre"
|
||||||
|
extension-element-prefixes="c"
|
||||||
exclude-result-prefixes="rtf"
|
exclude-result-prefixes="rtf"
|
||||||
>
|
>
|
||||||
|
|
||||||
@ -147,7 +130,7 @@ xhtml = '''\
|
|||||||
<xsl:text>generator</xsl:text>
|
<xsl:text>generator</xsl:text>
|
||||||
</xsl:attribute>
|
</xsl:attribute>
|
||||||
<xsl:attribute name="content">
|
<xsl:attribute name="content">
|
||||||
<xsl:text>http://rtf2xml.sourceforge.net/</xsl:text>
|
<xsl:text>http://calibre-ebook.com</xsl:text>
|
||||||
</xsl:attribute>
|
</xsl:attribute>
|
||||||
</xsl:element>
|
</xsl:element>
|
||||||
|
|
||||||
@ -233,9 +216,7 @@ xhtml = '''\
|
|||||||
<xsl:text>span.italic-bold{font-style:italic;font-weight:bold}
</xsl:text>
|
<xsl:text>span.italic-bold{font-style:italic;font-weight:bold}
</xsl:text>
|
||||||
<xsl:text>span.italic-underline{font-style:italic;text-decoration:underline}
</xsl:text>
|
<xsl:text>span.italic-underline{font-style:italic;text-decoration:underline}
</xsl:text>
|
||||||
<xsl:text>span.bold-underline{font-weight:bold;text-decoration:underline}
</xsl:text>
|
<xsl:text>span.bold-underline{font-weight:bold;text-decoration:underline}
</xsl:text>
|
||||||
<xsl:for-each select="//rtf:inline">
|
<xsl:text>
</xsl:text>
|
||||||
<xsl:call-template name="parse-inline"/>
|
|
||||||
</xsl:for-each>
|
|
||||||
</xsl:document>
|
</xsl:document>
|
||||||
</xsl:template>
|
</xsl:template>
|
||||||
|
|
||||||
@ -287,52 +268,6 @@ xhtml = '''\
|
|||||||
</xsl:if>
|
</xsl:if>
|
||||||
</xsl:template>
|
</xsl:template>
|
||||||
|
|
||||||
<xsl:template name="parse-inline">
|
|
||||||
<xsl:variable name="num-attrs" select="count(@*)"/>
|
|
||||||
<xsl:choose>
|
|
||||||
<xsl:when test="$num-attrs = 1 and @italics"/>
|
|
||||||
<xsl:when test="$num-attrs = 1 and @bold"/>
|
|
||||||
<xsl:when test="$num-attrs = 1 and @underline"/>
|
|
||||||
<xsl:when test="$num-attrs = 2 and @italics and @bold"/>
|
|
||||||
<xsl:when test="$num-attrs = 2 and @italcs and @underline"/>
|
|
||||||
<xsl:when test="$num-attrs = 2 and @bold and @underline"/>
|
|
||||||
<xsl:otherwise>
|
|
||||||
<xsl:text>span.</xsl:text>
|
|
||||||
<xsl:value-of select="generate-id(.)"/>
|
|
||||||
<xsl:text>{</xsl:text>
|
|
||||||
<xsl:if test="@italics = 'true'">
|
|
||||||
<xsl:text>font-style:italic;</xsl:text>
|
|
||||||
</xsl:if>
|
|
||||||
<xsl:if test="@italics = 'false'">
|
|
||||||
<xsl:text>font-style:normal;</xsl:text>
|
|
||||||
</xsl:if>
|
|
||||||
<xsl:if test="@bold = 'true'">
|
|
||||||
<xsl:text>font-weight:bold;</xsl:text>
|
|
||||||
</xsl:if>
|
|
||||||
<xsl:if test="@bold = 'false'">
|
|
||||||
<xsl:text>font-weight:normal;</xsl:text>
|
|
||||||
</xsl:if>
|
|
||||||
<xsl:if test="@underline and @underline != 'false'">
|
|
||||||
<xsl:text>text-decoration:underline;</xsl:text>
|
|
||||||
</xsl:if>
|
|
||||||
<xsl:if test="@underline= 'false'">
|
|
||||||
<xsl:text>text-decoration:none;</xsl:text>
|
|
||||||
</xsl:if>
|
|
||||||
<xsl:if test="@strike-through = 'true'">
|
|
||||||
<xsl:text>text-decoration:line-through;</xsl:text>
|
|
||||||
</xsl:if>
|
|
||||||
<xsl:if test="@strike-through = 'false'">
|
|
||||||
<xsl:text>text-decoration:none;</xsl:text>
|
|
||||||
</xsl:if>
|
|
||||||
<xsl:if test="@font-size">
|
|
||||||
<xsl:text>font-size:</xsl:text>
|
|
||||||
<xsl:value-of select="@font-size"/>
|
|
||||||
<xsl:text>pt;</xsl:text>
|
|
||||||
</xsl:if>
|
|
||||||
<xsl:text>}</xsl:text>
|
|
||||||
</xsl:otherwise>
|
|
||||||
</xsl:choose>
|
|
||||||
</xsl:template>
|
|
||||||
|
|
||||||
<xsl:template match="rtf:inline">
|
<xsl:template match="rtf:inline">
|
||||||
<xsl:variable name="num-attrs" select="count(@*)"/>
|
<xsl:variable name="num-attrs" select="count(@*)"/>
|
||||||
@ -345,45 +280,7 @@ xhtml = '''\
|
|||||||
<xsl:otherwise>
|
<xsl:otherwise>
|
||||||
<xsl:element name="span">
|
<xsl:element name="span">
|
||||||
<xsl:attribute name="class">
|
<xsl:attribute name="class">
|
||||||
<xsl:choose>
|
<c:inline-class/>
|
||||||
<xsl:when test="$num-attrs=1 and @italics='true'">
|
|
||||||
<xsl:text>italic</xsl:text>
|
|
||||||
</xsl:when>
|
|
||||||
<xsl:when test="$num-attrs=1 and @italics='false'">
|
|
||||||
<xsl:text>no-italic</xsl:text>
|
|
||||||
</xsl:when>
|
|
||||||
<xsl:when test="$num-attrs=1 and @bold='true'">
|
|
||||||
<xsl:text>bold</xsl:text>
|
|
||||||
</xsl:when>
|
|
||||||
<xsl:when test="$num-attrs=1 and @bold='true'">
|
|
||||||
<xsl:text>bold</xsl:text>
|
|
||||||
</xsl:when>
|
|
||||||
<xsl:when test="$num-attrs=1 and @bold='false'">
|
|
||||||
<xsl:text>no-bold</xsl:text>
|
|
||||||
</xsl:when>
|
|
||||||
<xsl:when test="$num-attrs=1 and @underlined">
|
|
||||||
<xsl:choose>
|
|
||||||
<xsl:when test="not(@underlined='false')">
|
|
||||||
<xsl:text>underline</xsl:text>
|
|
||||||
</xsl:when>
|
|
||||||
<xsl:otherwise>
|
|
||||||
<xsl:text>no-underline</xsl:text>
|
|
||||||
</xsl:otherwise>
|
|
||||||
</xsl:choose>
|
|
||||||
</xsl:when>
|
|
||||||
<xsl:when test="$num-attrs=2 and @bold='true' and @italics='true'">
|
|
||||||
<xsl:text>italic-bold</xsl:text>
|
|
||||||
</xsl:when>
|
|
||||||
<xsl:when test="$num-attrs=2 and @italics='true' and @underline and @underline != 'false'">
|
|
||||||
<xsl:text>italic-underline</xsl:text>
|
|
||||||
</xsl:when>
|
|
||||||
<xsl:when test="$num-attrs=2 and @bold='true' and @underline and @underline != 'false'">
|
|
||||||
<xsl:text>bold-underline</xsl:text>
|
|
||||||
</xsl:when>
|
|
||||||
<xsl:otherwise>
|
|
||||||
<xsl:value-of select="generate-id(.)"/>
|
|
||||||
</xsl:otherwise>
|
|
||||||
</xsl:choose>
|
|
||||||
</xsl:attribute>
|
</xsl:attribute>
|
||||||
<xsl:apply-templates/>
|
<xsl:apply-templates/>
|
||||||
</xsl:element>
|
</xsl:element>
|
||||||
@ -539,4 +436,3 @@ xhtml = '''\
|
|||||||
</xsl:template>
|
</xsl:template>
|
||||||
|
|
||||||
</xsl:stylesheet>
|
</xsl:stylesheet>
|
||||||
'''
|
|
||||||
@ -2,12 +2,48 @@ from __future__ import with_statement
|
|||||||
__license__ = 'GPL v3'
|
__license__ = 'GPL v3'
|
||||||
__copyright__ = '2008, Kovid Goyal <kovid at kovidgoyal.net>'
|
__copyright__ = '2008, Kovid Goyal <kovid at kovidgoyal.net>'
|
||||||
|
|
||||||
import os, glob, re
|
import os, glob, re, textwrap
|
||||||
|
|
||||||
from lxml import etree
|
from lxml import etree
|
||||||
|
|
||||||
from calibre.customize.conversion import InputFormatPlugin
|
from calibre.customize.conversion import InputFormatPlugin
|
||||||
|
|
||||||
|
class InlineClass(etree.XSLTExtension):
|
||||||
|
|
||||||
|
FMTS = ('italics', 'bold', 'underlined', 'strike-through', 'small-caps')
|
||||||
|
|
||||||
|
def __init__(self, log):
|
||||||
|
etree.XSLTExtension.__init__(self)
|
||||||
|
self.log = log
|
||||||
|
self.font_sizes = []
|
||||||
|
self.colors = []
|
||||||
|
|
||||||
|
def execute(self, context, self_node, input_node, output_parent):
|
||||||
|
classes = []
|
||||||
|
for x in self.FMTS:
|
||||||
|
cls = x if input_node.get(x, None) == 'true' else 'no-'+x
|
||||||
|
classes.append(cls)
|
||||||
|
none = True
|
||||||
|
for x in self.FMTS:
|
||||||
|
if 'no-'+x not in classes:
|
||||||
|
none = False
|
||||||
|
break
|
||||||
|
if none:
|
||||||
|
classes = ['none']
|
||||||
|
fs = input_node.get('font-size', False)
|
||||||
|
if fs:
|
||||||
|
if fs not in self.font_sizes:
|
||||||
|
self.font_sizes.append(fs)
|
||||||
|
classes.append('fs%d'%self.font_sizes.index(fs))
|
||||||
|
fc = input_node.get('font-color', False)
|
||||||
|
if fc:
|
||||||
|
if fc not in self.colors:
|
||||||
|
self.colors.append(fc)
|
||||||
|
classes.append('col%d'%self.colors.index(fc))
|
||||||
|
|
||||||
|
output_parent.text = ' '.join(classes)
|
||||||
|
|
||||||
|
|
||||||
class RTFInput(InputFormatPlugin):
|
class RTFInput(InputFormatPlugin):
|
||||||
|
|
||||||
name = 'RTF Input'
|
name = 'RTF Input'
|
||||||
@ -21,15 +57,15 @@ class RTFInput(InputFormatPlugin):
|
|||||||
parser = ParseRtf(
|
parser = ParseRtf(
|
||||||
in_file = stream,
|
in_file = stream,
|
||||||
out_file = ofile,
|
out_file = ofile,
|
||||||
# Convert symbol fonts to unicode equivelents. Default
|
# Convert symbol fonts to unicode equivalents. Default
|
||||||
# is 1
|
# is 1
|
||||||
convert_symbol = 1,
|
convert_symbol = 1,
|
||||||
|
|
||||||
# Convert Zapf fonts to unicode equivelents. Default
|
# Convert Zapf fonts to unicode equivalents. Default
|
||||||
# is 1.
|
# is 1.
|
||||||
convert_zapf = 1,
|
convert_zapf = 1,
|
||||||
|
|
||||||
# Convert Wingding fonts to unicode equivelents.
|
# Convert Wingding fonts to unicode equivalents.
|
||||||
# Default is 1.
|
# Default is 1.
|
||||||
convert_wingdings = 1,
|
convert_wingdings = 1,
|
||||||
|
|
||||||
@ -94,9 +130,41 @@ class RTFInput(InputFormatPlugin):
|
|||||||
#open(name+'.hex', 'wb').write(enc)
|
#open(name+'.hex', 'wb').write(enc)
|
||||||
return imap
|
return imap
|
||||||
|
|
||||||
|
def write_inline_css(self, ic):
|
||||||
|
font_size_classes = ['span.fs%d { font-size: %spt }'%(i, x) for i, x in
|
||||||
|
enumerate(ic.font_sizes)]
|
||||||
|
color_classes = ['span.col%d { color: %s }'%(i, x) for i, x in
|
||||||
|
enumerate(ic.colors)]
|
||||||
|
css = textwrap.dedent('''
|
||||||
|
span.none {
|
||||||
|
text-decoration: none; font-weight: normal;
|
||||||
|
font-style: normal; font-variant: normal
|
||||||
|
}
|
||||||
|
|
||||||
|
span.italics { font-style: italic }
|
||||||
|
span.no-italics { font-style: normal }
|
||||||
|
|
||||||
|
span.bold { font-weight: bold }
|
||||||
|
span.no-bold { font-weight: normal }
|
||||||
|
|
||||||
|
span.small-caps { font-variant: small-caps }
|
||||||
|
span.no-small-caps { font-variant: normal }
|
||||||
|
|
||||||
|
span.underlined { text-decoration: underline }
|
||||||
|
span.no-underlined { text-decoration: none }
|
||||||
|
|
||||||
|
span.strike-through { text-decoration: line-through }
|
||||||
|
span.no-strike-through { text-decoration: none }
|
||||||
|
|
||||||
|
''')
|
||||||
|
css += '\n'+'\n'.join(font_size_classes)
|
||||||
|
css += '\n' +'\n'.join(color_classes)
|
||||||
|
with open('styles.css', 'ab') as f:
|
||||||
|
f.write(css)
|
||||||
|
|
||||||
|
|
||||||
def convert(self, stream, options, file_ext, log,
|
def convert(self, stream, options, file_ext, log,
|
||||||
accelerators):
|
accelerators):
|
||||||
from calibre.ebooks.rtf.xsl import xhtml
|
|
||||||
from calibre.ebooks.metadata.meta import get_metadata
|
from calibre.ebooks.metadata.meta import get_metadata
|
||||||
from calibre.ebooks.metadata.opf2 import OPFCreator
|
from calibre.ebooks.metadata.opf2 import OPFCreator
|
||||||
from calibre.ebooks.rtf2xml.ParseRtf import RtfInvalidCodeException
|
from calibre.ebooks.rtf2xml.ParseRtf import RtfInvalidCodeException
|
||||||
@ -124,15 +192,18 @@ class RTFInput(InputFormatPlugin):
|
|||||||
if name is not None:
|
if name is not None:
|
||||||
pict.set('num', name)
|
pict.set('num', name)
|
||||||
self.log('Converting XML to HTML...')
|
self.log('Converting XML to HTML...')
|
||||||
styledoc = etree.fromstring(xhtml)
|
inline_class = InlineClass(self.log)
|
||||||
|
styledoc = etree.fromstring(P('templates/rtf.xsl', data=True))
|
||||||
|
|
||||||
transform = etree.XSLT(styledoc)
|
extensions = { ('calibre', 'inline-class') : inline_class }
|
||||||
|
transform = etree.XSLT(styledoc, extensions=extensions)
|
||||||
result = transform(doc)
|
result = transform(doc)
|
||||||
html = 'index.xhtml'
|
html = 'index.xhtml'
|
||||||
with open(html, 'wb') as f:
|
with open(html, 'wb') as f:
|
||||||
res = transform.tostring(result)
|
res = transform.tostring(result)
|
||||||
res = res[:100].replace('xmlns:html', 'xmlns') + res[100:]
|
res = res[:100].replace('xmlns:html', 'xmlns') + res[100:]
|
||||||
f.write(res)
|
f.write(res)
|
||||||
|
self.write_inline_css(inline_class)
|
||||||
stream.seek(0)
|
stream.seek(0)
|
||||||
mi = get_metadata(stream, 'rtf')
|
mi = get_metadata(stream, 'rtf')
|
||||||
if not mi.title:
|
if not mi.title:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user