89 lines
1.8 KiB
Lua
89 lines
1.8 KiB
Lua
--!strict
|
|
--!optimize 2
|
|
|
|
local Array = {}
|
|
Array.__index = Array
|
|
|
|
type constructor<T> = { inner: {T} }
|
|
export type Array<T> = typeof(setmetatable({} :: constructor<T>, Array))
|
|
|
|
type FnCall<T, R = T> = (T, number, Array<T>) -> R
|
|
|
|
function Array.new<T>(from: {T}?): Array<T>
|
|
return setmetatable({ inner = from or {} }, Array)
|
|
end
|
|
|
|
function Array:table<T>(): {T}
|
|
return self.inner
|
|
end
|
|
|
|
function Array:pop<T>(): T?
|
|
return table.remove(self.inner, #self.inner)
|
|
end
|
|
|
|
function Array:concat(sep: string?, i: number?, j: number?): string
|
|
return table.concat(self.inner, sep, i, j)
|
|
end
|
|
|
|
function Array:clone<T>(): Array<T>
|
|
return Array.new(table.clone(self.inner))
|
|
end
|
|
|
|
function Array:len(): number
|
|
return #self.inner
|
|
end
|
|
|
|
function Array:get<T, I>(index: I): T?
|
|
return self.inner[index]
|
|
end
|
|
|
|
function Array:push<T>(...: T): ()
|
|
for _, val: T in {...} do
|
|
table.insert(self.inner, val)
|
|
end
|
|
end
|
|
|
|
function Array:foreach<T>(callbackFn: FnCall<T, nil>): Array<T>
|
|
for ind: number, val: T in self.inner do
|
|
callbackFn(val, ind, self)
|
|
end
|
|
return self
|
|
end
|
|
|
|
function Array:map<T, M>(callbackFn: FnCall<T, M>): Array<M>
|
|
local n: {M} = {}
|
|
for ind: number, val: T in self.inner do
|
|
table.insert(n, callbackFn(val, ind, self))
|
|
end
|
|
return Array.new(n)
|
|
end
|
|
|
|
function Array:filter<T>(callbackFn: FnCall<T, boolean>): Array<T>
|
|
local n: {T} = {}
|
|
for ind: number, val: T in self.inner do
|
|
if callbackFn(val, ind, self) then
|
|
table.insert(n, val)
|
|
end
|
|
end
|
|
return Array.new(n)
|
|
end
|
|
|
|
function Array:binary_search<T>(x: T): T?
|
|
local size = #self.inner
|
|
if size == 0 then return nil end
|
|
local base = 1
|
|
|
|
while base < size do
|
|
size -= bit32.rshift(base + size, 1)
|
|
end
|
|
|
|
return nil
|
|
end
|
|
|
|
function Array:debug_display<T>(): string
|
|
return self:map(function(val: T, ind: number)
|
|
return ind .. ": " .. tostring(val) .. (ind ~= self:len() and ", " or "")
|
|
end):concat()
|
|
end
|
|
|
|
return Array |