Implement ReplayManager::StopRecording

happyhavoc · Nov 4, 2024 · f6b21ba · f6b21ba
1 parent 1900220
commit f6b21ba
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/config/implemented.csv b/config/implemented.csv
@@ -329,6 +329,7 @@ th06::ReplayManager::OnDraw
 th06::ReplayManager::AddedCallback
 th06::ReplayManager::AddedCallbackDemo
 th06::ReplayManager::DeletedCallback
+th06::ReplayManager::StopRecording
 th06::Player::AddedCallback
 th06::Player::DeletedCallback
 th06::Player::RegisterChain

diff --git a/config/stubbed.csv b/config/stubbed.csv
@@ -12,7 +12,6 @@ th06::ResultScreen::RegisterChain
 th06::ResultScreen::ReleaseScoreDat
 th06::ScreenEffect::RegisterChain
 th06::MainMenu::SelectRelated
-th06::ReplayManager::StopRecording
 th06::ReplayManager::SaveReplay
 th06::ScreenEffect::DrawSquare
 th06::TextHelper::ReleaseTextBuffer

diff --git a/src/ReplayManager.cpp b/src/ReplayManager.cpp
@@ -327,4 +327,20 @@ ZunResult ReplayManager::DeletedCallback(ReplayManager *mgr)
     g_ReplayManager = NULL;
     return ZUN_SUCCESS;
 }
+
+void ReplayManager::StopRecording()
+{
+    ReplayManager *mgr = g_ReplayManager;
+    if (mgr != NULL)
+    {
+        mgr->replayInputs += 1;
+        mgr->replayInputs->frameNum = mgr->frameId;
+        mgr->replayInputs->inputKey = 0;
+        mgr->replayInputs += 1;
+        mgr->replayInputs->frameNum = 9999999;
+        mgr->replayInputs->inputKey = 0;
+        mgr->replayInputStageBookmarks[g_GameManager.currentStage - 1] = mgr->replayInputs + 1;
+    }
+}
+
 }; // namespace th06