153 lines
3.5 KiB
Lua
153 lines
3.5 KiB
Lua
-- train_related.lua
|
|
-- Occupation of track sections - mainly implementation of train callbacks
|
|
|
|
--[[
|
|
Track section occupation is saved as follows
|
|
|
|
In train:
|
|
train.il_sections = {
|
|
[n] = {ts_id = <...> (origin = <sigd>)}
|
|
}
|
|
-- "origin" is the TCB (signal describer) the train initially entered this section
|
|
|
|
In track section
|
|
ts.trains = {
|
|
[n] = <train_id>
|
|
}
|
|
|
|
When any inconsistency is detected, we will assume the most restrictive setup.
|
|
It will be possible to indicate a section "free" via the GUI.
|
|
]]
|
|
|
|
local ildb = advtrains.interlocking.db
|
|
|
|
|
|
local function itexist(tbl, com)
|
|
for _,item in ipairs(tbl) do
|
|
if (item==com) then
|
|
return true
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
local function itkexist(tbl, ikey, com)
|
|
for _,item in ipairs(tbl) do
|
|
if item[ikey] == com then
|
|
return true
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
local function itremove(tbl, com)
|
|
local i=1
|
|
while i <= #tbl do
|
|
if tbl[i] == com then
|
|
table.remove(tbl, i)
|
|
else
|
|
i = i + 1
|
|
end
|
|
end
|
|
end
|
|
local function itkremove(tbl, ikey, com)
|
|
local i=1
|
|
while i <= #tbl do
|
|
if tbl[i][ikey] == com then
|
|
table.remove(tbl, i)
|
|
else
|
|
i = i + 1
|
|
end
|
|
end
|
|
end
|
|
|
|
local function setsection(tid, train, ts_id, ts, origin)
|
|
-- train
|
|
if not train.il_sections then train.il_sections = {} end
|
|
if not itkexist(train.il_sections, "ts_id", ts_id) then
|
|
table.insert(train.il_sections, {ts_id = ts_id, origin = origin})
|
|
end
|
|
|
|
-- ts
|
|
if not ts.trains then ts.trains = {} end
|
|
if not itexist(ts.trains, tid) then
|
|
table.insert(ts.trains, tid)
|
|
end
|
|
|
|
end
|
|
|
|
local function freesection(tid, train, ts_id, ts)
|
|
-- train
|
|
if not train.il_sections then train.il_sections = {} end
|
|
itkremove(train.il_sections, "ts_id", ts_id)
|
|
|
|
-- ts
|
|
if not ts.trains then ts.trains = {} end
|
|
itremove(ts.trains, tid)
|
|
|
|
end
|
|
|
|
|
|
-- This is regular operation
|
|
-- The train is on a track and drives back and forth
|
|
|
|
-- This sets the section for both directions, to be failsafe
|
|
advtrains.tnc_register_on_enter(function(pos, id, train, index)
|
|
local tcb = ildb.get_tcb(pos)
|
|
if tcb then
|
|
for connid=1,2 do
|
|
local ts = tcb[connid].ts_id and ildb.get_ts(tcb[connid].ts_id)
|
|
if ts then
|
|
setsection(id, train, tcb[connid].ts_id, ts, {p=pos, s=connid})
|
|
end
|
|
end
|
|
end
|
|
end)
|
|
|
|
|
|
-- this time, of course, only clear the backside (cp connid)
|
|
advtrains.tnc_register_on_leave(function(pos, id, train, index)
|
|
local tcb = ildb.get_tcb(pos)
|
|
if tcb and train.path_cp[index] then
|
|
local connid = train.path_cp[index]
|
|
local ts = tcb[connid].ts_id and ildb.get_ts(tcb[connid].ts_id)
|
|
if ts then
|
|
freesection(id, train, tcb[connid].ts_id, ts)
|
|
end
|
|
end
|
|
end)
|
|
|
|
-- those callbacks are needed to account for created and removed trains (also regarding coupling)
|
|
|
|
advtrains.te_register_on_create(function(id, train)
|
|
-- let's see what track sections we find here
|
|
local index = atround(train.index)
|
|
local pos = advtrains.path_get(train, index)
|
|
local ts_id, origin = ildb.get_ts_at_pos(pos)
|
|
if ts_id then
|
|
local ts = ildb.get_ts(ts_id)
|
|
if ts then
|
|
setsection(id, train, ts_id, ts, origin)
|
|
else
|
|
atwarn("ILDB corruption: TCB",origin," has invalid TS reference")
|
|
end
|
|
elseif ts_id==nil then
|
|
atwarn("Train",id,": Unable to determine whether to block a track section!")
|
|
else
|
|
atdebug("Train",id,": Outside of interlocked area!")
|
|
end
|
|
end)
|
|
|
|
advtrains.te_register_on_remove(function(id, train)
|
|
if train.il_sections then
|
|
for idx, item in ipairs(train.il_sections) do
|
|
|
|
local ts = item.ts_id and ildb.get_ts(item.ts_id)
|
|
|
|
if ts and ts.trains then
|
|
itremove(ts.trains, id)
|
|
end
|
|
end
|
|
train.il_sections = nil
|
|
end
|
|
end)
|