Software `/` code `/` prosody

Annotate

util/array.lua @ 13439:1e229d710a3c

util.signal: Add support for signalfd(2) on Linux signalfd allows handling signal events using the same method as sockets, via file descriptors. Thus all signal dispatch can go through the same main event loop as everything else, removing need for thread-scary signal handling where execution would just jump to the signal handler regardless of the state of Lua, and needing to keep track of Lua states/threads.

author	Kim Alvefur <zash@zash.se>
date	Sat, 24 Feb 2024 00:05:29 +0100
parent	13248:db433ed3135c

rev	line source
1522 569d58d21612 Add copyright header to those files missing one Matthew Wild <mwild1@gmail.com> parents: 1373 diff changeset	1 -- Prosody IM
2923 b7049746bd29 Update copyright headers for 2010 Matthew Wild <mwild1@gmail.com> parents: 1934 diff changeset	2 -- Copyright (C) 2008-2010 Matthew Wild
b7049746bd29 Update copyright headers for 2010 Matthew Wild <mwild1@gmail.com> parents: 1934 diff changeset	3 -- Copyright (C) 2008-2010 Waqas Hussain
5776 bd0ff8ae98a8 Remove all trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 5564 diff changeset	4 --
1522 569d58d21612 Add copyright header to those files missing one Matthew Wild <mwild1@gmail.com> parents: 1373 diff changeset	5 -- This project is MIT/X11 licensed. Please see the
569d58d21612 Add copyright header to those files missing one Matthew Wild <mwild1@gmail.com> parents: 1373 diff changeset	6 -- COPYING file in the source package for more information.
569d58d21612 Add copyright header to those files missing one Matthew Wild <mwild1@gmail.com> parents: 1373 diff changeset	7 --
569d58d21612 Add copyright header to those files missing one Matthew Wild <mwild1@gmail.com> parents: 1373 diff changeset	8
3540 bc139431830b Monster whitespace commit (beware the whitespace monster). Waqas Hussain <waqas20@gmail.com> parents: 2923 diff changeset	9 local t_insert, t_sort, t_remove, t_concat
bc139431830b Monster whitespace commit (beware the whitespace monster). Waqas Hussain <waqas20@gmail.com> parents: 2923 diff changeset	10 = table.insert, table.sort, table.remove, table.concat;
12975 d10957394a3c util: Prefix module imports with prosody namespace Kim Alvefur <zash@zash.se> parents: 12403 diff changeset	11 local t_move = require "prosody.util.table".move;
918 967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	12
4449 ca74d8ed1a15 util.array: Avoid globals. Waqas Hussain <waqas20@gmail.com> parents: 4448 diff changeset	13 local setmetatable = setmetatable;
10592 9918b4b0cd58 util.array: Fix equality metamethod in Lua 5.3 Kim Alvefur <zash@zash.se> parents: 9529 diff changeset	14 local getmetatable = getmetatable;
4449 ca74d8ed1a15 util.array: Avoid globals. Waqas Hussain <waqas20@gmail.com> parents: 4448 diff changeset	15 local math_random = math.random;
5857 8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	16 local math_floor = math.floor;
4449 ca74d8ed1a15 util.array: Avoid globals. Waqas Hussain <waqas20@gmail.com> parents: 4448 diff changeset	17 local pairs, ipairs = pairs, ipairs;
ca74d8ed1a15 util.array: Avoid globals. Waqas Hussain <waqas20@gmail.com> parents: 4448 diff changeset	18 local tostring = tostring;
6417 060b63a27e9b util.array: Add type() local Kim Alvefur <zash@zash.se> parents: 5857 diff changeset	19 local type = type;
4449 ca74d8ed1a15 util.array: Avoid globals. Waqas Hussain <waqas20@gmail.com> parents: 4448 diff changeset	20
1905 e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	21 local array = {};
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	22 local array_base = {};
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	23 local array_methods = {};
9528 1af7cc3b9747 util.array: Break long line Matthew Wild <mwild1@gmail.com> parents: 9487 diff changeset	24 local array_mt = {
1af7cc3b9747 util.array: Break long line Matthew Wild <mwild1@gmail.com> parents: 9487 diff changeset	25 __index = array_methods;
1af7cc3b9747 util.array: Break long line Matthew Wild <mwild1@gmail.com> parents: 9487 diff changeset	26 __name = "array";
13032 ea4923bebca9 util.array: Change tostring format to [a,b,c] Kim Alvefur <zash@zash.se> parents: 12975 diff changeset	27 __tostring = function (self) return "["..self:concat(", ").."]"; end;
9528 1af7cc3b9747 util.array: Break long line Matthew Wild <mwild1@gmail.com> parents: 9487 diff changeset	28 };
1905 e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	29
9487 4e38121a349d util.array: Add freeze metamethod Kim Alvefur <zash@zash.se> parents: 8521 diff changeset	30 function array_mt:__freeze() return self; end
4e38121a349d util.array: Add freeze metamethod Kim Alvefur <zash@zash.se> parents: 8521 diff changeset	31
5085 cbc7eb5cfa8c util.array: Accept an iterator to the array constructor Matthew Wild <mwild1@gmail.com> parents: 4449 diff changeset	32 local function new_array(self, t, _s, _var)
cbc7eb5cfa8c util.array: Accept an iterator to the array constructor Matthew Wild <mwild1@gmail.com> parents: 4449 diff changeset	33 if type(t) == "function" then -- Assume iterator
cbc7eb5cfa8c util.array: Accept an iterator to the array constructor Matthew Wild <mwild1@gmail.com> parents: 4449 diff changeset	34 t = self.collect(t, _s, _var);
cbc7eb5cfa8c util.array: Accept an iterator to the array constructor Matthew Wild <mwild1@gmail.com> parents: 4449 diff changeset	35 end
918 967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	36 return setmetatable(t or {}, array_mt);
967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	37 end
967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	38
13247 1bb4aa803b32 util.array: Fix new() library function Kim Alvefur <zash@zash.se> parents: 13138 diff changeset	39 function array.new(t)
1bb4aa803b32 util.array: Fix new() library function Kim Alvefur <zash@zash.se> parents: 13138 diff changeset	40 return setmetatable(t or {}, array_mt);
1bb4aa803b32 util.array: Fix new() library function Kim Alvefur <zash@zash.se> parents: 13138 diff changeset	41 end
13138 0b0cefce6e42 util.array: Expose new() on module table Kim Alvefur <zash@zash.se> parents: 11787 diff changeset	42
1373 120275376bbb util.array: Add support for + operator to create a new array from two arrays joined Matthew Wild <mwild1@gmail.com> parents: 1372 diff changeset	43 function array_mt.__add(a1, a2)
120275376bbb util.array: Add support for + operator to create a new array from two arrays joined Matthew Wild <mwild1@gmail.com> parents: 1372 diff changeset	44 local res = new_array();
120275376bbb util.array: Add support for + operator to create a new array from two arrays joined Matthew Wild <mwild1@gmail.com> parents: 1372 diff changeset	45 return res:append(a1):append(a2);
120275376bbb util.array: Add support for + operator to create a new array from two arrays joined Matthew Wild <mwild1@gmail.com> parents: 1372 diff changeset	46 end
120275376bbb util.array: Add support for + operator to create a new array from two arrays joined Matthew Wild <mwild1@gmail.com> parents: 1372 diff changeset	47
7976 2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	48 function array_mt.__eq(a, b)
10592 9918b4b0cd58 util.array: Fix equality metamethod in Lua 5.3 Kim Alvefur <zash@zash.se> parents: 9529 diff changeset	49 if getmetatable(a) ~= array_mt or getmetatable(b) ~= array_mt then
9918b4b0cd58 util.array: Fix equality metamethod in Lua 5.3 Kim Alvefur <zash@zash.se> parents: 9529 diff changeset	50 -- Lua 5.3+ calls this if both operands are tables, even if metatables differ
9918b4b0cd58 util.array: Fix equality metamethod in Lua 5.3 Kim Alvefur <zash@zash.se> parents: 9529 diff changeset	51 return false;
9918b4b0cd58 util.array: Fix equality metamethod in Lua 5.3 Kim Alvefur <zash@zash.se> parents: 9529 diff changeset	52 end
7976 2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	53 if #a == #b then
2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	54 for i = 1, #a do
2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	55 if a[i] ~= b[i] then
2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	56 return false;
2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	57 end
2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	58 end
2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	59 else
2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	60 return false;
2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	61 end
2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	62 return true;
2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	63 end
2a7ef5fcaa77 util.array: Add O(n) equality check metamethod Kim Alvefur <zash@zash.se> parents: 7700 diff changeset	64
9529 6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	65 function array_mt.__div(a1, func)
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	66 local a2 = new_array();
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	67 local o = 0;
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	68 for i = 1, #a1 do
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	69 local new_value = func(a1[i]);
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	70 if new_value ~= nil then
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	71 o = o + 1;
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	72 a2[o] = new_value;
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	73 end
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	74 end
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	75 return a2;
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	76 end
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	77
920 e302cbc9d036 util.array: Expose array.* functions, to be used for unwrapped arrays Matthew Wild <mwild1@gmail.com> parents: 918 diff changeset	78 setmetatable(array, { __call = new_array });
e302cbc9d036 util.array: Expose array.* functions, to be used for unwrapped arrays Matthew Wild <mwild1@gmail.com> parents: 918 diff changeset	79
4440 c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	80 -- Read-only methods
c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	81 function array_methods:random()
7019 abcd5ec3ee41 util.array: Fix minory style issues Kim Alvefur <zash@zash.se> parents: 6777 diff changeset	82 return self[math_random(1, #self)];
4440 c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	83 end
c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	84
9529 6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	85 -- Return a random value excluding the one at idx
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	86 function array_methods:random_other(idx)
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	87 local max = #self;
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	88 return self[((math.random(1, max-1)+(idx-1))%max)+1];
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	89 end
6a1e7723208b util.array: Add __div for parity with util.set Matthew Wild <mwild1@gmail.com> parents: 9528 diff changeset	90
4440 c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	91 -- These methods can be called two ways:
c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	92 -- array.method(existing_array, [params [, ...]]) -- Create new array for result
c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	93 -- existing_array:method([params, ...]) -- Transform existing array into result
c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	94 --
1905 e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	95 function array_base.map(outa, ina, func)
7019 abcd5ec3ee41 util.array: Fix minory style issues Kim Alvefur <zash@zash.se> parents: 6777 diff changeset	96 for k, v in ipairs(ina) do
1905 e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	97 outa[k] = func(v);
918 967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	98 end
1905 e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	99 return outa;
918 967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	100 end
967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	101
1905 e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	102 function array_base.filter(outa, ina, func)
1915 e9d5406caf8c util.array: Fix for array:filter() (in-place filtering) Matthew Wild <mwild1@gmail.com> parents: 1905 diff changeset	103 local inplace, start_length = ina == outa, #ina;
e9d5406caf8c util.array: Fix for array:filter() (in-place filtering) Matthew Wild <mwild1@gmail.com> parents: 1905 diff changeset	104 local write = 1;
7019 abcd5ec3ee41 util.array: Fix minory style issues Kim Alvefur <zash@zash.se> parents: 6777 diff changeset	105 for read = 1, start_length do
1915 e9d5406caf8c util.array: Fix for array:filter() (in-place filtering) Matthew Wild <mwild1@gmail.com> parents: 1905 diff changeset	106 local v = ina[read];
918 967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	107 if func(v) then
1915 e9d5406caf8c util.array: Fix for array:filter() (in-place filtering) Matthew Wild <mwild1@gmail.com> parents: 1905 diff changeset	108 outa[write] = v;
e9d5406caf8c util.array: Fix for array:filter() (in-place filtering) Matthew Wild <mwild1@gmail.com> parents: 1905 diff changeset	109 write = write + 1;
918 967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	110 end
967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	111 end
5776 bd0ff8ae98a8 Remove all trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 5564 diff changeset	112
1922 d5fe0f9b377a util.array: Small logic fix for array:filter() Matthew Wild <mwild1@gmail.com> parents: 1915 diff changeset	113 if inplace and write <= start_length then
7019 abcd5ec3ee41 util.array: Fix minory style issues Kim Alvefur <zash@zash.se> parents: 6777 diff changeset	114 for i = write, start_length do
1915 e9d5406caf8c util.array: Fix for array:filter() (in-place filtering) Matthew Wild <mwild1@gmail.com> parents: 1905 diff changeset	115 outa[i] = nil;
e9d5406caf8c util.array: Fix for array:filter() (in-place filtering) Matthew Wild <mwild1@gmail.com> parents: 1905 diff changeset	116 end
e9d5406caf8c util.array: Fix for array:filter() (in-place filtering) Matthew Wild <mwild1@gmail.com> parents: 1905 diff changeset	117 end
5776 bd0ff8ae98a8 Remove all trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 5564 diff changeset	118
1905 e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	119 return outa;
918 967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	120 end
967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	121
11787 3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	122 function array_base.slice(outa, ina, i, j)
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	123 if j == nil then
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	124 j = -1;
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	125 end
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	126 if j < 0 then
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	127 j = #ina + (j+1);
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	128 end
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	129 if i < 0 then
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	130 i = #ina + (i+1);
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	131 end
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	132 if i < 1 then
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	133 i = 1;
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	134 end
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	135 if j > #ina then
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	136 j = #ina;
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	137 end
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	138 if i > j then
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	139 for idx = 1, #outa do
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	140 outa[idx] = nil;
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	141 end
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	142 return outa;
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	143 end
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	144
12403 42b2713ab818 util.array: Take advantage of table.move() Matthew Wild <mwild1@gmail.com> parents: 11787 diff changeset	145
42b2713ab818 util.array: Take advantage of table.move() Matthew Wild <mwild1@gmail.com> parents: 11787 diff changeset	146 t_move(ina, i, j, 1, outa);
11787 3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	147 if ina == outa then
12403 42b2713ab818 util.array: Take advantage of table.move() Matthew Wild <mwild1@gmail.com> parents: 11787 diff changeset	148 -- Clear (nil) remainder of range
42b2713ab818 util.array: Take advantage of table.move() Matthew Wild <mwild1@gmail.com> parents: 11787 diff changeset	149 t_move(ina, #outa+1, #outa*2, 2+j-i, ina);
11787 3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	150 end
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	151 return outa;
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	152 end
3ae6fa901a8b util.array: Add :slice() method + tests Matthew Wild <mwild1@gmail.com> parents: 10895 diff changeset	153
1905 e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	154 function array_base.sort(outa, ina, ...)
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	155 if ina ~= outa then
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	156 outa:append(ina);
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	157 end
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	158 t_sort(outa, ...);
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	159 return outa;
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	160 end
918 967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	161
8017 ec7cab8e744d util.array: Add method for filtering out duplicate values Kim Alvefur <zash@zash.se> parents: 7976 diff changeset	162 function array_base.unique(outa, ina)
ec7cab8e744d util.array: Add method for filtering out duplicate values Kim Alvefur <zash@zash.se> parents: 7976 diff changeset	163 local seen = {};
ec7cab8e744d util.array: Add method for filtering out duplicate values Kim Alvefur <zash@zash.se> parents: 7976 diff changeset	164 return array_base.filter(outa, ina, function (item)
ec7cab8e744d util.array: Add method for filtering out duplicate values Kim Alvefur <zash@zash.se> parents: 7976 diff changeset	165 if seen[item] then
ec7cab8e744d util.array: Add method for filtering out duplicate values Kim Alvefur <zash@zash.se> parents: 7976 diff changeset	166 return false;
ec7cab8e744d util.array: Add method for filtering out duplicate values Kim Alvefur <zash@zash.se> parents: 7976 diff changeset	167 else
ec7cab8e744d util.array: Add method for filtering out duplicate values Kim Alvefur <zash@zash.se> parents: 7976 diff changeset	168 seen[item] = true;
ec7cab8e744d util.array: Add method for filtering out duplicate values Kim Alvefur <zash@zash.se> parents: 7976 diff changeset	169 return true;
ec7cab8e744d util.array: Add method for filtering out duplicate values Kim Alvefur <zash@zash.se> parents: 7976 diff changeset	170 end
ec7cab8e744d util.array: Add method for filtering out duplicate values Kim Alvefur <zash@zash.se> parents: 7976 diff changeset	171 end);
ec7cab8e744d util.array: Add method for filtering out duplicate values Kim Alvefur <zash@zash.se> parents: 7976 diff changeset	172 end
ec7cab8e744d util.array: Add method for filtering out duplicate values Kim Alvefur <zash@zash.se> parents: 7976 diff changeset	173
10895 5777968301e8 util.array: pluck: Support default value to avoid holes Matthew Wild <mwild1@gmail.com> parents: 10592 diff changeset	174 function array_base.pluck(outa, ina, key, default)
7019 abcd5ec3ee41 util.array: Fix minory style issues Kim Alvefur <zash@zash.se> parents: 6777 diff changeset	175 for i = 1, #ina do
10895 5777968301e8 util.array: pluck: Support default value to avoid holes Matthew Wild <mwild1@gmail.com> parents: 10592 diff changeset	176 local v = ina[i][key];
5777968301e8 util.array: pluck: Support default value to avoid holes Matthew Wild <mwild1@gmail.com> parents: 10592 diff changeset	177 if v == nil then
5777968301e8 util.array: pluck: Support default value to avoid holes Matthew Wild <mwild1@gmail.com> parents: 10592 diff changeset	178 v = default;
5777968301e8 util.array: pluck: Support default value to avoid holes Matthew Wild <mwild1@gmail.com> parents: 10592 diff changeset	179 end
5777968301e8 util.array: pluck: Support default value to avoid holes Matthew Wild <mwild1@gmail.com> parents: 10592 diff changeset	180 outa[i] = v;
4439 1c8d2c0d02db util.array: Add pluck() method to pick a given property from each item Matthew Wild <mwild1@gmail.com> parents: 4387 diff changeset	181 end
1c8d2c0d02db util.array: Add pluck() method to pick a given property from each item Matthew Wild <mwild1@gmail.com> parents: 4387 diff changeset	182 return outa;
1c8d2c0d02db util.array: Add pluck() method to pick a given property from each item Matthew Wild <mwild1@gmail.com> parents: 4387 diff changeset	183 end
1c8d2c0d02db util.array: Add pluck() method to pick a given property from each item Matthew Wild <mwild1@gmail.com> parents: 4387 diff changeset	184
5857 8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	185 function array_base.reverse(outa, ina)
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	186 local len = #ina;
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	187 if ina == outa then
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	188 local middle = math_floor(len/2);
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	189 len = len + 1;
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	190 local o; -- opposite
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	191 for i = 1, middle do
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	192 o = len - i;
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	193 outa[i], outa[o] = outa[o], outa[i];
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	194 end
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	195 else
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	196 local off = len + 1;
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	197 for i = 1, len do
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	198 outa[i] = ina[off - i];
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	199 end
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	200 end
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	201 return outa;
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	202 end
8613888d0f9e util.array: Improve array:reverse() and make it work as both method and non-mutating function Kim Alvefur <zash@zash.se> parents: 5776 diff changeset	203
4440 c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	204 --- These methods only mutate the array
7700 0d70410efdcf util.array: Remove unused arguments [luacheck] Kim Alvefur <zash@zash.se> parents: 7699 diff changeset	205 function array_methods:shuffle()
918 967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	206 local len = #self;
7019 abcd5ec3ee41 util.array: Fix minory style issues Kim Alvefur <zash@zash.se> parents: 6777 diff changeset	207 for i = 1, #self do
abcd5ec3ee41 util.array: Fix minory style issues Kim Alvefur <zash@zash.se> parents: 6777 diff changeset	208 local r = math_random(i, len);
918 967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	209 self[i], self[r] = self[r], self[i];
967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	210 end
1372 3b13bb57002e util.array: Make array:reverse() and array:shuffle() return the array to allow chaining Matthew Wild <mwild1@gmail.com> parents: 1371 diff changeset	211 return self;
918 967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	212 end
967edf874df7 util.array: New array library Matthew Wild <mwild1@gmail.com> parents: diff changeset	213
7699 9c40d0be2295 util.array: Rename arguments to avoid name clash [luacheck] Kim Alvefur <zash@zash.se> parents: 7020 diff changeset	214 function array_methods:append(ina)
12403 42b2713ab818 util.array: Take advantage of table.move() Matthew Wild <mwild1@gmail.com> parents: 11787 diff changeset	215 t_move(ina, 1, #ina, #self+1, self);
1371 9e45bdf55353 util.array: Add array:append() method, to append a new array to an existing one Matthew Wild <mwild1@gmail.com> parents: 1027 diff changeset	216 return self;
9e45bdf55353 util.array: Add array:append() method, to append a new array to an existing one Matthew Wild <mwild1@gmail.com> parents: 1027 diff changeset	217 end
9e45bdf55353 util.array: Add array:append() method, to append a new array to an existing one Matthew Wild <mwild1@gmail.com> parents: 1027 diff changeset	218
4440 c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	219 function array_methods:push(x)
4449 ca74d8ed1a15 util.array: Avoid globals. Waqas Hussain <waqas20@gmail.com> parents: 4448 diff changeset	220 t_insert(self, x);
4448 d745f4c28294 util.array: Make array:push() chainable. Waqas Hussain <waqas20@gmail.com> parents: 4440 diff changeset	221 return self;
4440 c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	222 end
c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	223
7020 6ab9c691c4c6 util.array: Just use table.remove as array:pop() Kim Alvefur <zash@zash.se> parents: 7019 diff changeset	224 array_methods.pop = t_remove;
4440 c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	225
c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	226 function array_methods:concat(sep)
4449 ca74d8ed1a15 util.array: Avoid globals. Waqas Hussain <waqas20@gmail.com> parents: 4448 diff changeset	227 return t_concat(array.map(self, tostring), sep);
4440 c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	228 end
c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	229
c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	230 function array_methods:length()
c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	231 return #self;
c60ed6732b34 util.array: Expand some of the more basic methods to act more sensibly than their names suggested Matthew Wild <mwild1@gmail.com> parents: 4439 diff changeset	232 end
1905 e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	233
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	234 --- These methods always create a new array
1027 fe2e3d3dba6a util.array: Add array.collect() to collect results from iterators into an array, and use module() to correctly set the module name Matthew Wild <mwild1@gmail.com> parents: 922 diff changeset	235 function array.collect(f, s, var)
4387 06161b0b83f2 util.array: Fix array.collect() for iterators that expect initial value of var to be preserved Matthew Wild <mwild1@gmail.com> parents: 3540 diff changeset	236 local t = {};
1027 fe2e3d3dba6a util.array: Add array.collect() to collect results from iterators into an array, and use module() to correctly set the module name Matthew Wild <mwild1@gmail.com> parents: 922 diff changeset	237 while true do
fe2e3d3dba6a util.array: Add array.collect() to collect results from iterators into an array, and use module() to correctly set the module name Matthew Wild <mwild1@gmail.com> parents: 922 diff changeset	238 var = f(s, var);
7019 abcd5ec3ee41 util.array: Fix minory style issues Kim Alvefur <zash@zash.se> parents: 6777 diff changeset	239 if var == nil then break; end
4449 ca74d8ed1a15 util.array: Avoid globals. Waqas Hussain <waqas20@gmail.com> parents: 4448 diff changeset	240 t_insert(t, var);
1027 fe2e3d3dba6a util.array: Add array.collect() to collect results from iterators into an array, and use module() to correctly set the module name Matthew Wild <mwild1@gmail.com> parents: 922 diff changeset	241 end
fe2e3d3dba6a util.array: Add array.collect() to collect results from iterators into an array, and use module() to correctly set the module name Matthew Wild <mwild1@gmail.com> parents: 922 diff changeset	242 return setmetatable(t, array_mt);
fe2e3d3dba6a util.array: Add array.collect() to collect results from iterators into an array, and use module() to correctly set the module name Matthew Wild <mwild1@gmail.com> parents: 922 diff changeset	243 end
fe2e3d3dba6a util.array: Add array.collect() to collect results from iterators into an array, and use module() to correctly set the module name Matthew Wild <mwild1@gmail.com> parents: 922 diff changeset	244
1905 e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	245 ---
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	246
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	247 -- Setup methods from array_base
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	248 for method, f in pairs(array_base) do
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	249 local base_method = f;
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	250 -- Setup global array method which makes new array
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	251 array[method] = function (old_a, ...)
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	252 local a = new_array();
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	253 return base_method(a, old_a, ...);
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	254 end
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	255 -- Setup per-array (mutating) method
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	256 array_methods[method] = function (self, ...)
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	257 return base_method(self, self, ...);
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	258 end
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	259 end
e3e0a17e0b33 util.array: Per-array methods now always mutate the array, array.* return a mutated copy, and most methods (e.g. sort) now return the array Matthew Wild <mwild1@gmail.com> parents: 1522 diff changeset	260
1027 fe2e3d3dba6a util.array: Add array.collect() to collect results from iterators into an array, and use module() to correctly set the module name Matthew Wild <mwild1@gmail.com> parents: 922 diff changeset	261 return array;

prosody

1e229d710a3c

util/array.lua

Software / code / prosody

Annotate

util/array.lua @ 13439:1e229d710a3c

Software `/` code `/` prosody