--whitelist for productivity module effect
for _, mod in pairs(data.raw.module) do
	if mod.effect ~= nil then
		for _, effect in pairs(mod.effect) do
			if effect[1] == productivity then
				if mod.limitation ~= nil then
					local limitation = mod.limitation
					--new stuff
					table.insert(limitation, "17-military-science-pack")
					table.insert(limitation, "17-chemical-science-pack")
					table.insert(limitation, "17-production-science-pack")
					table.insert(limitation, "17-utility-science-pack")
					table.insert(limitation, "17-low-density-structure")					
				end
			end
		end
	end
end

--overrides
local function recipe_change(table, condition, replace)
	for _=1, #table do
		if table[_][1] == condition then
			table[_] = replace
		end
	end
end

local function prerequisites(technology, condition, replace)
	for _=1, #technology do
		if technology[_] == condition then
			table.remove(technology, _)
			if replace ~= nil then
				table.insert(technology, replace)
			end
		end
	end
end

local function remove_science(technology, science_pack)
	for _, ingredient in pairs (technology) do
		if ingredient[1] == science_pack then
			table.remove(technology, _)
		end
	end
end

if settings.startup["017-recipes-changes"].value then
	recipe_change(data.raw["recipe"]["atomic-bomb"].ingredients, "processing-unit", {"rocket-control-unit", 15})
	recipe_change(data.raw["recipe"]["power-armor-mk2"].ingredients, "speed-module-3", {"speed-module-2", 5})
	recipe_change(data.raw["recipe"]["power-armor-mk2"].ingredients, "effectivity-module-3", {"effectivity-module-2", 5})
	local power_armor = data.raw["technology"]["power-armor-2"].prerequisites
	prerequisites(power_armor, "speed-module-3", "speed-module-2")
	prerequisites(power_armor, "effectivity-module-3", "effectivity-module-2")
	table.insert(power_armor, "advanced-electronics-2")
end

local r_effects = data.raw["technology"]["rocket-silo"].effects --remove the duplicate LDS effect in rocket-silo research
for _=1, #r_effects do
	if r_effects[_] then
		if r_effects[_].type == "unlock-recipe" and (r_effects[_].recipe == "low-density-structure" or r_effects[_].recipe == "rocket-fuel" or r_effects[_].recipe == "rocket-control-unit") then
			r_effects[_] = nil
		elseif settings.startup["017-techtree"].value and settings.startup["017-rocket-victory"].value then
			if r_effects[_].type == "unlock-recipe" and r_effects[_].recipe == "satellite" then
				r_effects[_] = nil
			end
		end
	end
end

data.raw["technology"]["nuclear-fuel-reprocessing"].unit.count = (1500-settings.startup["017-nuclear-reprocessing-discount"].value)

if settings.startup["017-pack-type-rebalancing"].value then
	remove_science(data.raw["technology"]["rocket-silo"].unit.ingredients, "military-science-pack")
	if not mods["Nuclear Fuel"] then
		remove_science(data.raw["technology"]["kovarex-enrichment-process"].unit.ingredients, "high-tech-science-pack")
	end
	if not mods["bobmodules"] then
		remove_science(data.raw["technology"]["speed-module-3"].unit.ingredients, "high-tech-science-pack")
		table.insert(data.raw["technology"]["speed-module-3"].unit.ingredients, {"production-science-pack", 1})
		remove_science(data.raw["technology"]["effectivity-module-3"].unit.ingredients, "high-tech-science-pack")
		table.insert(data.raw["technology"]["effectivity-module-3"].unit.ingredients, {"production-science-pack", 1})
		remove_science(data.raw["technology"]["effect-transmission"].unit.ingredients, "high-tech-science-pack")
		table.insert(data.raw["technology"]["effect-transmission"].unit.ingredients, {"production-science-pack", 1})
	end
	if not mods["bobslogistics"] then
		remove_science(data.raw["technology"]["logistic-system"].unit.ingredients, "production-science-pack")
	end
end

