From 932874882cab24fcf130933423fb400c17e27d56 Mon Sep 17 00:00:00 2001
From: Xu Liu <blinue@outlook.com>
Date: Mon, 27 Dec 2021 18:51:06 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E6=A8=A1=E6=8B=9F?=
 =?UTF-8?q?=E7=8B=AC=E5=8D=A0=E5=85=A8=E5=B1=8F=E7=9A=84=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Effects/FSR_EASU_DX10.hlsl              |  4 +-
 Magpie/MagWindow.cs                     |  7 ++-
 Magpie/MainWindow.xaml.cs               |  3 +-
 Magpie/Options/AdvancedOptionsPage.xaml |  9 +++-
 Magpie/Properties/Resources.Designer.cs |  9 ++++
 Magpie/Properties/Resources.resx        |  9 ++--
 Magpie/Properties/Resources.zh-CN.resx  |  9 ++--
 Magpie/Properties/Settings.Designer.cs  | 12 +++++
 Magpie/Properties/Settings.settings     |  3 ++
 Runtime/App.cpp                         | 10 +++-
 Runtime/App.h                           |  8 +++-
 Runtime/ExclModeHack.cpp                | 62 +++++++++++++++++++++++++
 Runtime/ExclModeHack.h                  | 15 ++++++
 Runtime/Runtime.vcxproj                 |  2 +
 Runtime/Runtime.vcxproj.filters         |  6 +++
 15 files changed, 152 insertions(+), 16 deletions(-)
 create mode 100644 Runtime/ExclModeHack.cpp
 create mode 100644 Runtime/ExclModeHack.h

diff --git a/Effects/FSR_EASU_DX10.hlsl b/Effects/FSR_EASU_DX10.hlsl
index 628fa0ca8..c562e8aa7 100644
--- a/Effects/FSR_EASU_DX10.hlsl
+++ b/Effects/FSR_EASU_DX10.hlsl
@@ -1,5 +1,5 @@
-// 适用于 DirectX 功能级别 10 的 FSR_EASU
-// 比原始版本稍慢
+// 閫傜敤浜� DirectX 鍔熻兘绾у埆 10 鐨� FSR_EASU
+// 姣斿師濮嬬増鏈◢鎱�
 
 //!MAGPIE EFFECT
 //!VERSION 1
