-- More trees!
--
-- This mod adds more types of trees to the game
-- at present, they consist of jungle trees and conifers
--
-- Much of the node definitions here came from cisoun's conifers mod and
-- bas080's jungle trees mod.
--
-- Brought together into one mod and made L-systems compatible by Vanessa
-- Ezekowitz.  Thrown together on 2013-01-09 :-)
--
-- Jungle tree axioms/rules tweaked by RealBadAngel
--
-- License: WTFPL for all parts (code and textures)
--

-- snow biomes mod uses this code to make biomes:
--
--	local perlin1 = env:get_perlin(112,3, 0.5, 150)
--	local test = perlin1:get2d({x=pos.x, y=pos.z})
--	if smooth and (test > 0.73 or (test > 0.43 and math.random(0,29) > (0.73 - test) * 100 )) then
--		in_biome = true
--	elseif not smooth and test > 0.53 then
--		in_biome = true
--	end
--	
-- We'll just save this for later use ;-)
--

moretrees = {}

-- Jungletree init stuff:

local JT_SPAWN_INTERVAL = 1000
local JT_SPAWN_CHANCE = 100

local JT_GROW_INTERVAL = 100
local JT_GROW_CHANCE = 10

local JT_RADIUS = 15
local JT_WATER_RADIUS = 15
local JT_WATER_COUNT = 10

local jungletree_seed_diff = plantslib.plantlife_seed_diff

-- Conifers init stuff:

local CONIFERS_SPAWN_SAPLING_INTERVAL = 1000
local CONIFERS_SPAWN_SAPLING_CHANCE = 100

local CONIFERS_GROW_SAPLING_INTERVAL = 100
local CONIFERS_GROW_SAPLING_CHANCE = 10

local CONIFERS_DISTANCE = 9 -- how far apart should conifer saplings spawn?
local CONIFERS_ALTITUDE = 25

local CONIFERS_REMOVE_TREES = false -- Remove trees above CONIFERS_ALTITUDE?
local CONIFERS_RTREES_INTERVAL = 360
local CONIFERS_RTREES_CHANCE = 10

local conifers_seed_diff = plantslib.plantlife_seed_diff + 30

-- Spawning functions

jungletrees_biome = {
	surface = "default:dirt_with_grass",	-- must grow on grass only
	avoid =	{"jungletree:sapling","default:jungletree"}, -- avoid spawning near these
	radius = JT_RADIUS,			-- Keep this much room around the above avoid items
	seed_diff = jungletree_seed_diff,	-- duh? :-)
	neighbors = nil,			-- we don't care about neighbors
	ncount = nil,				-- or the count thereof
	water_depth = nil,			-- or water depth
	min_elevation = -5,			-- must be 5m below sea level or higher
	max_elevation = 10,			-- but no higher than 10m
	near_nodes = {"default:water_source"},	-- Jungle trees must be near water
	near_nodes_size = JT_WATER_RADIUS,	-- within this radius of it (default 25)
	near_nodes_count = JT_WATER_COUNT,	-- with this many water nodes in the area
	temp_min = nil,				-- don't care about temperature
	temp_max = nil,				-- at either end of the scale
	exec_funct = "moretrees:grow_jungletree" -- name of the function to execute to grow a tree
}

conifers_biome = {
	surface = "default:dirt_with_grass",
	avoid =	{"conifers:sapling", "conifers:trunk"},
	radius = CONIFERS_DISTANCE,
	seed_diff = conifers_seed_diff,														
	neighbors = nil,
	ncount = nil,
	water_depth = nil,
	min_elevation = CONIFERS_ALTITUDE,
	max_elevation = nil,
	near_nodes = nil,
	near_nodes_size = nil,
	near_nodes_count = nil,
	temp_min = nil,
	temp_max = nil,
	exec_funct = "moretrees:grow_conifer"
}

plantslib:register_generate_plant(jungletrees_biome)
plantslib:register_generate_plant(conifers_biome)

-- growing functions

plantslib:grow_plants(
	JT_GROW_INTERVAL,
	JT_GROW_CHANCE,
	"jungletree:sapling",
	nil,
	nil,
	nil,
	nil,
	nil,
	nil,
	nil,
	nil,
	"moretrees:grow_jungletree",
	jungletree_seed_diff
)

plantslib:grow_plants(
	CONIFERS_GROW_SAPLING_INTERVAL,
	CONIFERS_GROW_SAPLING_CHANCE,
	"conifers:sapling",
	nil,
	nil,
	nil,
	nil,
	nil,
	nil,
	nil,
	nil,
	"moretrees:grow_conifer",
	conifers_seed_diff
)

-- L-System Tree definitions

local jungle_tree={
	axiom=nil,
	rules_a=nil,
	rules_b=nil,
	trunk="default:jungletree",
	leaves="jungletree:leaves_green",
	leaves2=nil,
	leaves2_chance=nil,
	angle=45,
	iterations=nil,
	random_level=2,
	trunk_type=nil,
	thin_branches=true;
	fruit_chance=15,
	fruit="vines:vine"
}

