Module:LootChest

local p = {

calc_average_amount_this_item_per_chest = function( 			min_stacksize, max_stacksize,			min_chest_numstacks, max_chest_numstacks, 			item_weight, chest_total_item_weight ) local avg_stacksize = ( min_stacksize + max_stacksize ) / 2 local avg_numstacks = ( min_chest_numstacks + max_chest_numstacks ) / 2 return avg_stacksize * avg_numstacks * item_weight / chest_total_item_weight end, calc_chance_any_of_this_item_per_chest = function( 			min_chest_numstacks, max_chest_numstacks,			item_weight, chest_total_item_weight ) local avg_numstacks = ( min_chest_numstacks + max_chest_numstacks ) / 2 local relativeweight = item_weight / chest_total_item_weight return 1 - math.pow( 1 - relativeweight, avg_numstacks ) end, calc_num_chests_to_find_this_item = function( 			min_chest_numstacks, max_chest_numstacks,			item_weight, chest_total_item_weight ) local avg_numstacks = ( min_chest_numstacks + max_chest_numstacks ) / 2 local relativeweight = item_weight / chest_total_item_weight return 1 / ( 1 - math.pow( 1 - relativeweight, avg_numstacks ) ) end, -- these define which sprite, label and link to use, in the table, -- and 'cannot_stack' and 'plural' dictate how to display the single-item summary -- NOTE: order in this list doesn't matter.

items = { ["acacia-wood"]        = { "block", link="Wood", title="Acacia or Dark Oak Wood", plural=false }, ["apple"]              = { "item" }, ["beetroot-seeds"]     = { "item", link="Beetroot Seeds", plural=false }, ["bone"]               = { "item" }, ["book"]               = { "item" }, ["bread"]              = { "item", plural=false }, ["bucket"]             = { "item" }, ["coal"]               = { "item", plural=false }, ["compass"]            = { "item" }, ["diamond"]            = { "item" }, ["disc-13"]            = { "item", title="Music Disc (13)", link="Music Disc", cannot_stack=true, plural=false }, ["disc-cat"]           = { "item", title="Music Disc (Cat)", link="Music Disc", cannot_stack=true, plural=false }, ["emerald"]            = { "item" }, ["empty-map"]          = { "item", link="Map", title="Empty Map" }, ["enchanted-book"]     = { "item", cannot_stack=true, note="enchantment1" }, ["ender-pearl"]        = { "item" }, ["flint-and-steel"]    = { "item", cannot_stack=true, plural=false }, ["golden-apple"]       = { "item", title="Golden Apple (Normal)", plural="Golden Apples (Normal)" }, ["golden-chestplate"]  = { "item", link="Armor", cannot_stack=true }, ["golden-sword"]       = { "item", link="Sword", cannot_stack=true }, ["gold-ingot"]         = { "item" }, ["gold-nugget"]        = { "item" }, ["gunpowder"]          = { "item", plural=false }, ["horse-armor-diamond"] = { "item", title="Diamond Horse Armor", link="Horse Armor", cannot_stack=true, plural=false }, ["horse-armor-gold"]   = { "item", title="Gold Horse Armor", link="Horse Armor", cannot_stack=true, plural=false }, ["horse-armor-iron"]   = { "item", title="Iron Horse Armor", link="Horse Armor", cannot_stack=true, plural=false }, ["iron-boots"]         = { "item", link="Armor", cannot_stack=true, plural=false }, ["iron-chestplate"]    = { "item", link="Armor", cannot_stack=true }, ["iron-helmet"]        = { "item", link="Armor", cannot_stack=true }, ["iron-ingot"]         = { "item" }, ["iron-leggings"]      = { "item", link="Armor", cannot_stack=true, plural=false }, ["iron-pickaxe"]       = { "item", link="Pickaxe", cannot_stack=true }, ["iron-sword"]         = { "item", link="Sword", cannot_stack=true }, ["lapis-lazuli"]       = { "item", plural=false }, ["melon-seeds"]        = { "item", plural=false }, ["name-tag"]           = { "item" }, ["nether-wart"]        = { "item", plural=false }, ["oak-sapling"]        = { "block", link="Sapling" }, ["oak-wood"]           = { "block", link="Wood", title="Oak, Spruce, Birch or Jungle Wood", plural=false }, ["obsidian"]           = { "block", plural=false }, ["paper"]              = { "item", plural=false }, ["pumpkin-seeds"]      = { "item", plural=false }, ["rail"]               = { "block", title="Rails", plural=false }, ["redstone"]           = { "item", plural=false }, ["rotten-flesh"]       = { "item", plural=false }, ["saddle"]             = { "item", cannot_stack=true }, ["stick"]              = { "item" }, ["stone-axe"]          = { "item", link="Axe", cannot_stack=true }, ["stone-pickaxe"]      = { "item", link="Pickaxe", cannot_stack=true }, ["string"]             = { "item", plural=false }, ["wheat"]              = { "item", plural=false }, ["wood-planks"]        = { "block", title="Any Wood Planks", plural=false }, ["wooden-axe"]         = { "item", link="Axe", cannot_stack=true }, ["wooden-pickaxe"]     = { "item", link="Pickaxe", cannot_stack=true },

["enchanted-diamond-boots"]     = { "item", id="diamond-boots", link="Diamond Boots", note="enchantment2", cannot_stack=true, plural=false }, ["enchanted-diamond-chestplate"] = { "item", id="diamond-chestplate", link="Diamond Chestplate", note="enchantment2", cannot_stack=true }, ["enchanted-diamond-helmet"]    = { "item", id="diamond-helmet", link="Diamond Helmet", note="enchantment2", cannot_stack=true }, ["enchanted-diamond-leggings"]  = { "item", id="diamond-leggings", link="Diamond Leggings", note="enchantment2", cannot_stack=true, plural=false }, ["enchanted-diamond-pickaxe"]   = { "item", id="diamond-pickaxe", link="Diamond Pickaxe", note="enchantment2", cannot_stack=true }, ["enchanted-diamond-shovel"]    = { "item", id="diamond-shovel", link="Diamond Shovel", note="enchantment2", cannot_stack=true }, ["enchanted-diamond-sword"]     = { "item", id="diamond-sword", link="Diamond Sword", note="enchantment2", cannot_stack=true }, ["enchanted-iron-boots"]        = { "item", id="iron-boots", link="Iron Boots", note="enchantment2", cannot_stack=true, plural=false }, ["enchanted-iron-chestplate"]   = { "item", id="iron-chestplate", link="Iron Chestplate", note="enchantment2", cannot_stack=true }, ["enchanted-iron-helmet"]       = { "item", id="iron-helmet", link="Iron Helmet", note="enchantment2", cannot_stack=true }, ["enchanted-iron-leggings"]     = { "item", id="iron-leggings", link="Iron Leggings", note="enchantment2", cannot_stack=true, plural=false }, ["enchanted-iron-pickaxe"]      = { "item", id="iron-pickaxe", link="Iron Pickaxe", note="enchantment2", cannot_stack=true }, ["enchanted-iron-shovel"]       = { "item", id="iron-shovel", link="Iron Shovel", note="enchantment2", cannot_stack=true }, ["enchanted-iron-sword"]        = { "item", id="iron-sword", link="Iron Sword", note="enchantment2", cannot_stack=true }, },

notes = { ["enchantment1"] = " ", ["enchantment2"] = " ", },	-- NOTE: order here doesn't matter. --		 * in the table, chests will sort in alphabetical order --      * in the table, items will sort by chance, then by avg#, then alphabetically. chests = { ["village-blacksmith"] = { numstacks  = {3,8}, totalweight = nil, header     = "Village", link       = "village blacksmith", items = { ["diamond"]            = {1,3,3}, ["iron-ingot"]         = {1,5,10}, ["gold-ingot"]         = {1,3,5}, ["bread"]              = {1,3,15}, ["apple"]              = {1,3,15}, ["iron-pickaxe"]       = {1,1,5}, ["iron-sword"]         = {1,1,5}, ["iron-chestplate"]    = {1,1,5}, ["iron-helmet"]        = {1,1,5}, ["iron-leggings"]      = {1,1,5}, ["iron-boots"]         = {1,1,5}, ["obsidian"]           = {3,7,5}, ["oak-sapling"]        = {3,7,5}, ["horse-armor-iron"]   = {1,1,1}, ["horse-armor-gold"]   = {1,1,1}, ["horse-armor-diamond"] = {1,1,1}, ["saddle"]             = {1,1,3} }		},		["stronghold-altar"] = { numstacks  = {2,3}, totalweight = nil, header     = "Altar", superheader = "Stronghold", link       = "stronghold altar", items      = { ["diamond"]            = {1,3,3}, ["iron-ingot"]         = {1,5,10}, ["gold-ingot"]         = {1,3,5}, ["bread"]              = {1,3,15}, ["apple"]              = {1,3,15}, ["iron-pickaxe"]       = {1,1,5}, ["iron-sword"]         = {1,1,5}, ["iron-chestplate"]    = {1,1,5}, ["iron-helmet"]        = {1,1,5}, ["iron-leggings"]      = {1,1,5}, ["iron-boots"]         = {1,1,5}, ["horse-armor-iron"]   = {1,1,1}, ["horse-armor-gold"]   = {1,1,1}, ["horse-armor-diamond"] = {1,1,1}, ["ender-pearl"]        = {1,1,10}, ["redstone"]           = {4,9,5}, ["golden-apple"]       = {1,1,1}, ["saddle"]             = {1,1,1}, ["enchanted-book"]     = {1,1,1} }		},		["stronghold-library"] = { numstacks  = {2,10}, totalweight = nil, header     = "Library", superheader = "Stronghold", link       = "stronghold library", items      = { ["enchanted-book"]     = {1,1,10}, ["book"]               = {1,3,20}, ["paper"]              = {2,7,20}, ["empty-map"]          = {1,1,1}, ["compass"]            = {1,1,1} }		},		["stronghold-storeroom"] = { numstacks  = {1,4}, totalweight = nil, header     = "Storeroom", superheader = "Stronghold", link       = "stronghold storeroom", items      = { ["iron-ingot"]         = {1,5,10}, ["gold-ingot"]         = {1,3,5}, ["bread"]              = {1,3,15}, ["apple"]              = {1,3,15}, ["iron-pickaxe"]       = {1,1,1}, ["redstone"]           = {4,9,5}, ["enchanted-book"]     = {1,1,1}, ["coal"]               = {3,8,10} }		},		["bonus"] = { numstacks  = {10,10}, totalweight = nil, header     = "Bonus", link       = "bonus", items      = { ["bread"]              = {2,3,3}, ["apple"]              = {2,3,5}, ["stick"]              = {1,3,10}, ["wood-planks"]        = {1,3,10}, ["oak-wood"]           = {1,3,10}, ["acacia-wood"]        = {1,3,10}, ["stone-axe"]          = {1,1,3}, ["wooden-axe"]         = {1,1,5}, ["stone-pickaxe"]      = {1,1,3}, ["wooden-pickaxe"]     = {1,1,5} }		},		["dungeon"] = { numstacks  = {8,8}, totalweight = nil, header     = "Dungeon", link       = "dungeon", items      = { ["iron-ingot"]         = {1,4,10}, ["bread"]              = {1,1,10}, ["horse-armor-iron"]   = {1,1,5}, ["horse-armor-gold"]   = {1,1,2}, ["horse-armor-diamond"] = {1,1,1}, ["redstone"]           = {1,4,10}, ["golden-apple"]       = {1,1,1}, ["saddle"]             = {1,1,10}, ["wheat"]              = {1,4,10}, ["gunpowder"]          = {1,4,10}, ["string"]             = {1,4,10}, ["bucket"]             = {1,1,10}, ["disc-13"]            = {1,1,4}, ["disc-cat"]           = {1,1,4}, ["name-tag"]           = {1,1,10}, ["enchanted-book"]     = {1,1,1} }		},		["mineshaft"] = { numstacks  = {3,6}, totalweight = nil, header     = "Mineshaft", link       = "abandoned mine shafts", items      = { ["diamond"]            = {1,2,3}, ["iron-ingot"]         = {1,5,10}, ["gold-ingot"]         = {1,3,5}, ["bread"]              = {1,3,15}, ["iron-pickaxe"]       = {1,1,1}, ["horse-armor-iron"]   = {1,1,1}, ["redstone"]           = {4,9,5}, ["lapis-lazuli"]       = {4,9,5}, ["saddle"]             = {1,1,3}, ["enchanted-book"]     = {1,1,1}, ["coal"]               = {3,8,10}, ["rail"]               = {4,8,1}, ["melon-seeds"]        = {2,4,10}, ["pumpkin-seeds"]      = {2,4,10} }		},		["nether-fortress"] = { numstacks  = {2,4}, totalweight = nil, header     = "Nether fortress", link       = "nether fortress", items      = { ["diamond"]            = {1,3,5}, ["iron-ingot"]         = {1,5,5}, ["gold-ingot"]         = {1,3,15}, ["obsidian"]           = {2,4,2}, ["horse-armor-iron"]   = {1,1,5}, ["horse-armor-gold"]   = {1,1,8}, ["horse-armor-diamond"] = {1,1,3}, ["saddle"]             = {1,1,10}, ["golden-sword"]       = {1,1,5}, ["golden-chestplate"]  = {1,1,5}, ["flint-and-steel"]    = {1,1,5}, ["nether-wart"]        = {3,7,5} }		},		["desert-and-jungle-temple"] = { numstacks  = {2,6}, totalweight = nil, header     = "Desert & jungle temple", link       = "desert and jungle temple", items      = { ["diamond"]            = {1,3,3}, ["iron-ingot"]         = {1,5,10}, ["gold-ingot"]         = {2,7,15}, ["emerald"]            = {1,3,2}, ["horse-armor-iron"]   = {1,1,1}, ["horse-armor-gold"]   = {1,1,1}, ["horse-armor-diamond"] = {1,1,1}, ["saddle"]             = {1,1,3}, ["enchanted-book"]     = {1,1,1}, ["bone"]               = {4,6,20}, ["rotten-flesh"]       = {3,7,16} }		},		["end-city"] = { numstacks  = {2,6}, totalweight = nil, header     = "End city", link       = "End city", items      = { ["diamond"]            = {2,7,5}, ["iron-ingot"]         = {4,8,10}, ["gold-ingot"]         = {2,7,15}, ["emerald"]            = {2,6,2}, ["beetroot-seeds"]     = {1,10,5}, ["saddle"]             = {1,1,3}, ["horse-armor-iron"]   = {1,1,1}, ["horse-armor-gold"]   = {1,1,1}, ["horse-armor-diamond"] = {1,1,1}, ["enchanted-diamond-sword"]     = {1,1,3}, ["enchanted-diamond-boots"]     = {1,1,3}, ["enchanted-diamond-chestplate"] = {1,1,3}, ["enchanted-diamond-leggings"]  = {1,1,3}, ["enchanted-diamond-helmet"]    = {1,1,3}, ["enchanted-diamond-pickaxe"]   = {1,1,3}, ["enchanted-diamond-shovel"]    = {1,1,3}, ["enchanted-iron-sword"]        = {1,1,3}, ["enchanted-iron-boots"]        = {1,1,3}, ["enchanted-iron-chestplate"]   = {1,1,3}, ["enchanted-iron-leggings"]     = {1,1,3}, ["enchanted-iron-helmet"]       = {1,1,3}, ["enchanted-iron-pickaxe"]      = {1,1,3}, ["enchanted-iron-shovel"]       = {1,1,3}, }		},		["igloo"] = { numstacks  = {2,8}, totalweight = nil, header     = "Igloo", link       = "igloo", items      = { ["apple"]              = {1,3,15}, ["coal"]               = {1,4,15}, ["gold-nugget"]        = {1,3,10}, ["stone-axe"]          = {1,1,2}, ["rotten-flesh"]       = {1,1,10}, ["emerald"]            = {1,1,1}, ["wheat"]              = {2,3,10} }		}	},

-- these values are used: -- * in place of the keys, when the key is used as a parameter -- chest-param -> internally-valid-chest-param synonyms = { ["desert"] = "desert-and-jungle-temple", ["jungle"] = "desert-and-jungle-temple", ["desert-temple"] = "desert-and-jungle-temple", ["jungle-temple"] = "desert-and-jungle-temple", ["nether"] = "nether-fortress", ["fortress"] = "nether-fortress", ["village"] = "village-blacksmith", ["blacksmith"] = "village-blacksmith", ["altar"] = "stronghold-altar", ["storeroom"] = "stronghold-storeroom", ["library"] = "stronghold-library" },

-- these values are used: -- * in the header-description of a table showing only a single chest -- * if the key is not here, but it is a valid chest parameter, --  that header-description will default to use the key string from p.chests, --  e.g. "nether-fortress" -- chest-param -> description-string display_names = { ["nether-fortress"] = "Nether fortress", ["nether"] = "Nether fortress", ["fortress"] = "Nether fortress", ["desert"] = "desert temple", ["jungle"] = "jungle temple", ["desert-temple"] = "desert temple", ["jungle-temple"] = "jungle temple" },	-- these descriptions are used: -- * in column titles, -- * and above the table when only a single column-type is chosen columns = { ["stacksize"] = 'the size of stacks (or for unstackable items, number) of this item found in this chest.', ["weight"] = 'the weight of this item relative to other items in the chest.', ["items"] = 'the number of items expected per chest, averaged over a large number of chests.', ["chance"] = 'the odds of getting any of this item in a single chest.', ["chests"] = 'the number of chests expected to search, to find any of this item.' },	current_frame = nil }