diff --git a/Magpie/MagWindow.cs b/Magpie/MagWindow.cs
index e7db0f06a..7b7b9d0d4 100644
--- a/Magpie/MagWindow.cs
+++ b/Magpie/MagWindow.cs
@@ -45,6 +45,7 @@ private enum FlagMasks : uint {
 			NoCursor = 0x1,
 			AdjustCursorSpeed = 0x2,
 			ShowFPS = 0x4,
+			SimulateExclusiveFullscreen = 0x8,
 			DisableLowLatency = 0x10,
 			BreakpointMode = 0x20,
 			DisableWindowResizing = 0x40,
@@ -194,7 +195,8 @@ private enum FlagMasks : uint {
 			bool disableDirectFlip,
 			bool confineCursorIn3DGames,
 			bool cropTitleBarOfUWP,
-			bool disableEffectCache
+			bool disableEffectCache,
+			bool simulateExclusiveFullscreen
 		) {
 			if (Running) {
 				Logger.Info("宸插瓨鍦ㄥ叏灞忕獥鍙ｏ紝鍙栨秷杩涘叆鍏ㄥ睆");
@@ -230,7 +232,8 @@ bool disableEffectCache
 				(disableDirectFlip ? (uint)FlagMasks.DisableDirectFlip : 0) |
 				(confineCursorIn3DGames ? (uint)FlagMasks.ConfineCursorIn3DGames : 0) |
 				(cropTitleBarOfUWP ? (uint)FlagMasks.CropTitleBarOfUWP : 0) |
-				(disableEffectCache ? (uint)FlagMasks.DisableEffectCache : 0);
+				(disableEffectCache ? (uint)FlagMasks.DisableEffectCache : 0) |
+				(simulateExclusiveFullscreen ? (uint)FlagMasks.SimulateExclusiveFullscreen : 0);
 
 			_ = runEvent.Set();
 			Running = true;
diff --git a/Magpie/MainWindow.xaml.cs b/Magpie/MainWindow.xaml.cs
index 30e9082da..4e36b0db6 100644
--- a/Magpie/MainWindow.xaml.cs
+++ b/Magpie/MainWindow.xaml.cs
@@ -297,7 +297,8 @@ public partial class MainWindow : Window {
 				Settings.Default.DisableDirectFlip,
 				Settings.Default.ConfineCursorIn3DGames,
 				Settings.Default.CropTitleBarOfUWP,
-				Settings.Default.DebugDisableEffectCache
+				Settings.Default.DebugDisableEffectCache,
+				Settings.Default.SimulateExclusiveFullscreen
 			);
 
 			prevSrcWindow = magWindow.SrcWindow;
diff --git a/Magpie/Options/AdvancedOptionsPage.xaml b/Magpie/Options/AdvancedOptionsPage.xaml
index f573f07fc..0e9d970dc 100644
--- a/Magpie/Options/AdvancedOptionsPage.xaml
+++ b/Magpie/Options/AdvancedOptionsPage.xaml
@@ -17,8 +17,15 @@
     </Page.Resources>
     
     <StackPanel>
+        <CheckBox Content="{x:Static props:Resources.UI_Options_Advanced_Simulate_Exclusive_Fullscreen}"
+                  IsChecked="{Binding Source={x:Static props:Settings.Default},Path=SimulateExclusiveFullscreen,Mode=TwoWay}"/>
+        
         <StackPanel>
-            <Label Content="{x:Static props:Resources.UI_Options_Advanced_Logging}" FontWeight="Bold" Padding="0" FontSize="15" />
+            <Label Content="{x:Static props:Resources.UI_Options_Advanced_Logging}"
+                   FontWeight="Bold"
+                   Padding="0"
+                   Margin="0,20,0,0"
+                   FontSize="15" />
             <StackPanel Orientation="Horizontal" Margin="0,10,0,0">
                 <Label Content="{x:Static p:Resources.UI_Options_Advanced_Logging_Level}" Padding="0" VerticalContentAlignment="Center" />
                 <ComboBox x:Name="cbbLoggingLevel" Margin="10,0,0,0" SelectedIndex="{Binding Source={x:Static props:Settings.Default},Path=LoggingLevel,Mode=TwoWay}">
diff --git a/Magpie/Properties/Resources.Designer.cs b/Magpie/Properties/Resources.Designer.cs
index 67657d569..7dc9282aa 100644
--- a/Magpie/Properties/Resources.Designer.cs
+++ b/Magpie/Properties/Resources.Designer.cs
@@ -456,6 +456,15 @@ public class Resources {
             }
         }
         
+        /// <summary>
+        ///   鏌ユ壘绫讳技 Simulate Exclusive Fullscreen 鐨勬湰鍦板寲瀛楃涓层��
+        /// </summary>
+        public static string UI_Options_Advanced_Simulate_Exclusive_Fullscreen {
+            get {
+                return ResourceManager.GetString("UI_Options_Advanced_Simulate_Exclusive_Fullscreen", resourceCulture);
+            }
+        }
+        
         /// <summary>
         ///   鏌ユ壘绫讳技 Application 鐨勬湰鍦板寲瀛楃涓层��
         /// </summary>
diff --git a/Magpie/Properties/Resources.resx b/Magpie/Properties/Resources.resx
index 0bec3f55d..4791a07fa 100644
--- a/Magpie/Properties/Resources.resx
+++ b/Magpie/Properties/Resources.resx
@@ -250,6 +250,9 @@
   <data name="UI_Options_Advanced_Show_All_Capture_Methods" xml:space="preserve">
     <value>Show All Capture Methods</value>
   </data>
+  <data name="UI_Options_Advanced_Simulate_Exclusive_Fullscreen" xml:space="preserve">
+    <value>Simulate Exclusive Fullscreen</value>
+  </data>
   <data name="UI_Options_Application" xml:space="preserve">
     <value>Application</value>
   </data>
@@ -334,12 +337,12 @@
   <data name="UI_Options_Scale_Multiple_Monitors_All" xml:space="preserve">
     <value>Use All Monitors</value>
   </data>
-  <data name="UI_Options_Scale_Multiple_Monitors_Nearest" xml:space="preserve">
-    <value>Use the Monitor Nearest to the Source Window</value>
-  </data>
   <data name="UI_Options_Scale_Multiple_Monitors_Intersect" xml:space="preserve">
     <value>Use the Monitors that the Source Window Intersects</value>
   </data>
+  <data name="UI_Options_Scale_Multiple_Monitors_Nearest" xml:space="preserve">
+    <value>Use the Monitor Nearest to the Source Window</value>
+  </data>
   <data name="UI_Options_Scale_No_VSync" xml:space="preserve">
     <value>Disable Vsync</value>
   </data>
diff --git a/Magpie/Properties/Resources.zh-CN.resx b/Magpie/Properties/Resources.zh-CN.resx
index 3531406f2..bf7b4a4ba 100644
--- a/Magpie/Properties/Resources.zh-CN.resx
+++ b/Magpie/Properties/Resources.zh-CN.resx
@@ -249,6 +249,9 @@
   <data name="UI_Options_Advanced_Show_All_Capture_Methods" xml:space="preserve">
     <value>鏄剧ず鎵�鏈夋崟鑾锋ā寮�</value>
   </data>
+  <data name="UI_Options_Advanced_Simulate_Exclusive_Fullscreen" xml:space="preserve">
+    <value>妯℃嫙鐙崰鍏ㄥ睆</value>
+  </data>
   <data name="UI_Options_Application" xml:space="preserve">
     <value>搴旂敤绋嬪簭</value>
   </data>
@@ -333,12 +336,12 @@
   <data name="UI_Options_Scale_Multiple_Monitors_All" xml:space="preserve">
     <value>浣跨敤鎵�鏈夋樉绀哄櫒</value>
   </data>
-  <data name="UI_Options_Scale_Multiple_Monitors_Nearest" xml:space="preserve">
-    <value>浣跨敤璺濈婧愮獥鍙ｆ渶杩戠殑鏄剧ず鍣�</value>
-  </data>
   <data name="UI_Options_Scale_Multiple_Monitors_Intersect" xml:space="preserve">
     <value>浣跨敤婧愮獥鍙ｈ法瓒婄殑鎵�鏈夋樉绀哄櫒</value>
   </data>
+  <data name="UI_Options_Scale_Multiple_Monitors_Nearest" xml:space="preserve">
+    <value>浣跨敤璺濈婧愮獥鍙ｆ渶杩戠殑鏄剧ず鍣�</value>
+  </data>
   <data name="UI_Options_Scale_No_VSync" xml:space="preserve">
     <value>鍏抽棴鍨傜洿鍚屾</value>
   </data>
diff --git a/Magpie/Properties/Settings.Designer.cs b/Magpie/Properties/Settings.Designer.cs
index e7aced4d2..8a8bf81cf 100644
--- a/Magpie/Properties/Settings.Designer.cs
+++ b/Magpie/Properties/Settings.Designer.cs
@@ -334,5 +334,17 @@ internal sealed partial class Settings : global::System.Configuration.Applicatio
                 this["MultiMonitorUsage"] = value;
             }
         }
