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         return self;
  46 end
  47
  48 function array:reverse()
  49         local len = #self-1;
  50         for i=len,1,-1 do
  51                 self:push(self[i]);
  52                 self:pop(i);
  53         end
  54         return self;
  55 end
  56
  57 function array:append(array)
  58         local len,len2  = #self, #array;
  59         for i=1,len2 do
  60                 self[len+i] = array[i];
  61         end
  62         return self;
  63 end
  64
  65 function array.collect(f, s, var)
  66         local t, var = {};
  67         while true do
  68                 var = f(s, var);
  69                 if var == nil then break; end
  70                 table.insert(t, var);
  71         end
  72         return setmetatable(t, array_mt);
  73 end
  74
  75 _G.array = array;
  76 module("array");
  77
  78 return array;