p.base = function( ... )

p.current_frame = mw.getCurrentFrame

local args = arg[1] if args == p.current_frame then args = require( 'Module:ProcessArgs' ).merge( true ) else args = arg end -- transform args into usable list local chests, columns = q.massage_args( args ) if #chests == 0 then return " Module:LootChest: no valid arguments " end

q.fill_in_chest_total_weights( chests ) q.fill_in_chest_item_details( chests ) -- construct an ordered list dictating the order of the rows local ordered_item_rows = q.construct_ordered_item_rows( chests ) return q.print_table( chests, columns, ordered_item_rows )

end

p.doc = function

local valid_args = {} for chest_name, val in pairs(p.chests) do		local synonyms = {} for syn, orig in pairs(p.synonyms) do			if orig == chest_name then table.insert( synonyms, syn ) end end if #synonyms > 0 then chest_name = chest_name .. " ( " .. table.concat( synonyms, ", " ) .. " )" end table.insert( valid_args, chest_name ) end table.sort( valid_args ) return table.concat( valid_args, ",\n " )

end

p.doc2 = function

local valid_args = {} for column_name, val in pairs(p.columns) do table.insert( valid_args, column_name .. ": " .. val ) end table.sort( valid_args ) return table.concat( valid_args, ",\n " )

end

