started to add treecapitator

mods/treecapitator/api.lua

@@ -0,0 +1,94 @@
+-- the table containing the tree definitions
+treecapitator.trees = {}
+
+-- Warn if a tree definition passed to treecapitator.register_tree contains
+-- invalid options, partly for backwards compatibility.
+-- Not all options are checked.
+local function check_tree_definition(tr)
+	if type(tr.trees) ~= "table" then
+		minetest.log("warning",
+			"[treecapitator] Invalid tree definition (trees field)")
+	end
+	if tr.type ~= "acacia" then
+		if type(tr.leaves) ~= "table" then
+			minetest.log("warning",
+				"[treecapitator] Invalid tree definition (leaves field)")
+		end
+		if type(tr.range) ~= "number" then
+			minetest.log("warning",
+				"[treecapitator] Invalid tree definition (range field)")
+		end
+	end
+end
+
+-- For the usage of this function, see trees.lua.
+local after_dig_wrap
+local after_dig_nodes = {}
+function treecapitator.register_tree(tree_definition)
+	check_tree_definition(tree_definition)
+	local tr = {}
+	for name, value in pairs(tree_definition) do
+		tr[name] = value
+	end
+	tr.fruits = tr.fruits or {}
+	tr.type = tr.type or "default"
+	treecapitator.trees[#treecapitator.trees+1] = tr
+	if treecapitator.after_register[tr.type] then
+		treecapitator.after_register[tr.type](tr)
+	end
+
+	for i = 1,#tr.trees do
+		local nodename = tr.trees[i]
+		local data = minetest.registered_nodes[nodename]
+		if not data then
+			error(nodename .. " has to be registered before calling " ..
+				"treecapitator.register_tree.")
+		end
+		local func = after_dig_wrap
+		local prev_after_dig = data.after_dig_node
+		if prev_after_dig
+		and not after_dig_nodes[nodename] then
+			func = function(pos, oldnode, oldmetadata, digger)
+				prev_after_dig(pos, oldnode, oldmetadata, digger)
+				treecapitator.capitate_tree(pos, digger)
+			end
+		end
+		minetest.override_item(nodename, {after_dig_node = func})
+		after_dig_nodes[nodename] = true
+	end
+end
+
+function treecapitator.capitation_allowed()
+	return not treecapitator.capitation_usually_disallowed
+end
+
+
+-- Example of overriding this function
+if treecapitator.no_hand_capitation then
+	-- disallow capitating trees if no proper tool is used
+	treecapitator.capitation_usually_disallowed = true
+	local allowed = treecapitator.capitation_allowed
+	function treecapitator.capitation_allowed(pos, digger)
+		local def = minetest.registered_nodes[
+			minetest.get_node{x=pos.x, y=pos.y+1, z=pos.z}.name
+		]
+		return def and def.groups and minetest.get_dig_params(def.groups,
+			digger:get_wielded_item():get_tool_capabilities()).wear > 0
+			or allowed(pos, digger)
+	end
+end
+
+-- test if trunk nodes were redefined
+minetest.after(2, function()
+	for nodename in pairs(after_dig_nodes) do
+		if not minetest.registered_nodes[nodename].after_dig_node then
+			error(nodename .. " didn't keep after_dig_node.")
+		end
+	end
+	after_dig_nodes = nil
+end)
+
+-- wrapping is necessary, someone may overwrite treecapitator.capitate_tree
+function after_dig_wrap(pos, _,_, digger)
+	treecapitator.capitate_tree(pos, digger)
+end