AwesomeFiles/config/awesome/modules/awedock/rubato/timed.lua
2022-07-11 20:06:55 -05:00

329 lines
10 KiB
Lua
Executable file

local easing = require(RUBATO_DIR.."easing")
local subscribable = require(RUBATO_DIR.."subscribable")
local gears = require "gears"
--- Get the slope (this took me forever to find).
-- i is intro duration
-- o is outro duration
-- t is total duration
-- d is distance to cover
-- F_1 is the value of the antiderivate at 1: F_1(1)
-- F_2 is the value of the outro antiderivative at 1: F_2(1)
-- b is the y-intercept
-- m is the slope
-- @see timed
local function get_slope(i, o, t, d, F_1, F_2, b)
return (d + i * b * (F_1 - 1)) / (i * (F_1 - 1) + o * (F_2 - 1) + t)
end
--- Get the dx based off of a bunch of factors
-- @see timed
local function get_dx(time, duration, intro, intro_e, outro, outro_e, m, b)
-- Intro math. Scales by difference between initial slope and target slope
if time <= intro then
return intro_e(time / intro) * (m - b) + b
-- Outro math
elseif (duration - time) <= outro then
return outro_e((duration - time) / outro) * m
-- Otherwise (it's in the plateau)
else return m end
end
--weak table for memoizing results
local simulate_easing_mem = {}
setmetatable(simulate_easing_mem, {__mode="kv"})
--- Simulates the easing to get the result to find an error coefficient
-- Uses the coefficient to adjust dx so that it's guaranteed to hit the target
-- This must be called when the sign of the target slope is changing
-- @see timed
local function simulate_easing(pos, duration, intro, intro_e, outro, outro_e, m, b, dt)
local ps_time = 0
local ps_pos = pos
local dx
-- Key for cacheing results
local key = string.format("%f %f %f %s %f %s %f %f",
pos, duration,
intro, tostring(intro_e),
outro, tostring(outro_e),
m, b)
-- Short circuits if it's already done the calculation
if simulate_easing_mem[key] then return simulate_easing_mem[key] end
-- Effectively runs the exact same code to find what the target will be
while duration - ps_time >= dt / 2 do
--increment time
ps_time = ps_time + dt
--get dx, but use the pseudotime as to not mess with stuff
dx = get_dx(ps_time, duration,
intro, intro_e,
outro, outro_e,
m, b)
--increment pos by dx
ps_pos = ps_pos + dx * dt
end
simulate_easing_mem[key] = ps_pos
return ps_pos
end
--- INTERPOLATE. bam. it still ends in a period. But this one is timed.
-- So documentation isn't super necessary here since it's all on the README and idk how to do
-- documentation correctly, so please see the README or read the code to better understand how
-- it works
local function timed(args)
local obj = subscribable()
--set up default arguments
obj.duration = args.duration or 1
obj.pos = args.pos or 0
obj.prop_intro = args.prop_intro or false
obj.intro = args.intro or 0.2
obj.inter = args.inter or args.intro
--set args.outro nicely based off how large args.intro is
if obj.intro > (obj.prop_intro and 0.5 or obj.duration) and not args.outro then
obj.outro = math.max((args.prop_intro and 1 or args.duration - args.intro), 0)
elseif not args.outro then obj.outro = obj.intro
else obj.outro = args.outro end
--assert that these values are valid
assert(obj.intro + obj.outro <= obj.duration or obj.prop_intro, "Intro and Outro must be less than or equal to total duration")
assert(obj.intro + obj.outro <= 1 or not obj.prop_intro, "Proportional Intro and Outro must be less than or equal to 1")
obj.easing = args.easing or easing.linear
obj.easing_outro = args.easing_outro or obj.easing
obj.easing_inter = args.easing_inter or obj.easing
--dev interface changes
obj.log = args.log or function() end
obj.awestore_compat = args.awestore_compat or false
--animation logic changes
obj.override_simulate = args.override_simulate or false
--[[ rapid_set is allowed by awestore but I don't like it, so it's bound to awestore_compat if not explicitly set
override_dt doesn't work well with big animations or scratchpads (blame awesome not me) (probably) so that too is
is tied to awestore_compat if not explicitly set, then to the default value ]]
obj.rapid_set = args.rapid_set == nil and obj.awestore_compat or args.rapid_set
obj.override_dt = args.override_dt == nil and (not obj.awestore_compat and RUBATO_OVERRIDE_DT) or args.override_dt
-- hidden properties
obj._props = {
target = obj.pos,
rate = args.rate or RUBATO_DEF_RATE
}
-- awestore compatibility
if obj.awestore_compat then
obj._initial = obj.pos
obj._last = 0
function obj:initial() return obj._initial end
function obj:last() return obj._last end
obj.started = subscribable()
obj.ended = subscribable()
end
-- Variables used in calculation, defined once bcz less operations
local time = 0 -- current time
local dt = 1 / obj._props.rate -- change in time
local dx = 0 -- value of slope at current time
local m -- slope
local b -- y-intercept
local is_inter --whether or not it's in an intermittent state
local last_frame_time --the time at the last frame
local frame_time = dt --duration of the last frame (placeholder)
-- Variables used in simulation
local ps_pos -- pseudoposition
local coef -- corrective coefficient TODO: apply to plateau
-- The timer that does all the animating
local timer = gears.timer { timeout = dt }
timer:connect_signal("timeout", function()
-- Find the correct dt if it's not already correct
if (obj.override_dt and last_frame_time) then
frame_time = os.clock() - last_frame_time
--[[ if frame time is bigger than dt, we must readjust dt by the difference
between dt and the last frame. Basically, dt + (frame_time - dt) which just
results in frame_time ]]
if (frame_time > timer.timeout) then dt = frame_time
else dt = timer.timeout end
end
-- for the next timeout event
if (obj.override_dt) then last_frame_time = os.clock() end
--increment time
time = time + dt
--get dx
dx = get_dx(time, obj.duration,
(is_inter and obj.inter or obj.intro) * (obj.prop_intro and obj.duration or 1),
is_inter and obj.easing_inter.easing or obj.easing.easing,
obj.outro * (obj.prop_intro and obj.duration or 1),
obj.easing_outro.easing,
m, b)
--increment pos by dx
--scale by dt and correct with coef if necessary
obj.pos = obj.pos + dx * dt * coef
--sets up when to stop by time
--weirdness is to try to get as close to duration as possible
if obj.duration - time < dt / 2 then
obj.pos = obj._props.target --snaps to target in case of small error
time = obj.duration --snaps time to duration
is_inter = false --resets intermittent
timer:stop() --stops itself
--run subscribed in functions
obj:fire(obj.pos, obj.duration, dx)
-- awestore compatibility...
if obj.awestore_compat then obj.ended:fire(obj.pos, obj.duration, dx) end
--otherwise it just fires normally
else obj:fire(obj.pos, time, dx) end
end)
-- Set target and begin interpolation
local function set(target_new)
--disallow setting it twice (because it makes it go wonky sometimes)
if not obj.rapid_set and obj._props.target == target_new then return end
--animation values
obj._props.target = target_new --sets target
time = 0 --resets time
coef = 1 --resets coefficient
--rate stuff
--both of these values would ideally be timer.timeout so that's their default
dt, frame_time = timer.timeout, timer.timeout
last_frame_time = nil --is given a value after the first frame
-- does annoying awestore compatibility
if obj.awestore_compat then
obj._last = obj._props.target
obj.started:fire(obj.pos, time, dx)
end
-- if it's already started, reflect that in is_inter
is_inter = timer.started
--set initial position if interrupting another animation
b = timer.started and dx or 0
--get the slope of the plateau
m = get_slope((is_inter and obj.inter or obj.intro) * (obj.prop_intro and obj.duration or 1),
obj.outro * (obj.prop_intro and obj.duration or 1),
obj.duration,
obj._props.target - obj.pos,
is_inter and obj.easing_inter.F or obj.easing.F,
obj.easing_outro.F,
b)
--if it will make a mistake (or override_simulate is true), fix it
--it should only make a mistake when switching direction
--b ~= zero protection so that I won't get any NaNs (because NaN ~= NaN)
if obj.override_simulate or (b ~= 0 and b / math.abs(b) ~= m / math.abs(m)) then
ps_pos = simulate_easing(obj.pos, obj.duration,
(is_inter and obj.inter or obj.intro) * (obj.prop_intro and obj.duration or 1),
is_inter and obj.easing_inter.easing or obj.easing.easing,
obj.outro * (obj.prop_intro and obj.duration or 1),
obj.easing_outro.easing,
m, b, dt)
--get coefficient by calculating ratio of theoretical range : experimental range
coef = (obj.pos - obj._props.target) / (obj.pos - ps_pos)
if coef ~= coef then coef = 1 end --check for div by 0 resulting in NaN
end
if not timer.started then timer:start() end
end
if obj.awestore_compat then function obj:set(target) set(target) end end
-- Functions for setting state
-- Completely resets the timer
function obj:reset()
timer:stop()
time = 0
obj._props.target = obj.pos
dx = 0
m = nil
b = nil
is_inter = false
coef = 1
dt = timer.timeout
end
-- Effectively pauses the timer
function obj:abort()
timer:stop()
is_inter = false
end
--subscribe stuff initially and add callback
obj.subscribe_callback = function(func) func(obj.pos, time, dt) end
if args.subscribed ~= nil then obj:subscribe(args.subscribed) end
-- Metatable for cooler api
local mt = {}
function mt:__index(key)
-- Returns the state value
if key == "state" then return timer.started
-- If it's in _props return it from props
elseif self._props[key] then return self._props[key]
-- Otherwise just be nice
else return rawget(self, key) end
end
function mt:__newindex(key, value)
-- Don't allow for setting state
if key == "state" then return
-- Changing target should call set
elseif key == "target" then set(value) --set target
-- Changing rate should also update timeout
elseif key == "rate" then
self._props.rate = value
timer.timeout = 1 / value
-- If it's in _props set it there
elseif self._props[key] ~= nil then self._props[key] = value
-- Otherwise just set it normally
else rawset(self, key, value) end
end
setmetatable(obj, mt)
return obj
end
return timed