-- Headshunt Controller
-- Note: may need customising due to track layout
local this_dir = true -- true = F.dir arrow points north. atc_arrow points towards yard

__approach_callback_mode = 1
if event.approach and not event.has_entered then
	atc_set_ars_disable(true)
	atc_set_lzb_tsr(1)
	return
end

if event.train then
	if not atc_arrow then --train has entered headshunt from yard. bounce or depart
		F.remove_rc({"TY_HEADSHUNT"})
		if not F.has_rc("TY_DEPART") then --if has TY_DEPART then another controller closer to the signal will remove to ensure signal engages the RC
			schedule_in(";01",atc_id)
			atc_set_ars_disable(true)
			return
		end
	else --train has bounced
		if F.has_rc("TY_AROUND") then
			if this_dir = F.dir() then --send train to Classification
				F.remove_rc({"TY_AROUND"})
				F.add_rc({"TY_PICKUP"})
				atc_set_ars_disable(false)
				return
			end --if at wrong end, do nothing. train will follow TY_AROUND ARS to relevant headshunt
		end -- if doesn't have TY_AROUND then let ARS classify the train
	end
	return
end

if event.schedule then
	if not atc_id then --bounce the train that just passed
		atc_send_to_train(event.msg,"S0WRD1S3")
	else
		if atc_id == event.msg then --train hasn't fully passed yet, wait a bit longer
			schedule_in(";01",atc_id)
		else -- somehow another train crossed the controller before the first one returned. Unlikely but possible
			atc_send_to_train(event.msg,"BBOL")
			atc_send("BBOL")
			F.error(true)
		end
	end
	return
end