warscript 0.0.1-dev.c37d12c → 0.0.1-dev.c47d565

package/engine/random.lua

@@ -1,9 +1,39 @@
 local ____exports = {}
 local ____math = require("math")
 local MAXIMUM_INTEGER = ____math.MAXIMUM_INTEGER
+local PI = ____math.PI
 local getRandomInt = GetRandomInt
 local getRandomReal = GetRandomReal
+local select = _G.select
+local cos = math.cos
+local sin = math.sin
+local sqrt = math.sqrt
 ____exports.randomAngle = function() return getRandomReal(0, 360) end
 ____exports.randomInteger = function(m, n) return n ~= nil and getRandomInt(m, n) or getRandomInt(0, m or MAXIMUM_INTEGER) end
 ____exports.randomFloat = function(m, n) return n ~= nil and getRandomReal(m, n) or getRandomReal(0, m or MAXIMUM_INTEGER) end
+____exports.randomXY = function(centerX, centerY, range)
+    local r = range * sqrt(getRandomReal(0, 1))
+    local t = getRandomReal(0, 1) * 2 * PI
+    return centerX + r * cos(t), centerY + r * sin(t)
+end
+____exports.randomElement = function(array)
+    return array[getRandomInt(1, #array)]
+end
+____exports.random = function(...)
+    return (select(
+        getRandomInt(
+            1,
+            select("#", ...)
+        ),
+        ...
+    ))
+end
+____exports.shuffle = function(array)
+    for i = #array - 1, 1, -1 do
+        local j = getRandomInt(0, i)
+        local value = array[i + 1]
+        array[i + 1] = array[j + 1]
+        array[j + 1] = value
+    end
+end
 return ____exports