Module:DiceUtils

From bg3.wiki
Jump to navigation Jump to search

Documentation for this module may be created at Module:DiceUtils/doc

local p = {}

function p.calculateDamage(frame)
	args = frame.args
	
	result = {}
	damage = {
		Acid = {},
		Cold = {},
		Fire = {},
		Force = {},
		Healing = {},
		Lightning = {},
		Necrotic = {},
		Poison = {},
		Radiant = {},
		Thunder = {},
		Psychic = {},
		Physical = {},
		Piercing = {},
		Bludgeoning = {},
		Slashing = {}
	}
	
	for damage_die in string.gmatch(args.dice, "([^,]+)") do
		idx = string.find(damage_die, ":")
		die = string.sub(damage_die, 1, idx - 1)
		d_type = string.sub(damage_die, idx + 1)
		
		table.insert(damage[d_type], die)
	end
	
	for d_type, dice in pairs(damage) do
		if not (next(dice) == nil) then
			min_damage = 0
			max_damage = 0
			avg_str = ""
			for i, die in ipairs(dice) do
				idx = string.find(die, "d")
				count = tonumber(string.sub(die, 1, idx -1))
				value = tonumber(string.sub(die, idx + 1))
				min_damage = min_damage + count
				max_damage = max_damage + value * count
				
				avg_str = string.format("(%d - %d)", min_damage, max_damage)
			end
			
			damage_str = string.format("%s %s %s", table.concat(dice, " + "), d_type, avg_str)
			
			table.insert(result, damage_str)
		end
	end
	
	return table.concat(result, " + ")

end

return p