--Environment Code
-------------------------------------------------------------------------------------
-- Yard Information
if event.init then
	S.yards = {
		--[[  yard info template
			yard_id = {
			yard_name = string,
			active_indicator_pos = POS(),
			dir_indicator_pos = POS(),
			error_indicator_pos = POS(),
			headshunt_max = number,
			notify = empty table,
			notify_pos = pos,

			-- reserved vars --
			last_id = string  (the id of the last train that entered the yard last)
			arrival_length = number
			arrival_time = string (RWT)
			departure_length = number or "?" (set when train arrives, until train departs)
			departure_time = string (RWT)
			rts = bool (whether the entering train will exit the way it came in)
		},
		]]--
		TY = {
			yard_name = "Trisiston",
			active_indicator_pos = POS(-4025,14,-2659),
			dir_indicator_pos = POS(-4025,12,-2665),
			error_indicator_pos = POS(-4025,13,-2671),
			headshunt_max = 5,
			notify = {},
			notify_pos = POS(-4023,12,-2660),
			arrival_time = rwt.now(),
			departure_time = rwt.now()
		},
		BY = {
			yard_name = "Banach",
			active_indicator_pos = POS(-2002,3,-1099),
			dir_indicator_pos = POS(-2009,3,-1101),
			error_indicator_pos = POS(-1999,3,-1099),
			headshunt_max = 5,
			notify = {},
			notify_pos = POS(-2004,2,-1101),
			arrival_time = rwt.now(),
			departure_time = rwt.now()
		},
		ARC = {
			yard_name = "Arcadius",
			active_indicator_pos = POS(-1952,16,840),
			dir_indicator_pos = POS(-1950,16,840),
			error_indicator_pos = POS(-1948,16,840),
			headshunt_max = 5,
			notify = {},
			notify_pos = POS(-1946,16,840),
			arrival_time = rwt.now(),
			departure_time = rwt.now()
		},
		IP = {
			yard_name = "Ipswich",
			active_indicator_pos = POS(1179,16,3848),
			dir_indicator_pos = POS(1177,16,3855),
			error_indicator_pos = POS(1179,16,3850),
			headshunt_max = 5,
			notify = {},
			notify_pos = POS(1178,16,3851),
			arrival_time = rwt.now(),
			departure_time = rwt.now()
		},
		CAN = {
			yard_name = "Cannery",
			active_indicator_pos = POS(-594,26,2486),
			dir_indicator_pos = POS(-594,26,2484),
			error_indicator_pos = POS(-594,26,2482),
			headshunt_max = 2,
			--notify = {},
			--notify_pos = POS(-594,26,2485),
			arrival_time = rwt.now(),
			departure_time = rwt.now()
		}
		-- HY = {
			-- active_indicator_pos = POS(-4025,14,-2659),
			-- dir_indicator_pos = POS(-4025,13,-2665),
			-- error_indicator_pos = POS(-4025,13,-2671),
		-- },
	}
	S.known_trains = {
		['120684'] =  "LHF #1", --Maverick2797
		['249165'] =  "LHF #2", --Maverick2797
		['734206'] =  "ARC-BY", --Maverick2797
		['588750'] =  "MMF-TY", --survivalg/erstazi
		['543381'] =  "WOA-IP", --Maverick2797
		['834721'] =  "S27-ARC", --Maverick2797
		['513598'] =  "S27EX-ARC",  --Maverick2797
		['590988'] =  "CAN-ARC", --Maverick2797
	}
end
------------------------------------------------------------------------------------
-- Utility Functions
F.indicator = function(indicator,set)
	if set ~= nil then
		if type(set) == string then
			setstate(indicator,set)
		else
			setstate(indicator,(set and "on") or "off")
		end
	end
	return (getstate(indicator) == "on") or false
end

F.get_rc_safe = function()
	return get_rc() or ""
end

F.has_rc = function(query,rc_list) -- query = string, single entry
	if not atc_id then return false end
	if rc_list == "" or query == nil or query=="" then return false end
	if not rc_list then rc_list = F.get_rc_safe() end
	for word in rc_list:gmatch("[^%s]+") do
		if word == query then return true end
	end
	return false
end

F.has_rc_match = function(query,rc_list) -- query = pattern string, single entry
	if not atc_id then return false end
	if rc_list == "" or query == nil or query=="" then return false end
	if not rc_list then rc_list = F.get_rc_safe() end

	local rc = {}
	for v in rc_list:gmatch("("..query..")") do
		table.insert(rc,v)
	end

	if rc[1] == true then
		return true, rc
	else
		return nil
	end
end

F.add_rc = function(rc_list) -- rc_list = string or table, eg: {"rc1","rc2"} OR "rc1 rc2"
	if not atc_id then return false end
	if type(rc_list) == "table" then
		rc_list = table.concat(rc_list," ")
	end
	set_rc(F.get_rc_safe().." "..rc_list)
	return true
end

F.remove_rc = function(rc_list,arrow_mode) -- rc_list = string eg: "rc1 rc2 rc3" OR table eg: {"rc1","rc2","rc3"}
	-- Arrow Modes:
	-- true: with arrow direction
	-- false: against arrow direction
	-- nil: ignores arrow direction

	if not atc_id then return false end
	if not rc_list then return false end

	if (arrow_mode == nil) or (atc_arrow == arrow_mode) then
		-- prep rc_list to useable format
		local rc_remove = {}
		if type(rc_list) == "string" then
			for word in rc_list:gmatch("[^%s]+") do
				rc_remove[word] = true
			end
		elseif type(rc_list) == "table" then
			for _,word in pairs(rc_list) do
				rc_remove[word] = true
			end
		end

		-- remove codes from train's rc
		local rc = F.get_rc_safe()
		local reinsert = {}
		for token in rc:gmatch("[^%s]+") do
			if not rc_remove[token] then
				table.insert(reinsert,token)
			end
		end
		-- insert new string to train's rc
		set_rc(table.concat(reinsert," "))
	end
	return reinsert
end

F.remove_rc_match = function(rc_list) -- rc_list = pattern string, single entry, eg: "rc_%d+"
	if not atc_id then return false end
	if not rc_list then return false end
	local rm = {}
	for v in F.get_rc_safe():gmatch("("..rc_list..")") do
		table.insert(rm,v)
	end
	F.remove_rc(rm)
	return rm
end

----------------------------------------------------------------------------------------------
-- Trackside Functions