Add process priority settings and handlers

Add client config options to control process priorities: ClHighProcessPriority (set DDNet process/thread to high) and ClDiscordNormalProcessPriority (lower Discord processes to normal). Implement Windows-specific logic to apply/reset the DDNet process/thread priority and to scan/process Discord executables and set their priority via a background thread. Expose a new component callback OnFocusChange and wire GameClient to detect window focus changes and notify components so priority can be reapplied on focus. Add UI controls in the settings (Performance) to toggle these options and a console chain handler to apply changes immediately. Threading and atomic flags are used to manage the Discord-priority worker; behavior is guarded by platform-specific (Windows) includes and APIs.

Author: qxdFox

src/engine/shared/config_variables_entity.h

@@ -416,6 +416,9 @@ MACRO_CONFIG_STR(ClPhysicBallsSkin, ec_physic_balls_skin, 24, "volleyball", CFGF
 // Moving Tiles
 MACRO_CONFIG_INT(ClShowMovingTilesEntities, ec_show_moving_tiles_entities, 1, 0, 1, CFGFLAG_CLIENT | CFGFLAG_SAVE, "Show server-side moving tiles in entities")
 
+MACRO_CONFIG_INT(ClHighProcessPriority, ec_high_process_priority, 1, 0, 1, CFGFLAG_CLIENT | CFGFLAG_SAVE, "Set DDNets process priority to High")
+MACRO_CONFIG_INT(ClDiscordNormalProcessPriority, ec_discord_normal_process_priority, 1, 0, 1, CFGFLAG_CLIENT | CFGFLAG_SAVE, "Set Discord process priority to normal")
+
 // Custom variables from my server for the editor
 // You can use these if you get my server from https://github.com/FoxNet-DDNet/FoxNet
 

src/game/client/component.h

@@ -305,6 +305,12 @@ class CComponent : public CComponentInterfaces
 	virtual void OnSelfDeath()
 	{
 	}
+	/*
+	 * Called when the current focus of the window changes, for example when the user alt-tabs out of the game.
+	*/
+	virtual void OnFocusChange(bool Focused)
+	{
+	}
 };
 
 #endif

src/game/client/components/entity/commands.cpp

@@ -434,11 +434,6 @@ void CEClient::ConReplyLast(IConsole::IResult *pResult, void *pUserData)
 	pSelf->GameClient()->m_Chat.SendChat(0, Text);
 }
 
-void CEClient::ConCrash(IConsole::IResult *pResult, void *pUserData)
-{
-	exit(666);
-}
-
 void CEClient::ConSpectateId(IConsole::IResult *pResult, void *pUserData)
 {
 	CEClient *pSelf = (CEClient *)pUserData;
@@ -449,6 +444,11 @@ void CEClient::ConSpectateId(IConsole::IResult *pResult, void *pUserData)
 	pSelf->GameClient()->m_Chat.SendChat(0, pCmd);
 }
 
+void CEClient::ConCrash(IConsole::IResult *pResult, void *pUserData)
+{
+	exit(666);
+}
+
 void CEClient::OnConsoleInit()
 {
 	IConfigManager *pConfigManager = Kernel()->RequestInterface<IConfigManager>();
@@ -498,7 +498,7 @@ void CEClient::OnConsoleInit()
 	Console()->Chain("ec_discord_rpc", ConchainDiscordUpdate, this);
 	Console()->Chain("ec_discord_map_status", ConchainDiscordUpdate, this);
 	Console()->Chain("ec_discord_online_status", ConchainDiscordUpdate, this);
-	Console()->Chain("ec_discord_offline_status", ConchainDiscordUpdate, this);
+	Console()->Chain("ec_high_process_priority", ConchainDDNetProcessPriority, this);
 }
 
 void CEClient::ConchainGoresMode(IConsole::IResult *pResult, void *pUserData, IConsole::FCommandCallback pfnCallback, void *pCallbackUserData)
@@ -527,6 +527,17 @@ void CEClient::ConchainDiscordUpdate(IConsole::IResult *pResult, void *pUserData
 	pSelf->Client()->DiscordRPCchange();
 }
 
