PushButtons traction cut off relay

Author: cesarBLG

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

@@ -96,11 +96,15 @@ public void Initialize()
                     switch(ScriptName)
                     {
                         case "Automatic":
-                            Script = new AutomaticTractionCutOffRelay() as TractionCutOffRelay;
+                            Script = new AutomaticTractionCutOffRelay();
                             break;
 
                         case "Manual":
-                            Script = new ManualTractionCutOffRelay() as TractionCutOffRelay;
+                            Script = new ManualTractionCutOffRelay();
+                            break;
+
+                        case "PushButtons":
+                            Script = new PushButtonsTractionCutOffRelay();
                             break;
 
                         default:
@@ -399,4 +403,124 @@ public override void HandleEvent(PowerSupplyEvent evt)
             }
         }
     }
+
+    public class PushButtonsTractionCutOffRelay : TractionCutOffRelay
+    {
+        private Timer ClosingTimer;
+        private TractionCutOffRelayState PreviousState;
+        private bool QuickPowerOn;
+
+        public override void Initialize()
+        {
+            ClosingTimer = new Timer(this);
+            ClosingTimer.Setup(ClosingDelayS());
+
+            SetDriverClosingAuthorization(true);
+        }
+
+        public override void Update(float elapsedSeconds)
+        {
+            SetClosingAuthorization(TCSClosingAuthorization() && CurrentDieselEngineState() == DieselEngineState.Running);
+
+            switch (CurrentState())
+            {
+                case TractionCutOffRelayState.Closed:
+                    if (!ClosingAuthorization() || DriverOpeningOrder())
+                    {
+                        SetCurrentState(TractionCutOffRelayState.Open);
+                    }
+                    break;
+
+                case TractionCutOffRelayState.Closing:
+                    if (ClosingAuthorization() && (DriverClosingOrder() || QuickPowerOn))
+                    {
+                        if (!ClosingTimer.Started)
+                        {
+                            ClosingTimer.Start();
+                        }
+
+                        if (ClosingTimer.Triggered)
+                        {
+                            QuickPowerOn = false;
+                            ClosingTimer.Stop();
+                            SetCurrentState(TractionCutOffRelayState.Closed);
+                        }
+                    }
+                    else
+                    {
+                        QuickPowerOn = false;
+                        ClosingTimer.Stop();
+                        SetCurrentState(TractionCutOffRelayState.Open);
+                    }
+                    break;
+
+                case TractionCutOffRelayState.Open:
+                    if (ClosingAuthorization() && (DriverClosingOrder() || QuickPowerOn))
+                    {
+                        SetCurrentState(TractionCutOffRelayState.Closing);
+                    }
+                    else if (QuickPowerOn)
+                    {
+                        QuickPowerOn = false;
+                    }
+                    break;
+            }
+
+            if (PreviousState != CurrentState())
+            {
+                switch (CurrentState())
+                {
+                    case TractionCutOffRelayState.Open:
+                        SignalEvent(Event.TractionCutOffRelayOpen);
+                        break;
+
+                    case TractionCutOffRelayState.Closing:
+                        SignalEvent(Event.TractionCutOffRelayClosing);
+                        break;
+
+                    case TractionCutOffRelayState.Closed:
+                        SignalEvent(Event.TractionCutOffRelayClosed);
+                        break;
+                }
+            }
+
+            PreviousState = CurrentState();
+        }
+
+        public override void HandleEvent(PowerSupplyEvent evt)
+        {
+            switch (evt)
+            {
+                case PowerSupplyEvent.CloseTractionCutOffRelayButtonPressed:
+                    SetDriverOpeningOrder(false);
+                    SetDriverClosingOrder(true);
+                    SignalEvent(Event.TractionCutOffRelayClosingOrderOn);
+                    Confirm(CabControl.TractionCutOffRelayClosingOrder, CabSetting.On);
+                    if (!ClosingAuthorization())
+                    {
+                        Message(ConfirmLevel.Warning, Simulator.Catalog.GetString("Traction cut-off relay closing not authorized"));
+                    }
+                    break;
+                case PowerSupplyEvent.CloseTractionCutOffRelayButtonReleased:
+                    SetDriverClosingOrder(false);
+                    break;
+                case PowerSupplyEvent.OpenTractionCutOffRelayButtonPressed:
+                    SetDriverClosingOrder(false);
+                    SetDriverOpeningOrder(true);
+                    SignalEvent(Event.TractionCutOffRelayClosingOrderOff);
+                    Confirm(CabControl.TractionCutOffRelayClosingOrder, CabSetting.Off);
+                    break;
+                case PowerSupplyEvent.OpenTractionCutOffRelayButtonReleased:
+                    SetDriverOpeningOrder(false);
+                    break;
+                case PowerSupplyEvent.QuickPowerOn:
+                    QuickPowerOn = true;
+                    break;
+                case PowerSupplyEvent.QuickPowerOff:
+                    QuickPowerOn = false;
+                    SetCurrentState(TractionCutOffRelayState.Open);
+                    break;
+            }
+        }
+    }
 }