Improve flight start detection

KSP-RO · Jan 21, 2024 · 5b88289 · 5b88289
1 parent 72c3ea0
commit 5b88289
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/TestFlightCore/TestFlightCore/TestFlightCore.cs b/TestFlightCore/TestFlightCore/TestFlightCore.cs
@@ -770,10 +770,19 @@ private void OnEngineActiveChange(ModuleEngines data)
             OnFlightStart();
         }
 
+        private void OnVesselSituationChange(GameEvents.HostedFromToAction<Vessel, Vessel.Situations> ev)
+        {
+            if (ev.from == Vessel.Situations.PRELAUNCH && ev.host == FlightGlobals.ActiveVessel)
+            {
+                OnFlightStart();
+            }
+        }
+
         private void OnFlightStart()
         {
             GameEvents.onStageActivate.Remove(OnStageActivate);
             GameEvents.onEngineActiveChange.Remove(OnEngineActiveChange);
+            GameEvents.onVesselSituationChange.Remove(OnVesselSituationChange);
             firstStaged = true;
             missionStartTime = Planetarium.GetUniversalTime();
         }
@@ -856,10 +865,17 @@ public override void Start()
                 GameEvents.onCrewTransferred.Add(OnCrewChange);
                 _OnCrewChange();
                 firstStaged = vessel.situation != Vessel.Situations.PRELAUNCH;
-                if (vessel.situation == Vessel.Situations.PRELAUNCH)
+
+                if (!firstStaged && part.isEngine())
+                {
+                    firstStaged = part.FindModulesImplementing<ModuleEngines>().Find(e => e.staged) != null;
+                }
+
+                if (!firstStaged)
                 {
                     GameEvents.onStageActivate.Add(OnStageActivate);
                     GameEvents.onEngineActiveChange.Add(OnEngineActiveChange);
+                    GameEvents.onVesselSituationChange.Add(OnVesselSituationChange);
                 }
                 else
                     missionStartTime = Planetarium.GetUniversalTime();
@@ -872,6 +888,7 @@ public override void OnDestroy()
             GameEvents.onCrewTransferred.Remove(OnCrewChange);
             GameEvents.onStageActivate.Remove(OnStageActivate);
             GameEvents.onEngineActiveChange.Remove(OnEngineActiveChange);
+            GameEvents.onVesselSituationChange.Remove(OnVesselSituationChange);
         }
 
         public override void OnAwake()