Add: Extend FEAT_TRAIN track_type for prop 34 (#390)

Author: glx22

nml/actions/action0properties.py

@@ -412,14 +412,26 @@ def prop_test(value):
     ]
 
 
+def single_or_list(prop_name, single_num, multiple_num, value):
+    if isinstance(value, ConstantNumeric):
+        return [Action0Property(single_num, value, 1)]
+    if isinstance(value, Array):
+        if len(value.values) == 1:
+            return [Action0Property(single_num, value.values[0].reduce_constant(), 1)]
+        return [VariableByteListProp(multiple_num, [[type.reduce_constant().value for type in value.values]])]
+    raise generic.ScriptError("'{}' must be a constant or an array of constants".format(prop_name))
+
+
 #
 # Feature 0x00 (Trains)
 #
 
 # fmt: off
 properties[0x00] = {
     **general_veh_props,
-    "track_type":                     {"size": 1, "num": 0x05},
+    "track_type": {
+        "custom_function": lambda value: single_or_list("track_type", 0x05, 0x34, value)
+    },
     "ai_special_flag":                {"size": 1, "num": 0x08},
     "speed": {
         "size": 2,

regression/013_train_callback.nml

@@ -136,6 +136,8 @@ item(FEAT_TRAINS, bulk_wagon) {
         speed: 0;
         refit_cost: 40;
         track_type: RAIL;
+        track_type: [RAIL];
+        track_type: [RAIL, ELRL];
         power: 0;
         running_cost_base: RUNNING_COST_STEAM;
         cargo_capacity: 30;

regression/expected/013_train_callback.grf

@@ -223,11 +223,11 @@ FF
 69 * 9 0D 7E \D<< 01 FF \dxFFFFFFF8
 
 70 * 12 06
-7F 01 FF \wx004D
-7E 01 FF \wx004F
+7F 01 FF \wx0053
+7E 01 FF \wx0055
 FF
 
-71 * 82 00 00 \b26 01 FF \wx0074
+71 * 88 00 00 \b28 01 FF \wx0074
 06 0F
 28 \wx0010
 1D \dx00000000
@@ -249,6 +249,9 @@ FF
 09 \wx0000
 1C 28
 05 00
+05 00
+34 \b2
+00 01
 0B \wx0000
 0E \dx00004C30
 14 1E