50 lines
1.4 KiB
Lua
50 lines
1.4 KiB
Lua
local min = math.min
|
|
local format = string.format
|
|
local concat = table.concat
|
|
local pdfvariable = pdf.variable
|
|
local function write(pdf, tree, total, max)
|
|
tree = tree or pdf.pages
|
|
if #tree == 0 then
|
|
local id = pdf:getobj()
|
|
pdf:indirect(id, '<</Type/Pages/Kids[]/Count 0>>')
|
|
return id
|
|
end
|
|
max = max or 6 -- These defaults only work on the lowest level
|
|
total = total or #tree
|
|
local remaining = total
|
|
-- if #tree == 1 then
|
|
-- retur
|
|
local newtree = {}
|
|
local parent = ""
|
|
for i=0,(#tree-1)//6 do
|
|
local id = tree[-i]
|
|
newtree[i+1] = id
|
|
if 0 == i % 6 and #tree > 6 then
|
|
local parentid = pdf:getobj()
|
|
newtree[-(i//6)] = parentid
|
|
parent = format("/Parent %i 0 R", parentid)
|
|
elseif #tree <= 6 then
|
|
parent = pdfvariable.pagesattr
|
|
end
|
|
pdf:indirect(id, format('<</Type/Pages%s/Kids[%s 0 R]/Count %i>>', parent, concat(tree, ' 0 R ', 6*i+1, min(#tree, 6*i+6)), min(remaining, max)))
|
|
remaining = remaining - max
|
|
end
|
|
if newtree[0] then
|
|
return write(pdf, newtree, total, max*6)
|
|
end
|
|
return newtree[1]
|
|
end
|
|
local function newpage(pdf)
|
|
local pageid = pdf:getobj()
|
|
local pagenumber = #pdf.pages
|
|
pdf.pages[pagenumber+1] = pageid
|
|
if 0 == pagenumber % 6 then
|
|
pdf.pages[-(pagenumber//6)] = pdf:getobj()
|
|
end
|
|
return pageid, pdf.pages[-(pagenumber//6)]
|
|
end
|
|
return {
|
|
write = write,
|
|
newpage = newpage,
|
|
}
|