From 6586a869cc0b3a84b540bf487aa99f95db927488 Mon Sep 17 00:00:00 2001
From: "LT_Schmiddy (Alex Schmid)" <schmidalex1996@gmail.com>
Date: Wed, 20 Nov 2024 16:25:10 -0500
Subject: [PATCH] Attempts to reset majora's mask internally. Not working yet.

---
 assets/config_menu.rml  |  7 +++++++
 include/zelda_config.h  |  3 +++
 patches/play_patches.c  | 13 +++++++++++++
 patches/play_patches.h  |  5 +++++
 patches/syms.ld         |  1 +
 src/game/recomp_api.cpp |  8 ++++++++
 src/ui/ui_config.cpp    | 27 +++++++++++++++++++++++++++
 7 files changed, 64 insertions(+)

diff --git a/assets/config_menu.rml b/assets/config_menu.rml
index 6de1fb4..2f569f6 100644
--- a/assets/config_menu.rml
+++ b/assets/config_menu.rml
@@ -73,6 +73,13 @@
 						</panel>
 					</tabset>
 					<div class="config__icon-buttons">
+            <button
+							class="icon-button"
+							onclick="open_reset_game_prompt"
+							id="config__reset_game-button"
+						>
+              <svg src="icons/Reset.svg" />
+            </button>
 						<button
 							class="icon-button"
 							onclick="open_quit_game_prompt"
diff --git a/include/zelda_config.h b/include/zelda_config.h
index 59b41f0..5a3c1a7 100644
--- a/include/zelda_config.h
+++ b/include/zelda_config.h
@@ -88,6 +88,9 @@ namespace zelda64 {
     void set_analog_cam_mode(AnalogCamMode mode);
 
     void open_quit_game_prompt();
+    void open_reset_game_prompt();
+
+    extern bool should_game_reset;
 };
 
 #endif
diff --git a/patches/play_patches.c b/patches/play_patches.c
index e3d7c84..a1f3ba4 100644
--- a/patches/play_patches.c
+++ b/patches/play_patches.c
@@ -1,6 +1,10 @@
 #include "play_patches.h"
 #include "z64debug_display.h"
 #include "input.h"
+#include "input.h"
+
+#include "z64game.h"
+#include "overlays/gamestates/ovl_title/z_title.h"
 
 extern Input D_801F6C18;
 
@@ -8,11 +12,20 @@ void controls_play_update(PlayState* play) {
     gSaveContext.options.zTargetSetting = recomp_get_targeting_mode();
 }
 
+void do_reset_game(PlayState* this) {
+    STOP_GAMESTATE(&this->state);
+    SET_NEXT_GAMESTATE(&this->state, ConsoleLogo_Init, sizeof(ConsoleLogoState));
+}
+
 // @recomp Patched to add hooks for various added functionality.
 RECOMP_PATCH void Play_Main(GameState* thisx) {
     static Input* prevInput = NULL;
     PlayState* this = (PlayState*)thisx;
 
+    if (recomp_should_reset_game()) {
+        do_reset_game(this);
+    }
+
     // @recomp
     debug_play_update(this);
     controls_play_update(this);
diff --git a/patches/play_patches.h b/patches/play_patches.h
index 19e7db2..cce59fb 100644
--- a/patches/play_patches.h
+++ b/patches/play_patches.h
@@ -2,6 +2,9 @@
 #define __PLAY_PATCHES_H__
 
 #include "patches.h"
+#include "patch_helpers.h"
+
+DECLARE_FUNC(bool, recomp_should_reset_game);
 
 void debug_play_update(PlayState* play);
 void camera_pre_play_update(PlayState* play);
@@ -11,4 +14,6 @@ void analog_cam_post_play_update(PlayState* play);
 void matrix_play_update(PlayState* play);
 void autosave_post_play_update(PlayState* play);
 
+
+
 #endif
diff --git a/patches/syms.ld b/patches/syms.ld
index a0e5d7d..e82160d 100644
--- a/patches/syms.ld
+++ b/patches/syms.ld
@@ -44,3 +44,4 @@ recomp_get_inverted_axes = 0x8F0000A4;
 recomp_high_precision_fb_enabled = 0x8F0000A8;
 recomp_get_resolution_scale = 0x8F0000AC;
 recomp_get_analog_inverted_axes = 0x8F0000B0;
+recomp_should_reset_game = 0x8F0000B4;
\ No newline at end of file
diff --git a/src/game/recomp_api.cpp b/src/game/recomp_api.cpp
index 28b4779..5bb0cf7 100644
--- a/src/game/recomp_api.cpp
+++ b/src/game/recomp_api.cpp
@@ -166,3 +166,11 @@ extern "C" void recomp_set_right_analog_suppressed(uint8_t* rdram, recomp_contex
 
     recomp::set_right_analog_suppressed(suppressed);
 }
+
+
+extern "C" void recomp_should_reset_game(uint8_t* rdram, recomp_context* ctx) {
+    bool retVal = zelda64::should_game_reset;
+    zelda64::should_game_reset = false;
+
+    _return(ctx, retVal);
+}
\ No newline at end of file
diff --git a/src/ui/ui_config.cpp b/src/ui/ui_config.cpp
index 12bb20e..c3b5542 100644
--- a/src/ui/ui_config.cpp
+++ b/src/ui/ui_config.cpp
@@ -161,6 +161,11 @@ static bool sample_positions_supported = false;
 
 static bool cont_active = true;
 
+namespace zelda64 {
+	bool should_game_reset;
+}
+
+
 static recomp::InputDevice cur_device = recomp::InputDevice::Controller;
 
 int recomp::get_scanned_input_index() {
@@ -279,6 +284,23 @@ void zelda64::open_quit_game_prompt() {
 	);
 }
 
+void zelda64::open_reset_game_prompt() {
+	prompt_context.open_prompt(
+		"Return to the main menu?",
+		"Any progress since your last save will be lost.",
+		"Quit",
+		"Cancel",
+		[]() {
+			should_game_reset = true;
+		},
+		[]() {},
+		recompui::ButtonVariant::Error,
+		recompui::ButtonVariant::Tertiary,
+		true,
+		"config__quit-game-button"
+	);
+}
+
 // These defaults values don't matter, as the config file handling overrides them.
 struct ControlOptionsContext {
 	int rumble_strength; // 0 to 100
@@ -559,6 +581,11 @@ public:
                 zelda64::open_quit_game_prompt();
 			});
 
+		recompui::register_event(listener, "open_reset_game_prompt",
+			[](const std::string& param, Rml::Event& event) {
+				zelda64::open_reset_game_prompt();
+			});
+
 		recompui::register_event(listener, "toggle_input_device",
 			[](const std::string& param, Rml::Event& event) {
 				cur_device = cur_device == recomp::InputDevice::Controller