local jt_axiom1 = "FFFA"
local jt_rules_a1 = "FFF[&&-FBf[&&&Ff]^^^Ff][&&+FBFf[&&&FFf]^^^Ff][&&---FBFf[&&&Ff]^^^Ff][&&+++FBFf[&&&Ff]^^^Ff]F/A"
local jt_rules_b1 = "[-Ff&f][+Ff&f]B"

local jt_axiom2 = "FFFFFA"
local jt_rules_a2 = "FFFFF[&&-FFFBF[&&&FFff]^^^FFf][&&+FFFBFF[&&&FFff]^^^FFf][&&---FFFBFF[&&&FFff]^^^FFf][&&+++FFFBFF[&&&FFff]^^^FFf]FF/A"
local jt_rules_b2 = "[-FFf&ff][+FFf&ff]B"

local conifer_tree={
	axiom="FFFAF[&&-F][&&+F][&&---F][&&+++F]Fff",
	rules_a=nil,
	rules_b=nil,
	trunk="conifers:trunk",
	leaves=nil,
	angle=45,
	iterations=7,
	random_level=5,
	thin_trunks=true
}

local ct_rules_a1 = "FF[FF][&&-FBF][&&+FBF][&&---FBF][&&+++FBF]F/A"
local ct_rules_b1 = "[-FBf][+FBf]"

local ct_rules_a2 = "FF[FF][&&-FBF][&&+FBF][&&---FBF][&&+++FBF]F/A"
local ct_rules_b2 = "[-fB][+fB]"

-- Code that actually spawns the trees!

function moretrees:grow_jungletree(pos)
	local r1 = math.random(2)
	local r2 = math.random(3)
	if r1 == 1 then
		jungle_tree["leaves2"] = "jungletree:leaves_red"
	else 
		jungle_tree["leaves2"] = "jungletree:leaves_yellow"
	end
	jungle_tree["leaves2_chance"] = math.random(25, 75)

	if r2 == 1 then
		jungle_tree["trunk_type"] = "single"
		jungle_tree["iterations"] = 2
		jungle_tree["axiom"] = jt_axiom1
		jungle_tree["rules_a"] = jt_rules_a1
		jungle_tree["rules_b"] = jt_rules_b1
	elseif r2 == 2 then
		jungle_tree["trunk_type"] = "double"
		jungle_tree["iterations"] = 4
		jungle_tree["axiom"] = jt_axiom2
		jungle_tree["rules_a"] = jt_rules_a2
		jungle_tree["rules_b"] = jt_rules_b2
	elseif r2 == 3 then
		jungle_tree["trunk_type"] = "crossed"
		jungle_tree["iterations"] = 4
		jungle_tree["axiom"] = jt_axiom2
		jungle_tree["rules_a"] = jt_rules_a2
		jungle_tree["rules_b"] = jt_rules_b2
	end

	minetest.env:remove_node(pos)
	local leaves = minetest.env:find_nodes_in_area({x = pos.x-1, y = pos.y, z = pos.z-1}, {x = pos.x+1, y = pos.y+10, z = pos.z+1}, "default:leaves")
	for leaf in ipairs(leaves) do
			minetest.env:remove_node(leaves[leaf])
	end
	minetest.env:spawn_tree(pos,jungle_tree)
end

function moretrees:grow_conifer(pos)
	if math.random(2) == 1 then
		conifer_tree["leaves"]="conifers:leaves"
	else
		conifer_tree["leaves"]="conifers:leaves_special"
	end
	if math.random(2) == 1 then
		conifer_tree["rules_a"] = ct_rules_a1
		conifer_tree["rules_b"] = ct_rules_b1
	else
		conifer_tree["rules_a"] = ct_rules_a2
		conifer_tree["rules_b"] = ct_rules_b2
	end

	minetest.env:remove_node(pos)
	local leaves = minetest.env:find_nodes_in_area({x = pos.x, y = pos.y, z = pos.z}, {x = pos.x, y = pos.y+5, z = pos.z}, "default:leaves")
	for leaf in ipairs(leaves) do
			minetest.env:remove_node(leaves[leaf])
	end
	minetest.env:spawn_tree(pos,conifer_tree)
end

-- Other stuff

-- Should we remove all the trees above the conifers altitude?
if CONIFERS_REMOVE_TREES == true then
	minetest.register_abm({
		nodenames = {
			"default:tree", 
			"default:leaves"
		},
		interval = CONIFERS_RTREES_INTERVAL,
		chance = CONIFERS_RTREES_CHANCE,
		
		action = function(pos, node, _, _)
			if minetest.env:get_node({x = pos.x, y = pos.y + 1, z = pos.z}).name == "air"
				and pos.y >= CONIFERS_ALTITUDE
			then
				minetest.env:add_node(pos , {name = "air"})
			end
		end
	})
