Initial commit

.gitignore

@@ -1 +0,0 @@
-tests/results.*

README.md

@@ -0,0 +1,3 @@
+# lua-array
+
+Generalized array implementation on top of Lua tables

luau/src.luau

@@ -1,176 +0,0 @@
---!strict
---!optimize 2
-
-local Array = {}
-Array.__index = Array
-
-type constructor<T> = { __inner: {T} }
-export type Array<T> = setmetatable<constructor<T>, typeof(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:into_table<T>(): {T}
-	return self.__inner
-end
-
-function Array:len(): number
-	return #self.__inner
-end
-
-function Array:pop<T>(): T?
-	return table.remove(self.__inner, self:len())
-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:is_empty(): boolean
-	return self:len() == 0
-end
-
-function Array:type_id(): string
-	return type(self.__inner[1])
-end
-
-function Array:get<T, I>(index: I): T?
-	return self.__inner[index]
-end
-
-function Array:get_base0<T>(index: number): T?
-	return self:get(index + 1)
-end
-
-function Array:join<T>(sep: T): Array<T>
-	return self
-end
-
-function Array:append<T>(other: Array<T>): Array<T>
-	return self:append_table(other:into_table())
-end
-
-function Array:append_table<T>(t: {T}): Array<T>
-	table.move(t, 1, #t, self:len() + 1, self.__inner)
-	return self
-end
-
-function Array:slice<T>(start: number, endj: number?): Array<T>
-	local slice_t: {T} = {}
-	table.move(self.__inner, start, endj or self:len(), 1, slice_t)
-	return Array.new(slice_t)
-end
-
-function Array:push<T>(...: T): ()
-	for _, val: T in {...} do
-		table.insert(self.__inner, val)
-	end
-end
-
-function Array:foreach<T>(f: FnCall<T, nil>): Array<T>
-	for ind: number, val: T in self.__inner do
-		f(val, ind, self)
-	end
-	return self
-end
-
-function Array:map<T, M>(f: FnCall<T, M>): Array<M>
-	local t_map: {M} = {}
-	for ind: number, val: T in self.__inner do
-		table.insert(t_map, f(val, ind, self))
-	end
-	return Array.new(t_map)
-end
-
-function Array:filter<T>(f: FnCall<T, boolean>): Array<T>
-	local t_filter: {T} = {}
-	for ind: number, val: T in self.__inner do
-		if f(val, ind, self) then
-			table.insert(t_filter, val)
-		end
-	end
-	return Array.new(t_filter)
-end
-
-function Array:reverse<T>(): Array<T>
-	local t_rev: {T} = {}
-	local t_len = self:len()
-	for i = 1, t_len do
-		t_rev[i] = self.__inner[t_len - i + 1]
-	end
-	self.__inner = t_rev
-	return self
-end
-
-function Array:sort<T>(): Array<T>
-	table.sort(self.__inner, function(a: T, z: T)
-		if type(a) == "number" then
-			return a < z
-		end
-		return tostring(a) < tostring(z)
-	end)
-	return self
-end
-
-function Array:reduce(f: (number, number) -> number): number
-	local init_v = self.__inner[1]
-	if type(init_v) == "string" then
-		return tonumber("nan") :: number
-	end
-	for _, val: number in self.__inner do
-		init_v = f(init_v, val)
-	end
-	return init_v
-end
-
-local function binary_search_by(self: Array<number>, x: number, eq: (number, number) -> number): number?
-	local size = self:len()
-	if size == 0 then return nil end
-	local base = 1
-	while base <= size do
-		local mid = eq(base, size)
-		if self.__inner[mid] == x then
-			return mid
-		elseif self.__inner[mid] < x then
-			base = mid + 1
-		else
-			size = mid - 1
-		end
-	end
-	return nil
-end
-
-function Array:binary_search(x: number): number?
-	if self:type_id() ~= "number" then
-		return nil
-	end
-	return binary_search_by(self, x, function(base, size)
-		return bit32.rshift(base + size, 1)
-	end)
-end
-
-function Array:binary_search_64(x: number): number?
-	if self:type_id() ~= "number" then
-		return nil
-	end
-	return binary_search_by(self, x, function(base, size)
-		return math.floor((base + size) / 2)
-	end)
-end
-
-function Array:tostring<T>(): string
-	return self:map(function(val: T, index: number)
-		local formatted_val = if type(val) == "string" then `"{val}"` else tostring(val)
-		local formatted_end = if index ~= self:len() then ", " else ""
-		return index .. ": " .. formatted_val .. formatted_end
-	end):concat()
-end
-
-return Array

tests/luau/map_filter.luau

@@ -1,17 +0,0 @@
---!strict
-
-local Array = require("../../luau/src")
-
-local numbers = Array.new()
-for i = 1, 100 do
-	numbers:push(i)
-end
-print(numbers:tostring(), "\nlength=", numbers:len())
-
-numbers:map(function(n: number)
-	return tostring(n)
-end):foreach(function(e) print("(:map)squared=", e) end)
-
-numbers:filter(function(n: number)
-	return bit32.band(n, 1) ~= 0
-end):foreach(function(e) print("(:filter)odd=", e) end)

tests/luau/reduce.luau

@@ -1,13 +0,0 @@
---!strict
-
-local Array = require("../../luau/src")
-
-local numbers = Array.new({1, 2, 3, 4})
-
-print(numbers:tostring(), "\nlength=", numbers:len())
-
-local sum = numbers:reduce(function(accumulator: number, currentValue: number)
-	return accumulator - currentValue
-end)
-
-print("sum=", sum)

tests/luau/reverse.luau

@@ -1,13 +0,0 @@
---!strict
-
-local Array = require("../../luau/src")
-
-local numbers = Array.new()
-for i = 1, 100 do
-	numbers:push(i)
-end
-print(numbers:tostring(), "\nlength=", numbers:len())
-
-numbers:reverse()
-
-print(numbers:tostring())

tests/luau/slice.luau

@@ -1,18 +0,0 @@
---!strict
-
-local Array = require("../../luau/src")
-
-local animals = Array.new({"ant", "bison", "camel", "duck", "elephant"})
-
-print(animals:tostring(), "\nlength=", animals:len())
-
-local slice = animals:slice(3)
-local slice2 = animals:slice(2,4)
-local slice3 = animals:slice(1,5)
-
-print("(:slice 3)" .. slice:tostring())
-print("(:slice 2,4)" .. slice2:tostring())
-print("(:slice 1,5)" .. slice3:tostring())
-
-animals:join({"hi", "shrek"})
-print("(:join)" .. animals:tostring())

tests/luau/sort.luau

@@ -1,11 +0,0 @@
---!strict
-
-local Array = require("../../luau/src")
-
-local months = Array.new({"March", "Jan", "Feb", "Dec"})
-
-print(months:tostring(), "\nlength=", months:len())
-
-months:sort()
-
-print(months:tostring())

tests/start.sh

@@ -1,15 +0,0 @@
-#!/bin/bash
-
-touch results.txt
-echo "" > results.txt
-
-test () {
-	echo -e "\n-- [Test: $1] --" >> results.txt
-	luau $2 >> results.txt
-}
-
-test "map_filter" "./luau/map_filter.luau"
-test "reverse" "./luau/reverse.luau"
-test "reduce" "./luau/reduce.luau"
-test "slice" "./luau/slice.luau"
-test "sort" "./luau/sort.luau"