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)
args = process_args.merge(true)
local tblName = args['name'] or ''
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
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
finalTable[i]['chance'] = string.format("%.2f%%",row['weight']*100/totalWeight)
end
return (p.formatTable(finalTable))
end
function p.formatTable(final)
local tblheader = '<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">' .. tblheader .. tr .. '</table>'
end
return p