Force AI trains to turn on power supply

Author: SteelFill

Source/Orts.Simulation/Simulation/AIs/AI.cs

@@ -927,11 +927,20 @@ public AITrain CreateAITrainDetail(Service_Definition sd, Traffic_Service_Defini
                         }
 
                         if (train.InitialSpeed != 0)
+                        {
                             car.SignalEvent(PowerSupplyEvent.RaisePantograph, 1);
+
+                            if (car is MSTSLocomotive loco)
+                                loco.SetPower(true);
+                        }
                     }
                     else
                     {
                         car.SignalEvent(PowerSupplyEvent.RaisePantograph, 1);
+
+                        if (car is MSTSLocomotive loco)
+                            loco.SetPower(true);
+
                         car.CarID = "AI" + train.Number.ToString() + " - " + (train.Cars.Count - 1).ToString();
                     }
 

Source/Orts.Simulation/Simulation/RollingStocks/MSTSDieselLocomotive.cs

@@ -1392,7 +1392,7 @@ public override void SwitchToPlayerControl()
         public override void SwitchToAutopilotControl()
         {
             SetDirection(Direction.Forward);
-            if (!LocomotivePowerSupply.MainPowerSupplyOn)
+            if (!LocomotivePowerSupply.MainPowerSupplyOn || !LocomotivePowerSupply.BatteryOn || !LocomotivePowerSupply.MasterKey.On)
             {
                 LocomotivePowerSupply.HandleEvent(PowerSupplyEvent.QuickPowerOn);
             }

Source/Orts.Simulation/Simulation/RollingStocks/MSTSLocomotive.cs

@@ -2047,6 +2047,7 @@ public override void Update(float elapsedClockSeconds)
 
             TrainControlSystem.Update(elapsedClockSeconds);
 
+            LocomotivePowerSupply?.Update(elapsedClockSeconds);
             UpdateControllers(elapsedClockSeconds);
 
             // Train Heading - only check the lead locomotive otherwise flipped locomotives further in consist will overwrite the train direction
@@ -2203,9 +2204,6 @@ public override void Update(float elapsedClockSeconds)
 
                     //Force to display
                     FilteredMotiveForceN = CurrentFilter.Filter(MotiveForceN, elapsedClockSeconds);
-
-                    // Only update power supply for player driven locomotives
-                    LocomotivePowerSupply?.Update(elapsedClockSeconds);
                     break;
                 default:
                     break;