Merge branch 'toc'

pdf.c

@@ -87,6 +87,53 @@ static int getNumberOfPages(lua_State *L) {
 	return 1;
 }
 
+/*
+ * helper function for getTableOfContent()
+ */
+static int walkTableOfContent(lua_State *L, fz_outline* ol, int *count, int depth) {
+	depth++;
+	while(ol) {
+		lua_pushnumber(L, *count);
+
+		/* set subtable */
+		lua_newtable(L);
+		lua_pushstring(L, "page");
+		lua_pushnumber(L, ol->dest.ld.gotor.page + 1);
+		lua_settable(L, -3);
+		lua_pushstring(L, "depth");
+		lua_pushnumber(L, depth); 
+		lua_settable(L, -3);
+		lua_pushstring(L, "title");
+		lua_pushstring(L, ol->title);
+		lua_settable(L, -3);
+
+		lua_settable(L, -3);
+
+		(*count)++;
+		if (ol->down) {
+			walkTableOfContent(L, ol->down, count, depth);
+		}
+		ol = ol->next;
+	}
+	return 0;
+}
+
+/*
+ * Return a table with (title,page) pair as entry:
+ * {"chapter1"=12, "chapter2"=20}
+ */
+static int getTableOfContent(lua_State *L) {
+	fz_outline *ol;
+	int count = 1;
+
+	PdfDocument *doc = (PdfDocument*) luaL_checkudata(L, 1, "pdfdocument");
+	ol = pdf_load_outline(doc->xref);
+
+	lua_newtable(L);
+	walkTableOfContent(L, ol, &count, 0);
+	return 1;
+}
+
 static int newDrawContext(lua_State *L) {
 	int rotate = luaL_optint(L, 1, 0);
 	double zoom = luaL_optnumber(L, 2, (double) 1.0);
@@ -309,6 +356,7 @@ static const struct luaL_reg pdf_func[] = {
 static const struct luaL_reg pdfdocument_meth[] = {
 	{"openPage", openPage},
 	{"getPages", getNumberOfPages},
+	{"getTOC", getTableOfContent},
 	{"close", closeDocument},
 	{"__gc", closeDocument},
 	{NULL, NULL}

pdfreader.lua

@@ -1,5 +1,6 @@
 require "keys"
 require "settings"
+require "tocmenu"
 
 PDFReader = {
 	-- "constants":
@@ -335,6 +336,18 @@ function PDFReader:inputloop()
 					self:setglobalzoommode(self.ZOOM_FIT_TO_PAGE_HEIGHT)
 				end
 
+			elseif ev.code == KEY_T then
+				-- show table of content menu
+				toc = self.doc:getTOC()
+				toc_menu = TOCMenu:new(toc)
+				--toc_menu:dump()
+				no = toc_menu:choose(0, fb.bb:getHeight())
+				if no then
+					self:goto(no)
+				else
+					self:goto(self.pageno)
+				end
+
 			elseif ev.code == KEY_J then
 				self:setrotate( self.globalrotate + 10 )
 			elseif ev.code == KEY_K then

tocmenu.lua

@@ -0,0 +1,169 @@
+require "rendertext"
+require "keys"
+require "graphics"
+
+TOCMenu = {
+	-- font for displaying file/dir names
+	face = freetype.newBuiltinFace("cjk", 22),
+	fhash = "s22",
+	-- font for page title
+	tface = freetype.newBuiltinFace("Helvetica-BoldOblique", 25),
+	tfhash = "hbo25",
+	-- font for paging display
+	sface = freetype.newBuiltinFace("sans", 16),
+	sfhash = "s16",
+	-- title height
+	title_H = 40,
+	-- spacing between lines
+	spacing = 36,
+	-- foot height
+	foot_H = 27,
+
+	-- state buffer
+	toc = {},
+	items = 14,
+	page = 1,
+	current = 1,
+	oldcurrent = 0,
+}
+
+function TOCMenu:new(toc)
+	--@TODO set font here in the future  21.02 2012
+	--clearglyphcache()
+	instance = self
+	instance.toc = toc
+	instance.items = #toc
+	return instance
+end
+
+function TOCMenu:dump()
+	for k,v in pairs(self.toc) do
+		print("TOC item: "..k)
+		for key,value in pairs(v) do
+			print("    "..key..": "..value)
+		end
+	end
+end
+
+function TOCMenu:choose(ypos, height)
+	local perpage = math.floor(height / self.spacing) - 2
+	local pagedirty = true
+	local markerdirty = false
+
+	local prevItem = function ()
+		if self.current == 1 then
+			if self.page > 1 then
+				self.current = perpage
+				self.page = self.page - 1
+				pagedirty = true
+			end
+		else
+			self.current = self.current - 1
+			markerdirty = true
+		end
+	end
+
+	local nextItem = function ()
+		if self.current == perpage then
+			if self.page < (self.items / perpage) then
+				self.current = 1
+				self.page = self.page + 1
+				pagedirty = true
+			end
+		else
+			if self.page ~= math.floor(self.items / perpage) + 1
+				or self.current + (self.page-1)*perpage < self.items then
+				self.current = self.current + 1
+				markerdirty = true
+			end
+		end
+	end
+
+
+	while true do
+		if pagedirty then
+			-- draw menu title
+			fb.bb:paintRect(0, ypos, fb.bb:getWidth(), self.title_H + 10, 0)
+			fb.bb:paintRect(30, ypos + 10, fb.bb:getWidth() - 60, self.title_H, 5)
+			x = fb.bb:getWidth() - 260 -- move text to the right
+			y = ypos + self.title_H
+			renderUtf8Text(fb.bb, x, y, self.tface, self.tfhash,
+				"Table of Contents", true)
+
+			-- draw font items
+			fb.bb:paintRect(0, ypos + self.title_H + 10, fb.bb:getWidth(), height - self.title_H, 0)
+			local c
+			for c = 1, perpage do
+				local i = (self.page - 1) * perpage + c 
+				if i <= self.items then
+					y = ypos + self.title_H + (self.spacing * c)
+					renderUtf8Text(fb.bb, 30, y, self.face, self.fhash,
+						("      "):rep(self.toc[i]["depth"]-1)..self.toc[i]["title"], true)
+				end
+			end
+
+			-- draw footer
+			y = ypos + self.title_H + (self.spacing * perpage) + self.foot_H
+			x = (fb.bb:getWidth() / 2) - 50
+			renderUtf8Text(fb.bb, x, y, self.sface, self.sfhash,
+				"Page "..self.page.." of "..(math.floor(self.items / perpage)+1), true)
+			markerdirty = true
+		end
+
+		if markerdirty then
+			if not pagedirty then
+				if self.oldcurrent > 0 then
+					y = ypos + self.title_H + (self.spacing * self.oldcurrent) + 10
+					fb.bb:paintRect(30, y, fb.bb:getWidth() - 60, 3, 0)
+					fb:refresh(1, 30, y, fb.bb:getWidth() - 60, 3)
+				end
+			end
+			-- draw new marker line
+			y = ypos + self.title_H + (self.spacing * self.current) + 10
+			fb.bb:paintRect(30, y, fb.bb:getWidth() - 60, 3, 15)
+			if not pagedirty then
+				fb:refresh(1, 30, y, fb.bb:getWidth() - 60, 3)
+			end
+			self.oldcurrent = self.current
+			markerdirty = false
+		end
+
+		if pagedirty then
+			fb:refresh(0, 0, ypos, fb.bb:getWidth(), height)
+			pagedirty = false
+		end
+
+		local ev = input.waitForEvent()
+		if ev.type == EV_KEY and ev.value == EVENT_VALUE_KEY_PRESS then
+			ev.code = adjustFWKey(ev.code)
+			if ev.code == KEY_FW_UP then
+				prevItem()
+			elseif ev.code == KEY_FW_DOWN then
+				nextItem()
+			elseif ev.code == KEY_PGFWD then
+				if self.page < (self.items / perpage) then
+					if self.current + self.page*perpage > self.items then
+						self.current = self.items - self.page*perpage
+					end
+					self.page = self.page + 1
+					pagedirty = true
+				else
+					self.current = self.items - (self.page-1)*perpage
+					markerdirty = true
+				end
+			elseif ev.code == KEY_PGBCK then
+				if self.page > 1 then
+					self.page = self.page - 1
+					pagedirty = true
+				else
+					self.current = 1
+					markerdirty = true
+				end
+			elseif ev.code == KEY_ENTER or ev.code == KEY_FW_PRESS then
+				return self.toc[perpage*(self.page-1)+self.current]["page"]
+			elseif ev.code == KEY_BACK then
+				return nil
+			end
+		end
+	end
+end