+        
+        [global::System.Configuration.UserScopedSettingAttribute()]
+        [global::System.Diagnostics.DebuggerNonUserCodeAttribute()]
+        [global::System.Configuration.DefaultSettingValueAttribute("False")]
+        public bool SimulateExclusiveFullscreen {
+            get {
+                return ((bool)(this["SimulateExclusiveFullscreen"]));
+            }
+            set {
+                this["SimulateExclusiveFullscreen"] = value;
+            }
+        }
     }
 }
diff --git a/Magpie/Properties/Settings.settings b/Magpie/Properties/Settings.settings
index d16ffa773..9187c04e2 100644
--- a/Magpie/Properties/Settings.settings
+++ b/Magpie/Properties/Settings.settings
@@ -80,5 +80,8 @@
     <Setting Name="MultiMonitorUsage" Type="System.UInt32" Scope="User">
       <Value Profile="(Default)">0</Value>
     </Setting>
+    <Setting Name="SimulateExclusiveFullscreen" Type="System.Boolean" Scope="User">
+      <Value Profile="(Default)">False</Value>
+    </Setting>
   </Settings>
 </SettingsFile>
\ No newline at end of file
diff --git a/Runtime/App.cpp b/Runtime/App.cpp
index 0debcb3a8..95efbd893 100644
--- a/Runtime/App.cpp
+++ b/Runtime/App.cpp
@@ -5,6 +5,7 @@
 #include "GDIFrameSource.h"
 #include "DwmSharedSurfaceFrameSource.h"
 #include "DesktopDuplicationFrameSource.h"
+#include "ExclModeHack.h"
 
 
 extern std::shared_ptr<spdlog::logger> logger;
