Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

PML Output: Create a multi-level toc.

Browse Source
This commit is contained in:
John Schember 2011-02-04 07:29:57 -05:00
parent de6e5f5cc0
commit 9b60e7172d
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/calibre/ebooks/pml/pmlml.py
View File

@ -87,14 +87,18 @@ class PMLMLizer(object):
# This is used for adding \CX tags chapter markers. This is separate # This is used for adding \CX tags chapter markers. This is separate
# from the optional inline toc. # from the optional inline toc.
self.toc = {} self.toc = {}
for item in oeb_book.toc: self.create_flat_toc(self.oeb_book.toc)
return self.pmlmlize_spine()
def create_flat_toc(self, nodes, level=0):
for item in nodes:
href, mid, id = item.href.partition('#') href, mid, id = item.href.partition('#')
self.get_anchor_id(href, id) self.get_anchor_id(href, id)
if not self.toc.get(href, None): if not self.toc.get(href, None):
self.toc[href] = {} self.toc[href] = {}
self.toc[href][id] = item.title self.toc[href][id] = (item.title, level)
self.create_flat_toc(item.nodes, level + 1)
return self.pmlmlize_spine()
def pmlmlize_spine(self): def pmlmlize_spine(self):
self.image_hrefs = {} self.image_hrefs = {}
@ -256,9 +260,10 @@ class PMLMLizer(object):
toc_page = page.href toc_page = page.href
if self.toc.get(toc_page, None): if self.toc.get(toc_page, None):
for toc_x in (toc_name, toc_id): for toc_x in (toc_name, toc_id):
toc_title = self.toc[toc_page].get(toc_x, None) toc_title, toc_depth = self.toc[toc_page].get(toc_x, (None, 0))
if toc_title: if toc_title:
text.append('\\C0="%s"' % toc_title) toc_depth = max(min(toc_depth, 4), 0)
text.append('\\C%s="%s"' % (toc_depth, toc_title))
# Process style information that needs holds a single tag # Process style information that needs holds a single tag
# Commented out because every page in an OEB book starts with this style # Commented out because every page in an OEB book starts with this style