Default: Add biome API tree schematics and enable growing by sapling ABMs
[oweals/minetest_game.git] / mods / default / trees.lua
index 5752fe0de23e1515d12615bc72c59251c8c45b66..b5b20ec53b584ae2ce569990fd669402ec77a7e3 100644 (file)
@@ -17,6 +17,8 @@ local function can_grow(pos)
        return true
 end
 
+-- Sapling ABMs
+
 minetest.register_abm({
        nodenames = {"default:sapling"},
        interval = 10,
@@ -26,14 +28,19 @@ minetest.register_abm({
                        return
                end
 
-               minetest.log("action", "A sapling grows into a tree at "..minetest.pos_to_string(pos))
-               default.grow_tree(pos, random(1, 4) == 1)
+               minetest.log("action", "A sapling grows into a tree at "..
+                       minetest.pos_to_string(pos))
+               if minetest.get_mapgen_params().mgname == "v6" then
+                       default.grow_tree(pos, random(1, 4) == 1)
+               else
+                       default.grow_new_apple_tree(pos)
+               end
        end
 })
 
 minetest.register_abm({
        nodenames = {"default:junglesapling"},
-       interval = 10,
+       interval = 11,
        chance = 50,
        action = function(pos, node)
                if not can_grow(pos) then
@@ -41,23 +48,64 @@ minetest.register_abm({
                end
 
                minetest.log("action", "A jungle sapling grows into a tree at "..
-                               minetest.pos_to_string(pos))
-               default.grow_jungletree(pos)
+                       minetest.pos_to_string(pos))
+               if minetest.get_mapgen_params().mgname == "v6" then
+                       default.grow_jungle_tree(pos)
+               else
+                       default.grow_new_jungle_tree(pos)
+               end
        end
 })
 
+minetest.register_abm({
+       nodenames = {"default:pine_sapling"},
+       interval = 12,
+       chance = 50,
+       action = function(pos, node)
+               if not can_grow(pos) then
+                       return
+               end
+
+               minetest.log("action", "A pine sapling grows into a tree at "..
+                       minetest.pos_to_string(pos))
+               if minetest.get_mapgen_params().mgname == "v6" then
+                       default.grow_pine_tree(pos)
+               else
+                       default.grow_new_pine_tree(pos)
+               end
+       end
+})
 
-local c_air = minetest.get_content_id("air")
-local c_apple = minetest.get_content_id("default:apple")
+minetest.register_abm({
+       nodenames = {"default:acacia_sapling"},
+       interval = 13,
+       chance = 50,
+       action = function(pos, node)
+               if not can_grow(pos) then
+                       return
+               end
+
+               minetest.log("action", "An acacia sapling grows into a tree at "..
+                       minetest.pos_to_string(pos))
+               default.grow_new_acacia_tree(pos)
+       end
+})
+
+-- Appletree, jungletree function
 
 local function add_trunk_and_leaves(data, a, pos, tree_cid, leaves_cid,
                height, size, iters, is_apple_tree)
        local x, y, z = pos.x, pos.y, pos.z
+       local c_air = minetest.get_content_id("air")
+       local c_ignore = minetest.get_content_id("ignore")
+       local c_apple = minetest.get_content_id("default:apple")
 
        -- Trunk
-       for y_dist = 0, height - 1 do
-               local vi = a:index(x, y + y_dist, z)
-               if y_dist == 0 or data[vi] == c_air or data[vi] == leaves_cid then
+       data[a:index(x, y, z)] = tree_cid -- Force-place lowest trunk node to replace sapling
+       for yy = y + 1, y + height - 1 do
+               local vi = a:index(x, yy, z)
+               local node_id = data[vi]
+               if node_id == c_air or node_id == c_ignore or node_id == leaves_cid then
                        data[vi] = tree_cid
                end
        end
@@ -67,7 +115,7 @@ local function add_trunk_and_leaves(data, a, pos, tree_cid, leaves_cid,
        for y_dist = -size, 1 do
                local vi = a:index(x - 1, y + height + y_dist, z + z_dist)
                for x_dist = -1, 1 do
-                       if data[vi] == c_air then
+                       if data[vi] == c_air or data[vi] == c_ignore then
                                if is_apple_tree and random(1, 8) == 1 then
                                        data[vi] = c_apple
                                else
@@ -89,7 +137,7 @@ local function add_trunk_and_leaves(data, a, pos, tree_cid, leaves_cid,
                for yi = 0, 1 do
                for zi = 0, 1 do
                        local vi = a:index(clust_x + xi, clust_y + yi, clust_z + zi)
-                       if data[vi] == c_air then
+                       if data[vi] == c_air or data[vi] == c_ignore then
                                if is_apple_tree and random(1, 8) == 1 then
                                        data[vi] = c_apple
                                else
@@ -102,9 +150,7 @@ local function add_trunk_and_leaves(data, a, pos, tree_cid, leaves_cid,
        end
 end
 
-
-local c_tree = minetest.get_content_id("default:tree")
-local c_leaves = minetest.get_content_id("default:leaves")
+-- Appletree
 
 function default.grow_tree(pos, is_apple_tree, bad)
        --[[
@@ -118,11 +164,13 @@ function default.grow_tree(pos, is_apple_tree, bad)
 
        local x, y, z = pos.x, pos.y, pos.z
        local height = random(4, 5)
+       local c_tree = minetest.get_content_id("default:tree")
+       local c_leaves = minetest.get_content_id("default:leaves")
 
        local vm = minetest.get_voxel_manip()
        local minp, maxp = vm:read_from_map(
-                       {x = pos.x - 2, y = pos.y, z = pos.z - 2},
-                       {x = pos.x + 2, y = pos.y + height + 1, z = pos.z + 2}
+               {x = pos.x - 2, y = pos.y, z = pos.z - 2},
+               {x = pos.x + 2, y = pos.y + height + 1, z = pos.z + 2}
        )
        local a = VoxelArea:new({MinEdge = minp, MaxEdge = maxp})
        local data = vm:get_data()
@@ -134,26 +182,30 @@ function default.grow_tree(pos, is_apple_tree, bad)
        vm:update_map()
 end
 
-local c_jungletree = minetest.get_content_id("default:jungletree")
-local c_jungleleaves = minetest.get_content_id("default:jungleleaves")
+-- Jungletree
 
-function default.grow_jungletree(pos, bad)
+function default.grow_jungle_tree(pos, bad)
        --[[
-               NOTE: Tree-placing code is currently duplicated in the engine
+               NOTE: Jungletree-placing code is currently duplicated in the engine
                and in games that have saplings; both are deprecated but not
                replaced yet
        --]]
        if bad then
-               error("Deprecated use of default.grow_jungletree")
+               error("Deprecated use of default.grow_jungle_tree")
        end
 
        local x, y, z = pos.x, pos.y, pos.z
        local height = random(8, 12)
+       local c_air = minetest.get_content_id("air")
+       local c_ignore = minetest.get_content_id("ignore")
+       local c_jungletree = minetest.get_content_id("default:jungletree")
+       local c_jungleleaves = minetest.get_content_id("default:jungleleaves")
 
        local vm = minetest.get_voxel_manip()
        local minp, maxp = vm:read_from_map(
-                       {x = pos.x - 3, y = pos.y - 1,  z = pos.z - 3},
-                       {x = pos.x + 3, y = pos.y + height + 1, z = pos.z + 3})
+               {x = pos.x - 3, y = pos.y - 1, z = pos.z - 3},
+               {x = pos.x + 3, y = pos.y + height + 1, z = pos.z + 3}
+       )
        local a = VoxelArea:new({MinEdge = minp, MaxEdge = maxp})
        local data = vm:get_data()
 
@@ -162,12 +214,12 @@ function default.grow_jungletree(pos, bad)
        -- Roots
        for z_dist = -1, 1 do
                local vi_1 = a:index(x - 1, y - 1, z + z_dist)
-               local vi_2 = a:index(x - 1, y,     z + z_dist)
+               local vi_2 = a:index(x - 1, y, z + z_dist)
                for x_dist = -1, 1 do
                        if random(1, 3) >= 2 then
-                               if data[vi_1] == c_air then
+                               if data[vi_1] == c_air or data[vi_1] == c_ignore then
                                        data[vi_1] = c_jungletree
-                               elseif data[vi_2] == c_air then
+                               elseif data[vi_2] == c_air or data[vi_2] == c_ignore then
                                        data[vi_2] = c_jungletree
                                end
                        end
@@ -181,3 +233,174 @@ function default.grow_jungletree(pos, bad)
        vm:update_map()
 end
 
+-- Pinetree from mg mapgen mod, design by sfan5, pointy top added by paramat
+
+local function add_pine_needles(data, vi, c_air, c_ignore, c_snow, c_pine_needles)
+       local node_id = data[vi]
+       if node_id == c_air or node_id == c_ignore or node_id == c_snow then
+               data[vi] = c_pine_needles
+       end
+end
+
+local function add_snow(data, vi, c_air, c_ignore, c_snow)
+       local node_id = data[vi]
+       if node_id == c_air or node_id == c_ignore then
+               data[vi] = c_snow
+       end
+end
+
+function default.grow_pine_tree(pos)
+       local x, y, z = pos.x, pos.y, pos.z
+       local maxy = y + random(9, 13) -- Trunk top
+
+       local c_air = minetest.get_content_id("air")
+       local c_ignore = minetest.get_content_id("ignore")
+       local c_pinetree = minetest.get_content_id("default:pinetree")
+       local c_pine_needles  = minetest.get_content_id("default:pine_needles")
+       local c_snow = minetest.get_content_id("default:snow")
+       local c_snowblock = minetest.get_content_id("default:snowblock")
+       local c_dirtsnow = minetest.get_content_id("default:dirt_with_snow")
+
+       local vm = minetest.get_voxel_manip()
+       local minp, maxp = vm:read_from_map(
+               {x = x - 3, y = y - 1, z = z - 3},
+               {x = x + 3, y = maxy + 3, z = z + 3}
+       )
+       local a = VoxelArea:new({MinEdge = minp, MaxEdge = maxp})
+       local data = vm:get_data()
+
+       -- Scan for snow nodes near sapling to enable snow on branches
+       local snow = false
+       for yy = y - 1, y + 1 do
+       for zz = z - 1, z + 1 do
+               local vi  = a:index(x - 1, yy, zz)
+               for xx = x - 1, x + 1 do
+                       local nodid = data[vi]
+                       if nodid == c_snow or nodid == c_snowblock or nodid == c_dirtsnow then
+                               snow = true
+                       end
+                       vi  = vi + 1
+               end
+       end
+       end
+
+       -- Upper branches layer
+       local dev = 3
+       for yy = maxy - 1, maxy + 1 do
+               for zz = z - dev, z + dev do
+                       local vi = a:index(x - dev, yy, zz)
+                       local via = a:index(x - dev, yy + 1, zz)
+                       for xx = x - dev, x + dev do
+                               if random() < 0.95 - dev * 0.05 then
+                                       add_pine_needles(data, vi, c_air, c_ignore, c_snow,
+                                               c_pine_needles)
+                                       if snow then
+                                               add_snow(data, via, c_air, c_ignore, c_snow)
+                                       end
+                               end
+                               vi  = vi + 1
+                               via = via + 1
+                       end
+               end
+               dev = dev - 1
+       end
+
+       -- Centre top nodes
+       add_pine_needles(data, a:index(x, maxy + 1, z), c_air, c_ignore, c_snow,
+               c_pine_needles)
+       add_pine_needles(data, a:index(x, maxy + 2, z), c_air, c_ignore, c_snow,
+               c_pine_needles) -- Paramat added a pointy top node
+       if snow then
+               add_snow(data, a:index(x, maxy + 3, z), c_air, c_ignore, c_snow)
+       end
+
+       -- Lower branches layer
+       local my = 0
+       for i = 1, 20 do -- Random 2x2 squares of needles
+               local xi = x + random(-3, 2)
+               local yy = maxy + random(-6, -5)
+               local zi = z + random(-3, 2)
+               if yy > my then
+                       my = yy
+               end
+               for zz = zi, zi+1 do
+                       local vi = a:index(xi, yy, zz)
+                       local via = a:index(xi, yy + 1, zz)
+                       for xx = xi, xi + 1 do
+                               add_pine_needles(data, vi, c_air, c_ignore, c_snow,
+                                       c_pine_needles)
+                               if snow then
+                                       add_snow(data, via, c_air, c_ignore, c_snow)
+                               end
+                               vi  = vi + 1
+                               via = via + 1
+                       end
+               end
+       end
+
+       local dev = 2
+       for yy = my + 1, my + 2 do
+               for zz = z - dev, z + dev do
+                       local vi = a:index(x - dev, yy, zz)
+                       local via = a:index(x - dev, yy + 1, zz)
+                       for xx = x - dev, x + dev do
+                               if random() < 0.95 - dev * 0.05 then
+                                       add_pine_needles(data, vi, c_air, c_ignore, c_snow,
+                                               c_pine_needles)
+                                       if snow then
+                                               add_snow(data, via, c_air, c_ignore, c_snow)
+                                       end
+                               end
+                               vi  = vi + 1
+                               via = via + 1
+                       end
+               end
+               dev = dev - 1
+       end
+
+       -- Trunk
+       data[a:index(x, y, z)] = c_pinetree -- Force-place lowest trunk node to replace sapling
+       for yy = y + 1, maxy do
+               local vi = a:index(x, yy, z)
+               local node_id = data[vi]
+               if node_id == c_air or node_id == c_ignore or node_id == c_pine_needles then
+                       data[vi] = c_pinetree
+               end
+       end
+
+       vm:set_data(data)
+       vm:write_to_map()
+       vm:update_map()
+end
+
+-- New tree
+
+function default.grow_new_apple_tree(pos)
+       local path = minetest.get_modpath("default") .. "/schematics/apple_tree.mts"
+       minetest.place_schematic({x = pos.x - 2, y = pos.y - 1, z = pos.z - 2},
+               path, 0, nil, false)
+end
+
+-- New jungle tree
+
+function default.grow_new_jungle_tree(pos)
+       local path = minetest.get_modpath("default") .. "/schematics/jungle_tree.mts"
+       minetest.place_schematic({x = pos.x - 2, y = pos.y - 1, z = pos.z - 2},
+               path, 0, nil, false)
+end
+
+-- New pine tree
+
+function default.grow_new_pine_tree(pos)
+       local path = minetest.get_modpath("default") .. "/schematics/pine_tree.mts"
+       minetest.place_schematic({x = pos.x - 2, y = pos.y - 1, z = pos.z - 2},
+               path, 0, nil, false)
+end
+
+-- New acacia tree
+
+function default.grow_new_acacia_tree(pos)
+       local path = minetest.get_modpath("default") .. "/schematics/acacia_tree.mts"
+       minetest.place_schematic({x = pos.x - 4, y = pos.y - 1, z = pos.z - 4},
+               path, random, nil, false)
+end