2
0
mirror of https://github.com/koreader/koreader synced 2024-11-13 19:11:25 +00:00
koreader/frontend/ui/uimanager.lua
NiLuJe 21210800c1
ReaderFooter/Header: Refine autorefresh repaint-or-not checks (#10045)
Use both a whitelist for targeted widget repaints, a blacklist for no repaint at all, and a fallback for a full in-order ReaderUI repaint when unsure.

Use a similar approach in ReaderHeader (i.e., prevent explicit refreshes while ReaderMenu is open).

Re #9979, re #9768
2023-02-02 00:29:23 +01:00

1659 lines
70 KiB
Lua

--[[--
This module manages widgets.
]]
local Device = require("device")
local Event = require("ui/event")
local Geom = require("ui/geometry")
local dbg = require("dbg")
local logger = require("logger")
local ffiUtil = require("ffi/util")
local util = require("util")
local time = require("ui/time")
local _ = require("gettext")
local Input = Device.input
local Screen = Device.screen
local DEFAULT_FULL_REFRESH_COUNT = 6
-- This is a singleton
local UIManager = {
-- trigger a full refresh when counter reaches FULL_REFRESH_COUNT
FULL_REFRESH_COUNT =
G_reader_settings:isTrue("night_mode") and G_reader_settings:readSetting("night_full_refresh_count") or G_reader_settings:readSetting("full_refresh_count") or DEFAULT_FULL_REFRESH_COUNT,
refresh_count = 0,
currently_scrolling = false,
-- How long to wait between ZMQ wakeups: 50ms.
ZMQ_TIMEOUT = 50 * 1000,
event_handlers = nil,
_now = time.now(),
_window_stack = {},
_task_queue = {},
_task_queue_dirty = false,
_dirty = {},
_zeromqs = {},
_refresh_stack = {},
_refresh_func_stack = {},
_entered_poweroff_stage = false,
_exit_code = nil,
_prevent_standby_count = 0,
_prev_prevent_standby_count = 0,
event_hook = require("ui/hook_container"):new()
}
function UIManager:init()
self.event_handlers = {
__default__ = function(input_event)
self:sendEvent(input_event)
end,
SaveState = function()
self:flushSettings()
end,
Power = function(input_event)
Device:onPowerEvent(input_event)
end,
-- This is for hotpluggable evdev input devices (e.g., USB OTG)
UsbDevicePlugIn = function(input_event)
-- Retrieve the argument set by Input:handleKeyBoardEv
local evdev = table.remove(Input.fake_event_args[input_event])
self:broadcastEvent(Event:new("EvdevInputInsert", evdev))
end,
UsbDevicePlugOut = function(input_event)
local evdev = table.remove(Input.fake_event_args[input_event])
self:broadcastEvent(Event:new("EvdevInputRemove", evdev))
end,
}
self.poweroff_action = function()
self._entered_poweroff_stage = true
logger.info("Powering off the device...")
self:broadcastEvent(Event:new("Close"))
local Screensaver = require("ui/screensaver")
Screensaver:setup("poweroff", _("Powered off"))
Screensaver:show()
self:nextTick(function()
Device:saveSettings()
Device:powerOff()
if Device:isKobo() then
self:quit(88)
else
self:quit()
end
end)
end
self.reboot_action = function()
self._entered_poweroff_stage = true
logger.info("Rebooting the device...")
self:broadcastEvent(Event:new("Close"))
local Screensaver = require("ui/screensaver")
Screensaver:setup("reboot", _("Rebooting…"))
Screensaver:show()
self:nextTick(function()
Device:saveSettings()
Device:reboot()
if Device:isKobo() then
self:quit(88)
else
self:quit()
end
end)
end
Device:_setEventHandlers(self)
-- A simple wrapper for UIManager:quit()
-- This may be overwritten by setRunForeverMode(); for testing purposes
self:unsetRunForeverMode()
end
--[[--
Registers and shows a widget.
Widgets are registered in a stack, from bottom to top in registration order,
with a few tweaks to handle modals & toasts:
toast widgets are stacked together on top,
then modal widgets are stacked together, and finally come standard widgets.
If you think about how painting will be handled (also bottom to top), this makes perfect sense ;).
For more details about refreshtype, refreshregion & refreshdither see the description of `setDirty`.
If refreshtype is omitted, no refresh will be enqueued at this time.
@param widget a @{ui.widget.widget|widget} object
@string refreshtype `"full"`, `"flashpartial"`, `"flashui"`, `"[partial]"`, `"[ui]"`, `"partial"`, `"ui"`, `"fast"`, `"a2"` (optional)
@param refreshregion a rectangle @{ui.geometry.Geom|Geom} object (optional, requires refreshtype to be set)
@int x horizontal screen offset (optional, `0` if omitted)
@int y vertical screen offset (optional, `0` if omitted)
@bool refreshdither `true` if widget requires dithering (optional, requires refreshtype to be set)
@see setDirty
]]
function UIManager:show(widget, refreshtype, refreshregion, x, y, refreshdither)
if not widget then
logger.dbg("attempted to show a nil widget")
return
end
logger.dbg("show widget:", widget.id or widget.name or tostring(widget))
local window = {x = x or 0, y = y or 0, widget = widget}
-- put this window on top of the topmost non-modal window
for i = #self._window_stack, 0, -1 do
local top_window = self._window_stack[i]
-- toasts are stacked on top of other toasts,
-- then come modals, and then other widgets
if top_window and top_window.widget.toast then
if widget.toast then
table.insert(self._window_stack, i + 1, window)
break
end
elseif widget.modal or not top_window or not top_window.widget.modal then
table.insert(self._window_stack, i + 1, window)
break
end
end
-- and schedule it to be painted
self:setDirty(widget, refreshtype, refreshregion, refreshdither)
-- tell the widget that it is shown now
widget:handleEvent(Event:new("Show"))
-- check if this widget disables double tap gesture
Input.disable_double_tap = widget.disable_double_tap ~= false
-- a widget may override tap interval (when it doesn't, nil restores the default)
Input.tap_interval_override = widget.tap_interval_override
end
--[[--
Unregisters a widget.
It will be removed from the stack.
Will flag uncovered widgets as dirty.
For more details about refreshtype, refreshregion & refreshdither see the description of `setDirty`.
If refreshtype is omitted, no extra refresh will be enqueued at this time, leaving only those from the uncovered widgets.
@param widget a @{ui.widget.widget|widget} object
@string refreshtype `"full"`, `"flashpartial"`, `"flashui"`, `"[partial]"`, `"[ui]"`, `"partial"`, `"ui"`, `"fast"`, `"a2"` (optional)
@param refreshregion a rectangle @{ui.geometry.Geom|Geom} object (optional, requires refreshtype to be set)
@bool refreshdither `true` if the refresh requires dithering (optional, requires refreshtype to be set)
@see setDirty
]]
function UIManager:close(widget, refreshtype, refreshregion, refreshdither)
if not widget then
logger.dbg("attempted to close a nil widget")
return
end
logger.dbg("close widget:", widget.name or widget.id or tostring(widget))
local dirty = false
-- First notify the closed widget to save its settings...
widget:handleEvent(Event:new("FlushSettings"))
-- ...and notify it that it ought to be gone now.
widget:handleEvent(Event:new("CloseWidget"))
-- Make sure it's disabled by default and check if there are any widgets that want it disabled or enabled.
Input.disable_double_tap = true
local requested_disable_double_tap = nil
local is_covered = false
local start_idx = 1
-- Then remove all references to that widget on stack and refresh.
for i = #self._window_stack, 1, -1 do
local w = self._window_stack[i].widget
if w == widget then
self._dirty[w] = nil
table.remove(self._window_stack, i)
dirty = true
else
if not is_covered then
-- If anything else on the stack not already hidden by (i.e., below) a fullscreen widget was dithered, honor the hint
if w.dithered then
refreshdither = true
logger.dbg("Lower widget", w.name or w.id or tostring(w), "was dithered, honoring the dithering hint")
end
-- Remember the uppermost widget that covers the full screen, so we don't bother calling setDirty on hidden (i.e., lower) widgets in the following dirty loop.
-- _repaint already does that later on to skip the actual paintTo calls, so this ensures we limit the refresh queue to stuff that will actually get painted.
if w.covers_fullscreen then
is_covered = true
start_idx = i
logger.dbg("Lower widget", w.name or w.id or tostring(w), "covers the full screen")
if i > 1 then
logger.dbg("not refreshing", i-1, "covered widget(s)")
end
end
end
-- Set double tap to how the topmost widget with that flag wants it
if requested_disable_double_tap == nil and w.disable_double_tap ~= nil then
requested_disable_double_tap = w.disable_double_tap
end
end
end
if requested_disable_double_tap ~= nil then
Input.disable_double_tap = requested_disable_double_tap
end
if self._window_stack[1] then
-- set tap interval override to what the topmost widget specifies (when it doesn't, nil restores the default)
Input.tap_interval_override = self._window_stack[#self._window_stack].widget.tap_interval_override
end
if dirty and not widget.invisible then
-- schedule the remaining visible (i.e., uncovered) widgets to be painted
for i = start_idx, #self._window_stack do
self:setDirty(self._window_stack[i].widget)
end
self:_refresh(refreshtype, refreshregion, refreshdither)
end
end
-- Schedule an execution task; task queue is in descending order
function UIManager:schedule(sched_time, action, ...)
local lo, hi = 1, #self._task_queue
-- Leftmost binary insertion
while lo <= hi do
-- NOTE: We should be (mostly) free from overflow here, thanks to LuaJIT's BitOp semantics.
-- For more fun details about this particular overflow,
-- c.f., https://ai.googleblog.com/2006/06/extra-extra-read-all-about-it-nearly.html
-- NOTE: For more fun reading about the binary search algo in general,
-- c.f., https://reprog.wordpress.com/2010/04/19/are-you-one-of-the-10-percent/
local mid = bit.rshift(lo + hi, 1)
local mid_time = self._task_queue[mid].time
if mid_time <= sched_time then
hi = mid - 1
else
lo = mid + 1
end
end
table.insert(self._task_queue, lo, {
time = sched_time,
action = action,
args = table.pack(...),
})
self._task_queue_dirty = true
end
dbg:guard(UIManager, 'schedule',
function(self, sched_time, action)
assert(sched_time >= 0, "Only positive time allowed")
assert(action ~= nil, "No action")
end)
--[[--
Schedules a task to be run a certain amount of seconds from now.
@number seconds scheduling delay in seconds (supports decimal values, 1ms resolution).
@func action reference to the task to be scheduled (may be anonymous)
@param ... optional arguments passed to action
@see unschedule
]]
function UIManager:scheduleIn(seconds, action, ...)
-- We might run significantly late inside an UI frame, so we can't use the cached value here.
-- It would also cause some bad interactions with the way nextTick & co behave.
local when = time.now() + time.s(seconds)
self:schedule(when, action, ...)
end
dbg:guard(UIManager, 'scheduleIn',
function(self, seconds, action)
assert(seconds >= 0, "Only positive seconds allowed")
end)
--[[--
Schedules a task for the next UI tick.
@func action reference to the task to be scheduled (may be anonymous)
@param ... optional arguments passed to action
@see scheduleIn
]]
function UIManager:nextTick(action, ...)
return self:scheduleIn(0, action, ...)
end
--[[--
Schedules a task to be run two UI ticks from now.
Useful to run UI callbacks ASAP without skipping repaints.
@func action reference to the task to be scheduled (may be anonymous)
@param ... optional arguments passed to action
@return A reference to the initial nextTick wrapper function,
necessary if the caller wants to unschedule action *before* it actually gets inserted in the task queue by nextTick.
@see nextTick
]]
function UIManager:tickAfterNext(action, ...)
-- We need to keep a reference to this anonymous function, as it is *NOT* quite `action` yet,
-- and the caller might want to unschedule it early...
local action_wrapper = function(...)
self:nextTick(action, ...)
end
self:nextTick(action_wrapper, ...)
return action_wrapper
end
--[[
-- NOTE: This appears to work *nearly* just as well, but does sometimes go too fast (might depend on kernel HZ & NO_HZ settings?)
function UIManager:tickAfterNext(action)
return self:scheduleIn(0.001, action)
end
--]]
function UIManager:debounce(seconds, immediate, action)
-- Ported from underscore.js
local args = nil
local previous_call_at = nil
local is_scheduled = false
local result = nil
local scheduled_action
scheduled_action = function()
local passed_from_last_call = time:now() - previous_call_at
if seconds > passed_from_last_call then
self:scheduleIn(seconds - passed_from_last_call, scheduled_action)
is_scheduled = true
else
is_scheduled = false
if not immediate then
result = action(unpack(args))
end
if not is_scheduled then
-- This check is needed because action can recursively call debounced_action_wrapper
args = nil
end
end
end
local debounced_action_wrapper = function(...)
args = table.pack(...)
previous_call_at = time:now()
if not is_scheduled then
self:scheduleIn(seconds, scheduled_action)
is_scheduled = true
if immediate then
result = action(unpack(args))
end
end
return result
end
return debounced_action_wrapper
end
--[[--
Unschedules a previously scheduled task.
In order to unschedule anonymous functions, store a reference.
@func action
@see scheduleIn
@usage
self.anonymousFunction = function() self:regularFunction() end
UIManager:scheduleIn(10.5, self.anonymousFunction)
UIManager:unschedule(self.anonymousFunction)
]]
function UIManager:unschedule(action)
local removed = false
for i = #self._task_queue, 1, -1 do
if self._task_queue[i].action == action then
table.remove(self._task_queue, i)
removed = true
end
end
return removed
end
dbg:guard(UIManager, 'unschedule',
function(self, action) assert(action ~= nil) end)
--[[--
Mark a window-level widget as dirty, enqueuing a repaint & refresh request for that widget, to be processed on the next UI tick.
The second parameter (refreshtype) can either specify a refreshtype
(optionally in combination with a refreshregion - which is suggested,
and an even more optional refreshdither flag if the content requires dithering);
or a function that returns a refreshtype, refreshregion tuple (or a refreshtype, refreshregion, refreshdither triple),
which will be called *after* painting the widget.
This is an interesting distinction, because a widget's geometry,
usually stored in a field named `dimen`, is (generally) only computed at painting time (e.g., during `paintTo`).
The TL;DR being: if you already know the region, you can pass everything by value directly,
(it'll make for slightly more readable debug logs),
but if the region will only be known after the widget has been painted, pass a function.
Note that, technically, it means that stuff passed by value will be enqueued earlier in the refresh stack.
In practice, since the stack of (both types of) refreshes is optimized into as few actual refresh ioctls as possible,
and that during the next `_repaint` tick (which is when `paintTo` for dirty widgets happens),
this shouldn't change much in the grand scheme of things, but it ought to be noted ;).
See `_repaint` for more details about how the repaint & refresh queues are processed,
and `handleInput` for more details about when those queues are actually drained.
What you should essentially remember is that `setDirty` doesn't actually "do" anything visible on its own.
It doesn't block, and when it returns, nothing new has actually been painted or refreshed.
It just appends stuff to the paint and/or refresh queues.
Here's a quick rundown of what each refreshtype should be used for:
* `full`: high-fidelity flashing refresh (e.g., large images).
Highest quality, but highest latency.
Don't abuse if you only want a flash (in this case, prefer `flashui` or `flashpartial`).
* `partial`: medium fidelity refresh (e.g., text on a white background).
Can be promoted to flashing after `FULL_REFRESH_COUNT` refreshes.
Don't abuse to avoid spurious flashes.
In practice, this means this should mostly always be limited to ReaderUI.
* `[partial]`: variant of partial that asks the driver not to merge this update with surrounding ones.
Equivalent to partial on platforms where this distinction is not implemented.
* `ui`: medium fidelity refresh (e.g., mixed content).
Should apply to most UI elements.
When in doubt, use this.
* `[ui]`: variant of ui that asks the driver not to merge this update with surrounding ones.
Equivalent to ui on platforms where this distinction is not implemented.
* `fast`: low fidelity refresh (e.g., monochrome content (technically, from any to B&W)).
Should apply to most highlighting effects achieved through inversion.
Note that if your highlighted element contains text,
you might want to keep the unhighlight refresh as `"ui"` instead, for crisper text.
(Or optimize that refresh away entirely, if you can get away with it).
* `a2`: low fidelity refresh (e.g., monochrome content (technically, from B&W to B&W only)).
Should be limited to very specific use-cases (e.g., keyboard)
* `flashui`: like `ui`, but flashing.
Can be used when showing a UI element for the first time, or when closing one, to avoid ghosting.
* `flashpartial`: like `partial`, but flashing (and not counting towards flashing promotions).
Can be used when closing an UI element (usually over ReaderUI), to avoid ghosting.
You can even drop the region in these cases, to ensure a fullscreen flash.
NOTE: On REAGL devices, `flashpartial` will NOT actually flash (by design).
As such, even onCloseWidget, you might prefer `flashui` in most instances.
NOTE: You'll notice a trend on UI elements that are usually shown *over* some kind of text (generally ReaderUI)
of using `"ui"` onShow & onUpdate, but `"partial"` onCloseWidget.
This is by design: `"partial"` is what the reader (ReaderUI) uses, as it's tailor-made for pure text
over a white background, so this ensures we resume the usual flow of the reader.
The same dynamic is true for their flashing counterparts, in the rare instances we enforce flashes.
Any kind of `"partial"` refresh *will* count towards a flashing promotion after `FULL_REFRESH_COUNT` refreshes,
so making sure your stuff only applies to the proper region is key to avoiding spurious large black flashes.
That said, depending on your use case, using `"ui"` onCloseWidget can be a perfectly valid decision,
and will ensure never seeing a flash because of that widget.
Remember that the FM uses `"ui"`, so, if said widgets are shown over the FM,
prefer using `"ui"` or `"flashui"` onCloseWidget.
The final parameter (refreshdither) is an optional hint for devices with hardware dithering support that this repaint
could benefit from dithering (e.g., because it contains an image).
As far as the actual lifecycle of a widget goes, the rules are:
* What you `show`, you `close`.
* If you know the dimensions of the widget (or simply of the region you want to refresh), you can pass it directly:
* to `show` (as `show` calls `setDirty`),
* to `close` (as `close` will also call `setDirty` on the remaining dirty and visible widgets,
and will also enqueue a refresh based on that if there are dirty widgets).
* Otherwise, you can use, respectively, a widget's `Show` & `CloseWidget` handlers for that via `setDirty` calls.
This can also be useful if *child* widgets have specific needs (e.g., flashing, dithering) that they want to inject in the refresh queue.
* Remember that events propagate children first (in array order, starting at the top), and that if *any* event handler returns true,
the propagation of that specific event for this widget tree stops *immediately*.
(This generally means that, unless you know what you're doing (e.g., a widget that will *always* be used as a parent),
you generally *don't* want to return true in `Show` or `CloseWidget` handlers).
* If any widget requires freeing non-Lua resources (e.g., FFI/C), having a `free` method called from its `CloseWidget` handler is ideal:
this'll ensure that *any* widget including it will be sure that resources are freed when it (or its parent) are closed.
* Note that there *is* a `Close` event, but it has very specific use-cases, generally involving *programmatically* `close`ing a `show`n widget:
* It is broadcast (e.g., sent to every widget in the window stack; the same rules about propagation apply, but only per *window-level widget*)
at poweroff/reboot.
* It can also be used as a keypress handler by @{ui.widget.container.inputcontainer|InputContainer}, generally bound to the Back key.
Please refrain from implementing custom `onClose` methods if that's not their intended purpose ;).
On the subject of widgets and child widgets,
you might have noticed an unspoken convention across the codebase of widgets having a field called `show_parent`.
Since handling this is entirely at the programmer's behest, here's how we usually use it:
Basically, we cascade a field named `show_parent` to every child widget that matter
(e.g., those that serve an UI purpose, as opposed to, say, a container).
This ensures that every subwidget can reference its actual parent
(ideally, all the way to the window-level widget it belongs to, i.e., the one that was passed to `show`, hence the name ;)),
to, among other things, flag the right widget for repaint via `setDirty` (c.f., those pesky debug warnings when that's done wrong ;p) when they want to request a repaint.
This is why you often see stuff doing, when instantiating a new widget, `FancyWidget:new{ show_parent = self.show_parent or self }`;
meaning, if I'm already a subwidget, cascade my parent, otherwise, it means I'm a window-level widget, so cascade myself as that widget's parent ;).
Another convention (that a few things rely on) is naming a (persistent) MovableContainer wrapping a full widget `movable`, accessible as an instance field.
This is useful when it's used for transparency purposes, which, e.g., `setDirty` and @{ui.widget.button|Button} rely on to handle updating translucent widgets properly,
by checking if self.show_parent.movable exists and is currently translucent ;).
When I mentioned passing the *right* widget to `setDirty` earlier, what I meant is that `setDirty` will only actually flag a widget for repaint
*if* that widget is a window-level widget (that is, a widget that was passed to `show` earlier and hasn't been `close`'d yet),
hence the `self.show_parent` convention detailed above to get at the proper widget from within a subwidget ;).
Otherwise, you'll notice in debug mode that a debug guard will shout at you if that contract is broken,
and what happens in practice is the same thing as if an explicit `nil` were passed: no widgets will actually be flagged for repaint,
and only the *refresh* matching the requested region *will* be enqueued.
This is why you'll find a number of valid use-cases for passing a `nil` here, when you *just* want a screen refresh without a repaint :).
The string `"all"` is also accepted in place of a widget, and will do the obvious thing: flag the *full* window stack, bottom to top, for repaint,
while still honoring the refresh region (e.g., this doesn't enforce a full-screen refresh).
@usage
UIManager:setDirty(self.widget, "partial")
UIManager:setDirty(self.widget, "partial", Geom:new{x=10,y=10,w=100,h=50})
UIManager:setDirty(self.widget, function() return "ui", self.someelement.dimen end)
@param widget a window-level widget object, `"all"`, or `nil`
@param refreshtype `"full"`, `"flashpartial"`, `"flashui"`, `"[partial]"`, `"[ui]"`, `"partial"`, `"ui"`, `"fast"`, `"a2"` (or a lambda, see description above)
@param refreshregion a rectangle @{ui.geometry.Geom|Geom} object (optional, omitting it means the region will cover the full screen)
@bool refreshdither `true` if widget requires dithering (optional)
]]
function UIManager:setDirty(widget, refreshtype, refreshregion, refreshdither)
local widget_name
if widget then
widget_name = widget.name or widget.id or tostring(widget)
if widget == "all" then
-- special case: set all top-level widgets as being "dirty".
for _, window in ipairs(self._window_stack) do
local w = window.widget
self._dirty[w] = true
-- If any of 'em were dithered, honor their dithering hint
if w.dithered then
-- NOTE: That works when refreshtype is NOT a function,
-- which is why _repaint does another pass of this check ;).
logger.dbg("setDirty on all widgets: found a dithered widget, infecting the refresh queue")
refreshdither = true
end
end
elseif not widget.invisible then
-- NOTE: If our widget is translucent, or belongs to a translucent MovableContainer,
-- we'll want to flag everything below it as dirty, too,
-- because doing transparency right requires having an up to date background against which to blend.
-- (The typecheck is because some widgets use an alpha boolean trap for internal alpha handling (e.g., ImageWidget)).
local handle_alpha = false
-- NOTE: We only ever check the dirty flag on top-level widgets, so only set it there!
-- Enable verbose debug to catch misbehaving widgets via our post-guard.
for i = #self._window_stack, 1, -1 do
local w = self._window_stack[i].widget
if handle_alpha then
self._dirty[w] = true
logger.dbg("setDirty: Marking as dirty widget:", w.name or w.id or tostring(w), "because it's below translucent widget:", widget_name)
-- Stop flagging widgets at the uppermost one that covers the full screen
if w.covers_fullscreen then
break
end
end
if w == widget then
self._dirty[widget] = true
-- We've got a match, now check if it's translucent...
handle_alpha = (widget.alpha and type(widget.alpha) == "number" and widget.alpha < 1 and widget.alpha > 0)
or (widget.movable and widget.movable.alpha and widget.movable.alpha < 1 and widget.movable.alpha > 0)
-- We shouldn't be seeing the same widget at two different spots in the stack, so, we're done,
-- except when we need to keep looping to flag widgets below us in order to handle a translucent widget...
if not handle_alpha then
break
end
end
end
-- Again, if it's flagged as dithered, honor that
if widget.dithered then
refreshdither = true
end
end
else
-- Another special case: if we did NOT specify a widget, but requested a full refresh nonetheless (i.e., a diagonal swipe),
-- we'll want to check the window stack in order to honor dithering...
if refreshtype == "full" then
for i = #self._window_stack, 1, -1 do
-- If any of 'em were dithered, honor their dithering hint
if self._window_stack[i].widget.dithered then
logger.dbg("setDirty full on no specific widget: found a dithered widget, infecting the refresh queue")
refreshdither = true
-- One is enough ;)
break
end
end
end
end
-- handle refresh information
if type(refreshtype) == "function" then
-- callback, will be issued after painting
table.insert(self._refresh_func_stack, refreshtype)
if dbg.is_on then
-- NOTE: It's too early to tell what the function will return (especially the region), because the widget hasn't been painted yet.
-- Consuming the lambda now also appears have nasty side-effects that render it useless later, subtly breaking a whole lot of things...
-- Thankfully, we can track them in _refresh()'s logging very soon after that...
logger.dbg("setDirty via a func from widget", widget_name)
end
else
-- otherwise, enqueue refresh
self:_refresh(refreshtype, refreshregion, refreshdither)
if dbg.is_on then
if refreshregion then
logger.dbg("setDirty", refreshtype, "from widget", widget_name, "w/ region", refreshregion.x, refreshregion.y, refreshregion.w, refreshregion.h, "dithering:", refreshdither)
else
logger.dbg("setDirty", refreshtype, "from widget", widget_name, "w/ NO region; dithering:", refreshdither)
end
end
end
end
--[[
-- NOTE: While nice in theory, this is *extremely* verbose in practice,
-- because most widgets will call setDirty at least once during their initialization,
-- and that happens before they make it to the window stack...
-- Plus, setDirty(nil, ...) is a completely valid use-case with documented semantics...
dbg:guard(UIManager, 'setDirty',
nil,
function(self, widget, refreshtype, refreshregion, refreshdither)
if not widget or widget == "all" then return end
-- when debugging, we check if we were handed a valid window-level widget,
-- which would be a widget that was previously passed to `show`.
local found = false
for i = 1, #self._window_stack do
if self._window_stack[i].widget == widget then
found = true
break
end
end
if not found then
dbg:v("INFO: invalid widget for setDirty()", debug.traceback())
end
end)
--]]
--[[--
Clear the full repaint & refresh queues.
NOTE: Beware! This doesn't take any prisonners!
You shouldn't have to resort to this unless in very specific circumstances!
plugins/coverbrowser.koplugin/covermenu.lua building a franken-menu out of buttondialogtitle & buttondialog
and wanting to avoid inheriting their original paint/refresh cycle being a prime example.
--]]
function UIManager:clearRenderStack()
logger.dbg("clearRenderStack: Clearing the full render stack!")
self._dirty = {}
self._refresh_func_stack = {}
self._refresh_stack = {}
end
function UIManager:insertZMQ(zeromq)
table.insert(self._zeromqs, zeromq)
return zeromq
end
function UIManager:removeZMQ(zeromq)
for i = #self._zeromqs, 1, -1 do
if self._zeromqs[i] == zeromq then
table.remove(self._zeromqs, i)
end
end
end
--[[--
Sets the full refresh rate for e-ink screens (`FULL_REFRESH_COUNT`).
This is the amount of `"partial"` refreshes before the next one gets promoted to `"full"`.
Also makes the refresh rate persistent in global reader settings.
@see setDirty
--]]
function UIManager:setRefreshRate(rate, night_rate)
logger.dbg("set screen full refresh rate", rate, night_rate)
if G_reader_settings:isTrue("night_mode") then
if night_rate then
self.FULL_REFRESH_COUNT = night_rate
end
else
if rate then
self.FULL_REFRESH_COUNT = rate
end
end
if rate then
G_reader_settings:saveSetting("full_refresh_count", rate)
end
if night_rate then
G_reader_settings:saveSetting("night_full_refresh_count", night_rate)
end
end
--- Returns the full refresh rate for e-ink screens (`FULL_REFRESH_COUNT`).
function UIManager:getRefreshRate()
return G_reader_settings:readSetting("full_refresh_count") or DEFAULT_FULL_REFRESH_COUNT, G_reader_settings:readSetting("night_full_refresh_count") or G_reader_settings:readSetting("full_refresh_count") or DEFAULT_FULL_REFRESH_COUNT
end
--- Toggles Night Mode (i.e., inverted rendering).
function UIManager:ToggleNightMode(night_mode)
if night_mode then
self.FULL_REFRESH_COUNT = G_reader_settings:readSetting("night_full_refresh_count") or G_reader_settings:readSetting("full_refresh_count") or DEFAULT_FULL_REFRESH_COUNT
else
self.FULL_REFRESH_COUNT = G_reader_settings:readSetting("full_refresh_count") or DEFAULT_FULL_REFRESH_COUNT
end
end
--- Get n.th topmost widget
function UIManager:getNthTopWidget(n)
n = n and n-1 or 0
if #self._window_stack - n < 1 then
-- No or not enough widgets in the stack, bye!
return nil
end
local widget = self._window_stack[#self._window_stack - n].widget
return widget
end
--- Top-to-bottom widgets iterator
--- NOTE: VirtualKeyboard can be instantiated multiple times, and is a modal,
-- so don't be suprised if you find a couple of instances of it at the top ;).
function UIManager:topdown_widgets_iter()
local n = #self._window_stack
local i = n + 1
return function()
i = i - 1
if i > 0 then
return self._window_stack[i].widget
end
end
end
--- Get the topmost visible widget
function UIManager:getTopmostVisibleWidget()
for i = #self._window_stack, 1, -1 do
local widget = self._window_stack[i].widget
-- Skip invisible widgets (e.g., TrapWidget)
if not widget.invisible then
return widget
end
end
end
--- Check if a widget is still in the window stack, or is a subwidget of a widget still in the window stack.
function UIManager:isSubwidgetShown(widget, max_depth)
for i = #self._window_stack, 1, -1 do
local w = self._window_stack[i].widget
local matched, depth = util.arrayReferences(w, widget, max_depth)
if matched then
return matched, depth, w
end
end
return false
end
--- Same as `isSubwidgetShown`, but only check window-level widgets (e.g., what's directly registered in the window stack), don't recurse.
function UIManager:isWidgetShown(widget)
for i = #self._window_stack, 1, -1 do
if self._window_stack[i].widget == widget then
return true
end
end
return false
end
--- Signals to quit.
-- An exit_code of false is not allowed.
function UIManager:quit(exit_code, implicit)
if exit_code == false then
logger.err("UIManager:quit() called with false")
return
end
-- Also honor older exit codes; default to 0
self._exit_code = exit_code or self._exit_code or 0
if not implicit then
-- Explicit call via UIManager:quit (as opposed to self:_gated_quit)
if exit_code then
logger.info("Preparing to quit UIManager with exit code:", exit_code)
else
logger.info("Preparing to quit UIManager")
end
end
self._task_queue_dirty = false
self._window_stack = {}
self._task_queue = {}
for i = #self._zeromqs, 1, -1 do
self._zeromqs[i]:stop()
end
self._zeromqs = {}
if self.looper then
self.looper:close()
self.looper = nil
end
return self._exit_code
end
dbg:guard(UIManager, 'quit',
function(self, exit_code)
assert(exit_code ~= false, "exit_code == false is not supported")
end)
-- Disable automatic UIManager quit; for testing purposes
function UIManager:setRunForeverMode()
self._gated_quit = function() return false end
end
-- Enable automatic UIManager quit; for testing purposes
function UIManager:unsetRunForeverMode()
self._gated_quit = function() return self:quit(nil, true) end
end
-- Ignore an empty window stack *once*; for startup w/ a missing last_file shenanigans...
function UIManager:runOnce()
-- We don't actually want to call self.quit, and we need to deal with a bit of trickery in there anyway...
self._gated_quit = function()
-- We need this set to break the loop in UIManager:run()
self._exit_code = 0
-- And this is to break the loop in UIManager:handleInput()
return true
end
-- The idea being that we want to *return* from this run call, but *without* quitting.
-- NOTE: This implies that calling run multiple times across a single session *needs* to be safe.
self:run()
-- Restore standard behavior
self:unsetRunForeverMode()
self._exit_code = nil
end
--[[--
Transmits an @{ui.event.Event|Event} to active widgets, top to bottom.
Stops at the first handler that returns `true`.
Note that most complex widgets are based on @{ui.widget.container.WidgetContainer|WidgetContainer},
which itself will take care of propagating an event to its members.
@param event an @{ui.event.Event|Event} object
]]
function UIManager:sendEvent(event)
local top_widget
local checked_widgets = {}
-- Toast widgets, which, by contract, must be at the top of the window stack, never stop event propagation.
for i = #self._window_stack, 1, -1 do
local widget = self._window_stack[i].widget
-- Whether it's a toast or not, we'll call handleEvent now,
-- so we'll want to skip it during the table walk later.
checked_widgets[widget] = true
if widget.toast then
-- We never stop event propagation on toasts, but we still want to send the event to them.
-- (In particular, because we want them to close on user input).
widget:handleEvent(event)
else
-- The first widget to consume events as designed is the topmost non-toast one
top_widget = widget
break
end
end
-- Extremely unlikely, but we can't exclude the possibility of *everything* being a toast ;).
-- In which case, the event has nowhere else to go, so, we're done.
if not top_widget then
return
end
if top_widget:handleEvent(event) then
return
end
if top_widget.active_widgets then
for _, active_widget in ipairs(top_widget.active_widgets) do
if active_widget:handleEvent(event) then
return
end
end
end
-- If the event was not consumed (no handler returned true), active widgets (from top to bottom) can access it.
-- NOTE: _window_stack can shrink/grow when widgets are closed (CloseWidget & Close events) or opened.
-- Simply looping in reverse would only cover the list shrinking, and that only by a *single* element,
-- something we can't really guarantee, hence the more dogged iterator below,
-- which relies on a hash check of already processed widgets (LuaJIT actually hashes the table's GC reference),
-- rather than a simple loop counter, and will in fact iterate *at least* #items ^ 2 times.
-- Thankfully, that list should be very small, so the overhead should be minimal.
local i = #self._window_stack
while i > 0 do
local widget = self._window_stack[i].widget
if not checked_widgets[widget] then
checked_widgets[widget] = true
-- Widget's active widgets have precedence to handle this event
-- NOTE: ReaderUI & FileManager *may* optionally register their modules as such
-- (currently, they only do that for the Screenshot module).
if widget.active_widgets then
for _, active_widget in ipairs(widget.active_widgets) do
if active_widget:handleEvent(event) then
return
end
end
end
if widget.is_always_active then
-- Widget itself is flagged always active, let it handle the event
-- NOTE: is_always_active widgets are currently widgets that want to show a VirtualKeyboard or listen to Dispatcher events
if widget:handleEvent(event) then
return
end
end
-- As mentioned above, event handlers might have shown/closed widgets,
-- so all bets are off on our old window tally being accurate, so let's take it from the top again ;).
i = #self._window_stack
else
i = i - 1
end
end
end
--[[--
Transmits an @{ui.event.Event|Event} to all registered widgets.
@param event an @{ui.event.Event|Event} object
]]
function UIManager:broadcastEvent(event)
-- Unlike sendEvent, we send the event to *all* (window-level) widgets (i.e., we don't stop, even if a handler returns true).
-- NOTE: Same defensive approach to _window_stack changing from under our feet as above.
local checked_widgets = {}
local i = #self._window_stack
while i > 0 do
local widget = self._window_stack[i].widget
if not checked_widgets[widget] then
checked_widgets[widget] = true
widget:handleEvent(event)
i = #self._window_stack
else
i = i - 1
end
end
end
--[[
function UIManager:getNextTaskTimes(count)
count = math.min(count or 1, #self._task_queue)
local times = {}
for i = 1, count do
times[i] = self._task_queue[i].time - time.now()
end
return times
end
--]]
function UIManager:getNextTaskTime()
local next_task = self._task_queue[#self._task_queue]
if next_task then
return next_task.time - time:now()
end
end
function UIManager:_checkTasks()
self._now = time.now()
local wait_until = nil
-- Tasks due for execution might themselves schedule more tasks (that might also be immediately due for execution ;)).
-- Flipping this switch ensures we'll consume all such tasks *before* yielding to input polling.
self._task_queue_dirty = false
while self._task_queue[1] do
local task_time = self._task_queue[#self._task_queue].time
if task_time <= self._now then
-- Remove the upcoming task, as it is due for execution...
local task = table.remove(self._task_queue)
-- ...so do it now.
-- NOTE: Said task's action might modify _task_queue.
-- To avoid race conditions and catch new upcoming tasks during this call,
-- we repeatedly check the head of the queue (c.f., #1758).
task.action(unpack(task.args))
else
-- As the queue is sorted in descending order, it's safe to assume all items are currently future tasks.
wait_until = task_time
break
end
end
return wait_until, self._now
end
--[[--
Returns a time (fts) corresponding to the last tick.
This is essentially a cached time.now(), computed at the top of every iteration of the main UI loop,
(right before checking/running scheduled tasks).
This is mainly useful to compute/schedule stuff in the same time scale as the UI loop (i.e., MONOTONIC),
without having to resort to a syscall.
It should never be significantly stale, assuming the UI is in use (e.g., there are input events),
unless you're blocking the UI for a significant amount of time in a single UI frame.
That is to say, its granularity is an UI frame.
Prefer the appropriate time function for your needs if you require perfect accuracy or better granularity
(e.g., when you're actually working on the event loop *itself* (UIManager, Input, GestureDetector),
or if you're dealing with intra-frame timers).
This is *NOT* wall clock time (REALTIME).
]]
function UIManager:getTime()
return self._now
end
--[[--
Returns a time (fts) corresponding to the last UI tick plus the time in standby.
]]
function UIManager:getElapsedTimeSinceBoot()
return self:getTime() + Device.total_standby_time + Device.total_suspend_time
end
-- precedence of refresh modes:
local refresh_modes = { a2 = 1, fast = 2, ui = 3, partial = 4, ["[ui]"] = 5, ["[partial]"] = 6, flashui = 7, flashpartial = 8, full = 9 }
-- NOTE: We might want to introduce a "force_a2" that points to fast, but has the highest priority,
-- for the few cases where we might *really* want to enforce fast (for stuff like panning or skimming?).
-- refresh methods in framebuffer implementation
local refresh_methods = {
a2 = Screen.refreshA2,
fast = Screen.refreshFast,
ui = Screen.refreshUI,
partial = Screen.refreshPartial,
["[ui]"] = Screen.refreshNoMergeUI,
["[partial]"] = Screen.refreshNoMergePartial,
flashui = Screen.refreshFlashUI,
flashpartial = Screen.refreshFlashPartial,
full = Screen.refreshFull,
}
--[[
Compares refresh mode.
Will return the mode that takes precedence.
]]
local function update_mode(mode1, mode2)
if refresh_modes[mode1] > refresh_modes[mode2] then
logger.dbg("update_mode: Update refresh mode", mode2, "to", mode1)
return mode1
else
return mode2
end
end
--[[
Compares dither hints.
Dither always wins.
]]
local function update_dither(dither1, dither2)
if dither1 and not dither2 then
logger.dbg("update_dither: Update dither hint", dither2, "to", dither1)
return dither1
else
return dither2
end
end
--[[--
Enqueues a refresh.
Widgets call this in their `paintTo()` method in order to notify
UIManager that a certain part of the screen is to be refreshed.
@string mode
refresh mode (`"full"`, `"flashpartial"`, `"flashui"`, `"[partial]"`, `"[ui]"`, `"partial"`, `"ui"`, `"fast"`, `"a2"`)
@param region
A rectangle @{ui.geometry.Geom|Geom} object that specifies the region to be updated.
Optional, update will affect whole screen if not specified.
Note that this should be the exception.
@bool dither
A hint to request hardware dithering (if supported).
Optional, no dithering requested if not specified or not supported.
@local Not to be used outside of UIManager!
]]
function UIManager:_refresh(mode, region, dither)
if not mode then
-- If we're trying to float a dither hint up from a lower widget after a close, mode might be nil...
-- So use the lowest priority refresh mode (short of fast, because that'd do half-toning).
if dither then
mode = "ui"
else
-- Otherwise, this is most likely from a `show` or `close` that wasn't passed specific refresh details,
-- (which is the vast majority of them), in which case we drop it to avoid enqueuing a useless full-screen refresh.
return
end
end
-- Downgrade all refreshes to "fast" when ReaderPaging or ReaderScrolling have set this flag
if self.currently_scrolling then
mode = "fast"
end
if not region and mode == "full" then
self.refresh_count = 0 -- reset counter on explicit full refresh
end
-- Handle downgrading flashing modes to non-flashing modes, according to user settings.
-- NOTE: Do it before "full" promotion and collision checks/update_mode.
if G_reader_settings:isTrue("avoid_flashing_ui") then
if mode == "flashui" then
mode = "ui"
logger.dbg("_refresh: downgraded flashui refresh to", mode)
elseif mode == "flashpartial" then
mode = "partial"
logger.dbg("_refresh: downgraded flashpartial refresh to", mode)
elseif mode == "partial" and region then
mode = "ui"
logger.dbg("_refresh: downgraded regional partial refresh to", mode)
end
end
-- special case: "partial" refreshes
-- will get promoted every self.FULL_REFRESH_COUNT refreshes
-- since _refresh can be called mutiple times via setDirty called in
-- different widgets before a real screen repaint, we should make sure
-- refresh_count is incremented by only once at most for each repaint
-- NOTE: Ideally, we'd only check for "partial"" w/ no region set (that neatly narrows it down to just the reader).
-- In practice, we also want to promote refreshes in a few other places, except purely text-poor UI elements.
-- (Putting "ui" in that list is problematic with a number of UI elements, most notably, ReaderHighlight,
-- because it is implemented as "ui" over the full viewport, since we can't devise a proper bounding box).
-- So we settle for only "partial", but treating full-screen ones slightly differently.
if mode == "partial" and not self.refresh_counted then
self.refresh_count = (self.refresh_count + 1) % self.FULL_REFRESH_COUNT
if self.refresh_count == self.FULL_REFRESH_COUNT - 1 then
-- NOTE: Promote to "full" if no region (reader), to "flashui" otherwise (UI)
if region then
mode = "flashui"
else
mode = "full"
end
logger.dbg("_refresh: promote refresh to", mode)
end
self.refresh_counted = true
end
-- if no region is specified, use the screen's dimensions
region = region or Geom:new{w=Screen:getWidth(), h=Screen:getHeight()}
-- if no dithering hint was specified, don't request dithering
dither = dither or false
-- NOTE: While, ideally, we shouldn't merge refreshes w/ different waveform modes,
-- this allows us to optimize away a number of quirks of our rendering stack
-- (e.g., multiple setDirty calls queued when showing/closing a widget because of update mechanisms),
-- as well as a few actually effective merges
-- (e.g., the disappearance of a selection HL with the following menu update).
for i, refresh in ipairs(self._refresh_stack) do
-- Check for collision with refreshes that are already enqueued
-- NOTE: intersect *means* intersect: we won't merge edge-to-edge regions (but the EPDC probably will).
if region:intersectWith(refresh.region) then
-- combine both refreshes' regions
local combined = region:combine(refresh.region)
-- update the mode, if needed
mode = update_mode(mode, refresh.mode)
-- dithering hints are viral, one is enough to infect the whole queue
dither = update_dither(dither, refresh.dither)
-- remove colliding refresh
table.remove(self._refresh_stack, i)
-- and try again with combined data
return self:_refresh(mode, combined, dither)
end
end
-- if we've stopped hitting collisions, enqueue the refresh
logger.dbg("_refresh: Enqueued", mode, "update for region", region.x, region.y, region.w, region.h, "dithering:", dither)
table.insert(self._refresh_stack, {mode = mode, region = region, dither = dither})
end
--[[--
Repaints dirty widgets.
This will also drain the refresh queue, effectively refreshing the screen region(s) matching those freshly repainted widgets.
There may be refreshes enqueued without any widgets needing to be repainted (c.f., `setDirty`'s behavior when passed a `nil` widget),
in which case, nothing is repainted, but the refreshes are still drained and executed.
@local Not to be used outside of UIManager!
--]]
function UIManager:_repaint()
-- flag in which we will record if we did any repaints at all
-- will trigger a refresh if set.
local dirty = false
-- remember if any of our repaints were dithered
local dithered = false
-- We don't need to call paintTo() on widgets that are under
-- a widget that covers the full screen
local start_idx = 1
for i = #self._window_stack, 1, -1 do
if self._window_stack[i].widget.covers_fullscreen then
start_idx = i
if i > 1 then
logger.dbg("not painting", i-1, "covered widget(s)")
end
break
end
end
-- Show IDs of covered widgets when debugging
--[[
if start_idx > 1 then
for i = 1, start_idx-1 do
local widget = self._window_stack[i].widget
logger.dbg("NOT painting widget:", widget.name or widget.id or tostring(widget))
end
end
--]]
for i = start_idx, #self._window_stack do
local window = self._window_stack[i]
local widget = window.widget
-- paint if current widget or any widget underneath is dirty
if dirty or self._dirty[widget] then
-- pass hint to widget that we got when setting widget dirty
-- the widget can use this to decide which parts should be refreshed
logger.dbg("painting widget:", widget.name or widget.id or tostring(widget))
Screen:beforePaint()
-- NOTE: Nothing actually seems to use the final argument?
-- Could be used by widgets to know whether they're being repainted because they're actually dirty (it's true),
-- or because something below them was (it's nil).
widget:paintTo(Screen.bb, window.x, window.y, self._dirty[widget])
-- and remove from list after painting
self._dirty[widget] = nil
-- trigger a repaint for every widget above us, too
dirty = true
-- if any of 'em were dithered, we'll want to dither the final refresh
if widget.dithered then
logger.dbg("_repaint: it was dithered, infecting the refresh queue")
dithered = true
end
end
end
-- execute pending refresh functions
for _, refreshfunc in ipairs(self._refresh_func_stack) do
local refreshtype, region, dither = refreshfunc()
-- honor dithering hints from *anywhere* in the dirty stack
dither = update_dither(dither, dithered)
if refreshtype then
self:_refresh(refreshtype, region, dither)
end
end
self._refresh_func_stack = {}
-- we should have at least one refresh if we did repaint. If we don't, we
-- add one now and log a warning if we are debugging
if dirty and not self._refresh_stack[1] then
logger.dbg("no refresh got enqueued. Will do a partial full screen refresh, which might be inefficient")
self:_refresh("partial")
end
-- execute refreshes:
for _, refresh in ipairs(self._refresh_stack) do
-- Honor dithering hints from *anywhere* in the dirty stack
refresh.dither = update_dither(refresh.dither, dithered)
-- If HW dithering is disabled, unconditionally drop the dither flag
if not Screen.hw_dithering then
refresh.dither = nil
end
dbg:v("triggering refresh", refresh)
--[[
-- Remember the refresh region
self._last_refresh_region = refresh.region:copy()
--]]
refresh_methods[refresh.mode](Screen,
refresh.region.x, refresh.region.y,
refresh.region.w, refresh.region.h,
refresh.dither)
end
-- Don't trigger afterPaint if we did not, in fact, paint anything
if dirty then
Screen:afterPaint()
end
self._refresh_stack = {}
self.refresh_counted = false
end
--- Explicitly drain the paint & refresh queues *now*, instead of waiting for the next UI tick.
function UIManager:forceRePaint()
self:_repaint()
end
--[[--
Ask the EPDC to *block* until our previous refresh ioctl has completed.
This interacts sanely with the existing low-level handling of this in `framebuffer_mxcfb`
(i.e., it doesn't even try to wait for a marker that fb has already waited for, and vice-versa).
Will return immediately if it has already completed.
If the device isn't a Linux + MXCFB device, this is a NOP.
]]
function UIManager:waitForVSync()
Screen:refreshWaitForLast()
end
--[[--
Yield to the EPDC.
This is a dumb workaround for potential races with the EPDC when we request a refresh on a specific region,
and then proceed to *write* to the framebuffer, in the same region, very, very, very soon after that.
This basically just puts ourselves to sleep for a very short amount of time, to let the kernel do its thing in peace.
@int sleep_us Amount of time to sleep for (in µs). (Optional, defaults to 1ms).
]]
function UIManager:yieldToEPDC(sleep_us)
if Device:hasEinkScreen() then
-- NOTE: Early empiric evidence suggests that going as low as 1ms is enough to do the trick.
-- Consider jumping to the jiffy resolution (100Hz/10ms) if it turns out it isn't ;).
ffiUtil.usleep(sleep_us or 1000)
end
end
--[[--
Used to repaint a specific sub-widget that isn't on the `_window_stack` itself.
Useful to avoid repainting a complex widget when we just want to invert an icon, for instance.
No safety checks on x & y *by design*. I want this to blow up if used wrong.
This is an explicit repaint *now*: it bypasses and ignores the paint queue (unlike `setDirty`).
@param widget a @{ui.widget.widget|widget} object
@int x left origin of widget (in the Screen buffer, e.g., `widget.dimen.x`)
@int y top origin of widget (in the Screen buffer, e.g., `widget.dimen.y`)
]]
function UIManager:widgetRepaint(widget, x, y)
if not widget then return end
logger.dbg("Explicit widgetRepaint:", widget.name or widget.id or tostring(widget), "@", x, y)
if widget.show_parent and widget.show_parent.cropping_widget then
-- The main widget parent of this subwidget has a cropping container: see if
-- this widget is a child of this cropping container
local cropping_widget = widget.show_parent.cropping_widget
if util.arrayReferences(cropping_widget, widget) then
-- Delegate the painting of this subwidget to its cropping widget container
cropping_widget:paintTo(Screen.bb, cropping_widget.dimen.x, cropping_widget.dimen.y)
return
end
end
widget:paintTo(Screen.bb, x, y)
end
--[[--
Same idea as `widgetRepaint`, but does a simple `bb:invertRect` on the Screen buffer, without actually going through the widget's `paintTo` method.
@param widget a @{ui.widget.widget|widget} object
@int x left origin of the rectangle to invert (in the Screen buffer, e.g., `widget.dimen.x`)
@int y top origin of the rectangle (in the Screen buffer, e.g., `widget.dimen.y`)
@int w width of the rectangle (optional, will use `widget.dimen.w` like `paintTo` would if omitted)
@int h height of the rectangle (optional, will use `widget.dimen.h` like `paintTo` would if omitted)
@see widgetRepaint
--]]
function UIManager:widgetInvert(widget, x, y, w, h)
if not widget then return end
logger.dbg("Explicit widgetInvert:", widget.name or widget.id or tostring(widget), "@", x, y)
if widget.show_parent and widget.show_parent.cropping_widget then
-- The main widget parent of this subwidget has a cropping container: see if
-- this widget is a child of this cropping container
local cropping_widget = widget.show_parent.cropping_widget
if util.arrayReferences(cropping_widget, widget) then
-- Invert only what intersects with the cropping container
local widget_region = Geom:new{x=x, y=y, w=w or widget.dimen.w, h=h or widget.dimen.h}
local crop_region = cropping_widget:getCropRegion()
local invert_region = crop_region:intersect(widget_region)
Screen.bb:invertRect(invert_region.x, invert_region.y, invert_region.w, invert_region.h)
return
end
end
Screen.bb:invertRect(x, y, w or widget.dimen.w, h or widget.dimen.h)
end
function UIManager:setInputTimeout(timeout)
self.INPUT_TIMEOUT = timeout or (200*1000)
end
function UIManager:resetInputTimeout()
self.INPUT_TIMEOUT = nil
end
-- NOTE: The Event hook mechanism used to dispatch for *every* event, and would actually pass the event along.
-- We've simplified that to once per input frame, and without passing anything (as we, in fact, have never made use of it).
function UIManager:handleInputEvent(input_event)
local handler = self.event_handlers[input_event]
if handler then
handler(input_event)
else
self.event_handlers.__default__(input_event)
end
end
-- Process all pending events on all registered ZMQs.
function UIManager:processZMQs()
if self._zeromqs[1] then
self.event_hook:execute("InputEvent")
end
for _, zeromq in ipairs(self._zeromqs) do
for input_event in zeromq.waitEvent, zeromq do
self:handleInputEvent(input_event)
end
end
end
function UIManager:handleInput()
local wait_until, now
-- run this in a loop, so that paints can trigger events
-- that will be honored when calculating the time to wait
-- for input events:
repeat
wait_until, now = self:_checkTasks()
--[[
dbg("---------------------------------------------------")
dbg("wait_until", wait_until)
dbg("now ", now)
dbg("#exec stack ", #self._task_queue)
dbg("#window stack", #self._window_stack)
dbg("#dirty stack ", util.tableSize(self._dirty))
dbg("dirty?", self._task_queue_dirty)
dbg("---------------------------------------------------")
--]]
-- stop when we have no window to show
if not self._window_stack[1] then
logger.info("no dialog left to show")
if self:_gated_quit() ~= false then
return nil
end
end
self:_repaint()
until not self._task_queue_dirty
-- NOTE: Compute deadline *before* processing ZMQs, in order to be able to catch tasks scheduled *during*
-- the final ZMQ callback.
-- This ensures that we get to honor a single ZMQ_TIMEOUT *after* the final ZMQ callback,
-- which gives us a chance for another iteration, meaning going through _checkTasks to catch said scheduled tasks.
-- Figure out how long to wait.
-- Ultimately, that'll be the earliest of INPUT_TIMEOUT, ZMQ_TIMEOUT or the next earliest scheduled task.
local deadline
-- Default to INPUT_TIMEOUT (which may be nil, i.e. block until an event happens).
local wait_us = self.INPUT_TIMEOUT
-- If we have any ZMQs registered, ZMQ_TIMEOUT is another upper bound.
if self._zeromqs[1] then
wait_us = math.min(wait_us or math.huge, self.ZMQ_TIMEOUT)
end
-- We pass that on as an absolute deadline, not a relative wait time.
if wait_us then
deadline = now + time.us(wait_us)
end
-- If there's a scheduled task pending, that puts an upper bound on how long to wait.
if wait_until and (not deadline or wait_until < deadline) then
-- ^ We don't have a TIMEOUT induced deadline, making the choice easy.
-- ^ We have a task scheduled for *before* our TIMEOUT induced deadline.
deadline = wait_until
end
-- Run ZMQs if any
self:processZMQs()
-- If allowed, entering standby (from which we can wake by input) must trigger in response to event
-- this function emits (plugin), or within waitEvent() right after (hardware).
-- Anywhere else breaks preventStandby/allowStandby invariants used by background jobs while UI is left running.
self:_standbyTransition()
if self._pm_consume_input_early then
-- If the PM state transition requires an early return from input polling, honor that.
-- c.f., UIManager:setPMInputTimeout (and AutoSuspend:AllowStandbyHandler).
deadline = now
self._pm_consume_input_early = false
end
-- wait for next batch of events
local input_events = Input:waitEvent(now, deadline)
-- delegate each input event to handler
if input_events then
-- Dispatch event hooks first, as some plugins (*cough* AutoSuspend *cough*)
-- rely on it to react properly to the actual event...
if input_events[1] then
self.event_hook:execute("InputEvent")
end
-- Handle the full batch of events
for __, ev in ipairs(input_events) do
self:handleInputEvent(ev)
end
end
if self.looper then
logger.info("handle input in turbo I/O looper")
self.looper:add_callback(function()
--- @fixme Force close looper when there is unhandled error,
-- otherwise the looper will hang. Any better solution?
xpcall(function() self:handleInput() end, function(err)
io.stderr:write(err .. "\n")
io.stderr:write(debug.traceback() .. "\n")
self.looper:close()
os.exit(1, true)
end)
end)
end
end
function UIManager:onRotation()
self:setDirty("all", "full")
self:forceRePaint()
end
function UIManager:initLooper()
if G_defaults:readSetting("DUSE_TURBO_LIB") and not self.looper then
TURBO_SSL = true -- luacheck: ignore
__TURBO_USE_LUASOCKET__ = true -- luacheck: ignore
local turbo = require("turbo")
self.looper = turbo.ioloop.instance()
end
end
--[[--
This is the main loop of the UI controller.
It is intended to manage input events and delegate them to dialogs.
--]]
function UIManager:run()
-- Tell PowerD that we're ready
Device:getPowerDevice():readyUI()
self:initLooper()
-- currently there is no Turbo support for Windows
-- use our own main loop
if not self.looper then
repeat
self:handleInput()
until self._exit_code
else
self.looper:add_callback(function() self:handleInput() end)
self.looper:start()
end
logger.info("Tearing down UIManager with exit code:", self._exit_code)
return self._exit_code
end
--[[--
Executes all the operations of a suspension (i.e., sleep) request.
This function usually puts the device into suspension.
]]
function UIManager:suspend()
-- Should always exist, as defined in `generic/device` or overwritten with `setEventHandlers`
if self.event_handlers.Suspend then
-- Give the other event handlers a chance to be executed.
-- `Suspend` and `Resume` events will be sent by the handler
UIManager:nextTick(self.event_handlers.Suspend)
end
end
function UIManager:askForReboot(message_text)
-- Should always exist, as defined in `generic/device` or overwritten with `setEventHandlers`
if self.event_handlers.Reboot then
-- Give the other event handlers a chance to be executed.
-- 'Reboot' event will be sent by the handler
UIManager:nextTick(self.event_handlers.Reboot, message_text)
end
end
function UIManager:askForPowerOff(message_text)
-- Should always exist, as defined in `generic/device` or overwritten with `setEventHandlers`
if self.event_handlers.PowerOff then
-- Give the other event handlers a chance to be executed.
-- 'PowerOff' event will be sent by the handler
UIManager:nextTick(self.event_handlers.PowerOff, message_text)
end
end
function UIManager:askForRestart(message_text)
-- Should always exist, as defined in `generic/device` or overwritten with `setEventHandlers`
if self.event_handlers.PowerOff then
-- Give the other event handlers a chance to be executed.
-- 'Restart' event will be sent by the handler
UIManager:nextTick(self.event_handlers.Restart, message_text)
end
end
--[[--
Release standby lock.
Called once we're done with whatever we were doing in the background.
Standby is re-enabled only after all issued prevents are paired with allowStandby for each one.
]]
function UIManager:allowStandby()
assert(self._prevent_standby_count > 0, "allowing standby that isn't prevented; you have an allow/prevent mismatch somewhere")
self._prevent_standby_count = self._prevent_standby_count - 1
logger.dbg("UIManager:allowStandby, counter decreased to", self._prevent_standby_count)
end
--[[--
Prevent standby.
i.e., something is happening in background, yet UI may tick.
]]
function UIManager:preventStandby()
self._prevent_standby_count = self._prevent_standby_count + 1
logger.dbg("UIManager:preventStandby, counter increased to", self._prevent_standby_count)
end
-- The allow/prevent calls above can interminently allow standbys, but we're not interested until
-- the state change crosses UI tick boundary, which is what self._prev_prevent_standby_count is tracking.
function UIManager:_standbyTransition()
if self._prevent_standby_count == 0 and self._prev_prevent_standby_count > 0 then
-- edge prevent->allow
logger.dbg("UIManager:_standbyTransition -> AllowStandby")
Device:setAutoStandby(true)
self:broadcastEvent(Event:new("AllowStandby"))
elseif self._prevent_standby_count > 0 and self._prev_prevent_standby_count == 0 then
-- edge allow->prevent
logger.dbg("UIManager:_standbyTransition -> PreventStandby")
Device:setAutoStandby(false)
self:broadcastEvent(Event:new("PreventStandby"))
end
self._prev_prevent_standby_count = self._prevent_standby_count
end
-- Used by a PM transition event handler to request an early return from input polling.
-- NOTE: We can't re-use setInputTimeout to avoid interactions with ZMQ...
function UIManager:consumeInputEarlyAfterPM(toggle)
self._pm_consume_input_early = toggle
end
--- Broadcasts a `FlushSettings` Event to *all* widgets.
function UIManager:flushSettings()
self:broadcastEvent(Event:new("FlushSettings"))
end
--- Sanely restart KOReader (on supported platforms).
function UIManager:restartKOReader()
-- This is just a magic number to indicate the restart request for shell scripts.
self:quit(85)
end
--- Sanely abort KOReader (e.g., exit sanely, but with a non-zero return code).
function UIManager:abort()
self:quit(1)
end
UIManager:init()
return UIManager