if settings.startup["017-techtree"].value then
	local data_technology = data.raw["technology"]
	
	local removing = {
		{data_technology["oil-processing"].effects, "lubricant"},
		{data_technology["nuclear-power"].effects, "centrifuge"},
		{data_technology["nuclear-power"].effects, "uranium-processing"},
	}
	for j=1, #removing do
		local effect_ = removing[j][1]
		local recipe_ = removing[j][2]
		for i=1, #effect_ do
			if effect_[i] then
				if effect_[i].type == "unlock-recipe" and effect_[i].recipe == recipe_ then
					effect_[i] = nil
				end
			end
		end
	end
	
	--misc. techtree changes
	local engine = data_technology["electric-engine"].prerequisites
	local rocket_silo = data_technology["rocket-silo"].prerequisites
	local kovarex_process = data_technology["kovarex-enrichment-process"].prerequisites
	local logistics = data_technology["logistics-3"].prerequisites
	local fission = data_technology["nuclear-power"]
	prerequisites(rocket_silo, "rocket-speed-5", nil)
	prerequisites(engine, "advanced-electronics", nil)
	engine[#engine+1] = "lubricant"
	rocket_silo[#rocket_silo+1] = "rocket-fuel"
	rocket_silo[#rocket_silo+1] = "rocket-control-unit"
	rocket_silo[#rocket_silo+1] = "low-density-structure"
	prerequisites(kovarex_process, "nuclear-power", nil)
	kovarex_process[#kovarex_process+1] = "rocket-fuel"
	kovarex_process[#kovarex_process+1] = "uranium-enrichment"
	prerequisites(logistics, "automation-3", nil)
	logistics[#logistics+1] = "lubricant"
	fission.prerequisites = {"uranium-enrichment"}
	fission.unit.time = settings.startup["017-nuclear-power-energy"].value
	fission.unit.count = settings.startup["017-nuclear-power-cost"].value
	
	data.raw["recipe"]["science-pack-2"].enabled = false
	
	if settings.startup["017-old-science"].value and settings.startup["017-techtree"].value then
		remove_effect_table = {
			{"military-2", "military-science-pack"},
			{"advanced-electronics", "science-pack-3"},
			{"advanced-material-processing-2", "production-science-pack"},
			{"advanced-electronics-2", "high-tech-science-pack"}
		}
		
		data_technology["military-science-pack"].effects = {
				{type = "unlock-recipe", recipe = "military-science-pack"},
				{type = "unlock-recipe", recipe = "17-military-science-pack"}
		}
		data_technology["chemical-science-pack"].effects = {
				{type = "unlock-recipe", recipe = "science-pack-3"},
				{type = "unlock-recipe", recipe = "17-chemical-science-pack"}
		}
		data_technology["production-science-pack"].effects = {
				{type = "unlock-recipe", recipe = "production-science-pack"},
				{type = "unlock-recipe", recipe = "17-production-science-pack"}
		}
		data_technology["utility-science-pack"].effects = {
				{type = "unlock-recipe", recipe = "high-tech-science-pack"},
				{type = "unlock-recipe", recipe = "17-utility-science-pack"}
		}
		if not mods["bobrevamp"] then
			data_technology["low-density-structure"].effects = {
					{type = "unlock-recipe", recipe = "low-density-structure"},
					{type = "unlock-recipe", recipe = "17-low-density-structure"}
			}
		end
		
		for i=1, #remove_effect_table do
			local effects = data_technology[remove_effect_table[i][1]].effects
			for _=1, #effects do
				if effects[_] then
					if effects[_].type == "unlock-recipe" and effects[_].recipe == remove_effect_table[i][2] then
						table.remove(effects, _)
					end
				end
			end
		end
	end
	if mods["bobrevamp"] then
		data_technology["low-density-structure"].effects = {
			{type = "unlock-recipe", recipe = "low-density-structure"},
		--	{type = "unlock-recipe", recipe = "17-low-density-structure"}
		}
	end
end

--player equipment changes
--https://forums.factorio.com/viewtopic.php?t=6059
if settings.startup["017-durability"].value then
	for _, axe in pairs(data.raw["mining-tool"]) do
		axe.durability = math.huge
	end

	for _, armor in pairs(data.raw["armor"]) do
		armor.durability = math.huge
	end
end

if settings.startup["017-axe"].value then
	local function fake_axe(item, recipe)
		local name = item.name .. "-item"
		return{
			{
				type = "item",
				name = name,
				localised_name = {"item-name." .. item.name},
				localised_description = {"item-description.fake-mining-tool"},
				icon = item.icon,
				icon_size = item.icon_size,
				flags = {"goes-to-main-inventory"},
				subgroup = "tool",
				order = item.order,
				stack_size = item.stack_size
			},
			{
				type = "recipe",
				name = name,
				ingredients = recipe.ingredients,
				result = name
			}
		}
	end
	local axe_item = {}
	local axe_recipe = {}
	local axe_recipe_consumers = {}
	for _, axe in pairs(data.raw["mining-tool"]) do
		local axe_name = axe.name
		axe_item[axe_name] = axe
		axe_recipe_consumers[axe_name] = false
		for i, recipe in pairs(data.raw["recipe"]) do
			if recipe.ingredients then
				for j, ingredients in pairs(recipe.ingredients) do
					if ingredients[1] == axe_name and ingredients[2] ~= 0 then
						ingredients[1] = axe_name .. "-item"
						axe_recipe_consumers[axe_name] = true
					end
				end
			end
			if recipe.result == axe_name then
				axe_recipe[axe_name] = recipe
			end
		end
	end
	for l, axe in pairs(axe_item) do
		if axe_recipe[axe.name] ~= nil then
			local axe_name = axe.name
			modify = false
			if axe_recipe_consumers[axe_name] == true then
				data:extend(fake_axe(axe, axe_recipe[axe_name]))
			end
			axe_recipe[axe_name].hidden = true
		end
	end
end

if settings.startup["017-equipment"].value and data.raw["recipe"]["belt-immunity-equipment"] == nil then
	data:extend({
		{
			type = "technology",
			name = "belt-immunity-equipment",
			icon_size = 128,
			icon = "__017_science__/graphics/technology/belt-immunity-equipment.png",
			prerequisites = {"modular-armor"},
			effects =
			{
				{
					type = "unlock-recipe",
					recipe = "belt-immunity-equipment"
				}
			},
			unit =
			{
				count = 50,
				ingredients = {{"science-pack-1", 1}, {"science-pack-2", 1}},
				time = 15
			},
			order = "g-ga"
		},
		{
			type = "recipe",
			name = "belt-immunity-equipment",
			enabled = false,
			energy_required = 10,
			ingredients =
			{
				{"advanced-circuit", 5},
				{"steel-plate", 10}
			},
			result = "belt-immunity-equipment"
		}
	})
end

local lds_recipes = {
	"energy-shield-mk2-equipment",
	"battery-mk2-equipment",
	"fusion-reactor-equipment",
	"personal-laser-defense-equipment",
	"discharge-defense-equipment", -- sorry for indirectly nerfing discharge-defense :(
	"exoskeleton-equipment",
	"personal-roboport-mk2-equipment",
	"power-armor-mk2"
}

if settings.startup["017-lds"].value then
	local recipe = data.raw["recipe"]
	for _, r in pairs(recipe) do
		for _, l in pairs(lds_recipes) do
			if r.name == l then
				modified = false
				for _=1, #r.ingredients do
					if r.ingredients[_][1] == "steel-plate" then
						local n = math.ceil(r.ingredients[_][2] / 2)
						table.remove(r.ingredients, _)
						r.ingredients[#r.ingredients+1] = {"low-density-structure", n}
						modified = true
					end
				end
				if modified == false then
					for _=1, #r.ingredients do
						if r.ingredients[_][1] == "processing-unit" then
							local n = math.ceil(r.ingredients[_][2] / settings.startup["017-lds-num"].value)
							r.ingredients[#r.ingredients+1] = {"low-density-structure", n}
						end
					end
				end
			end
		end
	end
end

if settings.startup["017-tank-gun-nerf"].value then
	data.raw["gun"]["tank-machine-gun"].attack_parameters.damage_modifier = 1
end

if settings.startup["017-ore-icons"].value then
	local item = data.raw["item"]
	item["coal"].icon="__base__/graphics/icons/icons-new/coal.png"
	item["copper-ore"].icon="__base__/graphics/icons/icons-new/copper-ore.png"
	item["iron-ore"].icon="__base__/graphics/icons/icons-new/iron-ore.png"
	item["stone"].icon="__base__/graphics/icons/icons-new/stone.png"
	item["uranium-ore"].icon="__base__/graphics/icons/icons-new/uranium-ore.png"
end

--HUGE THANKS to Dimava for the following changes
--Added options for some changes (for modded users)

--increase belt speed to 15x
for _, belt in pairs(data.raw["splitter"]) do
	if (belt.speed * 32) % 1 == 0 and belt.speed * 32 < 8 then
		belt.speed = belt.speed * 45 / 40;
	end
end
for _, belt in pairs(data.raw["transport-belt"]) do
	if (belt.speed * 32) % 1 == 0 and belt.speed * 32 < 8 then
		belt.speed = belt.speed * 45 / 40;
	end
end
for _, belt in pairs(data.raw["underground-belt"]) do
	if (belt.speed * 32) % 1 == 0 and belt.speed * 32 < 8 then
		belt.speed = belt.speed * 45 / 40;
	end
end
for _, belt in pairs(data.raw["loader"]) do
	if (belt.speed * 32) % 1 == 0 and belt.speed * 32 < 8 then
		belt.speed = belt.speed * 45 / 40;
	end
end

if settings.startup["017-drill"].value then
	--simplify drill stats
	data.raw["mining-drill"]["burner-mining-drill"].mining_speed = 0.25
	data.raw["mining-drill"]["burner-mining-drill"].mining_power = 3

	--simplify ore hardness
	for _, ore in pairs(data.raw.resource) do
		if ore.minable.hardness == 0.9 then
			ore.minable.hardness = 1
		end
	end
	data.raw.resource.stone.minable.hardness = 1
end

if settings.startup["017-smelting"].value then
	--decrease smelting times
	for _, recipe in pairs(data.raw.recipe) do
		if recipe.category=="smelting" then
			if recipe.energy_required then
				if (recipe.energy_required / 3.5) % 1 == 0 then
					recipe.energy_required = recipe.energy_required * 3.2 / 3.5
				end
			else
				if (recipe.normal.energy_required / 3.5) % 1 == 0 then
					recipe.normal.energy_required = recipe.normal.energy_required * 3.2 / 3.5
				end
				if (recipe.expensive.energy_required / 3.5) % 1 == 0 then
					recipe.expensive.energy_required = recipe.expensive.energy_required * 3.2 / 3.5
				end			
			end
		end
	end
end

if settings.startup["017-assem-lim"].value then
	--remove assembling-machine ingredient_count limits
	for _, machine in pairs(data.raw["assembling-machine"]) do
		if machine.ingredient_count < 10 then
			machine.ingredient_count = 10
		end
	end
end