control.lua

require("defines")
require("util")
local helpers = require("helpers")

--local copies of mod settings
local circuitAuto
local circuitSpeed
local quickDelete

--read local copies of mod settings
local function copy_settings()
	circuitAuto = settings.global[AutomaticTrainDeployment_defines.names.settings.circuitAuto].value
	circuitSpeed = settings.global[AutomaticTrainDeployment_defines.names.settings.circuitSpeed].value
	quickDelete = settings.global[AutomaticTrainDeployment_defines.names.settings.quickDelete].value
end

script.on_init(copy_settings)
script.on_load(copy_settings)

--update local copies of mod settings
script.on_event(defines.events.on_runtime_mod_setting_changed, function (event)
	if event.setting == AutomaticTrainDeployment_defines.names.settings.circuitAuto then
		circuitAuto = settings.global[AutomaticTrainDeployment_defines.names.settings.circuitAuto].value
	elseif event.setting == AutomaticTrainDeployment_defines.names.settings.circuitSpeed then
		circuitSpeed = settings.global[AutomaticTrainDeployment_defines.names.settings.circuitSpeed].value
	elseif event.setting == AutomaticTrainDeployment_defines.names.settings.quickDelete then
		quickDelete = settings.global[AutomaticTrainDeployment_defines.names.settings.quickDelete].value
	end
end)

--the trigger to do stuff
script.on_event(defines.events.on_train_changed_state, function (event)
	local trigger = event.train
	--game.players[1].print("state changed to " .. trigger.state)

	--ensure train is ok
	if not trigger or not trigger.valid then
		return
	end

	--Delete trains that reach a deleteStop
	if quickDelete and trigger.state == defines.train_state.arrive_station and trigger.path_end_stop and trigger.path_end_stop.name == AutomaticTrainDeployment_defines.names.entities.deleteStop then
		for _, cart in pairs(trigger.carriages) do
			cart.destroy()
		end

	elseif trigger.state == defines.train_state.wait_station and trigger.station and trigger.station.name == AutomaticTrainDeployment_defines.names.entities.deleteStop then
		for _, cart in pairs(trigger.carriages) do
			cart.destroy()
		end

	--is a train heading to a copy/dandom stop?
	elseif trigger.state == defines.train_state.on_the_path and trigger.path_end_stop and trigger.path_end_stop.valid and
		(trigger.path_end_stop.name == AutomaticTrainDeployment_defines.names.entities.copyStop or 
		trigger.path_end_stop.name == AutomaticTrainDeployment_defines.names.entities.randomStop) then

		local trigger_stop = trigger.path_end_stop
		local source, source_stop = helpers.find_source_train(trigger_stop)

		if source == nil then
			--game.players[1].print("no source train")
			return
		end
		--game.players[1].print("train!")

		if not helpers.copy_rolling_stock(source, source_stop, trigger_stop) then
			return --no new train was created don't mess with existing ones
		end

		--find the newly created train
		local new = helpers.train_by_station(trigger_stop)

		--abort if no train was created
		if new == nil then
			--game.players[1].print("no train was created")
			return
		end

		--randomize the schedule?
		if trigger_stop.name == AutomaticTrainDeployment_defines.names.entities.randomStop then
			new.schedule = helpers.randomize_schedule(source)
		end

		local auto = circuitAuto == "Auto"
		local speed = 0

		--allows train auto mode and speed to be set by signals
		if circuitAuto == "A > 0" or circuitSpeed then
			local green = trigger_stop.get_circuit_network(defines.wire_type.green)
			local red = trigger_stop.get_circuit_network(defines.wire_type.red)

			if circuitAuto == "A > 0" then
				auto = (((green and green.get_signal{ type = "virtual", name = "signal-A"}) or 0) + ((red and red.get_signal{ type = "virtual", name = "signal-A"}) or 0) > 0)
			end

			if circuitSpeed then
				speed = (((green and green.get_signal{ type = "virtual", name = "signal-S"}) or 0) + ((red and red.get_signal{ type = "virtual", name = "signal-S"}) or 0))
			end
		end

		--set the train to automatic
		new.manual_mode = not auto 
		new.speed = (circuitSpeed and speed > 0 and speed) or 0
	end
end)

--varoius console commands I used to debug things
--/c game.player.selected.connect_rolling_stock(defines.rail_direction.front)
--/c game.player.print(game.player.selected.train.state)
--/c game.player.print(game.player.selected.train.path_end_stop)
--/c game.player.print(game.player.selected.direction)
--/c game.player.print(game.player.surface.find_entities_filtered{force = game.player.force, position = {x = 5, y = -9}, type = "locomotive"})
--/c game.player.print(game.player.selected.burner.currently_burning)
--/c game.player.print(game.player.selected.get_control_behavior().disabled
--/c game.player.print(game.player.selected.train.front_stock == game.player.selected)
--/c game.player.print(game.player.selected.circuit_connection_definitions[1].target_circuit_id)