모듈:Random 문서 원본 보기

local p = {}

function p._rand(m, n, o)
	local seed, key
	
	if type(m) == "table" then
		seed = tonumber(m.seed)
		key = tonumber(m.key)
		m = nil
		n = nil
	elseif type(n) == "table" then
		seed = tonumber(n.seed)
		key = tonumber(n.key)
		m = tonumber(m)
		n = nil
	elseif type(o) == "table" then
		seed = tonumber(o.seed)
		key = tonumber(o.key)
		m = tonumber(m)
		n = tonumber(n)
	else
		m = tonumber(m)
		n = tonumber(n)
	end
	
	if (seed or 0) == 0 then
		seed = os.time() * ((key or 0) == 0 and os.clock() or key)
	elseif (key or 0) ~= 0 then
		seed = seed * key
	end
	
	math.randomseed(seed)
	
	if n ~= nil then
		return math.random(m, n)
	elseif m ~= nil then
		return math.random(m)
	else
		return math.random()
	end
end

function p._choose(frame)
	local len = 0
	
	for index, value in ipairs(frame.args) do
		len = index
	end
	
	return frame.args[p.random(len, frame.args)]
end

function p.choose(frame)
	return p._choose(frame:getParent())
end

-- 테이블을 문자열로 변환
function p.str_join(list, delimiter)
    local len = #list
    if len == 0 then
        return ""
    end
    local string = list[1]
    for i = 2, len do
        string = string .. delimiter .. list[i]
    end
    return string
end

-- 테이블의 일부를 추출
function p.table_slice(arr,start,stop)
    local res = {}
    local n = #arr

    start = start or 1
    stop = stop or n

    -- 음수
    if start < 0 then
        start = n + start + 1
    end
    if stop < 0 then
        stop = n + stop + 1
    end

    -- start보다 stop이 큰 경우
    if stop < start then
        stop = start
    end

    -- 루프
    for i = start, stop, 1 do
        res[#res+1] = arr[i]
    end

    return res

end


-- n개의 숫자를 무작위로 뽑아서 반환
function p._sampling(n, total)
    local numbers = {}
    for i = 1, total do
        numbers[i] = i
    end

    -- 숫자를 무작위로 섞기
    math.randomseed(os.time())
    for i = total, 1, -1 do
        local r = math.random(i)
        numbers[i], numbers[r] = numbers[r], numbers[i]
    end

    -- 처음 n개 숫자 선택
    local sampled = {}
    for i = 1, n do
        table.insert(sampled, numbers[i])
    end

    return sampled
end

-- {{#invoke:random|lotto}}
function p.lotto(frame)
    local n = tonumber(frame.args[1]) or 6
    local total = tonumber(frame.args[2]) or 45
    local bonus = tonumber(frame.args[3]) or 1
    local sep = frame.args[4] or ", "
    local bonusSep = frame.args[5] or " + "
    local sort = frame.args[6] or "true"

    if n and total and n <= total and n > 0 then
        local allNums = p._sampling(n + bonus, total)
        -- 추출한 숫자들 중 n개 추출 및 정렬
        local nums = p.table_slice(allNums, 1, n)
        if sort == "true" then
            table.sort(nums)
        end

        -- bonus만큼 추출
        local bonusNums = p.table_slice(allNums, n + 1, n + bonus)
        if sort == "true" then
            table.sort(bonusNums)
        end

        -- 문자열로 변환
        local result = p.str_join(nums, sep)
        -- 보너스 번호가 있으면 추가
        if bonus > 0 then
            result = result .. bonusSep .. p.str_join(bonusNums, sep)
        end

        return result
    else
        return "{{색깔|Error: 파라미터가 올바르지 않습니다. 1 <= n <= total를 만족해야 합니다.|red}}"
    end
end

return p