+void CEClient::ConchainDDNetProcessPriority(IConsole::IResult *pResult, void *pUserData, IConsole::FCommandCallback pfnCallback, void *pCallbackUserData)
+{
+	pfnCallback(pResult, pCallbackUserData);
+	CEClient *pSelf = (CEClient *)pUserData;
+	if(pResult->NumArguments())
+	{
+		bool Value = pResult->GetInteger(0) != 0;
+		pSelf->SetDDNetProcessPriority(Value);
+	}
+}
+
 void CEClient::ConfigSaveCallback(IConfigManager *pConfigManager, void *pUserData)
 {
 	CEClient *pThis = (CEClient *)pUserData;

src/game/client/components/entity/entity.cpp

@@ -28,6 +28,23 @@
 #include <algorithm>
 #include <cmath>
 #include <cstring>
+#if defined(CONF_FAMILY_WINDOWS)
+#include <Windows.h>
+#endif
+
+#if defined(CONF_FAMILY_WINDOWS)
+#include <TlHelp32.h>
+#include <processthreadsapi.h>
+#include <base/thread.h>
+
+static bool IsDiscordProcessName(const wchar_t *pProcessName)
+{
+	return _wcsicmp(pProcessName, L"Discord.exe") == 0 ||
+	       _wcsicmp(pProcessName, L"DiscordCanary.exe") == 0 ||
+	       _wcsicmp(pProcessName, L"DiscordPTB.exe") == 0 ||
+	       _wcsicmp(pProcessName, L"DiscordSystemHelper.exe") == 0;
+}
+#endif
 
 void CEClient::OnChatMessage(int ClientId, int Team, const char *pMsg)
 {
@@ -481,8 +498,97 @@ void CEClient::OnShutdown()
 	g_Config.m_ClKillCounter = m_KillCount;
 }
 
+void CEClient::SetDDNetProcessPriority(bool Set)
+{
+#if defined(CONF_FAMILY_WINDOWS)
+	if(!SetPriorityClass(GetCurrentProcess(), Set ? HIGH_PRIORITY_CLASS : NORMAL_PRIORITY_CLASS))
+	{
+		log_info("entity-client", Set ? "Failed to set process priority" : "Failed to reset process priority");
+		return;
+	}
+	if(!SetThreadPriority(GetCurrentThread(), Set ? THREAD_PRIORITY_HIGHEST : THREAD_PRIORITY_NORMAL))
+	{
+		log_info("entity-client", Set ? "Failed to set thread priority" : "Failed to reset thread priority");
+		return;
+	}
+#endif
+}
+
+void CEClient::DiscordPriorityThread(void *pUserData)
+{
+	CEClient *pSelf = (CEClient *)pUserData;
+	pSelf->SetDiscordProcessesNormalPriority();
+	pSelf->m_DiscordPriorityDelay.store(time_get() + time_freq() * 30);
+	pSelf->m_DiscordPriorityThreadRunning.store(false);
+}
+
+void CEClient::StartDiscordPriorityThread()
+{
+	// Don't start the thread if it's already running
+	if(m_DiscordPriorityThreadRunning.load())
+		return;
+
+	if(m_pDiscordPriorityThread)
+	{
+		thread_wait(m_pDiscordPriorityThread);
+		m_pDiscordPriorityThread = nullptr;
+	}
+
+	m_DiscordPriorityDelay.store(0);
+	m_DiscordPriorityThreadRunning.store(true);
+	m_pDiscordPriorityThread = thread_init(DiscordPriorityThread, this, "discord-priority");
+}
+
+void CEClient::SetDiscordProcessesNormalPriority()
+{
+#if defined(CONF_FAMILY_WINDOWS)
+	HANDLE hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
+	if(hSnapshot == INVALID_HANDLE_VALUE)
+	{
+		log_info("entity-client", "Failed to create process snapshot");
+		return;
+	}
+
+	PROCESSENTRY32 Entry;
+	mem_zero(&Entry, sizeof(Entry));
+	Entry.dwSize = sizeof(Entry);
+
+	int Changed = 0;
+	int Failed = 0;
+
+	if(Process32First(hSnapshot, &Entry))
+	{
+		do
+		{
+			if(!IsDiscordProcessName(Entry.szExeFile))
+				continue;
+
+			HANDLE hProcess = OpenProcess(PROCESS_SET_INFORMATION | PROCESS_QUERY_LIMITED_INFORMATION, FALSE, Entry.th32ProcessID);
+			if(!hProcess)
+			{
+				Failed++;
+				continue;
+			}
+
+			if(SetPriorityClass(hProcess, NORMAL_PRIORITY_CLASS))
+				Changed++;
+			else
+				Failed++;
+
+			CloseHandle(hProcess);
+		} while(Process32Next(hSnapshot, &Entry));
+	}
+
+	CloseHandle(hSnapshot);
+#endif
+}
+
 void CEClient::OnInit()
 {
+	SetDDNetProcessPriority(g_Config.m_ClHighProcessPriority);
+	if(g_Config.m_ClDiscordNormalProcessPriority)
+		StartDiscordPriorityThread();
+
 	// On client load
 	TextRender()->SetCustomFace(g_Config.m_ClCustomFont);
 
@@ -527,6 +633,12 @@ void CEClient::OnStateChange(int NewState, int OldState)
 
 void CEClient::OnRender()
 {
+	const int64_t DiscordPriorityDelay = m_DiscordPriorityDelay.load();
+	if(g_Config.m_ClDiscordNormalProcessPriority && !m_DiscordPriorityThreadRunning.load() && DiscordPriorityDelay < time_get())
+	{
+		StartDiscordPriorityThread();
+	}
+
 	UpdateRainbow();
 
 	if(Client()->State() == CClient::STATE_DEMOPLAYBACK)
@@ -545,3 +657,8 @@ void CEClient::OnSelfDeath()
 {
 	m_KillCount++;
 }
+
+void CEClient::OnFocusChange(bool IsFocused)
+{
+	SetDDNetProcessPriority(g_Config.m_ClHighProcessPriority);
+}

src/game/client/components/entity/entity.h

@@ -8,6 +8,7 @@
 
 #include <game/client/component.h>
 
+#include <atomic>
 #include <vector>
 
 class CEClient : public CComponent
@@ -58,13 +59,16 @@ class CEClient : public CComponent
 
 	static void ConReplyLast(IConsole::IResult *pResult, void *pUserData);
 
-	static void ConCrash(IConsole::IResult *pResult, void *pUserData);
-
 	static void ConSpectateId(IConsole::IResult *pResult, void *pUserData);
 
+	static void ConCrash(IConsole::IResult *pResult, void *pUserData);
+
 	static void ConchainGoresMode(IConsole::IResult *pResult, void *pUserData, IConsole::FCommandCallback pfnCallback, void *pCallbackUserData);
 	static void ConchainFastInputs(IConsole::IResult *pResult, void *pUserData, IConsole::FCommandCallback pfnCallback, void *pCallbackUserData);
 	static void ConchainDiscordUpdate(IConsole::IResult *pResult, void *pUserData, IConsole::FCommandCallback pfnCallback, void *pCallbackUserData);
+	static void ConchainDDNetProcessPriority(IConsole::IResult *pResult, void *pUserData, IConsole::FCommandCallback pfnCallback, void *pCallbackUserData);
+
+	static void DiscordPriorityThread(void *pUserData);
 
 public:
 	int m_KillCount;
@@ -131,6 +135,13 @@ class CEClient : public CComponent
 
 	bool m_FirstLaunch = false;
 
+	void SetDDNetProcessPriority(bool Set);
+	std::atomic<int64_t> m_DiscordPriorityDelay{0};
+	std::atomic_bool m_DiscordPriorityThreadRunning{false};
+	void *m_pDiscordPriorityThread = nullptr;
+	void StartDiscordPriorityThread();
+	void SetDiscordProcessesNormalPriority();
+
 private:
 	int Sizeof() const override { return sizeof(*this); }
 	void OnInit() override;
@@ -139,6 +150,7 @@ class CEClient : public CComponent
 	void OnNewSnapshot() override;
 	void OnShutdown() override;
 	void OnSelfDeath() override;
+	void OnFocusChange(bool IsFocused) override;
 };
 
-#endif // GAME_CLIENT_COMPONENTS_ENTITY_ENTITY_H
+#endif // GAME_CLIENT_COMPONENTS_ENTITY_ENTITY_H

src/game/client/components/entity/menus_entity.cpp

@@ -1,4 +1,3 @@
-
 #include <base/color.h>
 #include <base/math.h>
 #include <base/str.h>
@@ -2185,9 +2184,9 @@ void CMenus::RenderSettingsEClient(CUIRect MainView)
 	CUIRect Label, Button;
 	// left side in settings menu
 
-	CUIRect Automation, FreezeKill, ChatSettings, GoresMode,
-		MenuSettings, FastInput, AntiLatency, FrozenTeeHud, AntiPingSmoothing, GhostTools;
-	MainView.VSplitMid(&Automation, &GoresMode);
+	CUIRect Automation, FreezeKill, ChatSettings,
+		Performance, GoresMode, MenuSettings, FastInput, AntiLatency, FrozenTeeHud, AntiPingSmoothing, GhostTools;
+	MainView.VSplitMid(&Automation, &Performance);
 
 	/* Automation */
 	{
@@ -2555,9 +2554,32 @@ void CMenus::RenderSettingsEClient(CUIRect MainView)
 	}
 	// right side
 
+	/* Performance */
+	{
+		Performance.VMargin(5.0f, &Performance);
+		Performance.HSplitTop(80.0f, &Performance, &GoresMode);
+		if(s_ScrollRegion.AddRect(Performance))
+		{
+			Performance.Draw(BackgroundColor, IGraphics::CORNER_ALL, CornerRoundness);
+			Performance.VMargin(Margin, &Performance);
+
+			Performance.HSplitTop(HeaderHeight, &Button, &Performance);
+			Ui()->DoLabel(&Button, Localize("Performance"), HeaderSize, HeaderAlignment);
+
+			if(DoButton_CheckBoxAutoVMarginAndSet(&g_Config.m_ClHighProcessPriority, ("High DDNet Process Priority"), &g_Config.m_ClHighProcessPriority, &Performance, LineSize))
+				GameClient()->m_EClient.SetDDNetProcessPriority(g_Config.m_ClHighProcessPriority);
+
+			if(DoButton_CheckBoxAutoVMarginAndSet(&g_Config.m_ClDiscordNormalProcessPriority, ("Lower Discords Process Priority"), &g_Config.m_ClDiscordNormalProcessPriority, &Performance, LineSize))
+			{
+				if(g_Config.m_ClDiscordNormalProcessPriority)
+					GameClient()->m_EClient.StartDiscordPriorityThread();
+			}
+		}
+	}
+
 	/* Gores Mode */
 	{
-		GoresMode.VMargin(5.0f, &GoresMode);
+		GoresMode.HSplitTop(Margin, nullptr, &GoresMode);
 		GoresMode.HSplitTop(120.0f, &GoresMode, &FastInput);
 		if(s_ScrollRegion.AddRect(GoresMode))
 		{

src/game/client/gameclient.cpp

@@ -870,6 +870,16 @@ void CGameClient::OnRender()
 	// render all systems
 	for(auto &pComponent : m_vpAll)
 		pComponent->OnRender();
+	IEngineGraphics *pGraphics = ((IEngineGraphics *)Kernel()->RequestInterface<IEngineGraphics>());
+	if (pGraphics)
+	{
+		if(m_WasWindowActive != pGraphics->WindowActive())
+		{
+			for(auto &pComponent : m_vpAll)
+				pComponent->OnFocusChange(pGraphics->WindowActive());
+			m_WasWindowActive = pGraphics->WindowActive();
+		}
+	}
 
 	// clear all events/input for this frame
 	Input()->Clear();

src/game/client/gameclient.h

@@ -1092,6 +1092,8 @@ class CGameClient : public IGameClient
 	int GetClientId(const char *pName) override;
 	const char *GetClientName(int ClientId) override { return m_aClients[ClientId].m_aName; }
 	vec2 GetCursorWorldPos() const;
+
+	int m_WasWindowActive = 1;
 };
 
 ColorRGBA CalculateNameColor(ColorHSLA TextColorHSL);