p.doc3 = function

local valid_args = {} for item_name, val in pairs(p.items) do		table.insert( valid_args, item_name ) end table.sort( valid_args ) return table.concat( valid_args, ", " )

end

p.base2 = function( ... )

p.current_frame = mw.getCurrentFrame

local args = arg[1] if args == p.current_frame then args = require( 'Module:ProcessArgs' ).merge( true ) else args = arg end local itemname = args[1] if p.items[itemname] == nil then return ' unknown item "' .. itemname .. '" ' end local chances = q.single_item_find_values( itemname ) --	local html = {} local item_display_name = '' if p.items[itemname].plural ~= nil and p.items[itemname].plural ~= false then item_display_name = p.items[itemname].plural else if p.items[itemname].title ~= nil then item_display_name = p.items[itemname].title else item_display_name = string.gsub( itemname, '-', ' ' ) end if p.items[itemname].plural == nil or p.items[itemname].plural ~= false then item_display_name = q.single_item_plural( item_display_name ) end end item_display_name = q.capitalize( item_display_name ) table.insert( html, item_display_name ) table.insert( html, ' can be found ' )

local html_stacks = {} for stacksize, chest_details in pairs( chances ) do		local html_per_stack = { 'in ' } local c = 0 local nc = q.tablelength( chest_details ) for k, chest in pairs( chest_details ) do			c = c + 1 if q.tablelength( chances ) == 1 and c == nc and nc > 1 then table.insert( html_per_stack, 'and ' ) end table.insert( html_per_stack, string.format("%.1f", chest.chance*100) ) table.insert( html_per_stack, "% of " ) if chest.chest_name == 'mineshaft' then table.insert( html_per_stack, ' chest minecarts in ' ) table.insert( html_per_stack, p.chests[chest.chest_name].link ) table.insert( html_per_stack, ', ' ) else table.insert( html_per_stack, p.chests[chest.chest_name].link ) table.insert( html_per_stack, ' chests, ' ) end end if #chest_details > 1 then table.insert( html_per_stack, 'all ' ) end table.insert( html_per_stack, 'in ' ) if p.items[itemname].cannot_stack ~= nil then table.insert( html_per_stack, 'groups of ' ) else table.insert( html_per_stack, 'stacks of ' ) end table.insert( html_per_stack, stacksize ) table.insert( html_stacks, table.concat( html_per_stack ) ) end local stackwise_summaries = '' if #html_stacks == 1 then table.insert( html, html_stacks[1] ) else for i = 1, #html_stacks - 1 do			table.insert( html, html_stacks[ i ] ) table.insert( html, ', ' ) end table.insert( html, 'and ' ) table.insert( html, html_stacks[#html_stacks] ) end table.insert( html, '.' ) -- 	return table.concat( html ) end

