koreader/reader.lua

#!./kpdfview
--[[
    KindlePDFViewer: a reader implementation
    Copyright (C) 2011 Hans-Werner Hilse <hilse@web.de>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
]]--

require "alt_getopt"

KEY_PAGEUP = 109 -- nonstandard
KEY_PAGEDOWN = 124 -- nonstandard
KEY_BACK = 91 -- nonstandard
KEY_MENU = 139

-- DPad:
KEY_UP = 122 -- nonstandard
KEY_DOWN = 123 -- nonstandard
KEY_LEFT = 105
KEY_RIGHT = 106
KEY_BTN = 92 -- nonstandard

-- constants from <linux/input.h>
EV_KEY = 1

-- event values
EVENT_VALUE_KEY_PRESS = 1
EVENT_VALUE_KEY_REPEAT = 2
EVENT_VALUE_KEY_RELEASE = 0

-- option parsing:
longopts = {
	password = "p",
	goto = "g",
	gamma = "G",
	device = "d",
	help = "h"
}
optarg, optind = alt_getopt.get_opts(ARGV, "p:G:hg:d:", longopts)
if optarg["h"] or ARGV[optind] == nil then
	print("usage: ./reader.lua [OPTION] ... DOCUMENT.PDF")
	print("Read PDFs on your E-Ink reader")
	print("")
	print("-p, --password=PASSWORD   set password for reading PDF document")
	print("-g, --goto=page           start reading on page")
	print("-G, --gamma=GAMMA         set gamma correction")
	print("                          (floating point notation, e.g. \"1.5\")")
	print("-d, --device=DEVICE       set device specific configuration,")
	print("                          currently one of \"kdxg\" (default), \"k3\"")
	print("-h, --help                show this usage help")
	print("")
	print("This software is licensed under the GPLv3.")
	print("See http://github.com/hwhw/kindlepdfviewer for more info.")
	return
end

rcount = 5
rcountmax = 5

globalzoom = -1

if optarg["d"] == "k3" then
	-- for now, the only difference is the additional input device
	input.open("/dev/input/event0")
	input.open("/dev/input/event1")
	input.open("/dev/input/event2")
elseif optarg["d"] == "emu" then
	input.open("")
	-- SDL key codes
	KEY_PAGEDOWN = 112
	KEY_PAGEUP = 117
	KEY_BACK = 22 -- backspace
	KEY_MENU = 67 -- F1
	KEY_UP = 111
	KEY_DOWN = 116
	KEY_LEFT = 113
	KEY_RIGHT = 114
	KEY_BTN = 36 -- enter for now
else
	input.open("/dev/input/event0")
	input.open("/dev/input/event1")
end

doc = pdf.openDocument(ARGV[optind], optarg["p"] or "")

print("pdf has "..doc:getPages().." pages.")

fb = einkfb.open("/dev/fb0")
width, height = fb:getSize()

nulldc = pdf.newDC()

cache = {
	{ age = 0, no = 0, bb = blitbuffer.new(width, height), dc = pdf.newDC(), page = nil },
	{ age = 0, no = 0, bb = blitbuffer.new(width, height), dc = pdf.newDC(), page = nil },
	{ age = 0, no = 0, bb = blitbuffer.new(width, height), dc = pdf.newDC(), page = nil }
}
function freecache()
	for i = 1, #cache do
		if cache[i].page ~= nil then
			print("freeing slot="..i.." oldpage="..cache[i].no)
			cache[i].page:close()
			cache[i].page = nil
		end
	end
end
function checkcache(no)
	for i = 1, #cache do
		if cache[i].no == no and cache[i].page ~= nil then
			print("cache hit: slot="..i.." page="..no)
			return i
		end
	end
	print("cache miss")
	return nil
end
function cacheslot()
	freeslot = nil
	while freeslot == nil do
		for i = 1, #cache do
			if cache[i].age > 0 then
				print("aging slot="..i)
				cache[i].age = cache[i].age - 1
			else
				if cache[i].page ~= nil then
					print("freeing slot="..i.." oldpage="..cache[i].no)
					cache[i].page:close()
					cache[i].page = nil
				end
				freeslot = i
			end
		end
	end
	print("returning free slot="..freeslot)
	return freeslot
end

function draworcache(no)
	local slot = checkcache(no)
	if slot == nil then
		slot = cacheslot()
		cache[slot].no = no
		cache[slot].age = #cache
		cache[slot].page = doc:openPage(no)
		setzoom(slot)
		print("drawing page="..no.." to slot="..slot)
		cache[slot].page:draw(cache[slot].dc, cache[slot].bb, 0, 0)
	end
	return slot
end

function setzoom(cacheslot)
	local pwidth, pheight = cache[cacheslot].page:getSize(nulldc)

	-- default zoom: fit to page
	local zoom = width / pwidth
	local offset_x = 0
	local offset_y = (height - (zoom * pheight)) / 2
	if height / pheight < zoom then
		zoom = height / pheight
		offset_x = (width - (zoom * pwidth)) / 2
		offset_y = 0
	end

	cache[cacheslot].dc:setZoom(zoom)
	cache[cacheslot].dc:setOffset(offset_x, offset_y)

	-- set gamma here, we don't have any other good place for this right now:
	if optarg["G"] then
		print("gamma correction: "..optarg["G"])
		cache[cacheslot].dc:setGamma(optarg["G"])
	end
end

function show(no)
	local slot = draworcache(no)
	fb:blitFullFrom(cache[slot].bb)
	if rcount == rcountmax then
		print("full refresh")
		rcount = 1
		fb:refresh(0)
	else
		print("partial refresh")
		rcount = rcount + 1
		fb:refresh(1)
	end
	slot_visible = slot;
end

function goto(no)
	if no < 1 or no > doc:getPages() then
		return
	end
	pageno = no
	show(no)
	if no < doc:getPages() then
		-- always pre-cache next page
		draworcache(no+1)
	end
end

function modify_gamma(offset)
	local slot = slot_visible
	local gamma = cache[slot].dc:getGamma();
	if gamma == -1 then
		gamma = 1
	end
	local no = cache[slot].no
	print("modify_gamma "..no.." slot="..slot.." gamma="..gamma.." offset="..offset)
	gamma = gamma + offset;
	optarg["G"] = gamma; -- for next page
	freecache()
	goto(no)
end

function mainloop()
	while 1 do
		local ev = input.waitForEvent()
		if ev.type == EV_KEY and ev.value == EVENT_VALUE_KEY_PRESS then
			local secs, usecs = util.gettime()
			if ev.code == KEY_PAGEUP then
				print(cache)
				goto(pageno + 1)
				print(cache)
			elseif ev.code == KEY_PAGEDOWN then
				print(cache)
				goto(pageno - 1)
				print(cache)
			elseif ev.code == KEY_BACK then
				return
			elseif ev.code == KEY_UP then
				modify_gamma( 0.2 )
			elseif ev.code == KEY_DOWN then
				modify_gamma( -0.2 )
			end
			local nsecs, nusecs = util.gettime()
			local dur = (nsecs - secs) * 1000000 + nusecs - usecs
			print("E: T="..ev.type.." V="..ev.value.." C="..ev.code.." DUR="..dur)
		end
	end
end

goto(tonumber(optarg["g"]) or 1)

mainloop()