Categories: Improve automatic ore categorization

This function is now executed after registering all recipes
within unified_inventory to properly register all item drops
as ores.

category.lua

@@ -115,6 +115,11 @@ function unified_inventory.set_category_index(category_name, index)
 	update_category_list()
 end
 function unified_inventory.add_category_item(category_name, item)
+	if type(item) ~= "string" then
+		minetest.log("warning", "[unified_inventory] Cannot register category item: " .. dump(item))
+		return
+	end
+
 	ensure_category_exists(category_name)
 	unified_inventory.registered_category_items[category_name][item] = true
 end

default-categories.lua

@@ -1,4 +1,5 @@
 local S = minetest.get_translator("unified_inventory")
+local ui = unified_inventory
 
 unified_inventory.register_category('plants', {
 	symbol = "flowers:tulip",
@@ -25,74 +26,87 @@ unified_inventory.register_category('lighting', {
 	label = S("Lighting")
 })
 
-
+local function register_automatic_categorization()
-if unified_inventory.automatic_categorization then
+	-- Add biome nodes to environment category
-	minetest.register_on_mods_loaded(function()
+	for _,def in pairs(minetest.registered_biomes) do
-
+		local env_nodes = {
-		-- Add biome nodes to environment category
+			def.node_riverbed, def.node_top, def.node_filler, def.node_dust,
-		for _,def in pairs(minetest.registered_biomes) do
+		}
-			local env_nodes = {
+		for i,node in pairs(env_nodes) do
-				def.node_riverbed, def.node_top, def.node_filler, def.node_dust,
+			if node then
-			}
+				unified_inventory.add_category_item('environment', node)
-			for i,node in pairs(env_nodes) do
-				if node then
-					unified_inventory.add_category_item('environment', node)
-				end
 			end
 		end
+	end
 
-		-- Add minable ores to minerals and everything else (pockets of stone & sand variations) to environment
+	-- Preparation for ore registration: find all possible drops (digging)
-		for _,item in  pairs(minetest.registered_ores) do
+	local possible_node_dig_drops = {
-			if item.ore_type == "scatter" then
+		-- ["default:stone_with_coal"] = { "default:coal_lump", "mymod:raregem" }
-				-- The NodeResolver is run *after* minetest.register_on_mods_loaded, thus the
+		-- Ores may be contained multiple times, depending on drop chances.
-				-- existence of ore names were yet not checked or enforced.
+	}
-				local def = minetest.registered_nodes[item.ore] or {}
+	for itemname, recipes in pairs(ui.crafts_for.usage) do
-				local drop = def.drop
+		for _, recipe in ipairs(recipes) do
-				if drop and drop ~= "" then
+			if recipe.type == "digging" or recipe.type == "digging_chance" then
-					unified_inventory.add_category_item('minerals', item.ore)
+				if not possible_node_dig_drops[itemname] then
-					unified_inventory.add_category_item('minerals', drop)
+					possible_node_dig_drops[itemname] = {}
-				else
-					unified_inventory.add_category_item('environment', item.ore)
 				end
-			else
+				local stack = ItemStack(recipe.output)
-				unified_inventory.add_category_item('environment', item.ore)
+				table.insert(possible_node_dig_drops[itemname], stack:get_name())
 			end
 		end
+	end
 
-		-- Add items by item definition
+	-- Add minable ores to minerals and everything else (pockets of stone & sand variations) to environment
-		for name, def in pairs(minetest.registered_items) do
+	for _, odef in pairs(minetest.registered_ores) do
-			local group = def.groups or {}
+		local drops = possible_node_dig_drops[odef.ore]
-			if not group.not_in_creative_inventory then
+		if drops and odef.ore_type == "scatter" then
-				if group.stair or
+			ui.add_category_item('minerals', odef.ore)
-				   group.slab or
+			-- Register all possible drops as "minerals"
-				   group.wall or
+			ui.add_category_items('minerals', drops)
-				   group.fence then
+			possible_node_dig_drops[odef.ore] = {} -- mask as handled
-					unified_inventory.add_category_item('building', name)
+		else
-				elseif group.flora or
+			ui.add_category_item('environment', odef.ore)
-					   group.flower or
+		end
-					   group.seed or
+	end
-					   group.leaves or
+
-					   group.sapling or
+	-- Add items by item definition
-					   group.tree then
+	for name, def in pairs(minetest.registered_items) do
-					unified_inventory.add_category_item('plants', name)
+		local group = def.groups or {}
-				elseif def.type == 'tool' then
+		if not group.not_in_creative_inventory then
-					unified_inventory.add_category_item('tools', name)
+			if group.stair or
-				elseif def.liquidtype == 'source' then
+			   group.slab or
-					unified_inventory.add_category_item('environment', name)
+			   group.wall or
-				elseif def.light_source and def.light_source > 0 then
+			   group.fence then
-					unified_inventory.add_category_item('lighting', name)
+				unified_inventory.add_category_item('building', name)
-				elseif group.door or
+			elseif group.flora or
-					   minetest.global_exists("doors") and (
+				   group.flower or
-					     doors.registered_doors and doors.registered_doors[name..'_a'] or
+				   group.seed or
-					     doors.registered_trapdoors and doors.registered_trapdoors[name]
+				   group.leaves or
-					   ) then
+				   group.sapling or
-					unified_inventory.add_category_item('building', name)
+				   group.tree then
-				end
+				unified_inventory.add_category_item('plants', name)
+			elseif def.type == 'tool' then
+				unified_inventory.add_category_item('tools', name)
+			elseif def.liquidtype == 'source' then
+				unified_inventory.add_category_item('environment', name)
+			elseif def.light_source and def.light_source > 0 then
+				unified_inventory.add_category_item('lighting', name)
+			elseif group.door or
+				   minetest.global_exists("doors") and (
+					 doors.registered_doors and doors.registered_doors[name..'_a'] or
+					 doors.registered_trapdoors and doors.registered_trapdoors[name]
+				   ) then
+				unified_inventory.add_category_item('building', name)
 			end
 		end
-	end)
+	end
 end
 
+if ui.automatic_categorization then
+	ui.register_on_initialized(register_automatic_categorization)
+end
+
+
 -- [[
 unified_inventory.add_category_items('plants', {
 	"default:dry_grass_5",
@@ -259,23 +273,6 @@ unified_inventory.add_category_items('minerals', {
 	"default:coal_lump",
 	"default:bronzeblock",
 	"default:goldblock",
-
-	"stairs:slab_bronzeblock",
-	"stairs:slab_copperblock",
-	"stairs:slab_steelblock",
-	"stairs:slab_tinblock",
-	"stairs:stair_bronzeblock",
-	"stairs:stair_copperblock",
-	"stairs:stair_inner_bronzeblock",
-	"stairs:stair_inner_copperblock",
-	"stairs:stair_inner_steelblock",
-	"stairs:stair_inner_tinblock",
-	"stairs:stair_outer_bronzeblock",
-	"stairs:stair_outer_copperblock",
-	"stairs:stair_outer_steelblock",
-	"stairs:stair_outer_tinblock",
-	"stairs:stair_steelblock",
-	"stairs:stair_tinblock",
 })
 
 unified_inventory.add_category_items('building', {

register.lua

@@ -215,7 +215,7 @@ local function stack_image_button(x, y, w, h, buttonname_prefix, item)
 		local group_name = name:sub(7)
 		local group_item = ui.get_group_item(group_name)
 		show_is_group = not group_item.sole
-		displayitem = group_item.item or "unknown"
+		displayitem = group_item.item or name
 		selectitem = group_item.sole and displayitem or name
 	end
 	local label = show_is_group and "G" or ""