Added table.shuffle

ianmaclarty · Nov 30, 2016 · 3943a20 · 3943a20
1 parent 0319737
commit 3943a20
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/lua/extra.lua b/lua/extra.lua
@@ -125,6 +125,17 @@ function table.count(t)
     return count
 end
 
+function table.shuffle(t)
+  if type(t) ~= "table" then
+      error("table expected, but got a "..type(t), 2)
+  end
+
+  for a = #t, 2, -1 do
+    local b = math.random(a)
+    t[a], t[b] = t[b], t[a]
+  end
+end
+
 local
 function table_tostring(t, indent, seen, depth)
     seen = seen or {}
@@ -154,7 +165,7 @@ function table_tostring(t, indent, seen, depth)
                 str = str .. prefix .. tab .. table_tostring(t[i], indent + 1, seen, depth + 1) .. ",\n"
             end
         else
-            table.sort(keys, function(k1, k2) 
+            table.sort(keys, function(k1, k2)
                 local t1 = type(k1)
                 local t2 = type(k2)
                 if t1 == "string" and t2 == "string" then