ChaosMod: Refactor OnPresentCallback into an ChaosEvent

gta-chaos-mod · Oct 19, 2023 · 4b98b4b · 4b98b4b
1 parent f504863
commit 4b98b4b
Show file tree

Hide file tree

Showing 5 changed files with 28 additions and 12 deletions.
diff --git a/ChaosMod/ChaosMod.vcxproj b/ChaosMod/ChaosMod.vcxproj
@@ -495,6 +495,7 @@
     <ClInclude Include="Util\HelpText.h" />
     <ClInclude Include="Components\HelpTextQueue.h" />
     <ClInclude Include="Memory\Hooks\GetLabelTextHook.h" />
+    <ClInclude Include="Memory\Hooks\PresentHook.h" />
   </ItemGroup>
   <ItemGroup>
     <ResourceCompile Include="ChaosMod.rc" />

diff --git a/ChaosMod/Memory/Hooks/PresentHook.cpp b/ChaosMod/Memory/Hooks/PresentHook.cpp
@@ -1,5 +1,7 @@
 #include <stdafx.h>
 
+#include "PresentHook.h"
+
 #include "Memory/Hooks/Hook.h"
 #include "Memory/Hooks/ShaderHook.h"
 
@@ -10,7 +12,7 @@ HRESULT HK_IDXGISwapChain_Present(IDXGISwapChain *swapChain, UINT syncInterval,
 {
 	if (!(flags & DXGI_PRESENT_TEST))
 	{
-		Hooks::OnPresentCallback();
+		Hooks::OnPresent.Fire();
 	}
 
 	return OG_IDXGISwapChain_Present(swapChain, syncInterval, flags);

diff --git a/ChaosMod/Memory/Hooks/PresentHook.h b/ChaosMod/Memory/Hooks/PresentHook.h
@@ -0,0 +1,8 @@
+#pragma once
+
+#include "Util/Events.h"
+
+namespace Hooks
+{
+	inline ChaosEvent OnPresent;
+}
diff --git a/ChaosMod/Memory/Hooks/ScriptThreadRunHook.h b/ChaosMod/Memory/Hooks/ScriptThreadRunHook.h
@@ -1,6 +1,6 @@
 #pragma once
 
-#include <Util/Events.h>
+#include "Util/Events.h"
 
 namespace rage
 {

diff --git a/ChaosMod/Memory/Hooks/ShaderHook.cpp b/ChaosMod/Memory/Hooks/ShaderHook.cpp
@@ -2,9 +2,11 @@
 
 #include "ShaderHook.h"
 
-#include "Memory/Hooks/Hook.h"
 #include "Memory/Shader.h"
 
+#include "Memory/Hooks/Hook.h"
+#include "Memory/Hooks/PresentHook.h"
+
 #define SHADER_CACHE_MAX_ENTRIES 20
 
 using DWORD64 = unsigned long long;
@@ -41,6 +43,18 @@ void *HK_rage__CreateShader(const char *name, BYTE *data, DWORD size, DWORD type
 
 static bool OnHook()
 {
+	static CHAOS_EVENT_LISTENER(Hooks::OnPresent) onPresentListener;
+
+	onPresentListener.Register(Hooks::OnPresent,
+	                           []()
+	                           {
+		                           if (ms_RefreshShaders)
+		                           {
+			                           ms_RefreshShaders = false;
+			                           Memory::InvalidateShaderCache();
+		                           }
+	                           });
+
 	Handle handle;
 
 	//
@@ -128,13 +142,4 @@ namespace Hooks
 		ms_ShaderBytecode.clear();
 		ms_RefreshShaders = true;
 	}
-
-	void OnPresentCallback()
-	{
-		if (ms_RefreshShaders)
-		{
-			ms_RefreshShaders = false;
-			Memory::InvalidateShaderCache();
-		}
-	}
 }