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

...

Browse Source
This commit is contained in:
Kovid Goyal 2011-08-18 19:33:29 -06:00
parent 72227947d3
commit 5db35a67c7
1 changed files with 3 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/calibre/ebooks/mobi/writer2/indexer.py
View File

@ -109,20 +109,6 @@ class TAGX(object): # {{{
list(map(self.add_tag, (11, 0))) list(map(self.add_tag, (11, 0)))
return self.header(1) + bytes(self.byts) return self.header(1) + bytes(self.byts)
class TAGX_BOOK(TAGX):
BITMASKS = dict(TAGX.BITMASKS)
BITMASKS.update({x:(1 << i) for i, x in enumerate([1, 2, 3, 4, 21, 22, 23])})
@property
def hierarchical_book(self):
'''
TAGX block for the primary index header of a hierarchical book
'''
list(map(self.add_tag, (1, 2, 3, 4, 21, 22, 23, 0)))
return self.header(1) + bytes(self.byts)
@property @property
def flat_book(self): def flat_book(self):
''' '''
@ -244,17 +230,6 @@ class IndexEntry(object):
ans = buf.getvalue() ans = buf.getvalue()
return ans return ans
class BookIndexEntry(IndexEntry):
@property
def entry_type(self):
tagx = TAGX_BOOK()
ans = 0
for tag in self.tag_nums:
ans |= tagx.BITMASKS[tag]
return ans
class PeriodicalIndexEntry(IndexEntry): class PeriodicalIndexEntry(IndexEntry):
def __init__(self, offset, label_offset, class_offset, depth): def __init__(self, offset, label_offset, class_offset, depth):
@ -571,9 +546,7 @@ class Indexer(object): # {{{
tagx_block = TAGX().secondary tagx_block = TAGX().secondary
else: else:
tagx_block = (TAGX().periodical if self.is_periodical else tagx_block = (TAGX().periodical if self.is_periodical else
(TAGX_BOOK().hierarchical_book if TAGX().flat_book)
self.book_has_subchapters else
TAGX_BOOK().flat_book))
header_length = 192 header_length = 192
# Ident 0 - 4 # Ident 0 - 4
@ -659,7 +632,6 @@ class Indexer(object): # {{{
# }}} # }}}
def create_book_index(self): # {{{ def create_book_index(self): # {{{
self.book_has_subchapters = False
indices = [] indices = []
seen = set() seen = set()
id_offsets = self.serializer.id_offsets id_offsets = self.serializer.id_offsets
@ -679,7 +651,7 @@ class Indexer(object): # {{{
continue continue
seen.add(offset) seen.add(offset)
indices.append(BookIndexEntry(offset, label)) indices.append(IndexEntry(offset, label))
indices.sort(key=lambda x:x.offset) indices.sort(key=lambda x:x.offset)
@ -703,7 +675,7 @@ class Indexer(object): # {{{
next_offset = self.serializer.body_end_offset next_offset = self.serializer.body_end_offset
index.length = next_offset - index.offset index.length = next_offset - index.offset
# Set index and depth values # Set index values
for index, x in enumerate(indices): for index, x in enumerate(indices):
x.index = index x.index = index