Core.lua

if _G.DynamicSuspicionIndicators_Inited then
	return
end
_G.DynamicSuspicionIndicators_Inited = true

DynamicSuspicionIndicatorsManager = DynamicSuspicionIndicatorsManager or {}
local DI = DynamicSuspicionIndicatorsManager

DI.smooth_speed = 5
local HOST_COLLECT_INTERVAL = 0.1
DI.mod_path = "mods/DynamicSuspicionIndicators/"
DI.demo_mode = false

DI._log = function(msg)
	DI.Logger.dbg(msg)
end

dofile(DI.mod_path .. "modules/infra/logger.lua")
dofile(DI.mod_path .. "modules/settings/schema.lua")
dofile(DI.mod_path .. "modules/infra/color.lua")
dofile(DI.mod_path .. "modules/infra/assets.lua")
DI.Assets:register_textures(DI.mod_path)
dofile(DI.mod_path .. "modules/infra/game.lua")
dofile(DI.mod_path .. "modules/units.lua")
dofile(DI.mod_path .. "modules/suspicion/phase.lua")
dofile(DI.mod_path .. "modules/policy/target.lua")
dofile(DI.mod_path .. "modules/suspicion/records.lua")
dofile(DI.mod_path .. "modules/units/index.lua")
dofile(DI.mod_path .. "modules/integrations/clientside_stealth.lua")
dofile(DI.mod_path .. "modules/sync/codec.lua")
dofile(DI.mod_path .. "modules/sync/transport.lua")
dofile(DI.mod_path .. "modules/sync.lua")
dofile(DI.mod_path .. "modules/detection/host.lua")
dofile(DI.mod_path .. "modules/detection/client/sync_overlay.lua")
dofile(DI.mod_path .. "modules/detection/client/fallback.lua")
dofile(DI.mod_path .. "modules/detection/client.lua")
dofile(DI.mod_path .. "modules/hud/view.lua")
dofile(DI.mod_path .. "modules/hud/glyph.lua")
dofile(DI.mod_path .. "modules/hud/waypoint_vr.lua")
dofile(DI.mod_path .. "modules/hud/waypoint_render.lua")
dofile(DI.mod_path .. "modules/hud/waypoint.lua")
dofile(DI.mod_path .. "modules/hud/item.lua")
dofile(DI.mod_path .. "modules/hud/demo.lua")

local function _settings()
	return DI.SettingsSchema.normalized((_G.DP and _G.DP.settings) or {}, DI.smooth_speed)
end

function DI:_ensure_panel()
	if self._inited then
		return true
	end
	local G = DI.Game
	local hud_id = G.player_hud_id()
	if not (hud_id and setup) then
		return false
	end
	local hud = G.hud_script(hud_id)
	if not (hud and hud.panel) then
		return false
	end
	local panel = hud.panel:panel({ name = "detection_percent", layer = 1000 })
	local ws = G.gui_fullscreen_workspace()
	if not ws then
		return false
	end
	DI.HudItem.init_panel(panel, ws)
	if not setup._dp_update_orig then
		setup._dp_update_orig = setup.update
		setup.update = function(s, t, dt)
			setup._dp_update_orig(s, t, dt)
			DI:_update(t, dt)
		end
	end
	DI.Assets:register_textures(self.mod_path)
	DI.WaypointOverlay:install_hooks()
	if DI.Sync and DI.Sync.install then
		DI.Sync.install()
	end
	self._inited = true
	DI.Logger.info(string.format("loaded (hud %dx%d)", hud.panel:w(), hud.panel:h()))
	return true
end

function DI:_update(t, dt)
	if not self:_ensure_panel() then
		return
	end

	local G = DI.Game
	local R = DI.Records
	local cfg = _settings()
	local wo_deps = {
		npc_kind = DI.Units.npc_kind,
		records = R.records,
		cfg = cfg,
		hide_idle_observer = DI.ClientStealth.hides_unrecorded,
	}

	if DI.demo_mode then
		self._last_host_collect_t = nil
		DI.HudItem.destroy_all()
		DI.HudItem.place_demo(cfg)
		DI.WaypointOverlay:update({ npc_kind = DI.Units.npc_kind, records = {}, cfg = cfg })
		return
	end

	if not G.whisper_mode() then
		self._last_host_collect_t = nil
		DI.HudItem.destroy_all()
		DI.HudItem.destroy_demo()
		R.clear()
		wo_deps.records = R.records
		DI.WaypointOverlay:update(wo_deps)
		return
	end

	if G.is_client() then
		DI.Detection.collect_client(cfg, t)
	else
		local last_t = self._last_host_collect_t
		if not last_t or t < last_t or t - last_t >= HOST_COLLECT_INTERVAL then
			DI.Detection.collect(cfg)
			self._last_host_collect_t = t
		end
		if DI.Sync and DI.Sync.host_flush then
			DI.Sync.host_flush(t)
		end
	end

	local cam = G.camera()
	if not cam then
		return
	end

	local roles, allowed = R.tick(t, dt, cfg)
	wo_deps.records = R.records
	DI.HudItem.place_records(R.records, roles, allowed, cfg, cam)
	DI.HudItem.destroy_demo()
	DI.WaypointOverlay:update(wo_deps)
end

local function _reset_client_state()
	DI.ClientStealth.reset()
	local D = DI.Detection
	if D then
		D._client_peer_susp = {}
		D._client_obs_status = {}
		D._los_cache = {}
		D._client_sync_active = nil
		D._camera_susp_seq = 0
		D._camera_susp_seen = {}
		D._camera_fb_min_seq = 0
		D._cam_lvl_prev = {}
	end
	DI.Records._alert_flash = {}
	DI.Records._target_alerts = {}
	DI.Records._target_peaks = {}
	DI.Records._prev = {}
	DI.Records._smooth = {}
	DI.Records._smooth_phase = {}
	DI._last_host_collect_t = nil
	if DI.UnitIndex and DI.UnitIndex.invalidate then
		DI.UnitIndex.invalidate()
	end
	DI.WaypointOverlay:destroy_all()
end

DI.Game.on_event("GameSetupNewGame", "DI_ResetClientState", _reset_client_state)
DI.Game.on_event("LocalPlayerMovedToGameStateBase", "DI_ResetClientStateJoin", _reset_client_state)

DI:_ensure_panel()
DI.Logger.info("loaded")