hw: Call OnTASPlaybackStopped() one frame later

YaLTeR · Jul 8, 2023 · db91df7 · db91df7
1 parent 1f6731b
commit db91df7
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/BunnymodXT/modules/HwDLL.cpp b/BunnymodXT/modules/HwDLL.cpp
@@ -5935,12 +5935,8 @@ void HwDLL::InsertCommands()
 
 		// Ran through all frames.
 		if (currentFramebulk >= totalFramebulks) {
-			CallOnTASPlaybackStopped();
 			runningFrames = false;
 
-			RenderYawOverrides.clear();
-			RenderYawOverrideIndex = 0;
-
 			if (!exportFilename.empty()) {
 				auto error = exportResult.Save(exportFilename);
 				if (error.Code == HLTAS::ErrorCode::OK)
@@ -5959,6 +5955,16 @@ void HwDLL::InsertCommands()
 		}
 	} else {
 		if (wasRunningFrames) {
+			RenderYawOverrides.clear();
+			RenderYawOverrideIndex = 0;
+
+			if (bxt_on_tas_playback_frame) {
+				// We don't use the return value here because we stop anyway.
+				CallOnTASPlaybackFrame();
+			}
+
+			CallOnTASPlaybackStopped();
+
 			ORIG_Cbuf_InsertText("host_framerate 0;_bxt_min_frametime 0;bxt_taslog 0\n");
 			if (sensitivityToRestore != 0) {
 				std::ostringstream ss;