Module:EncounterTable
Jump to navigation
Jump to search
local p = {}
local process_args = require'Module:ProcessArgs'
local text_util = require'Module:Text Utility'
function p.splitEncounterTableArgs(row)
local TABLE_ARGS = {'table', 'item', 'quantity', 'weight'}
return text_util.splitArgs(row, TABLE_ARGS)
end
function p.start(frame)
local args = process_args.merge(true)
local tblName = args.name or ''
local pick = args.roll or 'once'
local result = {}
local totalWeight = 0
for i, row in ipairs(args) do
local newrow = p.splitEncounterTableArgs(row)
result[i] = newrow
totalWeight = totalWeight + tonumber(newrow.weight)
end
-- format table.
local finalTable = {}
local str = ''
for i, row in ipairs(result) do
finalTable[i] = {}
if row.table == 'none' then
if row.item == 'none' then
error'A table or item must be specified!'
else
finalTable[i].item = '[[' .. row.item .. ']]'
end
elseif row.item == 'none' then
local linkstr = ''
maintable = row.table:match'^(.+)%s+%w*$'
if maintable ~= nil and maintable ~= '' then
linkstr = maintable .. '#' .. row.table
else
linkstr = row.table
end
finalTable[i].item = 'Item from [[Data:' .. linkstr .. '|' .. row.table .. ']]'
else
error'A table and item cannot be specified at the same time!'
end
-- dice tooltip quantity
finalTable[i].quantity = frame:expandTemplate{
title = 'Dice tooltip',
args = {row.quantity},
}
--finalTable[i].quantity = row.quantity
finalTable[i].weight = row.weight
-- calculate final chance
if pick == 'once' then
finalTable[i].chance = ('%.2f%%'):format(row.weight * 100 / totalWeight)
elseif pick == 'each' then
finalTable[i].chance = ('%.2f%%'):format(row.weight)
else
error'"roll" parameter takes only "once" or "each"!'
end
end
return p.formatTable(finalTable)
end
function p.formatTable(final)
local TABLE_HEADER = '<tr><th>Item</th><th>Quantity</th><th>Weight</th><th>Chance</th></tr>'
local tr = ''
for _, v in ipairs(final) do
tr = tr .. ' <tr><td>' .. v.item .. '</td><td>' .. v.quantity .. '</td><td>' .. v.weight .. '</td><td>' .. v.chance .. '</td></tr>'
end
return '<table class="wikitable sortable">' .. TABLE_HEADER .. tr .. '</table>'
end
return p