Make battery react to quick power on/off

Author: cesarBLG

Source/Orts.Simulation/Simulation/RollingStocks/SubSystems/PowerSupplies/BatterySwitch.cs

@@ -43,6 +43,8 @@ public enum ModeType
         public bool CommandButtonOn { get; protected set; } = false;
         public bool CommandButtonOff { get; protected set; } = false;
         public bool On { get; protected set; } = false;
+        protected bool QuickPowerOn;
+        protected bool QuickPowerOff;
 
         public BatterySwitch(MSTSWagon wagon)
         {
@@ -201,6 +203,15 @@ public virtual void Update(float elapsedClockSeconds)
                 case ModeType.PushButtons:
                     if (On)
                     {
+                        if (QuickPowerOn)
+                        {
+                            QuickPowerOn = false;
+                            if (CommandButtonOn)
+                            {
+                                CommandButtonOn = false;
+                                Wagon.SignalEvent(Event.BatterySwitchCommandOff);
+                            }
+                        }
                         if (CommandButtonOff)
                         {
                             if (!Timer.Started)
@@ -225,6 +236,15 @@ public virtual void Update(float elapsedClockSeconds)
                     }
                     else
                     {
+                        if (QuickPowerOff)
+                        {
+                            QuickPowerOff = false;
+                            if (CommandButtonOff)
+                            {
+                                CommandButtonOff = false;
+                                Wagon.SignalEvent(Event.BatterySwitchCommandOff);
+                            }
+                        }
                         if (CommandButtonOn)
                         {
                             if (!Timer.Started)
@@ -302,6 +322,37 @@ public virtual void HandleEvent(PowerSupplyEvent evt)
                         Wagon.SignalEvent(Event.BatterySwitchCommandOff);
                     }
                     break;
+                case PowerSupplyEvent.QuickPowerOn:
+                    switch (Mode)
+                    {
+                        case ModeType.Switch:
+                            CommandSwitch = true;
+                            Wagon.SignalEvent(Event.BatterySwitchCommandOn);
+                            break;
+                        case ModeType.PushButtons:
+                            CommandButtonOn = true;
+                            Wagon.SignalEvent(Event.BatterySwitchCommandOn);
+                            QuickPowerOn = true;
+                            QuickPowerOff = false;
+                            break;
+
+                    }
+                    break;
+                case PowerSupplyEvent.QuickPowerOff:
+                    switch (Mode)
+                    {
+                        case ModeType.Switch:
+                            CommandSwitch = false;
+                            Wagon.SignalEvent(Event.BatterySwitchCommandOn);
+                            break;
+                        case ModeType.PushButtons:
+                            CommandButtonOff = true;
+                            Wagon.SignalEvent(Event.BatterySwitchCommandOn);
+                            QuickPowerOn = false;
+                            QuickPowerOff = true;
+                            break;
+                    }
+                    break;
             }
         }
     }

Source/Orts.Simulation/Simulation/RollingStocks/SubSystems/PowerSupplies/DieselPowerSupply.cs

@@ -320,7 +320,7 @@ public override void HandleEvent(PowerSupplyEvent evt)
             {
                 case PowerSupplyEvent.QuickPowerOn:
                     QuickPowerOn = true;
-                    SignalEventToBatterySwitch(PowerSupplyEvent.CloseBatterySwitch);
+                    SignalEventToBatterySwitch(PowerSupplyEvent.QuickPowerOn);
                     SignalEventToMasterKey(PowerSupplyEvent.TurnOnMasterKey);
                     SignalEventToDieselEngines(PowerSupplyEvent.StartEngine);
                     SignalEventToElectricTrainSupplySwitch(PowerSupplyEvent.SwitchOnElectricTrainSupply);
@@ -332,7 +332,7 @@ public override void HandleEvent(PowerSupplyEvent evt)
                     SignalEventToTractionCutOffRelay(PowerSupplyEvent.OpenTractionCutOffRelay);
                     SignalEventToDieselEngines(PowerSupplyEvent.StopEngine);
                     SignalEventToMasterKey(PowerSupplyEvent.TurnOffMasterKey);
-                    SignalEventToBatterySwitch(PowerSupplyEvent.OpenBatterySwitch);
+                    SignalEventToBatterySwitch(PowerSupplyEvent.QuickPowerOff);
                     break;
 
                 case PowerSupplyEvent.TogglePlayerEngine:

Source/Orts.Simulation/Simulation/RollingStocks/SubSystems/PowerSupplies/ElectricPowerSupply.cs

@@ -271,7 +271,7 @@ public override void HandleEvent(PowerSupplyEvent evt)
             {
                 case PowerSupplyEvent.QuickPowerOn:
                     QuickPowerOn = true;
-                    SignalEventToBatterySwitch(PowerSupplyEvent.CloseBatterySwitch);
+                    SignalEventToBatterySwitch(PowerSupplyEvent.QuickPowerOn);
                     SignalEventToMasterKey(PowerSupplyEvent.TurnOnMasterKey);
                     SignalEventToPantograph(PowerSupplyEvent.RaisePantograph, 1);
                     SignalEventToOtherTrainVehiclesWithId(PowerSupplyEvent.RaisePantograph, 1);
@@ -285,7 +285,7 @@ public override void HandleEvent(PowerSupplyEvent evt)
                     SignalEventToPantographs(PowerSupplyEvent.LowerPantograph);
                     SignalEventToOtherTrainVehicles(PowerSupplyEvent.LowerPantograph);
                     SignalEventToMasterKey(PowerSupplyEvent.TurnOffMasterKey);
-                    SignalEventToBatterySwitch(PowerSupplyEvent.OpenBatterySwitch);
+                    SignalEventToBatterySwitch(PowerSupplyEvent.QuickPowerOff);
                     break;
 
                 default: