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