Fix another one

pzgulyas · pzgulyas · commit 9358e1ce5df8 · 2025-04-08T18:21:15.000+02:00
diff --git a/Source/Orts.Simulation/Simulation/RollingStocks/SubSystems/TrainControlSystem.cs b/Source/Orts.Simulation/Simulation/RollingStocks/SubSystems/TrainControlSystem.cs
@@ -675,12 +675,15 @@ SignalFeatures NextGenericSignalFeatures(string signalFunctionTypeName, int item
                             if (head.Function == function)
                                 functionHead = head;
                     signalTypeName = functionHead.SignalTypeName;
-                    foreach (var key in functionHead.signalType.DrawStates.Keys)
+                    if (functionHead?.signalType?.DrawStates != null)
                     {
-                        if (functionHead.signalType.DrawStates[key].Index == functionHead.draw_state)
+                        foreach (var key in functionHead.signalType.DrawStates.Keys)
                         {
-                            drawStateName = functionHead.signalType.DrawStates[key].Name;
-                            break;
+                            if (functionHead.signalType.DrawStates[key].Index == functionHead.draw_state)
+                            {
+                                drawStateName = functionHead.signalType.DrawStates[key].Name;
+                                break;
+                            }
                         }
                     }
                     textAspect = functionHead?.TextSignalAspect ?? "";

Original file line number	Diff line number	Diff line change
`@@ -675,12 +675,15 @@ SignalFeatures NextGenericSignalFeatures(string signalFunctionTypeName, int item`
`675`	`675`	`if (head.Function == function)`
`676`	`676`	`functionHead = head;`
`677`	`677`	`signalTypeName = functionHead.SignalTypeName;`
`678`		`- foreach (var key in functionHead.signalType.DrawStates.Keys)`
	`678`	`+ if (functionHead?.signalType?.DrawStates != null)`
`679`	`679`	`{`
`680`		`- if (functionHead.signalType.DrawStates[key].Index == functionHead.draw_state)`
	`680`	`+ foreach (var key in functionHead.signalType.DrawStates.Keys)`
`681`	`681`	`{`
`682`		`- drawStateName = functionHead.signalType.DrawStates[key].Name;`
`683`		`- break;`
	`682`	`+ if (functionHead.signalType.DrawStates[key].Index == functionHead.draw_state)`
	`683`	`+ {`
	`684`	`+ drawStateName = functionHead.signalType.DrawStates[key].Name;`
	`685`	`+ break;`
	`686`	`+ }`
`684`	`687`	`}`
`685`	`688`	`}`
`686`	`689`	`textAspect = functionHead?.TextSignalAspect ?? "";`