From 40f3133e54dcfc372f947dbae67c609e2b3aa8f2 Mon Sep 17 00:00:00 2001
From: GradualSyrup <68757075+GradualSyrup@users.noreply.github.com>
Date: Sun, 10 Sep 2023 01:24:59 -0500
Subject: [PATCH] Add Support for Custom Stages in Training Mode (#628)

* Add patch for Custom Stages

* fmt

* Changelog
---
 README.md           | 1 +
 src/training/mod.rs | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/README.md b/README.md
index 3f98b96..b66ba74 100644
--- a/README.md
+++ b/README.md
@@ -513,6 +513,7 @@ These are the features that can be found [in the latest beta release](https://gi
 * **Input Recording**: Trigger a recording and play it back (default binds: `R+DPad Down`, `R+DPad Up` respectively)! These recordings can be customized in terms of length between 60 and 600 frames, looping, and more! They can be configured to run in familiar situations like Mash or Ledge options and more - @GradualSyrup, @jugeeya
 * **Input Display**: Show inputs in a per-status, frame counted log-style viewer! Choose between raw controller inputs and "Smash"-style inputs, being what the game actually understands and passes to fighters - @jugeeya, @xhudaman
 * **Auto-Updater**: Allow the modpack to automatically update itself when updates are available. Users can choose which update track they'd like from the Stable or Beta track in the menu - @asimon-1
+* **Custom Stages**: The Custom Stage Tab is now visible in Training Mode, allowing for labbing on Custom Stages - @GradualSyrup, @RayTwo
 * **Mash Overrides**: Specify which mash options to perform in specific scenarios - @GradualSyrup, @asimon-1
 * **Hide Tech Animations**: Animations and the CPU cursor are are hidden (character model/effects are invisible) during when the CPU is teching-- use this to practice true reaction tech chasing! The fixed camera is also modified on legal stages to aid this practice. - @GradualSyrup
 * **Customizable Button Configs**: Configure button combinations for save states and input recording in the menu itself. Please note that we now use raw inputs rather than Smash inputs, and save state save/load binds have moved to `L+DPad Down`, `L+DPad Up` respectively - @jugeeya
diff --git a/src/training/mod.rs b/src/training/mod.rs
index 5ba6e75..a663392 100644
--- a/src/training/mod.rs
+++ b/src/training/mod.rs
@@ -878,6 +878,11 @@ pub fn training_mods() {
         add_hook(params_main).unwrap();
     }
 
+    // Enable Custom Stages for Training Mode
+    // Specifically, we prevent a field in StageSelectInfo of the Scene that controls if the Custom Stage tab is loaded
+    //  from being set to false when we load the SSS in Training Mode
+    skyline::patching::Patch::in_text(0x184d1d8).nop().unwrap();
+
     skyline::install_hooks!(
         // Mash airdodge/jump
         handle_get_command_flag_cat,