When recording float params, return to the Record screen when recordi…

…ng is complete
acidbubbles · Nov 23, 2022 · 91e4f1c · 91e4f1c
1 parent ec73d50
commit 91e4f1c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/UI/Screens/RecordScreen.cs b/src/UI/Screens/RecordScreen.cs
@@ -194,8 +194,12 @@ private IEnumerator OnRecordCo(int timeMode, bool recordExtendsLength, bool hide
                 showStartMarkers
             );
 
+            var changedScreen = false;
             if (!targets.OfType<FreeControllerV3AnimationTarget>().Any() && targets.OfType<JSONStorableFloatAnimationTarget>().Any())
+            {
                 ChangeScreen(TargetsScreen.ScreenName);
+                changedScreen = true;
+            }
 
             while (enumerator.MoveNext())
                 yield return enumerator.Current;
@@ -210,6 +214,12 @@ private IEnumerator OnRecordCo(int timeMode, bool recordExtendsLength, bool hide
             animationEditContext.ResetScrubberRange();
             if (hideMenuDuringRecording)
                 SuperController.singleton.ShowMainHUDAuto();
+            if (changedScreen)
+            {
+                yield return 0;
+                // Note: The listener has been disposed, we have to ask the plugin
+                plugin.ChangeScreen(ScreenName, null);
+            }
         }
     }
 }