end

-- Nodes for jungle trees

minetest.register_node(":jungletree:sapling", {
	description = "Jungle Tree Sapling",	
	drawtype = "plantlike",	
	visual_scale = 1.0,	
	tiles = {"jungletree_sapling.png"},	
	inventory_image = "jungletree_sapling.png",	
	wield_image = "default_sapling.png",	
	paramtype = "light",	
	walkable = false,	
	groups = {snappy=2,dig_immediate=3,flammable=2},
})

local leaves = {"green","yellow","red"}
for color = 1, 3 do
	local leave_name = ":jungletree:leaves_"..leaves[color]
	minetest.register_node(leave_name, {
		description = "Jungle Tree Leaves",
		drawtype = "allfaces_optional",
		tiles = {"jungletree_leaves_"..leaves[color]..".png"},
		paramtype = "light",
		groups = {snappy=3, leafdecay=3, flammable=2},
		drop = {
			max_items = 1,
			items = {
				{
					-- player will get sapling with 1/20 chance
					items = {'jungletree:sapling'},
					rarity = 20,
				},
				{
					-- player will get leaves only if he get no saplings,
					-- this is because max_items is 1
					items = {"jungletree:leaves_"..leaves[color]},
				}
			}
		},
		sounds = default.node_sound_leaves_defaults(),
	})
end

-- Nodes for conifers

minetest.register_node(":conifers:trunk", {
	description = "Conifer trunk",
	tile_images = { 
		"conifers_trunktop.png", 
		"conifers_trunktop.png", 
		"conifers_trunk.png", 
		"conifers_trunk.png", 
		"conifers_trunk.png", 
		"conifers_trunk.png" 
	},
	paramtype = "facedir_simple",
	is_ground_content = true,
	groups = {
		tree = 1,
		snappy = 2,
		choppy = 2,
		oddly_breakable_by_hand = 1,
		flammable = 2
	},
	sounds = default.node_sound_wood_defaults()
})

minetest.register_node(":conifers:trunk_reversed", {
	description = "Conifer reversed trunk",
	tile_images = { 
		"conifers_trunk_reversed.png", 
		"conifers_trunk_reversed.png",
		"conifers_trunktop.png", 
		"conifers_trunktop.png", 
		"conifers_trunk_reversed.png", 
		"conifers_trunk_reversed.png" 
	},
	--inventory_image = minetest.inventorycube(
		--"conifers_trunk.png",
		--"conifers_trunktop.png",
		--"conifers_trunk.png"
	--),
	paramtype = "facedir_simple",
	material = minetest.digprop_woodlike(1.0),
	legacy_facedir_simple = true,
	is_ground_content = true,
	groups = {
		tree = 1,
		snappy = 2,
		choppy = 2,
		oddly_breakable_by_hand = 1,
		flammable = 2
	},
	sounds = default.node_sound_wood_defaults()
})

minetest.register_node(":conifers:leaves", {
	description = "Conifer leaves",
	drawtype = "allfaces_optional",
	visual_scale = 1.3,
	tile_images = { "conifers_leaves.png" },
	paramtype = "light",
	groups = {
		snappy = 3,
		leafdecay = 3,
		flammable = 2
	},
	drop = {
		max_items = 1,
		items = {
			{
				-- player will get sapling with 1/20 chance
				items = {'conifers:sapling'},
				rarity = 20,
			},
			{
				-- player will get leaves only if he get no saplings,
				-- this is because max_items is 1
				items = {'conifers:leaves'},
			}
		}
	},
	sounds = default.node_sound_leaves_defaults()
})

minetest.register_node(":conifers:leaves_special", {
	description = "Bright conifer leaves",
	drawtype = "allfaces_optional",
	visual_scale = 1.3,
	tile_images = { "conifers_leaves_special.png" },
	paramtype = "light",
	groups = {
		snappy = 3,
		leafdecay = 3,
		flammable = 2
	},
	drop = {
		max_items = 1,
		items = {
			{
				-- player will get sapling with 1/20 chance
				items = {'conifers:sapling'},
				rarity = 20,
			},
			{
				-- player will get leaves only if he get no saplings,
				-- this is because max_items is 1
				items = {'conifers:leaves'},
			}
		}
	},
	sounds = default.node_sound_leaves_defaults()
})

minetest.register_node(":conifers:sapling", {
	description = "Conifer sapling",
	drawtype = "plantlike",
	visual_scale = 1.0,
	tile_images = {"conifers_sapling.png"},
	inventory_image = "conifers_sapling.png",
	wield_image = "conifers_sapling.png",
	paramtype = "light",
	walkable = false,
	groups = {
		snappy = 2,
		dig_immediate = 3,
		flammable = 2
	},
	sounds = default.node_sound_defaults(),
})

dofile(minetest.get_modpath("moretrees").."/crafts.lua")

print("[Moretrees] Loaded (2013-01-18)")