@@ -142,8 +143,13 @@ bool App::Run(
 
 	_hwndSrcClient = IsCropTitleBarOfUWP() ? FindClientWindow(hwndSrc) : hwndSrc;
 
+	// 妯℃嫙鐙崰鍏ㄥ睆
+	// 蹇呴』鍦ㄤ富绐楀彛鍒涘缓鍓嶏紝鍚﹀垯 SHQueryUserNotificationState 鍙兘杩斿洖 QUNS_BUSY 鑰屼笉鏄� QUNS_RUNNING_D3D_FULL_SCREEN
+	ExclModeHack exclMode;
+
 	if (!_CreateHostWnd()) {
 		SPDLOG_LOGGER_CRITICAL(logger, "鍒涘缓涓荤獥鍙ｅけ璐�");
+		_OnQuit();
 		return false;
 	}
 
@@ -234,7 +240,7 @@ void App::_Run() {
 		MSG msg;
 		while (PeekMessage(&msg, nullptr, 0, 0, PM_REMOVE)) {
 			if (msg.message == WM_QUIT) {
-				_OnClose();
+				_OnQuit();
 				return;
 			}
 
@@ -516,7 +522,7 @@ LRESULT App::_HostWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
 	return DefWindowProc(hWnd, message, wParam, lParam);
 }
 
-void App::_OnClose() {
+void App::_OnQuit() {
 	// 閲婃斁璧勬簮
 	_frameSource = nullptr;
 	_renderer = nullptr;
diff --git a/Runtime/App.h b/Runtime/App.h
index d250707cf..7359b4e71 100644
--- a/Runtime/App.h
+++ b/Runtime/App.h
@@ -129,6 +129,10 @@ class App {
 		return _flags & (UINT)_FlagMasks::DisableEffectCache;
 	}
 
+	bool IsSimulateExclusiveFullscreen() const {
+		return _flags & (UINT)_FlagMasks::SimulateExclusiveFullscreen;
+	}
+
 	const char* GetErrorMsg() const {
 		return _errorMsg;
 	}
@@ -157,7 +161,7 @@ class App {
 
 	LRESULT _HostWndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam);
 
-	void _OnClose();
+	void _OnQuit();
 
 	const char* _errorMsg = ErrorMessages::GENERIC;
 
@@ -184,7 +188,7 @@ class App {
 		NoCursor = 0x1,
 		AdjustCursorSpeed = 0x2,
 		ShowFPS = 0x4,
-		
+		SimulateExclusiveFullscreen = 0x8,
 		DisableLowLatency = 0x10,
 		BreakpointMode = 0x20,
 		DisableWindowResizing = 0x40,
diff --git a/Runtime/ExclModeHack.cpp b/Runtime/ExclModeHack.cpp
new file mode 100644
index 000000000..b71c23be9
--- /dev/null
+++ b/Runtime/ExclModeHack.cpp
@@ -0,0 +1,62 @@
+#include "pch.h"
+#include "ExclModeHack.h"
+#include "App.h"
+#include <shellapi.h>
+
+
+// 妯℃嫙 D3D 鐙崰鍏ㄥ睆妯″紡锛屼互璧峰埌鍏嶆墦鎵扮殑鏁堟灉
+// SHQueryUserNotificationState 閫氬父琚敤鏉ユ娴嬫槸鍚︽湁 D3D 娓告垙鐙崰鍏ㄥ睆锛屼互纭畾鏄惁搴旇鍚戠敤鎴锋帹閫侀�氱煡/寮圭獥
+// 姝ゅ嚱鏁板唴閮ㄤ娇鐢ㄥ悕涓� __DDrawExclMode__ 鐨� mutex 妫�娴嬬嫭鍗犲叏灞忥紝鍥犳杩欓噷鐩存帴鑾峰彇璇� mutex 浠ユā鎷熺嫭鍗犲叏灞�
+// 鎰熻阿 @codehz 鎻愪緵鐨勬�濊矾 https://github.com/Blinue/Magpie/issues/245
+ExclModeHack::ExclModeHack() {
+	if (!App::GetInstance().IsSimulateExclusiveFullscreen()) {
+		return;
+	}
+
+	QUERY_USER_NOTIFICATION_STATE state;
+	HRESULT hr = SHQueryUserNotificationState(&state);
+	if (FAILED(hr)) {
+		SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("SHQueryUserNotificationState 澶辫触", hr));
+		return;
+	}
+	if (state != QUNS_ACCEPTS_NOTIFICATIONS) {
+		SPDLOG_LOGGER_INFO(logger, "宸插浜庡厤鎵撴壈鐘舵��");
+		return;
+	}
+
+	_exclModeMutex.reset(Utils::SafeHandle(
+		OpenMutex(SYNCHRONIZE, FALSE, L"__DDrawExclMode__")));
+	if (!_exclModeMutex) {
+		SPDLOG_LOGGER_ERROR(logger, MakeWin32ErrorMsg("OpenMutex 澶辫触"));
+		return;
+	}
+
+	DWORD result = WaitForSingleObject(_exclModeMutex.get(), 0);
+	if (result != WAIT_OBJECT_0) {
+		SPDLOG_LOGGER_ERROR(logger, "鑾峰彇 __DDrawExclMode__ 澶辫触");
+		_exclModeMutex.reset();
+		return;
+	}
+
+	hr = SHQueryUserNotificationState(&state);
+	if (FAILED(hr)) {
+		SPDLOG_LOGGER_ERROR(logger, MakeComErrorMsg("SHQueryUserNotificationState 澶辫触", hr));
+		ReleaseMutex(_exclModeMutex.get());
+		_exclModeMutex.reset();
+		return;
+	}
+	if (state != QUNS_RUNNING_D3D_FULL_SCREEN) {
+		SPDLOG_LOGGER_INFO(logger, "妯℃嫙鐙崰鍏ㄥ睆澶辫触");
+		ReleaseMutex(_exclModeMutex.get());
+		_exclModeMutex.reset();
+		return;
+	}
+
+	SPDLOG_LOGGER_INFO(logger, "妯℃嫙鐙崰鍏ㄥ睆鎴愬姛");
+}
+
+ExclModeHack::~ExclModeHack() {
+	if (_exclModeMutex) {
+		ReleaseMutex(_exclModeMutex.get());
+	}
+}
diff --git a/Runtime/ExclModeHack.h b/Runtime/ExclModeHack.h
new file mode 100644
index 000000000..51b5fda73
--- /dev/null
+++ b/Runtime/ExclModeHack.h
@@ -0,0 +1,15 @@
+#pragma once
+#include "pch.h"
+#include "Utils.h"
+
+
+class ExclModeHack {
+public:
+	ExclModeHack();
+
+	~ExclModeHack();
+
+private:
+	Utils::ScopedHandle _exclModeMutex;
+};
+
diff --git a/Runtime/Runtime.vcxproj b/Runtime/Runtime.vcxproj
index 5c112d97d..eedb8e36b 100644
--- a/Runtime/Runtime.vcxproj
+++ b/Runtime/Runtime.vcxproj
@@ -148,6 +148,7 @@
     <ClInclude Include="EffectCompiler.h" />
     <ClInclude Include="EffectDesc.h" />
     <ClInclude Include="ErrorMessages.h" />
+    <ClInclude Include="ExclModeHack.h" />
     <ClInclude Include="FrameRateDrawer.h" />
     <ClInclude Include="FrameSourceBase.h" />
     <ClInclude Include="framework.h" />
@@ -171,6 +172,7 @@
     <ClCompile Include="EffectDesc.cpp" />
     <ClCompile Include="EffectDrawer.cpp" />
     <ClCompile Include="EffectCompiler.cpp" />
+    <ClCompile Include="ExclModeHack.cpp" />
     <ClCompile Include="FrameRateDrawer.cpp" />
     <ClCompile Include="FrameSourceBase.cpp" />
     <ClCompile Include="GDIFrameSource.cpp" />
diff --git a/Runtime/Runtime.vcxproj.filters b/Runtime/Runtime.vcxproj.filters
index 97a74cc3d..66a310c48 100644
--- a/Runtime/Runtime.vcxproj.filters
+++ b/Runtime/Runtime.vcxproj.filters
@@ -58,6 +58,9 @@
     <ClCompile Include="DesktopDuplicationFrameSource.cpp">
       <Filter>鎹曡幏</Filter>
     </ClCompile>
+    <ClCompile Include="ExclModeHack.cpp">
+      <Filter>搴旂敤绋嬪簭</Filter>
+    </ClCompile>
   </ItemGroup>
   <ItemGroup>
     <ClInclude Include="resource.h">
@@ -126,6 +129,9 @@
     <ClInclude Include="DesktopDuplicationFrameSource.h">
       <Filter>鎹曡幏</Filter>
     </ClInclude>
+    <ClInclude Include="ExclModeHack.h">
+      <Filter>搴旂敤绋嬪簭</Filter>
+    </ClInclude>
   </ItemGroup>
   <ItemGroup>
     <None Include="cpp.hint" />