p.base2_test = function

items = {} for item_name, v in pairs( p.items ) do table.insert( items, p.base2(item_name) .. '\n\n' ) end

table.sort( items ) return table.concat(items) end

q = {

tablelength = function(T) local count = 0 for _ in pairs(T) do count = count + 1 end return count end, single_item_find_values = function( itemname ) local chances = {}

for chest_name, chest in pairs( p.chests ) do			local chestitem = chest.items[itemname] if chestitem ~= nil then local stacksize = chestitem[1] if chestitem[1] ~= chestitem[2] then stacksize = stacksize .. "–" .. chestitem[2] end if chances[stacksize] == nil then chances[stacksize] = {} end local itemweight = chestitem[3] local chest_total_item_weight = 0 for itemname, item in pairs(chest.items) do					chest_total_item_weight = chest_total_item_weight + item[3] end local chance = p.calc_chance_any_of_this_item_per_chest( 					chest.numstacks[1], chest.numstacks[2],					itemweight, chest_total_item_weight ) table.insert( chances[stacksize], { ["chance"]=chance, ["chest_name"]=chest_name } ) end end return chances end, single_item_plural = function( itemname ) if string.sub( itemname, -2 ) == 'ss' or string.sub( itemname, -2 ) == 'ch' or string.sub( itemname, -2 ) == 'sh' or string.sub( itemname, -1 ) == 's' then return itemname .. 'es' end return itemname .. 's'	end, massage_args = function( args )

-- find what columns to put local columns = {} for k, _arg in pairs(args) do			if p.columns[_arg] ~= nil then columns[_arg] = true end end if q.tablelength(columns) == 0 then for column_name, v in pairs(p.columns) do				columns[column_name] = true end end -- find what chests to show

local chests = {} for k, _arg in pairs(args) do			if p.chests[_arg] ~= nil then table.insert( chests, _arg ) elseif p.synonyms[_arg] ~= nil then table.insert( chests, p.synonyms[_arg] ) end if p.display_names[_arg] ~= nil then local chestname = _arg if p.chests[chestname] == nil then chestname = p.synonyms[_arg] end p.chests[chestname].display_name = p.display_names[_arg] end end if #chests == 0 then for chest_name, chest in pairs(p.chests) do				table.insert( chests, chest_name ) end end table.sort( chests ) return chests, columns end,

sort_items = function( e1, e2 ) if e1.chanceany ~= e2.chanceany then return ( e1.chanceany > e2.chanceany ) end if e1.avgamount ~= e2.avgamount then return ( e1.avgamount > e2.avgamount ) end if e1.material == nil then e1.material = 0 if string.find( e1.itemname, "leather" ) ~= nil then e1.material = 1 end if string.find( e1.itemname, "iron" ) ~= nil then e1.material = 2 end if string.find( e1.itemname, "gold" ) ~= nil then e1.material = 3 end if string.find( e1.itemname, "diamond" ) ~= nil then e1.material = 4 end e1.armor = 0 if string.find( e1.itemname, "helmet" ) ~= nil or string.find( e1.itemname, "cap" ) ~= nil then e1.armor = 1 end if string.find( e1.itemname, "chestplate" ) ~= nil or string.find( e1.itemname, "tunic" ) ~= nil then e1.armor = 2 end if string.find( e1.itemname, "leggings" ) ~= nil or string.find( e1.itemname, "pants" ) ~= nil then e1.armor = 3 end if string.find( e1.itemname, "boots" ) ~= nil then e1.armor = 4 end end if e2.material == nil then e2.material = 0 if string.find( e2.itemname, "leather" ) ~= nil then e2.material = 1 end if string.find( e2.itemname, "iron" ) ~= nil then e2.material = 2 end if string.find( e2.itemname, "gold" ) ~= nil then e2.material = 3 end if string.find( e2.itemname, "diamond" ) ~= nil then e2.material = 4 end e2.armor = 0 if string.find( e2.itemname, "helmet" ) ~= nil or string.find( e2.itemname, "cap" ) ~= nil then e2.armor = 1 end if string.find( e2.itemname, "chestplate" ) ~= nil or string.find( e2.itemname, "tunic" ) ~= nil then e2.armor = 2 end if string.find( e2.itemname, "leggings" ) ~= nil or string.find( e2.itemname, "pants" ) ~= nil then e2.armor = 3 end if string.find( e2.itemname, "boots" ) ~= nil then e2.armor = 4 end end if e1.material ~= e2.material then return ( e1.material < e2.material ) end if e1.armor ~= e2.armor then return ( e1.armor < e2.armor ) end return ( e1.itemname < e2.itemname ) end, fill_in_chest_total_weights = function( chest_names )

for k, chest_name in pairs(chest_names) do			local chest = p.chests[chest_name] if chest == nil then break end local total_weight = 0 for itemname, item in pairs(chest.items) do				total_weight = total_weight + item[3] end chest.totalweight = total_weight end

end,

fill_in_chest_item_details = function( chest_names )

for key, chest_name in pairs(chest_names) do			local chest = p.chests[chest_name] if chest == nil then break end for item_name, item in pairs(chest.items) do				local min_stacksize = item[1] local max_stacksize = item[2] local min_chest_numstacks = chest.numstacks[1] local max_chest_numstacks = chest.numstacks[2] local item_weight = item[3] p.chests[chest_name].items[item_name].avgamount = p.calc_average_amount_this_item_per_chest( 						min_stacksize, max_stacksize, 						min_chest_numstacks, max_chest_numstacks, 						item_weight, chest.totalweight ) p.chests[chest_name].items[item_name].chanceany = p.calc_chance_any_of_this_item_per_chest( 						min_chest_numstacks, max_chest_numstacks, 						item_weight, chest.totalweight ) p.chests[chest_name].items[item_name].numchests = p.calc_num_chests_to_find_this_item( 						min_chest_numstacks, max_chest_numstacks, 						item_weight, chest.totalweight ) p.chests[chest_name].items[item_name].itemname = item_name -- for sorting later end end end,

construct_ordered_items_from_first_chest = function( chest_names )

local items_from_first_table = {} local item_chests = {} local item_names_ordered = {} for item_name, item in pairs( p.chests[chest_names[1]].items ) do			table.insert( items_from_first_table, item ) end table.sort( items_from_first_table, q.sort_items ) for k, item in pairs( items_from_first_table ) do			table.insert( item_names_ordered, item.itemname ) item_chests[item.itemname] = true end return item_names_ordered, item_chests

end,

get_ordered_items_from_other_chests = function( chest_names, item_chests )

local items_not_from_first_table = {} for chest_idx = 2, #chest_names do			for item_name, item in pairs( p.chests[chest_names[chest_idx]].items ) do				if item_chests[item_name] == nil then p.items[item_name].itemname = item_name table.insert( items_not_from_first_table, p.chests[chest_names[chest_idx]].items[item_name] ) item_chests[item_name] = true end end end

table.sort( items_not_from_first_table, q.sort_items ) return items_not_from_first_table end,

add_other_items_to_first_list = function( chest_names, item_names_ordered, item_chests, items_not_from_first_table )

for k, item in pairs( items_not_from_first_table ) do			table.insert( item_names_ordered, item.itemname ) end return item_names_ordered end,

set_up_ordered_item_rows = function( chest_names, item_names_ordered )

for k, itemname in pairs(item_names_ordered) do			item_names_ordered[k] = {itemname} for chest_idx = 1, #chest_names do				local item_data = p.chests[chest_names[chest_idx]].items[itemname] if item_data == nil then table.insert( item_names_ordered[k], false ) else table.insert( item_names_ordered[k], item_data ) end end end

return item_names_ordered end,

construct_ordered_item_rows = function( chest_names )

-- for the first chest, sort its by chance desc, then by avg amount desc, then alphabetically asc local item_names_ordered, item_chests = q.construct_ordered_items_from_first_chest( chest_names ) if #chest_names > 1 then -- after that, sort all the remaining items in list order local items_not_from_first_table = q.get_ordered_items_from_other_chests( chest_names, item_chests ) item_names_ordered = q.add_other_items_to_first_list( chest_names, item_names_ordered, item_chests, items_not_from_first_table ) end

-- set up item_names_ordered so that each is a row, representing chest values item_names_ordered = q.set_up_ordered_item_rows( chest_names, item_names_ordered ) return item_names_ordered end,

print_table = function( chest_names, columns, ordered_item_rows )

local html = {} local use_superheader = false local superheader_sizes = {} for i = 1, #chest_names do			sh = p.chests[chest_names[i]].superheader if sh ~= nil then if superheader_sizes[sh] == nil then superheader_sizes[sh] = 0 end superheader_sizes[sh] = superheader_sizes[sh] + 1 use_superheader = true end end local rowspan = ( 1 + q.tern( #chest_names > 1, 1, 0 ) + q.tern( use_superheader, 1, 0 ) ) local hide_col_description = rowspan > 1 and q.tablelength(columns) == 1 if q.tablelength(columns) == 1 then for column_name, v in pairs(columns) do				table.insert( html, "Values represent " ) table.insert( html, p.columns[column_name] ) table.insert( html, "\n" ) end end

if #chest_names == 1 then if q.tablelength(columns) == 1 then table.insert( html, " " ) end local chest_name = chest_names[1] local chest_min = p.chests[chest_name].numstacks[1] local chest_max = p.chests[chest_name].numstacks[2]

local display_name = p.chests[chest_name].display_name

chest_name = chest_name:gsub( "-", " " ) table.insert( html, "Each " ) table.insert( html, display_name or chest_name ) table.insert( html, " chest contains " ) table.insert( html, q.tern( chest_min == chest_max, chest_min, "between " .. chest_min .. " and " .. chest_max ) ) table.insert( html, " item stacks, with the following distribution: \n" ) end table.insert( html, " " ) return table.concat( html ) end, titlecase = function( str ) local buf = {} for word in string.gfind(str, "%S+") do			if word == "and" then table.insert( buf, word ) else local first, rest = string.sub( word, 1, 1 ), string.sub( word, 2 ) table.insert( buf, string.upper(first) .. string.lower(rest) ) end end return table.concat( buf, " " ) end, capitalize = function( str ) return ( string.lower(str):gsub( "^%l", string.upper ) ) end, tern = function( cond, T , F ) if cond then return T else return F end end }

string.lpad = function(str, len, char) if char == nil then char = ' ' end return string.rep(char, len - #(''..str)) .. str end

return p