local log = require("scada-common.log") local mqueue = require("scada-common.mqueue") local ppm = require("scada-common.ppm") local tcd = require("scada-common.tcd") local util = require("scada-common.util") local coordinator = require("coordinator.coordinator") local iocontrol = require("coordinator.iocontrol") local renderer = require("coordinator.renderer") local sounder = require("coordinator.sounder") local apisessions = require("coordinator.session.apisessions") local core = require("graphics.core") local log_render = coordinator.log_render local log_sys = coordinator.log_sys local log_comms = coordinator.log_comms local threads = {} local MAIN_CLOCK = 0.5 -- (2Hz, 10 ticks) local RENDER_SLEEP = 100 -- (100ms, 2 ticks) local MQ__RENDER_CMD = { START_MAIN_UI = 1, MON_CONNECT = 2, MON_DISCONNECT = 3, MON_RESIZE = 4, UPDATE = 5 } -- main thread ---@nodiscard ---@param smem crd_shared_memory function threads.thread__main(smem) ---@class parallel_thread local public = {} -- execute thread function public.exec() iocontrol.fp_rt_status("main", true) log.debug("main thread start") local loop_clock = util.new_clock(MAIN_CLOCK) -- start clock loop_clock.start() log_sys("system started successfully") -- load in from shared memory local crd_state = smem.crd_state local nic = smem.crd_sys.nic local coord_comms = smem.crd_sys.coord_comms local conn_watchdog = smem.crd_sys.conn_watchdog -- event loop while true do local event, param1, param2, param3, param4, param5 = util.pull_event() -- handle event if event == "peripheral_detach" then local type, device = ppm.handle_unmount(param1) if type ~= nil and device ~= nil then if type == "modem" then -- we only really care if this is our wireless modem -- if it is another modem, handle other peripheral losses separately if nic.is_modem(device) then nic.disconnect() log_sys("comms modem disconnected") local other_modem = ppm.get_wireless_modem() if other_modem then log_sys("found another wireless modem, using it for comms") nic.connect(other_modem) else -- close out main UI renderer.close_ui() -- alert user to status log_sys("awaiting comms modem reconnect...") iocontrol.fp_has_modem(false) end else log_sys("non-comms modem disconnected") end elseif type == "monitor" then smem.q.mq_render.push_data(MQ__RENDER_CMD.MON_DISCONNECT, device) elseif type == "speaker" then log_sys("lost alarm sounder speaker") iocontrol.fp_has_speaker(false) end end elseif event == "peripheral" then local type, device = ppm.mount(param1) if type ~= nil and device ~= nil then if type == "modem" then if device.isWireless() and not nic.is_connected() then -- reconnected modem log_sys("comms modem reconnected") nic.connect(device) iocontrol.fp_has_modem(true) elseif device.isWireless() then log.info("unused wireless modem reconnected") else log_sys("wired modem reconnected") end elseif type == "monitor" then smem.q.mq_render.push_data(MQ__RENDER_CMD.MON_CONNECT, { name = param1, device = device }) elseif type == "speaker" then log_sys("alarm sounder speaker reconnected") sounder.reconnect(device) iocontrol.fp_has_speaker(true) end end elseif event == "monitor_resize" then smem.q.mq_render.push_data(MQ__RENDER_CMD.MON_RESIZE, param1) elseif event == "timer" then if loop_clock.is_clock(param1) then -- main loop tick -- toggle heartbeat iocontrol.heartbeat() -- maintain connection if nic.is_connected() then local ok, start_ui = coord_comms.try_connect() if not ok then crd_state.link_fail = true log_sys("supervisor connection failed, shutting down...") log.fatal("failed to connect to supervisor") break elseif start_ui then log_sys("supervisor connected, dispatching main UI start") smem.q.mq_render.push_command(MQ__RENDER_CMD.START_MAIN_UI) --@todo if not ui_ok then break end end end -- iterate sessions and free any closed sessions apisessions.iterate_all() apisessions.free_all_closed() if renderer.ui_ready() then -- update clock used on main and flow monitors iocontrol.get_db().facility.ps.publish("date_time", os.date(smem.date_format)) end loop_clock.start() elseif conn_watchdog.is_timer(param1) then -- supervisor watchdog timeout log_comms("supervisor server timeout") -- close connection, main UI, and stop sounder coord_comms.close() renderer.close_ui() sounder.stop() else -- a non-clock/main watchdog timer event -- check API watchdogs apisessions.check_all_watchdogs(param1) -- notify timer callback dispatcher tcd.handle(param1) end elseif event == "modem_message" then -- got a packet local packet = coord_comms.parse_packet(param1, param2, param3, param4, param5) -- handle then check if it was a disconnect if coord_comms.handle_packet(packet) then log_comms("supervisor closed connection") -- close connection, main UI, and stop sounder coord_comms.close() renderer.close_ui() sounder.stop() end elseif event == "monitor_touch" or event == "mouse_click" or event == "mouse_up" or event == "mouse_drag" or event == "mouse_scroll" or event == "double_click" then -- handle a mouse event renderer.handle_mouse(core.events.new_mouse_event(event, param1, param2, param3)) elseif event == "speaker_audio_empty" then -- handle speaker buffer emptied sounder.continue() end -- check for termination request if event == "terminate" or ppm.should_terminate() then crd_state.shutdown = true log.info("terminate requested, main thread exiting") -- handle closing supervisor connection coord_comms.try_connect(true) if coord_comms.is_linked() then log_comms("terminate requested, closing supervisor connection...") else crd_state.link_fail = true end coord_comms.close() log_comms("supervisor connection closed") -- handle API sessions log_comms("closing api sessions...") apisessions.close_all() log_comms("api sessions closed") break end end end -- execute the thread in a protected mode, retrying it on return if not shutting down function public.p_exec() local crd_state = smem.crd_state while not crd_state.shutdown do local status, result = pcall(public.exec) if status == false then log.fatal(util.strval(result)) end iocontrol.fp_rt_status("main", false) -- if status is true, then we are probably exiting, so this won't matter -- if not, we need to restart the clock -- this thread cannot be slept because it will miss events (namely "terminate" otherwise) if not crd_state.shutdown then log.info("main thread restarting now...") end end end return public end -- coordinator renderer thread ---@nodiscard ---@param smem crd_shared_memory function threads.thread__render(smem) ---@class parallel_thread local public = {} -- execute thread function public.exec() iocontrol.fp_rt_status("render", true) log.debug("render thread start") -- load in from shared memory local crd_state = smem.crd_state local render_queue = smem.q.mq_render local last_update = util.time() -- thread loop while true do -- check for messages in the message queue while render_queue.ready() and not crd_state.shutdown do local msg = render_queue.pop() if msg ~= nil then if msg.qtype == mqueue.TYPE.COMMAND then -- received a command if msg.message == MQ__RENDER_CMD.START_MAIN_UI then -- start up the main UI log_render("starting main UI...") local draw_start = util.time_ms() local ui_message crd_state.ui_ok, ui_message = renderer.try_start_ui() if not crd_state.ui_ok then log_render(util.c("main UI error: ", ui_message)) log.fatal(util.c("main GUI render failed with error ", ui_message)) else log_render("main UI draw took " .. (util.time_ms() - draw_start) .. "ms") end elseif msg.message == MQ__RENDER_CMD.UPDATE then -- new data end elseif msg.qtype == mqueue.TYPE.DATA then -- received data local cmd = msg.message ---@type queue_data if cmd.key == MQ__RENDER_CMD.MON_CONNECT then -- monitor connected if renderer.handle_reconnect(cmd.val.name, cmd.val.device) then log_sys(util.c("configured monitor ", cmd.val.name, " reconnected")) else log_sys(util.c("unused monitor ", cmd.val.name, " connected")) end elseif cmd.key == MQ__RENDER_CMD.MON_DISCONNECT then -- monitor disconnected if renderer.handle_disconnect(cmd.val) then log_sys("lost a configured monitor") else log_sys("lost an unused monitor") end elseif cmd.key == MQ__RENDER_CMD.MON_RESIZE then -- monitor resized local is_used, is_ok = renderer.handle_resize(cmd.val) if is_used then log_sys(util.c("configured monitor ", cmd.val, " resized, ", util.trinary(is_ok, "display fits", "display does not fit"))) end end elseif msg.qtype == mqueue.TYPE.PACKET then -- received a packet end end -- quick yield util.nop() end -- check for termination request if crd_state.shutdown then log.info("render thread exiting") break end -- delay before next check last_update = util.adaptive_delay(RENDER_SLEEP, last_update) end end -- execute the thread in a protected mode, retrying it on return if not shutting down function public.p_exec() local crd_state = smem.crd_state while not crd_state.shutdown do local status, result = pcall(public.exec) if status == false then log.fatal(util.strval(result)) end iocontrol.fp_rt_status("render", false) if not crd_state.shutdown then log.info("render thread restarting in 5 seconds...") util.psleep(5) end end end return public end return threads