From f86b55c4818c2b04bbb54ea6a36e4afcab939bdc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Henrik=20Rydg=C3=A5rd?= <hrydgard@gmail.com>
Date: Tue, 29 Oct 2024 15:50:47 +0100
Subject: [PATCH 1/2] Postshader: Work around a play-reported race condition
 (haven't figured out how to hit it, though).

---
 UI/DisplayLayoutScreen.cpp | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/UI/DisplayLayoutScreen.cpp b/UI/DisplayLayoutScreen.cpp
index f6a4717f57..cf739f1b1a 100644
--- a/UI/DisplayLayoutScreen.cpp
+++ b/UI/DisplayLayoutScreen.cpp
@@ -378,14 +378,22 @@ void DisplayLayoutScreen::CreateViews() {
 
 			auto removeButton = shaderRow->Add(new Choice(ImageID("I_TRASHCAN"), new LinearLayoutParams(0.0f)));
 			removeButton->OnClick.Add([=](EventParams &e) -> UI::EventReturn {
-				g_Config.vPostShaderNames.erase(g_Config.vPostShaderNames.begin() + i);
-				System_PostUIMessage(UIMessage::GPU_CONFIG_CHANGED);
-				RecreateViews();
+				if (i < g_Config.vPostShaderNames.size()) {
+					// Protect against possible race conditions.
+					g_Config.vPostShaderNames.erase(g_Config.vPostShaderNames.begin() + i);
+					System_PostUIMessage(UIMessage::GPU_CONFIG_CHANGED);
+					RecreateViews();
+				}
 				return UI::EVENT_DONE;
 			});
 
 			auto moreButton = shaderRow->Add(new Choice(ImageID("I_THREE_DOTS"), new LinearLayoutParams(0.0f)));
 			moreButton->OnClick.Add([=](EventParams &e) -> UI::EventReturn {
+				if (i >= g_Config.vPostShaderNames.size()) {
+					// Protect against possible race conditions.
+					return UI::EVENT_DONE;
+				}
+
 				PopupContextMenuScreen *contextMenu = new UI::PopupContextMenuScreen(postShaderContextMenu, ARRAY_SIZE(postShaderContextMenu), I18NCat::DIALOG, moreButton);
 				screenManager()->push(contextMenu);
 				const ShaderInfo *info = GetPostShaderInfo(g_Config.vPostShaderNames[i]);

From b226dd329f0930ffc0246243dae3158580e3db83 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Henrik=20Rydg=C3=A5rd?= <hrydgard@gmail.com>
Date: Tue, 29 Oct 2024 16:15:46 +0100
Subject: [PATCH 2/2] Postshader: Fix issue where some changes weren't
 reflected until resizing the window

---
 UI/DisplayLayoutScreen.cpp | 20 ++++++++++++--------
 UI/DisplayLayoutScreen.h   |  2 +-
 2 files changed, 13 insertions(+), 9 deletions(-)

diff --git a/UI/DisplayLayoutScreen.cpp b/UI/DisplayLayoutScreen.cpp
index cf739f1b1a..c4ba612fb4 100644
--- a/UI/DisplayLayoutScreen.cpp
+++ b/UI/DisplayLayoutScreen.cpp
@@ -147,11 +147,7 @@ void DisplayLayoutScreen::dialogFinished(const Screen *dialog, DialogResult resu
 	RecreateViews();
 }
 
-UI::EventReturn DisplayLayoutScreen::OnPostProcShaderChange(UI::EventParams &e) {
-	// Remove the virtual "Off" entry. TODO: Get rid of it generally.
-	g_Config.vPostShaderNames.erase(std::remove(g_Config.vPostShaderNames.begin(), g_Config.vPostShaderNames.end(), "Off"), g_Config.vPostShaderNames.end());
-	FixPostShaderOrder(&g_Config.vPostShaderNames);
-
+static void NotifyPostChanges() {
 	System_PostUIMessage(UIMessage::GPU_CONFIG_CHANGED);
 	System_PostUIMessage(UIMessage::GPU_RENDER_RESIZED);  // To deal with shaders that can change render resolution like upscaling.
 	System_PostUIMessage(UIMessage::POSTSHADER_UPDATED);
@@ -159,6 +155,13 @@ UI::EventReturn DisplayLayoutScreen::OnPostProcShaderChange(UI::EventParams &e)
 	if (gpu) {
 		gpu->NotifyConfigChanged();
 	}
+}
+
+UI::EventReturn DisplayLayoutScreen::OnPostProcShaderChange(UI::EventParams &e) {
+	// Remove the virtual "Off" entry. TODO: Get rid of it generally.
+	g_Config.vPostShaderNames.erase(std::remove(g_Config.vPostShaderNames.begin(), g_Config.vPostShaderNames.end(), "Off"), g_Config.vPostShaderNames.end());
+	FixPostShaderOrder(&g_Config.vPostShaderNames);
+	NotifyPostChanges();
 	return UI::EVENT_DONE;
 }
 
@@ -378,10 +381,11 @@ void DisplayLayoutScreen::CreateViews() {
 
 			auto removeButton = shaderRow->Add(new Choice(ImageID("I_TRASHCAN"), new LinearLayoutParams(0.0f)));
 			removeButton->OnClick.Add([=](EventParams &e) -> UI::EventReturn {
+				// Protect against possible race conditions.
 				if (i < g_Config.vPostShaderNames.size()) {
-					// Protect against possible race conditions.
 					g_Config.vPostShaderNames.erase(g_Config.vPostShaderNames.begin() + i);
-					System_PostUIMessage(UIMessage::GPU_CONFIG_CHANGED);
+					FixPostShaderOrder(&g_Config.vPostShaderNames);
+					NotifyPostChanges();
 					RecreateViews();
 				}
 				return UI::EVENT_DONE;
@@ -415,7 +419,7 @@ void DisplayLayoutScreen::CreateViews() {
 						return UI::EVENT_DONE;
 					}
 					FixPostShaderOrder(&g_Config.vPostShaderNames);
-					System_PostUIMessage(UIMessage::GPU_CONFIG_CHANGED);
+					NotifyPostChanges();
 					RecreateViews();
 					return UI::EVENT_DONE;
 				});
diff --git a/UI/DisplayLayoutScreen.h b/UI/DisplayLayoutScreen.h
index d256b94f6d..9f89462bba 100644
--- a/UI/DisplayLayoutScreen.h
+++ b/UI/DisplayLayoutScreen.h
@@ -50,7 +50,7 @@ private:
 	UI::ChoiceStrip *mode_ = nullptr;
 	UI::Choice *postProcChoice_ = nullptr;
 	std::string shaderNames_[256];
-	std::deque<bool> settingsVisible_;  // vector<bool> is an insane bitpacked specialization!
+	std::deque<bool> settingsVisible_;  // vector<bool> is an insane bitpacked specialization! Not to be used with checkboxes!
 };
 
 class PostProcScreen : public UI::ListPopupScreen {