util/array.lua

   1 local array = {};
   2
   3 local array_mt = { __index = array, __tostring = function (array) return array:concat(", "); end };
   4 local function new_array(_, t)
   5         return setmetatable(t or {}, array_mt);
   6 end
   7
   8 setmetatable(array, { __call = new_array });
   9
  10 function array:map(func, t2)
  11         local t2 = t2 or array{};
  12         for k,v in ipairs(self) do
  13                 t2[k] = func(v);
  14         end
  15         return t2;
  16 end
  17
  18 function array:filter(func, t2)
  19         local t2 = t2 or array{};
  20         for k,v in ipairs(self) do
  21                 if func(v) then
  22                         t2:push(v);
  23                 end
  24         end
  25         return t2;
  26 end
  27
  28
  29 array.push = table.insert;
  30 array.pop = table.remove;
  31 array.sort = table.sort;
  32 array.concat = table.concat;
  33 array.length = function (t) return #t; end
  34
  35 function array:random()
  36         return self[math.random(1,#self)];
  37 end
  38
  39 function array:shuffle()
  40         local len = #self;
  41         for i=1,#self do
  42                 local r = math.random(i,len);
  43                 self[i], self[r] = self[r], self[i];
  44         end
  45 end
  46
  47 _G.array = array