130 lines
3.6 KiB
Lua
130 lines
3.6 KiB
Lua
--- General stateless utility algorithms
|
|
--
|
|
-- @module gear.algo
|
|
-- @copyright 2022 The DoubleFourteen Code Forge
|
|
-- @author Lorenzo Cogotti
|
|
|
|
local floor = math.floor
|
|
local min, max = math.min, math.max
|
|
|
|
local algo = {}
|
|
|
|
|
|
--- Clamp x within range [a,b] (where b >= a).
|
|
--
|
|
-- @param x (number) value to clamp.
|
|
-- @param a (number) interval lower bound (inclusive).
|
|
-- @param b (number) interval upper bound (includive).
|
|
-- @return (number) clamped value.
|
|
function algo.clamp(x, a, b) return min(max(x, a), b) end
|
|
|
|
--- Fast remove from array.
|
|
--
|
|
-- Replace 'array[i]' with last array's element and
|
|
-- discard array's tail.
|
|
function algo.removefast(array, i)
|
|
local n = #array
|
|
|
|
array[i] = array[n] -- NOP if i == n
|
|
array[n] = nil
|
|
end
|
|
|
|
local function lt(a, b) return a < b end
|
|
|
|
--- Sort array using Insertion Sort - O(n^2).
|
|
--
|
|
-- Provides the most basic sorting algorithm around.
|
|
-- Performs better than regular table.sort() for small arrays
|
|
-- (~100 elements).
|
|
--
|
|
-- @param array (table) array to be sorted.
|
|
-- @param less (function|nil) comparison function, takes 2 arguments,
|
|
-- returns true if its first argument is less than its second argument, false otherwise.
|
|
-- Defaults to operator <.
|
|
function algo.insertionsort(array, less)
|
|
less = less or lt
|
|
|
|
for i = 2,#array do
|
|
local val = array[i]
|
|
local j = i
|
|
|
|
while j > 1 and less(val, array[j-1]) do
|
|
array[j] = array[j-1]
|
|
j = j - 1
|
|
end
|
|
|
|
array[j] = val
|
|
end
|
|
end
|
|
|
|
--- Binary search last element where
|
|
-- what <= array[i] - also known as lower bound.
|
|
--
|
|
-- @param array (table) an array sorted according to the less function.
|
|
-- @param what the comparison argument.
|
|
-- @param less (function|nil) sorting criterium, a function taking 2 arguments,
|
|
-- returns true if the first argument is less than the second argument,
|
|
-- false otherwise. Defaults to using the < operator.
|
|
--
|
|
-- @return (number) the greatest index i, where what <= array[i].
|
|
-- If no such element exists, it returns an out of bounds index
|
|
-- such that array[i] == nil.
|
|
function algo.bsearchl(array, what, less)
|
|
less = less or lt
|
|
|
|
local lo, hi = 1, #array
|
|
local ofs, mid = -1, hi
|
|
|
|
while mid > 0 do
|
|
mid = floor(hi / 2)
|
|
|
|
-- array[lo+mid] <= what <-> what >= array[lo+mid]
|
|
-- <-> not what < array[lo+mid]
|
|
if not less(what, array[lo+mid]) then
|
|
lo = lo + mid
|
|
ofs = 0 -- at least one element where array[lo+mid] <= what
|
|
end
|
|
|
|
hi = hi - mid
|
|
end
|
|
|
|
return lo + ofs
|
|
end
|
|
|
|
--- Binary search first element where
|
|
-- what >= array[i] - also known as upper bound.
|
|
--
|
|
-- @param array (array) an array sorted according to the less function.
|
|
-- @param what the comparison argument.
|
|
-- @param less (function|nil) sorting criterium, a function taking 2 arguments,
|
|
-- returns true if the first argument is less than the second argument,
|
|
-- false otherwise. Defaults to using the < operator.
|
|
--
|
|
-- @return (number) the smallest index i, where what >= array[i].
|
|
-- If no such element exists, it returns an out of bounds index
|
|
-- such that array[i] == nil.
|
|
function algo.bsearchr(array, what, less)
|
|
less = less or lt
|
|
|
|
local lo, hi = 1, #array
|
|
local ofs, mid = -1, hi
|
|
|
|
while mid > 0 do
|
|
mid = floor(hi / 2)
|
|
|
|
-- array[lo+mid] >= what <-> not array[lo+mid] < what
|
|
if not less(array[lo+mid], what) then
|
|
ofs = 0
|
|
else
|
|
lo = lo + mid
|
|
ofs = 1
|
|
end
|
|
|
|
hi = hi - mid
|
|
end
|
|
|
|
return lo + ofs
|
|
end
|
|
|
|
return algo
|