renun/koreader
1
0
Fork 0
mirror of https://github.com/koreader/koreader.git synced 2025-08-10 00:52:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

count only leaf nodes of toc tree when level is zero

Browse Source
This commit is contained in:
chrox
2014-10-11 21:10:07 +08:00
parent 2b26e599d9
commit de38ea1199
2 changed files with 31 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
frontend/apps/reader/modules/readertoc.lua
View File

@@ -101,7 +101,8 @@ end
--[[
TOC ticks is a list of page number in ascending order of TOC nodes at certain level
positive level counts nodes of the depth level (level 1 for depth 1)
non-positive level counts nodes of reversed depth level (level -1 for max_depth-1)
negative level counts nodes of reversed depth level (level -1 for max_depth)
zero level counts leaf nodes of the toc tree
--]]
function ReaderToc:getTocTicks(level)
if self.ticks[level] then return self.ticks[level] end
@@ -110,15 +111,26 @@ function ReaderToc:getTocTicks(level)
local ticks = {}
if #self.toc > 0 then
local depth = nil
if level > 0 then
depth = level
if level == 0 then
local depth = 0
for i = #self.toc, 1, -1 do
local v = self.toc[i]
if v.depth >= depth then
table.insert(ticks, v.page)
end
depth = v.depth
end
else
depth = self:getMaxDepth() + level
end
for _, v in ipairs(self.toc) do
if v.depth == depth then
table.insert(ticks, v.page)
local depth = nil
if level > 0 then
depth = level
else
depth = self:getMaxDepth() + level + 1
end
for _, v in ipairs(self.toc) do
if v.depth == depth then
table.insert(ticks, v.page)
end
end
end
-- normally the ticks are sorted already but in rare cases